From 4d7aadeda14cdc41606faaf2bf397a3904847d7d Mon Sep 17 00:00:00 2001 From: nihui Date: Fri, 5 Jun 2020 16:49:23 +0800 Subject: [PATCH 1/2] Update tf_generated_ops.td --- tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td index 1c9297485b4..467a119c174 100644 --- a/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td +++ b/tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td @@ -8651,11 +8651,11 @@ particular, begin = [1, 2, x, x, 0, x] # x denotes don't care (usually 0) end = [2, 4, x, x, -3, x] strides = [1, 1, x, x, -1, 1] -begin_mask = 1<<4 | 1 << 5 = 48 +begin_mask = 1<<4 | 1<<5 = 48 end_mask = 1<<5 = 32 ellipsis_mask = 1<<3 = 8 -new_axis_mask = 1<<2 4 -shrink_axis_mask = 1<<0 +new_axis_mask = 1<<2 = 4 +shrink_axis_mask = 1<<0 = 1 ``` In this case if `foo.shape` is (5, 5, 5, 5, 5, 5) the final shape of From 8b81960fcb60222b03f04751e5f99a36b24d27ca Mon Sep 17 00:00:00 2001 From: nihui Date: Sat, 6 Jun 2020 12:00:15 +0800 Subject: [PATCH 2/2] Update api_def_StridedSlice.pbtxt --- tensorflow/core/api_def/base_api/api_def_StridedSlice.pbtxt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/core/api_def/base_api/api_def_StridedSlice.pbtxt b/tensorflow/core/api_def/base_api/api_def_StridedSlice.pbtxt index 9a89a4e8e75..2714e31ac28 100644 --- a/tensorflow/core/api_def/base_api/api_def_StridedSlice.pbtxt +++ b/tensorflow/core/api_def/base_api/api_def_StridedSlice.pbtxt @@ -123,11 +123,11 @@ particular, begin = [1, 2, x, x, 0, x] # x denotes don't care (usually 0) end = [2, 4, x, x, -3, x] strides = [1, 1, x, x, -1, 1] -begin_mask = 1<<4 | 1 << 5 = 48 +begin_mask = 1<<4 | 1<<5 = 48 end_mask = 1<<5 = 32 ellipsis_mask = 1<<3 = 8 -new_axis_mask = 1<<2 4 -shrink_axis_mask = 1<<0 +new_axis_mask = 1<<2 = 4 +shrink_axis_mask = 1<<0 = 1 ``` In this case if `foo.shape` is (5, 5, 5, 5, 5, 5) the final shape of