| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BatchMatmul.cpp | 65 uint32_t numDims = getNumberOfDimensions(LHSTensorShape); in computeOutputDimensions() local
77 uint32_t numDims = getNumberOfDimensions(shape); in swapRowColumnDims() local
213 uint32_t numDims = getNumberOfDimensions(inputLHSTensorShape); in prepare() local
|
| D | MirrorPad.cpp | 130 int numDims = 0; member
195 const int numDims = int(inputDims); in eval() local
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | Broadcast.cpp | 14 const unsigned int numDims = GetNumDimensions(); in BroadcastLoop() local
36 const unsigned int numDims = GetNumDimensions(); in BroadcastLoop() local
|
| H A D | Reduce.cpp | 19 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()
|
| H A D | Debug.cpp | 30 const unsigned int numDims = inputInfo.GetNumDimensions(); in PrintOutput() local
|
| H A D | Slice.cpp | 21 const unsigned int numDims = inputShape.GetNumDimensions(); in Slice() local
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonArgMinMaxWorkload.cpp | 38 auto numDims = input.GetNumDimensions(); in NeonArgMinMaxWorkloadValidate() local
68 auto numDims = info.m_InputTensorInfos[0].GetNumDimensions(); in NeonArgMinMaxWorkload() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClArgMinMaxWorkload.cpp | 38 auto numDims = input.GetNumDimensions(); in ClArgMinMaxWorkloadValidate() local
69 auto numDims = info.m_InputTensorInfos[0].GetNumDimensions(); in ClArgMinMaxWorkload() local
|
| /aosp_15_r20/external/armnn/src/armnnUtils/ |
| H A D | Transpose.cpp | 28 const size_type numDims = srcShape.GetNumDimensions(); in TransposeLoop() local
102 const unsigned int numDims = mappings.GetSize(); in TransposeTensorShape() local
|
| H A D | Permute.cpp | 28 const size_type numDims = dstShape.GetNumDimensions(); in PermuteLoop() local
103 const unsigned int numDims = mappings.GetSize(); in Permuted() local
|
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | Graph.cpp | 87 unsigned int numDims = inputTensorShape.GetNumDimensions(); in Print() local
103 unsigned int numDims = outputTensorShape.GetNumDimensions(); in Print() local
|
| H A D | SerializeLayerParameters.cpp | 271 uint32_t numDims = desc.GetNumDimensions(); in Serialize() local
598 uint32_t numDims = desc.GetNumDimensions(); in Serialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| H A D | EagerOperationBuilder.java | 254 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()
|
| H A D | GraphOperationBuilder.java | 338 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()
|
| H A D | EagerOperation.java | 169 private static native int numDims(long handle); in numDims() method in EagerOperation
|
| /aosp_15_r20/external/armnn/shim/sl/canonical/ |
| H A D | CanonicalUtils.hpp | 108 const unsigned int numDims = tensorShape.GetNumDimensions(); in ComputeShape() local
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | ElementwiseBaseLayer.cpp | 35 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
|
| H A D | LogicalBinaryLayer.cpp | 41 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
|
| H A D | ComparisonLayer.cpp | 47 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
|
| H A D | ElementwiseBinaryLayer.cpp | 43 unsigned int numDims = input0.GetNumDimensions(); in InferOutputShapes() local
|
| H A D | ConcatLayer.cpp | 208 unsigned int numDims = m_Param.GetNumDimensions(); in InferOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/ |
| H A D | tensor_handle.go | 92 func (th *TensorHandle) numDims() (int, error) { func
|
| /aosp_15_r20/external/android-nn-driver/ |
| H A D | Utils.hpp | 182 const unsigned int numDims = tensorShape.GetNumDimensions(); in ComputeShape() local
|
| /aosp_15_r20/external/armnn/delegate/common/src/ |
| H A D | DelegateUtils.hpp | 85 int numDims = armnn::numeric_cast<int>(numDimensions); in ComputeWrappedIndex() local
|
| /aosp_15_r20/external/armnn/delegate/common/src/test/ |
| H A D | DelegateTestInterpreter.hpp | 153 int32_t numDims = TfLiteTensorNumDims(outputTensor); in GetOutputShape() local
|