Lines Matching refs:kInputTensor
37 constexpr uint32_t kInputTensor = 0; variable
194 auto inputType = context->getInputType(kInputTensor); in validate()
231 Shape input = context->getInputShape(kInputTensor); in prepare()
248 if (context->getInputType(kInputTensor) == OperandType::TENSOR_FLOAT16) { in prepare()
277 switch (context->getInputType(kInputTensor)) { in execute()
279 return roiPooling(context->getInputBuffer<_Float16>(kInputTensor), in execute()
280 context->getInputShape(kInputTensor), in execute()
291 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
292 context->getInputShape(kInputTensor), in execute()
303 return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
304 context->getInputShape(kInputTensor), in execute()
315 return roiPooling(context->getInputBuffer<int8_t>(kInputTensor), in execute()
316 context->getInputShape(kInputTensor), in execute()