Lines Matching refs:kInputTensor
39 constexpr uint32_t kInputTensor = 0; variable
96 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
426 auto inputType = context->getInputType(kInputTensor); in validate()
472 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
520 Shape input = context->getInputShape(kInputTensor); in prepare()
583 switch (context->getInputType(kInputTensor)) { in execute()
585 return depthwiseConv(context->getInputBuffer<float>(kInputTensor), in execute()
586 context->getInputShape(kInputTensor), in execute()
598 return depthwiseConv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
599 context->getInputShape(kInputTensor), in execute()
614 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
615 context->getInputShape(kInputTensor), in execute()
629 return depthwiseConv(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
630 context->getInputShape(kInputTensor), in execute()
648 context->getInputBuffer<int8_t>(kInputTensor), in execute()
649 context->getInputShape(kInputTensor), in execute()
664 return depthwiseConv(context->getInputBuffer<int8_t>(kInputTensor), in execute()
665 context->getInputShape(kInputTensor), in execute()