| /external/armnn/src/backends/reference/workloads/ |
| D | Splitter.hpp | 21 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in Splitter() local
23 for (unsigned int index = 0; index < inputInfo0.GetNumElements(); ++index) in Splitter()
28 unsigned int dimensionStride = inputInfo0.GetNumElements(); in Splitter()
30 for (unsigned int i = 0; i<inputInfo0.GetNumDimensions(); i++) in Splitter()
32 dimensionStride /= inputInfo0.GetShape()[i]; in Splitter()
43 ARMNN_ASSERT(outputInfo.GetNumDimensions() == inputInfo0.GetNumDimensions()); in Splitter()
|
| D | RefGatherNdWorkload.cpp | 31 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in Execute() local
35 …std::unique_ptr<Decoder<float>> params_decoderPtr = MakeDecoder<float>(inputInfo0, inputs[0]->Map(… in Execute()
42 …std::map<std::string, unsigned int> keyIndices = CalculateGatherNdKeyIndices(inputInfo0, inputInfo… in Execute()
47 TensorShape paramsShape = inputInfo0.GetShape(); in Execute()
75 armnn::TensorInfo params_K_C_Info = inputInfo0; in Execute()
|
| D | RefGatherWorkload.cpp | 31 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in Execute() local
35 std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputInfo0, inputs[0]->Map()); in Execute()
43 …Gather(inputInfo0, inputInfo1, outputInfo, decoder, indicesData, encoder, m_Data.m_Parameters.m_Ax… in Execute()
|
| D | RefComparisonWorkload.cpp | 35 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in PostAllocationConfigure() local
39 m_Input0 = MakeDecoder<InType>(inputInfo0); in PostAllocationConfigure()
62 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in Execute() local
66 const TensorShape& inShape0 = inputInfo0.GetShape(); in Execute()
|
| D | RefLogicalBinaryWorkload.cpp | 40 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in Execute() local
44 const TensorShape& inShape0 = inputInfo0.GetShape(); in Execute()
48 std::unique_ptr<Decoder<InType>> input0 = MakeDecoder<InType>(inputInfo0, inputs[0]->Map()); in Execute()
|
| D | RefElementwiseBinaryWorkload.cpp | 29 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in ExecuteFunction() local
33 const TensorShape& inShape0 = inputInfo0.GetShape(); in ExecuteFunction()
37 std::unique_ptr<Decoder<DataType>> input0 = MakeDecoder<DataType>(inputInfo0, inputs[0]->Map()); in ExecuteFunction()
|
| D | RefElementwiseWorkload.cpp | 47 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]); in Execute() local
51 const TensorShape& inShape0 = inputInfo0.GetShape(); in Execute()
55 std::unique_ptr<Decoder<InType>> input0 = MakeDecoder<InType>(inputInfo0, inputs[0]->Map()); in Execute()
|
| /external/armnn/src/armnn/optimizations/ |
| D | AddBroadcastReshapeLayer.hpp | 34 … const TensorInfo& inputInfo0 = layer.GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(); in Run() local
37 if (inputInfo0.GetNumDimensions() == inputInfo1.GetNumDimensions()) in Run()
44 TensorInfo inputInfo = inputInfo0; in Run()
46 if (inputInfo0.GetNumDimensions() < inputInfo1.GetNumDimensions()) in Run()
49 reshapeInfo = inputInfo0; in Run()
|
| /external/armnn/delegate/common/src/ |
| D | DelegateUtils.hpp | 40 void ExpandTensorRankToEqual(armnn::TensorInfo& inputInfo0, in ExpandTensorRankToEqual() argument
43 unsigned int inputDimensions0 = inputInfo0.GetNumDimensions(); in ExpandTensorRankToEqual()
54 armnn::TensorInfo& smallInfo = input0IsSmaller ? inputInfo0 : inputInfo1; in ExpandTensorRankToEqual()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadUtils.cpp | 300 std::map<std::string, unsigned int> CalculateGatherNdKeyIndices(TensorInfo inputInfo0, TensorInfo i… in CalculateGatherNdKeyIndices() argument
303 for (unsigned int i = 0; i < inputInfo0.GetNumDimensions(); ++i) in CalculateGatherNdKeyIndices()
305 paramsShape.push_back(inputInfo0.GetShape()[i]); in CalculateGatherNdKeyIndices()
|
| D | WorkloadUtils.hpp | 259 std::map<std::string, unsigned int> CalculateGatherNdKeyIndices(TensorInfo inputInfo0, TensorInfo i…
|
| /external/android-nn-driver/ |
| D | ConversionUtils_1_3.hpp | 186 const TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertLogicalBinary() local
201 inputInfo0, in ConvertLogicalBinary()
|
| D | ConversionUtils.hpp | 311 const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); in BroadcastTensor() local
314 unsigned int inputDimensions0 = inputInfo0.GetNumDimensions(); in BroadcastTensor()
1933 const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertArgMinMax() local
1950 inputInfo0, in ConvertArgMinMax()
|
| D | ConversionUtils_1_2.hpp | 315 const TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertComparison_1_2() local
330 inputInfo0, in ConvertComparison_1_2()
|
| /external/armnn/shim/sl/canonical/ |
| D | ConversionUtils.hpp | 285 const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); in BroadcastTensor() local
288 unsigned int inputDimensions0 = inputInfo0.GetNumDimensions(); in BroadcastTensor()
|
| D | Converter.cpp | 203 const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertAdd() local
218 inputInfo0, in ConvertAdd()
291 const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertArgMinMax() local
308 inputInfo0, in ConvertArgMinMax()
353 const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertBatchMatMul() local
356 unsigned int rankInput0 = inputInfo0.GetNumDimensions(); in ConvertBatchMatMul()
411 inputInfo0, in ConvertBatchMatMul()
610 const TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertComparison() local
625 inputInfo0, in ConvertComparison()
2791 const TensorInfo& inputInfo0 = input0.GetTensorInfo(); in ConvertLogicalBinary() local
[all …]
|
| /external/armnn/src/backends/tosaCommon/test/ |
| D | OneToOneMappingTests.cpp | 116 TensorInfo inputInfo0 = TensorInfo(inputTensorShapes[0], DataType::Float32, 0.0f, 0, true); variable
120 input0->GetOutputSlot(0).SetTensorInfo(inputInfo0);
|
| /external/armnn/src/backends/reference/test/ |
| D | RefCreateWorkloadTests.cpp | 31 const TensorInfo& inputInfo0, in CheckInputsOutput() argument
39 CHECK((inputHandle0->GetTensorInfo() == inputInfo0)); in CheckInputsOutput()
|