Home
last modified time | relevance | path

Searched defs:numDimensions (Results 1 – 25 of 146) sorted by relevance

123456

/aosp_15_r20/external/armnn/src/armnn/
H A DTensor.cpp30 TensorShape::TensorShape(unsigned int numDimensions, bool initDimensionsSpecificity) in TensorShape()
40 TensorShape::TensorShape(const unsigned int numDimensions, const unsigned int* const dimensionSizes) in TensorShape()
59 TensorShape::TensorShape(unsigned int numDimensions, in TensorShape()
83 auto numDimensions = static_cast<unsigned int>(dimensionSizeList.size()); in TensorShape() local
219 void TensorShape::SetNumDimensions(unsigned int numDimensions, bool initDimensionsSpecificity) in SetNumDimensions()
283 void TensorShape::CheckValidNumDimensions(unsigned int numDimensions) in CheckValidNumDimensions()
359 TensorInfo::TensorInfo(unsigned int numDimensions, in TensorInfo()
384 TensorInfo::TensorInfo(unsigned int numDimensions, in TensorInfo()
H A DDescriptors.cpp89 OriginsDescriptor::OriginsDescriptor(uint32_t numViews, uint32_t numDimensions /*= 4*/) in OriginsDescriptor()
221 ViewsDescriptor::ViewsDescriptor(uint32_t numViews, uint32_t numDimensions /*= 4*/) in ViewsDescriptor()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLogSoftmax.cpp18 inline bool ValidateAxis(int axis, unsigned int numDimensions) in ValidateAxis()
34 const unsigned int numDimensions = inputInfo.GetNumDimensions(); in LogSoftmax() local
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonStridedSliceWorkload.cpp34 auto numDimensions = armnn::numeric_cast<int>(input.GetNumDimensions()); in NeonStridedSliceWorkloadValidate() local
72 …auto numDimensions = armnn::numeric_cast<int>(info.m_InputTensorInfos[0].GetNumDimensions()); in NeonStridedSliceWorkload() local
H A DNeonArgMinMaxWorkload.cpp21 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int axisIndex) in CalcAclAxis()
H A DNeonSplitterWorkload.cpp25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClStridedSliceWorkload.cpp40 auto numDimensions = armnn::numeric_cast<int>(input.GetNumDimensions()); in ClStridedSliceWorkloadValidate() local
79 …auto numDimensions = armnn::numeric_cast<int>(info.m_InputTensorInfos[0].GetNumDimensions()); in ClStridedSliceWorkload() local
H A DClArgMinMaxWorkload.cpp21 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int axisIndex) in CalcAclAxis()
H A DClSplitterWorkload.cpp24 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()
/aosp_15_r20/external/ow2-asm/asm-commons/src/main/java/org/objectweb/asm/commons/
H A DMethodRemapper.java200 public void visitMultiANewArrayInsn(final String descriptor, final int numDimensions) { in visitMultiANewArrayInsn()
201 super.visitMultiANewArrayInsn(remapper.mapDesc(descriptor), numDimensions); in visitMultiANewArrayInsn() local
H A DInstructionAdapter.java673 public void visitMultiANewArrayInsn(final String descriptor, final int numDimensions) { in visitMultiANewArrayInsn()
1288 public void multianewarray(final String descriptor, final int numDimensions) { in multianewarray()
H A DCodeSizeEvaluator.java202 public void visitMultiANewArrayInsn(final String descriptor, final int numDimensions) { in visitMultiANewArrayInsn()
/aosp_15_r20/external/ow2-asm/asm/src/main/java/org/objectweb/asm/
H A DType.java241 final int numDimensions = getDimensions(); in getElementType() local
656 int numDimensions = 1; in getDimensions() local
/aosp_15_r20/external/ow2-asm/asm-tree/src/main/java/org/objectweb/asm/tree/
H A DMultiANewArrayInsnNode.java53 public MultiANewArrayInsnNode(final String descriptor, final int numDimensions) { in MultiANewArrayInsnNode()
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
H A Darmnn_tensor.i84 auto numDimensions = $self->GetNumDimensions(); in __str__() local
/aosp_15_r20/external/armnn/src/armnn/optimizations/
H A DPermuteAsReshape.hpp44 const unsigned int numDimensions = permutation.GetSize(); in IsReshape() local
H A DAddBroadcastReshapeLayer.hpp53 uint32_t numDimensions = inputInfo.GetNumDimensions(); in Run() local
H A DTransposeAsReshape.hpp44 const unsigned int numDimensions = permutation.GetSize(); in IsReshape() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
H A DTensor.java80 int numDimensions(); in numDimensions() method
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
H A DShape.java67 public int numDimensions() { in numDimensions() method in Shape
H A DTensor.java333 public int numDimensions() { in numDimensions() method in Tensor
720 private static int numDimensions(Object o, DataType dtype) { in numDimensions() method in Tensor
/aosp_15_r20/external/armnn/delegate/common/src/
H A DDelegateUtils.hpp83 unsigned int ComputeWrappedIndex(int index, unsigned int numDimensions) in ComputeWrappedIndex()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DConcatenation.cpp149 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare() local
/aosp_15_r20/external/ow2-asm/asm-analysis/src/main/java/org/objectweb/asm/tree/analysis/
H A DSimpleVerifier.java252 int numDimensions = 0; in merge() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DFakeQuantPerChannelAffine.cpp217 auto numDimensions = X.ndimension(); in _fake_quantize_learnable_per_channel_affine_backward() local

123456