Home
last modified time | relevance | path

Searched refs:strided_slice (Results 1 – 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc37 namespace strided_slice { namespace
113 ::tflite::strided_slice::StartForAxis(op_params, input_shape, idx); in ResizeOutputTensor()
114 int32_t end = ::tflite::strided_slice::StopForAxis(op_params, input_shape, in ResizeOutputTensor()
247 nullptr, nullptr, strided_slice::Prepare, in Register_STRIDED_SLICE_REF()
248 strided_slice::Eval<strided_slice::kReference>}; in Register_STRIDED_SLICE_REF()
254 nullptr, nullptr, strided_slice::Prepare, in Register_STRIDED_SLICE()
255 strided_slice::Eval<strided_slice::kGenericOptimized>}; in Register_STRIDED_SLICE()
DAndroid.bp154 "strided_slice.cc",
/external/tensorflow/tensorflow/compiler/tests/
Dslice_ops_test.py134 o = array_ops.strided_slice(i, [2], [6], [2])
149 o = array_ops.strided_slice(i, begin, end, [1])
163 o = array_ops.strided_slice(i, [6], [2], [-2])
176 o = array_ops.strided_slice(i, [-1, 0], [0, 3])
190 o = array_ops.strided_slice(i, [0, 0], [-1, 3], [-1, 1])
206 o = array_ops.strided_slice(i, begin, end, [1, 1])
220 o = array_ops.strided_slice(i, [0, 2, 2], [2, 3, 6], [1, 1, 2])
241 o = array_ops.strided_slice(i, [2, 2, 6], [0, 0, 2], [-1, -1, -2])
Dnary_ops_test.py162 self._testNAry(lambda x: array_ops.strided_slice(*x),
171 lambda x: array_ops.strided_slice(*x), [
178 self._testNAry(lambda x: array_ops.strided_slice(*x),
186 self._testNAry(lambda x: array_ops.strided_slice(*x),
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc30 namespace strided_slice { namespace
87 using ::tflite::strided_slice::StartForAxis; in CheckOutputSize()
88 using ::tflite::strided_slice::StopForAxis; in CheckOutputSize()
180 return {/*init=*/strided_slice::Init, in Register_STRIDED_SLICE()
182 /*prepare=*/strided_slice::Prepare, in Register_STRIDED_SLICE()
183 /*invoke=*/strided_slice::Eval, in Register_STRIDED_SLICE()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc56 tflite::strided_slice::BuildStridedSliceParams( in StridedSlice()
61 int start_index = tflite::strided_slice::StartForAxis( in StridedSlice()
64 stop_for_axis[axis] = tflite::strided_slice::StopForAxis( in StridedSlice()
89 if (!tflite::strided_slice::LoopCondition(src_coord[axis], stop, in StridedSlice()
93 src_coord[axis] = tflite::strided_slice::StartForAxis( in StridedSlice()
Dpropagate_fixed_sizes.cc1697 tflite::strided_slice::BuildStridedSliceParams( in ProcessStridedSliceOperator()
1700 int start_index = tflite::strided_slice::StartForAxis( in ProcessStridedSliceOperator()
1702 int stop_index = tflite::strided_slice::StopForAxis( in ProcessStridedSliceOperator()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1769 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1770 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1775 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1776 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1777 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1778 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1779 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1780 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1781 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
1782 strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf.float32,…
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dstrided_slice.h34 using strided_slice::LoopCondition; in StridedSlice()
35 using strided_slice::StartForAxis; in StridedSlice()
36 using strided_slice::StopForAxis; in StridedSlice()
52 strided_slice::StridedSlicePadIndices(&params_copy, 5); in StridedSlice()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic_test.cc40 strided_slice::StridedSlicePadIndices(&op_params, 4); in RunStridedSlicePadIndices()
Dstrided_slice_logic.h26 namespace strided_slice {
DBUILD508 "reference/strided_slice.h",
617 "reference/strided_slice.h",
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dstrided_slice.py60 out = tf.strided_slice(
/external/tensorflow/tensorflow/python/ops/signal/
Dshape_ops.py194 subframes = array_ops.reshape(array_ops.strided_slice(
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp137 "kernels/strided_slice.cc",
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
DBUILD140 "//tensorflow/lite/delegates/gpu/common/tasks:strided_slice",
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
DBUILD878 name = "strided_slice",
879 srcs = ["strided_slice.cc"],
880 hdrs = ["strided_slice.h"],
895 ":strided_slice",
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples_lib.py140 from tensorflow.lite.testing.op_tests.strided_slice import make_strided_slice_tests, make_strided_s…
DBUILD641 "strided_slice",
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1206 func @strided_slice(%arg0: tensor<12x2x2x5xf32>, %arg1: tensor<1xi32>, %arg2: tensor<1xi32>, %arg3:…
1209 // CHECK-LABEL: strided_slice
1210 …// CHECK: "tfl.strided_slice"(%arg0, %arg1, %arg2, %arg3) {begin_mask = 0 : i32, ellipsis_mask = 0…
1223 …// CHECK-NEXT: "tfl.strided_slice"(%arg0, [[BEGIN]], [[END]], [[STRIDES]]) {begin_mask = 0 : i32, …
1230 …// CHECK: "tfl.strided_slice"(%arg0, %arg1, %arg2, %arg3) {begin_mask = 0 : i32, ellipsis_mask = 0…
1240 …// CHECK-NEXT: "tfl.strided_slice"(%arg0, [[BEGIN]], [[END]], [[STRIDES]]) {begin_mask = 0 : i32, …
1250 …// CHECK-NEXT: "tfl.strided_slice"(%arg0, [[BEGIN]], [[END]], [[STRIDES]]) {begin_mask = 0 : i32, …
1263 …// CHECK-NEXT: "tfl.strided_slice"(%arg0, [[BEGIN]], [[END]], [[STRIDES]]) {begin_mask = 0 : i32, …
1270 // CHECK-NOT: "tfl.strided_slice"
1277 // CHECK-NOT: "tfl.strided_slice"
Dfuse-tftext.mlir675 …ggedFromNestedRowSplits/RaggedFromRowSplits/RowPartitionFromRowSplits/strided_slice:0) = "> : tens…
705 …<"y (WhitespaceTokenize/RaggedFromNestedRowSplits/RaggedFromRowSplits/strided_slice:0) = "> : tens…
719 …edFromNestedRowSplits/RaggedFromRowSplits_1/RowPartitionFromRowSplits/strided_slice:0) = "> : tens…
748 …x (WhitespaceTokenize/RaggedFromNestedRowSplits/RaggedFromRowSplits_1/strided_slice:0) = "> : tens…
763 …edFromNestedRowSplits_1/RaggedFromRowSplits/RowPartitionFromRowSplits/strided_slice:0) = "> : tens…
793 …y (WhitespaceTokenize/RaggedFromNestedRowSplits_1/RaggedFromRowSplits/strided_slice:0) = "> : tens…
807 …edFromNestedRowSplits_2/RaggedFromRowSplits/RowPartitionFromRowSplits/strided_slice:0) = "> : tens…
837 …y (WhitespaceTokenize/RaggedFromNestedRowSplits_2/RaggedFromRowSplits/strided_slice:0) = "> : tens…
1807 …ggedFromNestedRowSplits/RaggedFromRowSplits/RowPartitionFromRowSplits/strided_slice:0) = "> : tens…
1837 …nize/WhitespaceTokenize/RaggedFromNestedRowSplits/RaggedFromRowSplits/strided_slice:0) = "> : tens…
[all …]
/external/tensorflow/tensorflow/lite/micro/tools/make/
DMakefile356 tensorflow/lite/micro/kernels/strided_slice.cc \
443 tensorflow/lite/kernels/internal/reference/strided_slice.h \
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py305 return None, None, None, None, array_ops.strided_slice(
334 dy = Apply(array_ops.strided_slice,
Dcheck_ops.py1964 …diff = lambda: array_ops.strided_slice(x, [1], [1] + s_len)- array_ops.strided_slice(x, [0], s_len)
Darray_ops.py1040 return strided_slice(
1113 def strided_slice(input_, function
1213 op = gen_array_ops.strided_slice(

123