Lines Matching refs:kInputTensor

180     Shape input = context->getInputShape(kInputTensor);  in prepare()
197     if (context->getInputType(kInputTensor) == OperandType::TENSOR_FLOAT16) {  in prepare()
226     switch (context->getInputType(kInputTensor)) {  in execute()
228             return roiPooling(context->getInputBuffer<_Float16>(kInputTensor),  in execute()
229                               context->getInputShape(kInputTensor),  in execute()
240             return roiPooling(context->getInputBuffer<float>(kInputTensor),  in execute()
241                               context->getInputShape(kInputTensor),  in execute()
252             return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor),  in execute()
253                               context->getInputShape(kInputTensor),  in execute()
264             return roiPooling(context->getInputBuffer<int8_t>(kInputTensor),  in execute()
265                               context->getInputShape(kInputTensor),  in execute()