Searched refs:inputTensorNum (Results 1 – 4 of 4) sorted by relevance
47 for (uint32_t inputTensorNum = 1, inputTensorCount = context->getNumInputs() - 1; in prepare() local48 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() local123 inputData[inputTensorNum] = context->getInputBuffer<T>(kInputFirstTensor + inputTensorNum); in pack()
48 for (uint32_t inputTensorNum = 0, inputTensorCount = context->getNumInputs() - 1; in validate() local49 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()
28 const int32_t inputTensorNum, in CreatePackTfLiteModel() argument48 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()
28 const int32_t inputTensorNum, in CreateConcatTfLiteModel() argument49 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()