Lines Matching refs:kInputTensor
39 constexpr uint32_t kInputTensor = 0; variable
196 auto inputType = context->getInputType(kInputTensor); in validate()
232 Shape input = context->getInputShape(kInputTensor); in prepare()
249 if (context->getInputType(kInputTensor) == OperandType::TENSOR_FLOAT16) { in prepare()
278 switch (context->getInputType(kInputTensor)) { in execute()
280 return roiPooling(context->getInputBuffer<_Float16>(kInputTensor), in execute()
281 context->getInputShape(kInputTensor), in execute()
292 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
304 return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
305 context->getInputShape(kInputTensor), in execute()
316 return roiPooling(context->getInputBuffer<int8_t>(kInputTensor), in execute()
317 context->getInputShape(kInputTensor), in execute()