Home
last modified time | relevance | path

Searched defs:num_dimensions (Results 1 – 21 of 21) sorted by relevance

/external/ComputeLibrary/arm_compute/core/
DDimensions.h122 unsigned int num_dimensions() const in num_dimensions() function
128 void set_num_dimensions(size_t num_dimensions) in set_num_dimensions()
DTensorInfo.h253 size_t num_dimensions() const override in num_dimensions() function
DSubTensorInfo.h144 size_t num_dimensions() const override in num_dimensions() function
/external/tensorflow/tensorflow/python/ops/
Drandom_grad.py33 def add_leading_unit_dimensions(x, num_dimensions): # pylint: disable=invalid-name argument
/external/ComputeLibrary/src/core/CL/
DICLKernel.h102 …ed int &idx, const ICLArray<T> *array, const Strides &strides, unsigned int num_dimensions, const … in add_1D_array_argument()
376 …signed &idx, const ICLArray<T> *array, const Strides &strides, unsigned int num_dimensions, const … in add_array_argument()
/external/tensorflow/tensorflow/lite/kernels/
Dtile.cc41 int num_dimensions) { in MultiplyShapeDims()
61 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
Dreshape.cc108 int num_dimensions = params->num_dimensions; in GetOutputShapeFromParam() local
Dgather.cc103 const int num_dimensions = in Prepare() local
Dtopk_v2.cc48 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
/external/eigen/test/
Dnum_dimensions.cpp63 EIGEN_DECLARE_TEST(num_dimensions) in EIGEN_DECLARE_TEST() argument
/external/ComputeLibrary/src/graph/nodes/
DFullyConnectedLayer.cpp52 unsigned int num_dimensions = input_descriptor.shape.num_dimensions(); in compute_weights_descriptor() local
/external/ComputeLibrary/tests/validation/reference/
DUnstack.cpp38 …rdinates expand_coordinates(Coordinates in_coord, size_t axis, size_t slice, size_t num_dimensions) in expand_coordinates()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dreshape_builder.cc41 int num_dimensions = params->num_dimensions; in PopulateShapeFromParam() local
/external/ComputeLibrary/src/runtime/
DTensorAllocator.cpp49 for(size_t num_dimensions = child_dims; num_dimensions > 0; --num_dimensions) in validate_subtensor_shape() local
/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dimage_classifier.cc222 const int num_dimensions = output_tensor->dims->size; in CheckAndSetOutputs() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc364 const int num_dimensions = window.dimensions_size(); in GetGpuConvConfig() local
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h336 int num_dimensions; member
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc3067 ArrayRef<int64_t> output_shape, int num_dimensions, in ComputePermutation()
3103 const int num_dimensions = input_tensor.getType().getRank(); in fold() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc2116 int num_dimensions = params->num_dimensions; in Validate() local
3150 int num_dimensions = params->num_dimensions; in Map() local
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc88 size_t num_dimensions = flat_vector->size(); in FlatBufferIntVectorToArray() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1148 const int64_t num_dimensions = lengths.size(); in ComputeStrides() local