| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | slice_spmd_expander.cc | 40 Status GetSliceOpArguments(mlir::TF::SliceOp slice_op, in GetSliceOpArguments()
54 mlir::Operation* slice_op, mlir::Value value, const Layout& layout, in VerifySliceLayout()
385 StatusOr<Layout> ApplyNewAndShrinkMasksToLayout(SliceOpT slice_op, in ApplyNewAndShrinkMasksToLayout()
450 auto slice_op = mlir::cast<mlir::TF::SliceOp>(op); in ExpandOp() local
549 auto slice_op = mlir::cast<mlir::TF::SliceOp>(op); in ComputeLayoutForward() local
560 auto slice_op = mlir::cast<mlir::TF::SliceOp>(op); in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_slice_attributes.cc | 32 auto* slice_op = slice_it->get(); in Run() local
|
| D | remove_trivial_slice.cc | 56 auto* slice_op = reshape_it->get(); in Run() local
|
| D | resolve_strided_slice_attributes.cc | 45 auto* slice_op = slice_it->get(); in Run() local
|
| D | unroll_batch_matmul.cc | 64 auto* slice_op = new SliceOperator; in SliceInput() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 192 auto slice_op = rewriter->create<TFL::SliceOp>( in SliceOutputs() local
367 LogicalResult PadSlice::matchAndRewrite(TFL::SliceOp slice_op, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 66 Output slice_op; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 355 auto slice_op = GetSliceOpForConv2DBackPropFilter(old_filter_shape, in HandleConv2DBackPropFilter() local
|
| D | legalize_hlo.cc | 1022 mhlo::SliceOp slice_op, OpAdaptor adaptor, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 1534 auto slice_op = in ComputeSliceK() local
|
| D | lower_static_tensor_list.cc | 864 auto slice_op = in CreateCondFalseBranch() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | array_ops_test.py | 1178 def run_and_time(self, slice_op): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 2154 auto slice_op = CreateOpAndInfer<tosa::SliceOp>( in convertSplitVOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 252 auto slice_op = dyn_cast_or_null<StridedSliceOp>(values()[0].getDefiningOp()); in fold() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5527 auto slice_op = rewriter.create<mhlo::SliceOp>( in matchAndRewrite() local
5607 Value slice_op = rewriter.create<RealDynamicSliceOp>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 3297 auto slice_op = in TEST_F() local
|