Home
last modified time | relevance | path

Searched defs:end_mask (Results 1 – 25 of 52) sorted by relevance

123

/external/ComputeLibrary/src/runtime/CL/functions/
DCLStridedSlice.cpp38 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()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEStridedSlice.cpp37 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()
/external/ComputeLibrary/src/core/utils/helpers/
Dtensor_transform.cpp74 int32_t end_mask, int32_t shrink_axis_mask) in calculate_end_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()
175 int32_t end_mask = 0; in construct_slice_end_mask() local
/external/ComputeLibrary/tests/validation/fixtures/
DSliceOperationsFixtures.h114 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()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEStridedSliceKernel.cpp43 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()
/external/ComputeLibrary/src/core/CL/kernels/
DCLStridedSliceKernel.cpp41 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()
/external/armnn/src/backends/neon/workloads/
DNeonStridedSliceWorkload.cpp36 int32_t end_mask = ConvertMaskToACLFormat(descriptor.m_EndMask, numDimensions); in NeonStridedSliceWorkloadValidate() local
74 int32_t end_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_EndMask, numDimensions); in NeonStridedSliceWorkload() local
/external/armnn/src/backends/cl/workloads/
DClStridedSliceWorkload.cpp42 int32_t end_mask = ConvertMaskToACLFormat(descriptor.m_EndMask, numDimensions); in ClStridedSliceWorkloadValidate() local
81 int32_t end_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_EndMask, numDimensions); in ClStridedSliceWorkload() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h125 const auto end_mask = params.end_mask; in StopForAxis() local
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc187 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
Dstrided_slice_op_test.cc110 const int32_t end_mask = 0; in BM_ValidateStridedSliceOp() local
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc568 int32_t end_mask, int32_t ellipsis_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
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dstrided_slice_builder.cc52 int end_mask = params->end_mask; in PopulateSubGraph() local
/external/ComputeLibrary/tests/validation/reference/
DSliceOperations.cpp82 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in strided_slice()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dslice_ops.cc171 const auto end_mask = std::bitset<32>(strided_slice_spec.end_dense_mask); in HandleDynamicStridedSliceInput() local
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc39 const int32 begin_mask, end_mask; member
52 int32 end_mask; member
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dstrided_slice_test.cc33 int end_mask, int shrink_axis_mask) { in StridedSliceOpModel()
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dslice_spmd_expander.cc138 uint64_t* end_mask, in GetInputOrientedData()
287 uint64_t end_mask = 0; in GetStridedSliceIntermediateLayout() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc167 int64_t begin_mask, int64_t end_mask, in CreateStridedSliceOp()
788 int64_t end_mask; in ConvertKerasLSTMLayer() local
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc160 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in TryToInferTensorOutputFromStridedSliceNode() local
Dshape_refiner.cc607 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc559 int64_t end_mask = strided_slice_op.end_mask(); in RewriteEllipsisMask() local
721 int end_mask = strided_slice_op.end_mask(); in matchAndRewrite() local
/external/cronet/net/disk_cache/blockfile/
Dblock_files.cc132 uint8_t end_mask = (0xf << (4 - bits_at_end)) & 0xf; in DeleteMapBlock() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc150 bool end_mask = (1 << i) & shape_spec.end_dense_mask; in EmitDynamicSlice() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1816 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()

123