| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | Dimensions.h | 143 unsigned int num_dimensions() const in num_dimensions() function
149 void set_num_dimensions(size_t num_dimensions) in set_num_dimensions()
|
| H A D | TensorInfo.h | 228 size_t num_dimensions() const override in num_dimensions() function
|
| H A D | SubTensorInfo.h | 151 size_t num_dimensions() const override in num_dimensions() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/helpers/ |
| H A D | WindowHelpers.cpp | 241 const auto num_dimensions = std::max(src0.num_dimensions(), src1.num_dimensions()); in calculate_squashed_or_max_window() local
290 const auto num_dimensions = src.num_dimensions(); in calculate_squashed_or_max_window() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/ |
| H A D | ICLKernel.h | 155 …ed int &idx, const ICLArray<T> *array, const Strides &strides, unsigned int num_dimensions, const … in add_1D_array_argument()
496 …signed &idx, const ICLArray<T> *array, const Strides &strides, unsigned int num_dimensions, const … in add_array_argument()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | tile.cc | 41 int num_dimensions) { in MultiplyShapeDims()
61 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
|
| H A D | gather.cc | 103 const int num_dimensions = in Prepare() local
|
| H A D | reshape.cc | 108 int num_dimensions = params->num_dimensions; in GetOutputShapeFromParam() local
|
| H A D | topk_v2.cc | 48 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | random_grad.py | 29 def add_leading_unit_dimensions(x, num_dimensions): # pylint: disable=invalid-name argument
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/core/ |
| H A D | one_dim_grouping_aggregator.h | 127 int num_dimensions = tensor->shape().dim_sizes().size(); in AggregateTensors() local
|
| /aosp_15_r20/external/eigen/test/ |
| H A D | num_dimensions.cpp | 63 EIGEN_DECLARE_TEST(num_dimensions) in EIGEN_DECLARE_TEST() argument
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Unstack.cpp | 38 …rdinates expand_coordinates(Coordinates in_coord, size_t axis, size_t slice, size_t num_dimensions) in expand_coordinates()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | reshape_builder.cc | 41 int num_dimensions = params->num_dimensions; in PopulateShapeFromParam() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/ |
| H A D | TensorAllocator.cpp | 48 for(size_t num_dimensions = child_dims; num_dimensions > 0; --num_dimensions) in validate_subtensor_shape() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | FullyConnectedLayer.cpp | 61 unsigned int num_dimensions = input_descriptor.shape.num_dimensions(); in compute_weights_descriptor() local
|
| /aosp_15_r20/art/runtime/mirror/ |
| H A D | array.cc | 86 int num_dimensions = dimensions->GetLength(); in CreateMultiArray() local
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| H A D | image_classifier.cc | 222 const int num_dimensions = output_tensor->dims->size; in CheckAndSetOutputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_runner.cc | 350 const int num_dimensions = window.dimensions_size(); in GetGpuConvConfig() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 188 inline int ReverseMaskBits(int mask, int num_dimensions) { in ReverseMaskBits()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | post_quantize.cc | 242 const int num_dimensions = input_tensor.getType().getRank(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | light_outside_compilation.cc | 377 int num_dimensions = allocated_shape.dims(); in PopulateMetadataBufferIfNeeded() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 3272 ArrayRef<int64_t> output_shape, int num_dimensions, in ComputePermutation()
3309 const int num_dimensions = input_tensor.getType().getRank(); in fold() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | builtin_op_data.h | 336 int num_dimensions; member
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | builtin_op_data.h | 344 int num_dimensions; member
|