Lines Matching refs:kInputTensor
37 constexpr uint32_t kInputTensor = 0; variable
80 const OperandType inputType = context->getInputType(kInputTensor); in validate()
90 const Shape& input = context->getInputShape(kInputTensor); in validate()
100 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
103 const Shape& input = context->getInputShape(kInputTensor); in prepare()
140 switch (context->getInputType(kInputTensor)) { in execute()
142 return transposeGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
143 context->getInputShape(kInputTensor), in execute()
149 return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
150 context->getInputShape(kInputTensor), in execute()
156 return transposeGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
163 return transposeGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
164 context->getInputShape(kInputTensor), in execute()