| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLStridedSlice.cpp | 38 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
48 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate()
74 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
81 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
94 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEStridedSlice.cpp | 37 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
48 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate()
71 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
89 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NEStridedSliceKernel.cpp | 43 int32_t begin_mask, int32_t end_mask, int32_t 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()
98 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
117 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | SliceOperationsFixtures.h | 114 int32_t begin_mask, int32_t end_mask, int32_t shrink_mask, in setup()
130 int32_t begin_mask, int32_t end_mask, int32_t shrink_mask, in compute_target()
163 int32_t begin_mask, int32_t end_mask, int32_t shrink_mask, in compute_reference()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_grad_test.cc | 567 const Tensor& dy, int32_t begin_mask, in StridedSliceGrad()
610 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in StridedSliceGradGrad()
664 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
691 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 0, in TEST() local
719 int begin_mask = 4, end_mask = 2, new_axis_mask = 8, shrink_axis_mask = 1, in TEST() local
747 int begin_mask = 0, end_mask = 0, new_axis_mask = 0, shrink_axis_mask = 1, in TEST() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLStridedSliceKernel.cpp | 41 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate_arguments()
80 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in configure()
160 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/utils/helpers/ |
| H A D | tensor_transform.cpp | 39 …ex(TensorShape input_shape, int index, Coordinates starts, Coordinates strides, int32_t begin_mask) in calculate_start_on_index()
122 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | strided_slice_op.cc | 187 int32 begin_mask, end_mask; member in tensorflow::StridedSliceOp
281 int32 begin_mask, end_mask; member in tensorflow::StridedSliceGradOp
408 int32 begin_mask, end_mask; member in tensorflow::StridedSliceAssignOp
|
| H A D | strided_slice_op_test.cc | 109 const int32_t begin_mask = 0; in BM_ValidateStridedSliceOp() local
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonStridedSliceWorkload.cpp | 35 int32_t begin_mask = ConvertMaskToACLFormat(descriptor.m_BeginMask, numDimensions); in NeonStridedSliceWorkloadValidate() local
73 … int32_t begin_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_BeginMask, numDimensions); in NeonStridedSliceWorkload() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClStridedSliceWorkload.cpp | 41 int32_t begin_mask = ConvertMaskToACLFormat(descriptor.m_BeginMask, numDimensions); in ClStridedSliceWorkloadValidate() local
80 … int32_t begin_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_BeginMask, numDimensions); in ClStridedSliceWorkload() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | strided_slice_logic.h | 77 const auto begin_mask = params.begin_mask; in StartForAxis() local
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 39 const int32 begin_mask, end_mask; member
51 int32 begin_mask; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | slice_spmd_expander.cc | 136 uint64_t* begin_mask, in GetInputOrientedData()
286 uint64_t begin_mask = 0; in GetStridedSliceIntermediateLayout() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 149 bool begin_mask = (1 << i) & shape_spec.begin_dense_mask; in EmitDynamicSlice() local
480 bool begin_mask = (1 << i) & shape_spec.begin_dense_mask; in CompileAsDynamicUpdateSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | lstm_utils.cc | 167 int64_t begin_mask, int64_t end_mask, in CreateStridedSliceOp()
787 int64_t begin_mask; in ConvertKerasLSTMLayer() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | strided_slice_builder.cc | 51 int begin_mask = params->begin_mask; in PopulateSubGraph() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | SliceOperations.cpp | 82 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in strided_slice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | slice_ops.cc | 170 const auto begin_mask = std::bitset<32>(strided_slice_spec.begin_dense_mask); in HandleDynamicStridedSliceInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | strided_slice_test.cc | 32 std::initializer_list<int> strides_data, int begin_mask, in StridedSliceOpModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | prepare_tf.cc | 558 int64_t begin_mask = strided_slice_op.begin_mask(); in RewriteEllipsisMask() local
720 int begin_mask = strided_slice_op.begin_mask(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 1816 int32_t begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; member
1827 int32_t begin_mask, end_mask, shrink_axis_mask; member
1901 MutableArrayRef<int64_t> input_shape, int32_t begin_mask, int32_t end_mask, in CalculateSlicedShapeFromDenseIndices()
1967 int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, in CalculateSlicedShapeFromSparseIndices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | eval_const_tensor.cc | 160 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in TryToInferTensorOutputFromStridedSliceNode() local
|
| H A D | shape_refiner.cc | 607 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | Types.h | 1938 : _begin_mask(begin_mask), _end_mask(end_mask), _shrink_axis_mask(shrink_axis_mask) in _begin_mask() argument
1943 int32_t begin_mask() const in begin_mask() function
|