Home
last modified time | relevance | path

Searched refs:slice_op (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_trivial_slice.cc56 auto* slice_op = reshape_it->get(); in Run() local
57 if (slice_op->type != OperatorType::kSlice) { in Run()
61 if (!IsSliceTrivial(*model, *slice_op, this)) { in Run()
65 AddMessageF("Removing trivial %s", LogName(*slice_op)); in Run()
67 CHECK_EQ(slice_op->inputs.size(), 3); in Run()
Dresolve_slice_attributes.cc32 auto* slice_op = slice_it->get(); in Run() local
33 if (slice_op->type != OperatorType::kSlice) return ::tensorflow::Status::OK(); in Run()
35 auto* op = static_cast<SliceOperator*>(slice_op); in Run()
Dresolve_strided_slice_attributes.cc45 auto* slice_op = slice_it->get(); in Run() local
46 if (slice_op->type != OperatorType::kStridedSlice) in Run()
49 auto* op = static_cast<StridedSliceOperator*>(slice_op); in Run()
/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py986 def run_and_time(self, slice_op): argument
989 _ = slice_op.eval()
993 slice_op.eval()
1010 slice_op = helper[::2, ::1, ::2]
1011 self.run_and_time(slice_op)
1017 slice_op = helper[3::1, 3::1, 3::1]
1018 self.run_and_time(slice_op)
1023 slice_op = var[3::1, 3::1, 3::1]
1024 self.run_and_time(slice_op)
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass.cc66 Output slice_op; member
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD86 "slice_op.cc",
/external/tensorflow/tensorflow/core/kernels/
DBUILD135 "slice_op.h",
140 "slice_op.h",
902 ":slice_op",
1108 name = "slice_op",
1109 prefix = "slice_op",
1807 ":slice_op",
1825 ":slice_op",
5535 "slice_op.cc",
5536 "slice_op.h",
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt235 tensorflow/core/kernels/slice_op.cc