Home
last modified time | relevance | path

Searched refs:inputTensorNum (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DPack.cpp47 for (uint32_t inputTensorNum = 1, inputTensorCount = context->getNumInputs() - 1; in prepare() local
48 inputTensorNum < inputTensorCount; ++inputTensorNum) { in prepare()
50 context->getInputShape(kInputFirstTensor + inputTensorNum))) in prepare()
51 << "Input tensor #" << inputTensorNum in prepare()
122 for (uint32_t inputTensorNum = 0; inputTensorNum < inputTensorCount; ++inputTensorNum) { in pack() local
123 inputData[inputTensorNum] = context->getInputBuffer<T>(kInputFirstTensor + inputTensorNum); in pack()
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DPack.cpp48 for (uint32_t inputTensorNum = 0, inputTensorCount = context->getNumInputs() - 1; in validate() local
49 inputTensorNum < inputTensorCount; ++inputTensorNum) { in validate()
50 const Shape inputShape = context->getInputShape(kInputFirstTensor + inputTensorNum); in validate()
52 << "Input tensor #" << inputTensorNum << " type " << inputShape.type in validate()
57 << "Input tensor #" << inputTensorNum << " scale " << inputShape.scale in validate()
60 << "Input tensor #" << inputTensorNum << " offset " << inputShape.offset in validate()
66 << "Input tensor #" << inputTensorNum << " has inconsistent rank"; in validate()
/aosp_15_r20/external/armnn/delegate/test/
H A DPackTestHelper.hpp28 const int32_t inputTensorNum, in CreatePackTfLiteModel() argument
48 const std::vector<int32_t> operatorOutputs{inputTensorNum}; in CreatePackTfLiteModel()
50 const std::vector<int> subgraphOutputs{inputTensorNum}; in CreatePackTfLiteModel()
52 std::vector<flatbuffers::Offset<Tensor>> tensors(inputTensorNum + 1); in CreatePackTfLiteModel()
53 for (int i = 0; i < inputTensorNum; ++i) in CreatePackTfLiteModel()
69 tensors[inputTensorNum] = CreateTensor(flatBufferBuilder, in CreatePackTfLiteModel()
80 CreatePackOptions(flatBufferBuilder, inputTensorNum, axis).Union(); in CreatePackTfLiteModel()
H A DControlTestHelper.hpp28 const int32_t inputTensorNum, in CreateConcatTfLiteModel() argument
49 const std::vector<int32_t> operatorOutputs{inputTensorNum}; in CreateConcatTfLiteModel()
51 const std::vector<int> subgraphOutputs{inputTensorNum}; in CreateConcatTfLiteModel()
53 std::vector<flatbuffers::Offset<Tensor>> tensors(inputTensorNum + 1); in CreateConcatTfLiteModel()
54 for (int i = 0; i < inputTensorNum; ++i) in CreateConcatTfLiteModel()
70 tensors[inputTensorNum] = CreateTensor(flatBufferBuilder, in CreateConcatTfLiteModel()