Lines Matching refs:kInputTensor
43 constexpr uint32_t kInputTensor = 0; variable
239 auto inputType = context->getInputType(kInputTensor); in validate()
254 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
273 Shape input = context->getInputShape(kInputTensor); in prepare()
290 switch (context->getInputType(kInputTensor)) { in execute()
292 return softmaxFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
298 return softmaxFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
299 context->getInputShape(kInputTensor), in execute()
304 return softmaxQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
305 context->getInputShape(kInputTensor), in execute()
310 return softmaxQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
311 context->getInputShape(kInputTensor), in execute()