Searched refs:inputTensorInfos (Results 1 – 7 of 7) sorted by relevance
| /external/armnn/delegate/classic/src/ |
| D | Pack.hpp | 36 std::vector<armnn::TensorInfo> inputTensorInfos; in VisitPackOperator() local
46 inputTensorInfos.emplace_back(inputTensorInfo); in VisitPackOperator()
51 std::transform(inputTensorInfos.begin(), in VisitPackOperator()
52 inputTensorInfos.end(), in VisitPackOperator()
73 desc.m_InputShape = inputTensorInfos[0].GetShape(); in VisitPackOperator()
|
| D | Control.hpp | 71 std::vector<armnn::TensorInfo> inputTensorInfos; in VisitConcatenationOperator() local
81 inputTensorInfos.emplace_back(inputTensorInfo); in VisitConcatenationOperator()
86 std::transform(inputTensorInfos.begin(), in VisitConcatenationOperator()
87 inputTensorInfos.end(), in VisitConcatenationOperator()
|
| /external/armnn/tests/TfLiteBenchmark-Armnn/ |
| D | TfLiteBenchmark-Armnn.cpp | 152 std::vector<armnn::TensorInfo> inputTensorInfos; in main() local
162 inputTensorInfos.push_back(inputTensorInfo); in main()
181 unsigned int nb_inputs = armnn::numeric_cast<unsigned int>(inputTensorInfos.size()); in main()
186 std::vector<float> in_data(inputTensorInfos.at(i).GetNumElements()); in main()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | ConcatTestImpl.cpp | 27 const std::vector<TensorInfo> & inputTensorInfos, in CreateDescriptorForConcat() argument
31 shapes.reserve(inputTensorInfos.size()); in CreateDescriptorForConcat()
32 for (const TensorInfo& it: inputTensorInfos) in CreateDescriptorForConcat()
47 const std::vector<TensorInfo> & inputTensorInfos, in NeedPermuteForConcat() argument
56 for (auto && tensorInfo : inputTensorInfos) in NeedPermuteForConcat()
177 std::vector<TensorInfo> & inputTensorInfos, in PermuteInputsForConcat() argument
185 ARMNN_ASSERT_MSG(inputTensorInfos.size() > 1, in PermuteInputsForConcat()
197 for (auto && tensorInfo : inputTensorInfos) in PermuteInputsForConcat()
227 inputTensorInfos[nthInput] = newTensorInfo; in PermuteInputsForConcat()
299 …std::vector<TensorInfo> inputTensorInfos(inputTensorInfosOrig.begin(), inputTensorInfosOrig.end()); in Concatenate() local
[all …]
|
| /external/armnn/src/armnn/test/ |
| D | InferOutputTests.hpp | 368 std::vector<armnn::TensorInfo> inputTensorInfos; in CreateStackLayerHelper() local
371 inputTensorInfos.push_back(armnn::TensorInfo(inputShapes[i], armnn::DataType::Float32)); in CreateStackLayerHelper()
377 Connect(inputs[i], stackLayer, inputTensorInfos[i], 0, i); in CreateStackLayerHelper()
|
| /external/android-nn-driver/ |
| D | ConversionUtils.hpp | 2194 std::vector<const armnn::TensorInfo*> inputTensorInfos; in ConvertConcatenation() local
2195 std::transform(inputHandles.begin(), inputHandles.end(), std::back_inserter(inputTensorInfos), in ConvertConcatenation()
2206 inputTensorInfos, in ConvertConcatenation()
|
| /external/armnn/shim/sl/canonical/ |
| D | Converter.cpp | 871 std::vector<const armnn::TensorInfo*> inputTensorInfos; in ConvertConcatenation() local
872 std::transform(inputHandles.begin(), inputHandles.end(), std::back_inserter(inputTensorInfos), in ConvertConcatenation()
883 inputTensorInfos, in ConvertConcatenation()
|