Lines Matching refs:kInputTensor
40 constexpr uint32_t kInputTensor = 0; variable
236 auto inputType = context->getInputType(kInputTensor); in validate()
250 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
267 Shape input = context->getInputShape(kInputTensor); in prepare()
284 switch (context->getInputType(kInputTensor)) { in execute()
286 return softmaxFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
287 context->getInputShape(kInputTensor), in execute()
292 return softmaxFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
298 return softmaxQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
299 context->getInputShape(kInputTensor), in execute()
304 return softmaxQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
305 context->getInputShape(kInputTensor), in execute()