| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | strided_slice.h | 26 class StridedSlice : public GPUOperation {
28 StridedSlice(const OperationDef& definition, const SliceAttributes& attr);
33 StridedSlice(StridedSlice&& operation);
34 StridedSlice& operator=(StridedSlice&& operation);
35 StridedSlice(const StridedSlice&) = delete;
36 StridedSlice& operator=(const StridedSlice&) = delete;
44 StridedSlice CreateStridedSlice(const OperationDef& definition,
|
| D | strided_slice.cc | 77 StridedSlice::StridedSlice(const OperationDef& definition, in StridedSlice() function in tflite::gpu::StridedSlice
84 StridedSlice::StridedSlice(StridedSlice&& operation) in StridedSlice() function in tflite::gpu::StridedSlice
87 StridedSlice& StridedSlice::operator=(StridedSlice&& operation) { in operator =()
95 std::string StridedSlice::GetStridedSliceCode(const OperationDef& op_def, in GetStridedSliceCode()
155 absl::Status StridedSlice::BindArguments(ArgumentsBinder* args) { in BindArguments()
169 int3 StridedSlice::GetGridSize() const { in GetGridSize()
176 StridedSlice CreateStridedSlice(const OperationDef& definition, in CreateStridedSlice()
178 return StridedSlice(definition, attr); in CreateStridedSlice()
|
| D | strided_slice_test_util.cc | 51 StridedSlice operation = CreateStridedSlice(op_def, attr); in StridedSliceTest()
53 src_tensor, absl::make_unique<StridedSlice>(std::move(operation)), in StridedSliceTest()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op_gpu_impl.h | 34 template struct functor::StridedSlice<GPUDevice, T, 1>; \
35 template struct functor::StridedSlice<GPUDevice, T, 2>; \
36 template struct functor::StridedSlice<GPUDevice, T, 3>; \
37 template struct functor::StridedSlice<GPUDevice, T, 4>; \
38 template struct functor::StridedSlice<GPUDevice, T, 5>; \
39 template struct functor::StridedSlice<GPUDevice, T, 6>; \
40 template struct functor::StridedSlice<GPUDevice, T, 7>; \
41 template struct functor::StridedSlice<GPUDevice, T, 8>; \
|
| D | strided_slice_op_impl.h | 108 functor::StridedSlice<Device, Proxy, NDIM>()( in HandleStridedSliceCase()
191 void StridedSlice<GPUDevice, T, NDIM>::operator()( \
197 extern template struct StridedSlice<GPUDevice, T, NDIM>; \
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_strided_slice.cc | 28 void StridedSlice(StridedSliceOperator const& op, Array const& input_array, in StridedSlice() function
145 StridedSlice<ArrayDataType::kFloat>(*op, input_array, &output_array); in Run()
148 StridedSlice<ArrayDataType::kUint8>(*op, input_array, &output_array); in Run()
151 StridedSlice<ArrayDataType::kInt32>(*op, input_array, &output_array); in Run()
154 StridedSlice<ArrayDataType::kInt64>(*op, input_array, &output_array); in Run()
157 StridedSlice<ArrayDataType::kComplex64>(*op, input_array, &output_array); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | fuse-tftext.mlir | 29 …%25 = "tf.StridedSlice"(%24#0, %15, %5, %16) {begin_mask = 1 : i64, device = "", ellipsis_mask = 0…
31 …%27 = "tf.StridedSlice"(%24#0, %16, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
34 …%30 = "tf.StridedSlice"(%29#0, %5, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0…
41 …%37 = "tf.StridedSlice"(%36#1, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
46 …%42 = "tf.StridedSlice"(%36#1, %16, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
47 …%43 = "tf.StridedSlice"(%36#1, %15, %5, %16) {begin_mask = 1 : i64, device = "", ellipsis_mask = 0…
54 …%50 = "tf.StridedSlice"(%49, %5, %15, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 :…
56 …%52 = "tf.StridedSlice"(%51, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 …
63 …%59 = "tf.StridedSlice"(%58, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = 0 …
65 …%61 = "tf.StridedSlice"(%36#4, %15, %16, %16) {begin_mask = 0 : i64, device = "", ellipsis_mask = …
[all …]
|
| D | prepare-tf.mlir | 459 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
464 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_…
472 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst_0, %cst_1) {begin_mask = 6 : i64, ellipsis_mask = 1 : i64…
478 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST_0]], %[[CST_1]]) {begi…
485 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, …
490 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_…
497 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
502 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_…
509 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask = 0 : i64,…
514 …// CHECK: %0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask …
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | strided_slice.h | 30 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
97 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
103 StridedSlice<T>(op_params, unextended_input_shape, unextended_output_shape, in StridedSlice()
108 inline void StridedSlice(const tflite::StridedSliceParams& op_params, in StridedSlice() function
114 StridedSlice<T>(op_params, unextended_input_shape, unextended_output_shape, in StridedSlice()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_StridedSliceGrad.pbtxt | 3 summary: "Returns the gradient of `StridedSlice`."
5 Since `StridedSlice` cuts out pieces of its `input` which is size
12 shape of `StridedSlice`'s `input`.
|
| D | api_def_TensorStridedSliceUpdate.pbtxt | 7 `strides` etc. work exactly as in `StridedSlice`.
|
| D | api_def_StridedSliceAssign.pbtxt | 7 `begin`, `end`, `strides`, etc. work exactly as in `StridedSlice`.
|
| D | api_def_ResourceStridedSliceAssign.pbtxt | 7 `begin, `end`, `strides`, etc. work exactly as in `StridedSlice`.
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | opens2s_gnmt_mixed_precision.golden_summary | 51 StridedSlice 16
105 StridedSlice 1
149 StridedSlice 2
155 StridedSlice 1
203 StridedSlice 1
273 StridedSlice 1
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | reshape_vectorizer.cc | 40 ops::StridedSlice(*s, shape, const_vec_1, const_vec_1, const_vec_1, in GetVectorizedShape()
41 ops::StridedSlice::Attrs().BeginMask(1)); in GetVectorizedShape()
|
| D | cwise_op_vectorizer.cc | 90 Output concat_pre = ops::StridedSlice( in ExpandDimsForBroadcast()
92 ops::StridedSlice::Attrs().BeginMask(1)); in ExpandDimsForBroadcast()
95 Output concat_post = ops::StridedSlice( in ExpandDimsForBroadcast()
97 ops::StridedSlice::Attrs().EndMask(1)); in ExpandDimsForBroadcast()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice.cc | 150 reference_ops::StridedSlice(op_params, in Eval()
157 reference_ops::StridedSlice( in Eval()
164 reference_ops::StridedSlice(op_params, in Eval()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_StridedSlice.pbtxt | 2 graph_op_name: "StridedSlice"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_StridedSlice.pbtxt | 2 graph_op_name: "StridedSlice"
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | strided_slice_test.cc | 30 TEST_F(OpenCLOperationTest, StridedSlice) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 370 auto y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1}); in TEST_F()
377 y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1}, in TEST_F()
378 StridedSlice::BeginMask(1 << 1).EndMask(1 << 2)); in TEST_F()
383 y = StridedSlice(scope_, x, {0, 2, 1, 1}, {0, 6, 3, 3}, {1, 2, 1, 1}, in TEST_F()
384 StridedSlice::NewAxisMask(1 << 0)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | StridedSlice.pbtxt | 2 name: "StridedSlice"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | StridedSlice.pbtxt | 2 name: "StridedSlice"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice.cc | 197 optimized_ops::StridedSlice<data_type>( \ in Eval()
201 reference_ops::StridedSlice<data_type>( \ in Eval()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | canonicalize.mlir | 546 // %dim0 = tf.StridedSlice(%shape, 0, 1, 1) // get unknown dim0 value
556 …%4 = "tf.StridedSlice"(%3, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<2xi32>, tensor<1xi32…
563 …%8 = "tf.StridedSlice"(%7, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<3xi32>, tensor<1xi32…
569 …%14 = "tf.StridedSlice"(%7, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<3xi32>, tensor<1xi3…
575 …%16 = "tf.StridedSlice"(%3, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<2xi32>, tensor<1xi3…
581 …%19 = "tf.StridedSlice"(%18, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<*xi32>, tensor<1xi…
1420 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
1431 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
1442 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
1453 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
[all …]
|