From d77db2a476454eb549512a7097813cb4a5b352b1 Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Wed, 4 Nov 2020 08:20:09 -0800 Subject: Clean input argument rank limitations Comparison ops, similar to the other elementwise ops, can operate on tensors of any rank. For those operators with rank limitations, make them consistent and remove "dims". Signed-off-by: Eric Kunze Change-Id: I33eb00122cec4c6625f080f9ed640f1d874fae87 --- chapters/activation_funcs.adoc | 8 ++++---- chapters/comparison.adoc | 18 +++++++++--------- chapters/data_layout.adoc | 10 +++++----- chapters/reduction.adoc | 12 ++++++------ 4 files changed, 24 insertions(+), 24 deletions(-) diff --git a/chapters/activation_funcs.adoc b/chapters/activation_funcs.adoc index 05a693b..f699f90 100644 --- a/chapters/activation_funcs.adoc +++ b/chapters/activation_funcs.adoc @@ -18,7 +18,7 @@ Clamp to an arbitrary minimum and maximum value. Note that the maximum and minim |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|Input|shape|Input tensor from 1 to 4 dims +|Input|in_t*|Input|shape|Input tensor with rank from 1 to 4 |Attribute|in_t|min_val|-|minimum clip value |Attribute|in_t|max_val|-|maximum clip value |Output|out_t*|Output|shape|Output tensor of same type and shape as input @@ -53,7 +53,7 @@ ReLU with a scalar maximum value. |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|Input|shape|Input tensor +|Input|in_t*|Input|shape|Input tensor with rank from 1 to 4 |Attribute|in_t|max_val|-|maximum clip value |Output|out_t*|Output|shape|Output tensor of same type and shape as input |=== @@ -103,7 +103,7 @@ generate_lookup_table(&sigmoid_table, &sigmoid_reference); |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|Input|shape|Input tensor from 1 to 4 dims +|Input|in_t*|Input|shape|Input tensor with rank from 1 to 4 |Output|out_t*|Output|shape|Output tensor of same type and shape as input |=== @@ -142,7 +142,7 @@ generate_lookup_table(&tanh_table, &tanh_reference); |Argument|Type|Name|Shape|Description -|Input|in_t*|Input|shape|Input tensor from 1 to 4 dims +|Input|in_t*|Input|shape|Input tensor with rank from 1 to 4 |Output|out_t*|Output|shape|Output tensor of same type and shape as input |=== diff --git a/chapters/comparison.adoc b/chapters/comparison.adoc index 3e69900..f9439fc 100644 --- a/chapters/comparison.adoc +++ b/chapters/comparison.adoc @@ -18,9 +18,9 @@ Elementwise comparison operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims -|Input|in_t*|input2|shape2|Input tensor with the same rank as Input 0 -|Output|out_t*|output|shape|Output tensor of same type, as the input tensors, with broadcast shape if necessary +|Input|in_t*|input1|shape1|Input tensor +|Input|in_t*|input2|shape2|Input tensor with the same rank as input1 +|Output|out_t*|output|shape|Output tensor with broadcast shape if necessary |=== *Operation Function:* @@ -54,9 +54,9 @@ Elementwise greater than comparison operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims -|Input|in_t*|input2|shape2|Input tensor with the same rank as Input 0 -|Output|out_t*|output|shape|Output tensor of same type, as the input tensors, with broadcast shape if necessary +|Input|in_t*|input1|shape1|Input tensor +|Input|in_t*|input2|shape2|Input tensor with the same rank as input1 +|Output|out_t*|output|shape|Output tensor with broadcast shape if necessary |=== *Operation Function:* @@ -90,9 +90,9 @@ Elementwise comparison operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims -|Input|in_t*|input2|shape2|Input tensor with the same rank as Input 0 -|Output|out_t*|output|shape|Output tensor of same type, as the input tensors, with broadcast shape if necessary +|Input|in_t*|input1|shape1|Input tensor +|Input|in_t*|input2|shape2|Input tensor with the same rank as input1 +|Output|out_t*|output|shape|Output tensor with broadcast shape if necessary |=== *Operation Function:* diff --git a/chapters/data_layout.adoc b/chapters/data_layout.adoc index bec551b..4e6d24d 100644 --- a/chapters/data_layout.adoc +++ b/chapters/data_layout.adoc @@ -17,7 +17,7 @@ Concatenate two tensors along a given axis. No data conversion happens during a |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims +|Input|in_t*|input1|shape1|Input tensor with rank from 1 to 4 |Input|in_t*|input2|shape2|Input tensor with rank matching input1 |Attribute|int|axis|-|Axis along which concatenation is to occur. |Output|out_t*|output|shape|Output tensor of same type, size as the input tensor @@ -106,7 +106,7 @@ Returns a tensor with the same type/values as the input, with a new shape specif |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims +|Input|in_t*|input1|shape1|Input tensor |Attribute|int|new_shape|[rank(output)]|List of values, with each element giving the size of the result tensor for the given dimension. At most one dimension may be given as-1 to automatically calculate the dimension size. |Output|out_t*|output|shape|Output tensor of same type, size as the input tensor |=== @@ -180,7 +180,7 @@ Extracts a slice of the input tensor 0 on the given axis, beginning at the start |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims +|Input|in_t*|input1|shape1|Input tensor with rank from 1 to 4 |Attribute|int|start|[rank(input1)]|List of integer coordinates, of length equal to the rank of input 0. Start coordinate for slicing. |Attribute|int|size|[rank(input1)]|List of integer size values, of length equal to the rank of input 0. Size of the input to be used. |Output|out_t*|output|shape|Output tensor of same type as the input tensor @@ -221,7 +221,7 @@ Replicates input 0 multiplies times along each dimension. |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims +|Input|in_t*|input1|shape1|Input tensor with rank from 1 to 4 |Attribute|int|multiplies|[rank(shape1)]|Number of times to replicate input1 in each dimension |Output|out_t*|output|shape|Output tensor of same type, rank as the input tensor |=== @@ -262,7 +262,7 @@ Permutes the dimensions based on perm. |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input1|shape1|Input tensor from 1 to 4 dims +|Input|in_t*|input1|shape1|Input tensor with rank from 1 to 4 |Attribute|int|perms|[rank(input1)]|List of integers of length equal to the rank of input1. |Output|out_t*|output|shape|Output tensor of same type, rank as the input tensor |=== diff --git a/chapters/reduction.adoc b/chapters/reduction.adoc index 0739dfd..9aaf9a1 100644 --- a/chapters/reduction.adoc +++ b/chapters/reduction.adoc @@ -18,7 +18,7 @@ Reduce a tensor along the given axis with a logical AND operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input|in_shape|Input tensor from 1 to 4 dims +|Input|in_t*|input|in_shape|Input tensor with rank from 1 to 4 |Attribute|int|axis|-|Axis to reduce |Output|out_t*|output|out_shape|Output tensor. Same rank as the input tensor. |=== @@ -59,7 +59,7 @@ Reduce a tensor along the given axis with a logical OR operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input|in_shape|Input tensor from 1 to 4 dims +|Input|in_t*|input|in_shape|Input tensor with rank from 1 to 4 |Attribute|int|axis|-|Axis to reduce |Output|out_t*|output|out_shape|Output tensor. Same rank as the input tensor. |=== @@ -100,7 +100,7 @@ Reduce a tensor along the given axis with a maximum operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input|in_shape|Input tensor from 1 to 4 dims +|Input|in_t*|input|in_shape|Input tensor with rank from 1 to 4 |Attribute|int|axis|-|Axis to reduce |Output|out_t*|output|out_shape|Output tensor. Same rank as the input tensor. |=== @@ -143,7 +143,7 @@ Reduce a tensor along the given axis with a minimum operation |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input|in_shape|Input tensor from 1 to 4 dims +|Input|in_t*|input|in_shape|Input tensor with rank from 1 to 4 |Attribute|int|axis|-|Axis to reduce |Output|out_t*|output|out_shape|Output tensor. Same rank as the input tensor. |=== @@ -191,7 +191,7 @@ Reduce a tensor along the given axis by computing the product of the axis. |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input|in_shape|Input tensor from 1 to 4 dims +|Input|in_t*|input|in_shape|Input tensor with rank from 1 to 4 |Attribute|int|axis|-|Axis to reduce |Output|out_t*|output|out_shape|Output tensor. Same rank as the input tensor. |=== @@ -232,7 +232,7 @@ Reduce a tensor along the given axis by computing the sum of the axis. |=== |Argument|Type|Name|Shape|Description -|Input|in_t*|input|in_shape|Input tensor from 1 to 4 dims +|Input|in_t*|input|in_shape|Input tensor with rank from 1 to 4 |Attribute|int|axis|-|Axis to reduce |Output|out_t*|output|out_shape|Output tensor. Same rank as the input tensor. |=== -- cgit v1.2.1