Home
last modified time | relevance | path

Searched defs:numDims (Results 1 – 25 of 35) sorted by relevance

12

/external/armnn/src/backends/reference/workloads/
DBroadcast.cpp14 const unsigned int numDims = GetNumDimensions(); in BroadcastLoop() local
36 const unsigned int numDims = GetNumDimensions(); in BroadcastLoop() local
DReduce.cpp19 bool NextIndex(const unsigned int numDims, const armnn::TensorShape& dims, std::vector<unsigned int… in NextIndex()
40 unsigned int ReducedOutputOffset(const unsigned int numDims, in ReducedOutputOffset()
DDebug.cpp30 const unsigned int numDims = inputInfo.GetNumDimensions(); in PrintOutput() local
DSlice.cpp21 const unsigned int numDims = inputShape.GetNumDimensions(); in Slice() local
/external/armnn/src/backends/cl/workloads/
DClArgMinMaxWorkload.cpp38 auto numDims = input.GetNumDimensions(); in ClArgMinMaxWorkloadValidate() local
69 auto numDims = info.m_InputTensorInfos[0].GetNumDimensions(); in ClArgMinMaxWorkload() local
/external/armnn/src/backends/neon/workloads/
DNeonArgMinMaxWorkload.cpp38 auto numDims = input.GetNumDimensions(); in NeonArgMinMaxWorkloadValidate() local
68 auto numDims = info.m_InputTensorInfos[0].GetNumDimensions(); in NeonArgMinMaxWorkload() local
/external/armnn/src/armnnUtils/
DPermute.cpp28 const size_type numDims = dstShape.GetNumDimensions(); in PermuteLoop() local
103 const unsigned int numDims = mappings.GetSize(); in Permuted() local
DTranspose.cpp28 const size_type numDims = srcShape.GetNumDimensions(); in TransposeLoop() local
102 const unsigned int numDims = mappings.GetSize(); in TransposeTensorShape() local
/external/armnn/shim/sl/canonical/
DCanonicalUtils.hpp108 const unsigned int numDims = tensorShape.GetNumDimensions(); in ComputeShape() local
/external/armnn/src/armnn/layers/
DElementwiseBaseLayer.cpp35 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
DLogicalBinaryLayer.cpp41 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
DComparisonLayer.cpp47 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
DElementwiseBinaryLayer.cpp43 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
DConcatLayer.cpp208 unsigned int numDims = m_Param.GetNumDimensions(); in InferOutputShapes() local
/external/tensorflow/tensorflow/go/
Dtensor_handle.go92 func (th *TensorHandle) numDims() (int, error) { func
/external/android-nn-driver/
DUtils.hpp182 const unsigned int numDims = tensorShape.GetNumDimensions(); in ComputeShape() local
/external/armnn/delegate/common/src/
DDelegateUtils.hpp85 int numDims = armnn::numeric_cast<int>(numDimensions); in ComputeWrappedIndex() local
/external/armnn/delegate/common/src/test/
DDelegateTestInterpreter.hpp153 int32_t numDims = TfLiteTensorNumDims(outputTensor); in GetOutputShape() local
/external/armnn/src/armnn/
DGraph.cpp87 unsigned int numDims = inputTensorShape.GetNumDimensions(); in Print() local
103 unsigned int numDims = outputTensorShape.GetNumDimensions(); in Print() local
DSerializeLayerParameters.cpp271 uint32_t numDims = desc.GetNumDimensions(); in Serialize() local
598 uint32_t numDims = desc.GetNumDimensions(); in Serialize() local
DDescriptors.cpp480 auto numDims = tensorShape.GetNumDimensions(); in GetAxesToMul() local
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DEagerOperationBuilder.java254 private static native void setAttrShape(long opHandle, String name, long[] shape, int numDims); in setAttrShape()
257 long opHandle, String name, long[] shapes, int[] numDims); in setAttrShapeList()
DEagerOperation.java169 private static native int numDims(long handle); in numDims() method in EagerOperation
DGraphOperationBuilder.java338 private static native void setAttrShape(long handle, String name, long[] shape, int numDims); in setAttrShape()
341 long handle, String name, long[] shapes, int[] numDims); in setAttrShapeList()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DConcatTestImpl.cpp74 unsigned int numDims = inputShape.GetNumDimensions(); in ExpandTensorShapeTo3dForPermute() local
188 unsigned int numDims = 0; in PermuteInputsForConcat() local

12