Home
last modified time | relevance | path

Searched refs:begin_mask (Results 1 – 25 of 140) sorted by relevance

123456

/external/ComputeLibrary/src/runtime/CL/functions/
DCLStridedSlice.cpp38 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
40 …ARM_COMPUTE_LOG_PARAMS(input, output, starts, ends, strides, begin_mask, end_mask, shrink_axis_mas… in configure()
42 …k->configure(compile_context, input, output, starts, ends, strides, begin_mask, end_mask, shrink_a… in configure()
48 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate() argument
50 …return CLStridedSliceKernel::validate(input, output, starts, ends, strides, begin_mask, end_mask, … in validate()
74 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
76 …ry::get().get_compile_context(), input, output, starts, ends, strides, begin_mask, end_mask, shrin… in configure()
81 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
89 …ontext, _impl->src->info(), _impl->dst->info(), starts, ends, strides, begin_mask, end_mask, shrin… in configure()
94 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate() argument
[all …]
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEStridedSlice.cpp37 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
39 …ARM_COMPUTE_LOG_PARAMS(input, output, starts, ends, strides, begin_mask, end_mask, shrink_axis_mas… in configure()
42 k->configure(input, output, starts, ends, strides, begin_mask, end_mask, shrink_axis_mask); in configure()
48 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate() argument
50 …return NEStridedSliceKernel::validate(input, output, starts, ends, strides, begin_mask, end_mask, … in validate()
71 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
76 …_impl->op->configure(input->info(), output->info(), starts, ends, strides, begin_mask, end_mask, s… in configure()
89 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate() argument
91 …return experimental::NEStridedSlice::validate(input, output, starts, ends, strides, begin_mask, en… in validate()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1763 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1764 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1765 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1766 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1767 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1768 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=False,dtype=t…
1769 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,…
1771 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf…
1772 DISABLED_strided_slice/strided_slice_begin=[0,0,0,0],begin_mask=None,constant_indices=True,dtype=tf…
[all …]
/external/ComputeLibrary/src/core/NEON/kernels/
DNEStridedSliceKernel.cpp43 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate_arguments() argument
60 … begin_mask, end_mask, shrink_axis_mask); in validate_arguments()
76 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate_and_configure_window() argument
81 … begin_mask, end_mask, shrink_axis_mask); in validate_and_configure_window()
98 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
101 …ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input, output, starts, ends, strides, begin_mask, en… in configure()
108 begin_mask, end_mask, shrink_axis_mask); in configure()
110 … = validate_and_configure_window(input, output, starts, ends, strides, begin_mask, end_mask, shrin… in configure()
117 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate() argument
119 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(input, output, starts, ends, strides, begin_mask, e… in validate()
[all …]
DNEStridedSliceKernel.h73 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask);
91 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask);
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h62 p->begin_mask <<= pad_count; in StridedSlicePadIndices()
64 p->begin_mask |= (1 << pad_count) - 1; in StridedSlicePadIndices()
77 const auto begin_mask = params.begin_mask; in StartForAxis() local
88 if (begin_mask & 1 << axis) { in StartForAxis()
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
198 op_params.begin_mask = begin_mask; in BuildStridedSliceParams()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dstrided_slice_builder.cc51 int begin_mask = params->begin_mask; in PopulateSubGraph() local
55 begin_mask = begin_mask << (4 - original_mask_size); in PopulateSubGraph()
59 kScalarShape, reinterpret_cast<char*>(&begin_mask), sizeof(begin_mask)); in PopulateSubGraph()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLStridedSlice.h79 int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
97 int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
115 … int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
148 int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
166 … int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
/external/ComputeLibrary/tests/validation/fixtures/
DSliceOperationsFixtures.h114 int32_t begin_mask, int32_t end_mask, int32_t shrink_mask, in setup() argument
117 …_target = compute_target(shape, starts, ends, strides, begin_mask, end_mask, shrink_mask, data_… in setup()
118 …_reference = compute_reference(shape, starts, ends, strides, begin_mask, end_mask, shrink_mask, da… in setup()
130 int32_t begin_mask, int32_t end_mask, int32_t shrink_mask, in compute_target() argument
139 … strided_slice.configure(&src, &dst, starts, ends, strides, begin_mask, end_mask, shrink_mask); in compute_target()
163 int32_t begin_mask, int32_t end_mask, int32_t shrink_mask, in compute_reference() argument
172 … return reference::strided_slice(src, starts, ends, strides, begin_mask, end_mask, shrink_mask); in compute_reference()
/external/ComputeLibrary/src/core/CL/kernels/
DCLStridedSliceKernel.cpp41 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate_arguments() argument
58 … begin_mask, end_mask, shrink_axis_mask); in validate_arguments()
80 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure() argument
84 …ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input, output, starts, ends, strides, begin_mask, en… in configure()
94 begin_mask, end_mask, shrink_axis_mask); in configure()
99 … begin_mask, end_mask, shrink_axis_mask); in configure()
160 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate() argument
162 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(input, output, starts, ends, strides, begin_mask, e… in validate()
DCLStridedSliceKernel.h58 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask);
76 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask);
/external/ComputeLibrary/tests/validation/reference/
DSliceOperations.cpp82 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in strided_slice() argument
100 …= compute_strided_slice_output_shape(src_shape, starts, ends, strides, begin_mask, end_mask, shrin… in strided_slice()
111 … begin_mask, end_mask, shrink_axis_mask); in strided_slice()
118 … begin_mask, end_mask, shrink_axis_mask, true)); in strided_slice()
134 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask);
137 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask);
/external/ComputeLibrary/src/core/utils/helpers/
Dtensor_transform.cpp39 …ex(TensorShape input_shape, int index, Coordinates starts, Coordinates strides, int32_t begin_mask) in calculate_start_on_index() argument
54 if(arm_compute::helpers::bit_ops::is_bit_set(begin_mask, index)) in calculate_start_on_index()
122 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in calculate_strided_slice_coords() argument
130 const int start_i = calculate_start_on_index(input_shape, i, starts, strides, begin_mask); in calculate_strided_slice_coords()
140 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask, bool return_unshrinked) in compute_strided_slice_output_shape() argument
148 const int start = calculate_start_on_index(input_shape, i, starts, strides, begin_mask); in compute_strided_slice_output_shape()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
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 …]
Dprepare-tf.mlir257 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
262 …SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 3 : i64, ellip…
270 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst_0, %cst_1) {begin_mask = 6 : i64, ellipsis_mask = 1 : i64…
276 …ICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST_0]], %[[CST_1]]) {begin_mask = 7 : i64, ellip…
283 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 2 : i64, …
288 …SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 6 : i64, ellip…
295 …%0 = "tf.StridedSlice"(%arg0, %cst, %cst, %cst_0) {begin_mask = 0 : i64, ellipsis_mask = 1 : i64, …
300 …SLICE:.*]] = "tf.StridedSlice"(%arg0, %[[CST]], %[[CST]], %[[CST_0]]) {begin_mask = 0 : i64, ellip…
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 …]
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEStridedSlice.h76 int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
94 … int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
126 int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
144 … int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
/external/ComputeLibrary/arm_compute/core/utils/helpers/
Dtensor_transform.h55 …x(TensorShape input_shape, int index, Coordinates starts, Coordinates strides, int32_t begin_mask);
91 … int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0);
113 … int32_t begin_mask = 0, int32_t end_mask = 0, int32_t shrink_axis_mask = 0,
/external/armnn/src/backends/neon/workloads/
DNeonStridedSliceWorkload.cpp35 int32_t begin_mask = ConvertMaskToACLFormat(descriptor.m_BeginMask, numDimensions); in NeonStridedSliceWorkloadValidate() local
44 begin_mask, in NeonStridedSliceWorkloadValidate()
73 … int32_t begin_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_BeginMask, numDimensions); in NeonStridedSliceWorkload() local
88 begin_mask, in NeonStridedSliceWorkload()
/external/armnn/src/backends/cl/workloads/
DClStridedSliceWorkload.cpp41 int32_t begin_mask = ConvertMaskToACLFormat(descriptor.m_BeginMask, numDimensions); in ClStridedSliceWorkloadValidate() local
50 begin_mask, in ClStridedSliceWorkloadValidate()
80 … int32_t begin_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_BeginMask, numDimensions); in ClStridedSliceWorkload() local
96 begin_mask, in ClStridedSliceWorkload()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc88 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
108 context->input(0).shape(), begin_mask, end_mask, in Compute()
187 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
196 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
235 input_shape, begin_mask, end_mask, ellipsis_mask, new_axis_mask, in Compute()
281 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
290 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceAssignOp()
347 old_lhs->shape(), begin_mask, end_mask, ellipsis_mask, in Compute()
408 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc39 const int32 begin_mask, end_mask; member
51 int32 begin_mask; member
90 dense->begin_mask = 0; in BuildDenseSpec()
118 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
143 if (sparse.begin_mask & (1 << i)) { in BuildDenseSpec()
144 dense->begin_mask |= (1 << full_index); in BuildDenseSpec()
292 {dense_spec.begin_mask & (1 << i), dense_spec.end_mask & (1 << i)}}; in ValidateStridedSliceOp()
314 (dense_spec.begin_mask & (1 << i)) && (dense_spec.end_mask & (1 << i)); in ValidateStridedSliceOp()
394 shape_spec->begin_dense_mask = dense_spec.begin_mask; in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc567 const Tensor& dy, int32_t begin_mask, in StridedSliceGrad() argument
584 {"begin_mask", begin_mask}, in StridedSliceGrad()
610 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in StridedSliceGradGrad() argument
627 {"begin_mask", begin_mask}, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
667 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
679 begin_mask, end_mask, ellipsis_mask, in TEST()
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
694 StridedSliceGrad(x, start, stop, strides, dy, begin_mask, end_mask, in TEST()
706 begin_mask, end_mask, ellipsis_mask, in TEST()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice.cc77 op_params.begin_mask = 0; in BuildStridedSliceParams()
137 op_params.begin_mask |= (1 << i); in BuildStridedSliceParams()
154 op_params.begin_mask |= (1 << i); in BuildStridedSliceParams()
163 if (op_context->params->begin_mask & (1 << orig_idx)) { in BuildStridedSliceParams()
164 op_params.begin_mask |= (1 << i); in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StridedSlice.pbtxt9 `begin_mask` then `begin[k]` is ignored and the full range of the
31 name: "begin_mask"
42 analogous to `begin_mask`
102 the second to last. `begin_mask` controls whether to replace the explicitly
126 begin_mask = 1<<4 | 1<<5 = 48
155 The appropriate begin_mask bit is set to indicate the start range is the
160 receive 0, 0, and 1, respectively. The appropriate bits in `begin_mask` and
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DStridedSlice.pbtxt38 name: "begin_mask"
111 name: "begin_mask"

123456