| /external/ComputeLibrary/src/runtime/CL/functions/ |
| 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()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| 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()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| 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()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| 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()
|
| /external/ComputeLibrary/src/core/utils/helpers/ |
| D | tensor_transform.cpp | 74 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()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonStridedSliceWorkload.cpp | 37 int32_t shrink_axis_mask = ConvertMaskToACLFormat(descriptor.m_ShrinkAxisMask, numDimensions); in NeonStridedSliceWorkloadValidate() local
75 …int32_t shrink_axis_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_ShrinkAxisMask, numDimensi… in NeonStridedSliceWorkload() local
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClStridedSliceWorkload.cpp | 43 int32_t shrink_axis_mask = ConvertMaskToACLFormat(descriptor.m_ShrinkAxisMask, numDimensions); in ClStridedSliceWorkloadValidate() local
82 …int32_t shrink_axis_mask = ConvertMaskToACLFormat(m_Data.m_Parameters.m_ShrinkAxisMask, numDimensi… in ClStridedSliceWorkload() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 126 const auto shrink_axis_mask = params.shrink_axis_mask; in StopForAxis() local
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 41 const int32 new_axis_mask, shrink_axis_mask; member
75 int32 shrink_axis_mask; member
174 int32_t new_axis_mask, int32_t shrink_axis_mask, in ValidateStridedSliceOp()
426 int32_t new_axis_mask, int32_t shrink_axis_mask, in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 188 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceOp
282 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceGradOp
409 int32 ellipsis_mask, new_axis_mask, shrink_axis_mask; member in tensorflow::StridedSliceAssignOp
|
| D | strided_slice_op_test.cc | 113 const int32_t shrink_axis_mask = 0; in BM_ValidateStridedSliceOp() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad_test.cc | 570 int32_t shrink_axis_mask) { in StridedSliceGrad()
611 int32_t new_axis_mask, int32_t shrink_axis_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/ |
| D | strided_slice_builder.cc | 53 int shrink_axis_mask = params->shrink_axis_mask; in PopulateSubGraph() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | SliceOperations.cpp | 82 int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in strided_slice()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | strided_slice_test.cc | 33 int end_mask, int shrink_axis_mask) { in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | slice_spmd_expander.cc | 393 const llvm::SmallVector<int64_t, 4> shrink_axis_mask = in ApplyNewAndShrinkMasksToLayout() local
621 const llvm::SmallVector<int64_t, 4> shrink_axis_mask = in ExpandOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 169 int64_t shrink_axis_mask, OpBuilder* builder) { in CreateStridedSliceOp()
789 int64_t shrink_axis_mask; in ConvertKerasLSTMLayer() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 160 int32 begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in TryToInferTensorOutputFromStridedSliceNode() local
|
| D | shape_refiner.cc | 607 int begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask; in PartialStridedSliceShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| 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
1902 int32_t shrink_axis_mask, MutableArrayRef<int64_t> begin, in CalculateSlicedShapeFromDenseIndices()
1968 int32_t new_axis_mask, int32_t shrink_axis_mask, in CalculateSlicedShapeFromSparseIndices()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | strided_slice_test.cc | 39 int shrink_axis_mask) { in StridedSliceOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 506 uint64_t shrink_axis_mask = strided_slice_op.shrink_axis_mask(); in RewriteEllipsisMask() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 440 int64_t shrink_axis_mask; in StridedSliceGradHelper() local
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 1050 … int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) in compute_strided_slice_shape()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 401 int shrink_axis_mask; member
|