experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update tf_generated_ops.td

This commit is contained in:
nihui 2020-06-05 16:49:23 +08:00 committed by GitHub
parent a39ac9de5f
commit 4d7aadeda1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
View File

@ -8654,8 +8654,8 @@ strides = [1, 1, x, x, -1, 1]
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