Lines Matching refs:kInputTensor

78     const Shape& inputShape = context->getInputShape(kInputTensor);  in prepare()
113     switch (context->getInputType(kInputTensor)) {  in execute()
115             return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor),  in execute()
116                                context->getInputShape(kInputTensor),  in execute()
124             return evalGeneric(context->getInputBuffer<float>(kInputTensor),  in execute()
125                                context->getInputShape(kInputTensor),  in execute()
133             return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor),  in execute()
134                                context->getInputShape(kInputTensor),  in execute()
142             return evalGeneric(context->getInputBuffer<uint8_t>(kInputTensor),  in execute()
143                                context->getInputShape(kInputTensor),  in execute()
151             return evalGeneric(context->getInputBuffer<int8_t>(kInputTensor),  in execute()
152                                context->getInputShape(kInputTensor),  in execute()