Lines Matching refs:kInputTensor
45 constexpr uint32_t kInputTensor = 0; variable
107 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
539 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
548 auto inputType = context->getInputType(kInputTensor); in validate()
589 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
637 Shape input = context->getInputShape(kInputTensor); in prepare()
704 switch (context->getInputType(kInputTensor)) { in execute()
706 return conv(context->getInputBuffer<float>(kInputTensor), in execute()
707 context->getInputShape(kInputTensor), in execute()
718 return conv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
719 context->getInputShape(kInputTensor), in execute()
733 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
734 context->getInputShape(kInputTensor), in execute()
748 return conv(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
749 context->getInputShape(kInputTensor), in execute()
766 context->getInputBuffer<int8_t>(kInputTensor), in execute()
767 context->getInputShape(kInputTensor), in execute()
782 return conv(context->getInputBuffer<int8_t>(kInputTensor), in execute()
783 context->getInputShape(kInputTensor), in execute()