Home
last modified time | relevance | path

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

123

/external/armnn/src/armnn/
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()
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()
/external/armnn/src/backends/reference/workloads/
DLogSoftmax.cpp18 inline bool ValidateAxis(int axis, unsigned int numDimensions) in ValidateAxis()
34 const unsigned int numDimensions = inputInfo.GetNumDimensions(); in LogSoftmax() local
/external/armnn/src/backends/cl/workloads/
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
DClArgMinMaxWorkload.cpp21 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int axisIndex) in CalcAclAxis()
DClSplitterWorkload.cpp24 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()
/external/armnn/src/backends/neon/workloads/
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
DNeonArgMinMaxWorkload.cpp21 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int axisIndex) in CalcAclAxis()
DNeonSplitterWorkload.cpp25 unsigned int CalcAclAxis(unsigned int numDimensions, unsigned int splitAxis) in CalcAclAxis()
/external/ow2-asm/asm-tree/src/main/java/org/objectweb/asm/tree/
DMultiANewArrayInsnNode.java53 public MultiANewArrayInsnNode(final String descriptor, final int numDimensions) { in MultiANewArrayInsnNode()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_tensor.i84 auto numDimensions = $self->GetNumDimensions(); in __str__() local
/external/armnn/src/armnn/optimizations/
DPermuteAsReshape.hpp44 const unsigned int numDimensions = permutation.GetSize(); in IsReshape() local
DAddBroadcastReshapeLayer.hpp53 uint32_t numDimensions = inputInfo.GetNumDimensions(); in Run() local
DTransposeAsReshape.hpp44 const unsigned int numDimensions = permutation.GetSize(); in IsReshape() local
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DShape.java67 public int numDimensions() { in numDimensions() method in Shape
/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DTensor.java80 int numDimensions(); in numDimensions() method
/external/ow2-asm/asm-commons/src/main/java/org/objectweb/asm/commons/
DMethodRemapper.java200 public void visitMultiANewArrayInsn(final String descriptor, final int numDimensions) { in visitMultiANewArrayInsn()
201 super.visitMultiANewArrayInsn(remapper.mapDesc(descriptor), numDimensions); in visitMultiANewArrayInsn() local
DCodeSizeEvaluator.java202 public void visitMultiANewArrayInsn(final String descriptor, final int numDimensions) { in visitMultiANewArrayInsn()
DAdviceAdapter.java425 public void visitMultiANewArrayInsn(final String descriptor, final int numDimensions) { in visitMultiANewArrayInsn()
/external/armnn/delegate/common/src/
DDelegateUtils.hpp83 unsigned int ComputeWrappedIndex(int index, unsigned int numDimensions) in ComputeWrappedIndex()
/external/ow2-asm/asm/src/main/java/org/objectweb/asm/
DType.java241 final int numDimensions = getDimensions(); in getElementType() local
656 int numDimensions = 1; in getDimensions() local
/external/ow2-asm/asm-analysis/src/main/java/org/objectweb/asm/tree/analysis/
DSimpleVerifier.java252 int numDimensions = 0; in merge() local
/external/pytorch/aten/src/ATen/native/quantized/
DFakeQuantPerChannelAffine.cpp217 auto numDimensions = X.ndimension(); in _fake_quantize_learnable_per_channel_affine_backward() local
/external/armnn/src/armnn/layers/
DSplitterLayer.cpp65 unsigned int numDimensions = desc.GetNumDimensions(); in CreateTensors() local
/external/ow2-asm/asm-analysis/src/test/java/org/objectweb/asm/tree/analysis/
DMethodNodeBuilder.java86 MethodNodeBuilder multiANewArrayInsn(final String descriptor, final int numDimensions) { in multiANewArrayInsn()

123