Home
last modified time | relevance | path

Searched defs:slice_op (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_slice_attributes.cc32 auto* slice_op = slice_it->get(); in Run() local
Dremove_trivial_slice.cc56 auto* slice_op = reshape_it->get(); in Run() local
Dresolve_strided_slice_attributes.cc45 auto* slice_op = slice_it->get(); in Run() local
Dunroll_batch_matmul.cc62 auto* slice_op = new SliceOperator; in SliceInput() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc108 auto slice_op = rewriter.create<TF::SliceOp>(loc, slice_result_type, in sliceInput() local
Dtpu_space_to_depth_pass.cc400 auto slice_op = GetSliceOpForConv2DBackPropFilter(old_filter_shape, in HandleConv2DBackPropFilter() local
Dlegalize_hlo.cc271 mhlo::SliceOp slice_op, ArrayRef<Value> args, in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass.cc66 Output slice_op; member
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc701 auto slice_op = in CreateCondFalseBranch() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc1898 auto slice_op = rewriter.create<tosa::SliceOp>( in convertSplitOp() local
1978 auto slice_op = rewriter.create<tosa::SliceOp>( in convertSplitVOp() local
/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py1076 def run_and_time(self, slice_op): argument
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1043 lmhlo::SliceOp slice_op, ArrayRef<Value> args, in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc234 auto slice_op = dyn_cast_or_null<StridedSliceOp>(values()[0].getDefiningOp()); in fold() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc3297 auto slice_op = in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4831 auto slice_op = rewriter.create<mhlo::SliceOp>( in matchAndRewrite() local