Home
last modified time | relevance | path

Searched refs:StridedSlice (Results 1 – 25 of 128) sorted by relevance

123456

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dstrided_slice.h28 class StridedSlice : public GPUOperation {
30 StridedSlice(const OperationDef& definition, const SliceAttributes& attr);
35 StridedSlice(StridedSlice&& operation);
36 StridedSlice& operator=(StridedSlice&& operation);
37 StridedSlice(const StridedSlice&) = delete;
38 StridedSlice& operator=(const StridedSlice&) = delete;
46 StridedSlice CreateStridedSlice(const OperationDef& definition,
H A Dstrided_slice.cc78 StridedSlice::StridedSlice(const OperationDef& definition, in StridedSlice() function in tflite::gpu::StridedSlice
85 StridedSlice::StridedSlice(StridedSlice&& operation) in StridedSlice() function in tflite::gpu::StridedSlice
88 StridedSlice& StridedSlice::operator=(StridedSlice&& operation) { in operator =()
96 std::string StridedSlice::GetStridedSliceCode(const OperationDef& op_def, in GetStridedSliceCode()
156 absl::Status StridedSlice::BindArguments(ArgumentsBinder* args) { in BindArguments()
170 int3 StridedSlice::GetGridSize() const { in GetGridSize()
177 StridedSlice CreateStridedSlice(const OperationDef& definition, in CreateStridedSlice()
179 return StridedSlice(definition, attr); in CreateStridedSlice()
H A Dstrided_slice_test_util.cc52 StridedSlice operation = CreateStridedSlice(op_def, attr); in StridedSliceTest()
54 src_tensor, std::make_unique<StridedSlice>(std::move(operation)), in StridedSliceTest()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
H A Dfuse-tftext.mlir29 …%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 …]
H A Dprepare-tf.mlir257 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
262 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_…
270 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst_0, %cst_1) {begin_mask = 6 : i64, ellipsis_mask = 1 : i64…
276 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST_0]], %[[CST_1]]) {begi…
283 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, …
288 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_…
295 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
300 …// CHECK: %[[STRIDED_SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_…
307 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask = 0 : i64,…
312 …// CHECK: %0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 15 : i64, ellipsis_mask …
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstrided_slice_op_gpu_impl.h34 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>; \
H A Dstrided_slice_op_impl.h107 functor::StridedSlice<Device, Proxy, NDIM>()( in HandleStridedSliceCase()
193 void StridedSlice<GPUDevice, T, NDIM>::operator()( \
199 extern template struct StridedSlice<GPUDevice, T, NDIM>; \
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dopens2s_gnmt_mixed_precision.golden_summary50 StridedSlice 12
106 StridedSlice 2
126 StridedSlice 1
151 StridedSlice 2
158 StridedSlice 1
207 StridedSlice 1
236 StridedSlice 2
281 StridedSlice 1
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_strided_slice.cc28 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/tests/
H A Dstrided_slice_with_same_dimension_removal.mlir16 …%StridedSlice, %ctl_7 = StridedSlice(%VariableV2, %Const, %Const_1, %Const_3) name("s1") {Index = …
17 // CHECK: StridedSlice(%[[VAR2]], {{.*}} name("s2")
18 …%StridedSlice_8, %ctl_9 = StridedSlice(%VariableV2_5, %Const, %Const_1, %Const_3) name("s2") {Inde…
19 …%Add, %ctl_10 = Add(%StridedSlice, %StridedSlice_8) name("out") {T = f32} : (tensor<3x5x2xf32>, te…
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dstrided_slice.h30 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_StridedSliceGrad.pbtxt3 summary: "Returns the gradient of `StridedSlice`."
5 Since `StridedSlice` cuts out pieces of its `input` which is size
12 shape of `StridedSlice`'s `input`.
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DStridedSlice.pbtxt2 name: "StridedSlice"
74 name: "StridedSlice"
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad_test.cc378 auto y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1}); in TEST_F()
385 y = StridedSlice(scope_, x, {2, 1, 1}, {6, 3, 3}, {2, 1, 1}, in TEST_F()
386 StridedSlice::BeginMask(1 << 1).EndMask(1 << 2)); in TEST_F()
391 y = StridedSlice(scope_, x, {0, 2, 1, 1}, {0, 6, 3, 3}, {1, 2, 1, 1}, in TEST_F()
392 StridedSlice::NewAxisMask(1 << 0)); in TEST_F()
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DStridedSliceLayer.cpp18 : LayerWithParameters(1, 1, LayerType::StridedSlice, param, name) in StridedSliceLayer()
39 return factory.CreateWorkload(LayerType::StridedSlice, descriptor, PrepInfoAndDesc(descriptor)); in CreateWorkload()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dcanonicalize.mlir629 // %dim0 = tf.StridedSlice(%shape, 0, 1, 1) // get unknown dim0 value
639 …%4 = "tf.StridedSlice"(%3, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<2xi32>, tensor<1xi32…
646 …%8 = "tf.StridedSlice"(%7, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<3xi32>, tensor<1xi32…
652 …%14 = "tf.StridedSlice"(%7, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<3xi32>, tensor<1xi3…
658 …%16 = "tf.StridedSlice"(%3, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<2xi32>, tensor<1xi3…
664 …%19 = "tf.StridedSlice"(%18, %0, %1, %1) {shrink_axis_mask = 1 : i64} : (tensor<*xi32>, tensor<1xi…
1593 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
1604 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
1615 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
1626 …%3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = …
[all …]
/aosp_15_r20/external/armnn/docs/
H A D05_02_deserializer_serializer.dox79 - StridedSlice
171 - StridedSlice
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
H A Dlegalize-tf-prefer-tf2xla.mlir73 // StridedSlice op legalizations.
90 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
H A Dlegalize-tf-no-tf2xla-fallback.mlir3198 // StridedSlice op legalizations.
3215 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
3228 // CHECK: "tf.StridedSlice"
3229 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
3250 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
3263 // CHECK: tf.StridedSlice
3264 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
3282 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
3296 %output = "tf.StridedSlice"(%input, %begin, %end, %strides)
3307 // For StridedSlice
[all …]
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DCMakeLists.txt204 StridedSlice.hpp
205 StridedSlice.cpp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/analysis/
H A Dcost_analysis.mlir90 …%9 = "tf.StridedSlice"(%8, %5, %4, %4) {begin_mask = 0 : i64, device = "/job:localhost/replica:0/t…
92 …%10 = "tf.StridedSlice"(%8, %3, %6, %4) {begin_mask = 0 : i64, device = "/job:localhost/replica:0/…
96 …%12 = "tf.StridedSlice"(%8, %6, %1, %4) {begin_mask = 0 : i64, device = "/job:localhost/replica:0/…
/aosp_15_r20/external/armnn/src/armnn/test/
H A DTestNameAndDescriptorLayerVisitor.cpp309 TEST_CASE_CHECK_LAYER_VISITOR_NAME_AND_DESCRIPTOR(StridedSlice, CheckStridedSliceLayerVisitorNameAn…
366 TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR_AND_DESCRIPTOR(StridedSlice,
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dstrided_slice.cc278 optimized_ops::StridedSlice<data_type>( \ in Eval()
282 reference_ops::StridedSlice<data_type>( \ in Eval()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Dsimple_selectors.cc189 StridedSlice operation = CreateStridedSlice(op_def, attr); in SelectStridedSlice()
190 *ptr = std::make_unique<StridedSlice>(std::move(operation)); in SelectStridedSlice()
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/java_api/
H A Dapi_def_StridedSlice.pbtxt2 graph_op_name: "StridedSlice"

123456