Home
last modified time | relevance | path

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

/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()
/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()