Lines Matching refs:kInputTensor
38 constexpr uint32_t kInputTensor = 0; variable
81 const OperandType inputType = context->getInputType(kInputTensor); in validate()
92 const Shape& input = context->getInputShape(kInputTensor); in validate()
104 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
107 const Shape& input = context->getInputShape(kInputTensor); in prepare()
144 switch (context->getInputType(kInputTensor)) { in execute()
146 return transposeGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
147 context->getInputShape(kInputTensor), in execute()
153 return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
154 context->getInputShape(kInputTensor), in execute()
160 return transposeGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
161 context->getInputShape(kInputTensor), in execute()
167 return transposeGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
168 context->getInputShape(kInputTensor), in execute()