Lines Matching refs:kInputTensor

76     NN_RET_CHECK(!context->isOmittedInput(kInputTensor));  in prepare()
79     const Shape& input = context->getInputShape(kInputTensor);  in prepare()
116     switch (context->getInputType(kInputTensor)) {  in execute()
118             return transposeGeneric(context->getInputBuffer<float>(kInputTensor),  in execute()
119                                     context->getInputShape(kInputTensor),  in execute()
125             return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor),  in execute()
126                                     context->getInputShape(kInputTensor),  in execute()
132             return transposeGeneric(context->getInputBuffer<uint8_t>(kInputTensor),  in execute()
133                                     context->getInputShape(kInputTensor),  in execute()
139             return transposeGeneric(context->getInputBuffer<int8_t>(kInputTensor),  in execute()
140                                     context->getInputShape(kInputTensor),  in execute()