Searched refs:strided_slice_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 527 TF::StridedSliceOp strided_slice_op = llvm::cast<TF::StridedSliceOp>(op); in RewriteNewAxisMask() local
528 uint64_t new_axis_mask = strided_slice_op.new_axis_mask(); in RewriteNewAxisMask()
531 Value original_input = strided_slice_op.input(); in RewriteNewAxisMask()
555 Location loc = strided_slice_op.getLoc(); in RewriteNewAxisMask()
572 uint64_t revised_begin_mask = strided_slice_op.begin_mask(); in RewriteNewAxisMask()
573 uint64_t revised_end_mask = strided_slice_op.end_mask(); in RewriteNewAxisMask()
576 revised_begin_mask |= strided_slice_op.new_axis_mask(); in RewriteNewAxisMask()
577 revised_end_mask |= strided_slice_op.new_axis_mask(); in RewriteNewAxisMask()
581 op, strided_slice_op.getType(), reshape, strided_slice_op.begin(), in RewriteNewAxisMask()
582 strided_slice_op.end(), strided_slice_op.strides(), in RewriteNewAxisMask()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | BUILD | 117 "strided_slice_op.cc",
118 "strided_slice_op.h",
178 "strided_slice_op.h",
225 "strided_slice_op.cc",
355 "strided_slice_op.h",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 114 "strided_slice_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 152 name = "strided_slice_op",
154 "strided_slice_op.cc",
167 "strided_slice_op.h",
172 "strided_slice_op.h",
988 ":strided_slice_op",
1190 deps = ARRAY_DEPS + [":strided_slice_op"],
2197 ":strided_slice_op",
5855 "strided_slice_op.cc",
5856 "strided_slice_op.h",
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 3555 auto strided_slice_op = ops::StridedSlice( in TEST_F() local
3558 auto z = ops::Identity(scope.WithOpName("z"), strided_slice_op); in TEST_F()
3646 auto strided_slice_op = ops::StridedSlice( in TEST_F() local
3649 auto z = ops::Identity(scope.WithOpName("z"), strided_slice_op); in TEST_F()
3707 auto strided_slice_op = ops::StridedSlice( in TEST_F() local
3710 auto z = ops::Identity(scope.WithOpName("z"), strided_slice_op); in TEST_F()
|