Lines Matching refs:kInputTensor

44 constexpr uint32_t kInputTensor = 0;  variable
362 auto inputType = context->getInputType(kInputTensor); in validate()
378 const Shape& input = context->getInputShape(kInputTensor); in validate()
388 auto inputType = context->getInputType(kInputTensor); in validateHardSwish()
400 Shape input = context->getInputShape(kInputTensor); in prepare()
436 switch (context->getInputType(kInputTensor)) { in executeRelu()
438 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
439 context->getInputShape(kInputTensor), in executeRelu()
443 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
444 context->getInputShape(kInputTensor), in executeRelu()
448 return reluQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu()
449 context->getInputShape(kInputTensor), in executeRelu()
453 return reluQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu()
454 context->getInputShape(kInputTensor), in executeRelu()
465 switch (context->getInputType(kInputTensor)) { in executeRelu1()
467 return relu1Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu1()
468 context->getInputShape(kInputTensor), in executeRelu1()
472 return relu1Float(context->getInputBuffer<float>(kInputTensor), in executeRelu1()
473 context->getInputShape(kInputTensor), in executeRelu1()
477 return relu1Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu1()
478 context->getInputShape(kInputTensor), in executeRelu1()
482 return relu1Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu1()
483 context->getInputShape(kInputTensor), in executeRelu1()
494 switch (context->getInputType(kInputTensor)) { in executeRelu6()
496 return relu6Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu6()
497 context->getInputShape(kInputTensor), in executeRelu6()
501 return relu6Float(context->getInputBuffer<float>(kInputTensor), in executeRelu6()
502 context->getInputShape(kInputTensor), in executeRelu6()
506 return relu6Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu6()
507 context->getInputShape(kInputTensor), in executeRelu6()
511 return relu6Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu6()
512 context->getInputShape(kInputTensor), in executeRelu6()
523 switch (context->getInputType(kInputTensor)) { in executeLogistic()
525 return logisticFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeLogistic()
526 context->getInputShape(kInputTensor), in executeLogistic()
530 return logisticFloat(context->getInputBuffer<float>(kInputTensor), in executeLogistic()
531 context->getInputShape(kInputTensor), in executeLogistic()
535 return logisticQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeLogistic()
536 context->getInputShape(kInputTensor), in executeLogistic()
540 return logisticQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeLogistic()
541 context->getInputShape(kInputTensor), in executeLogistic()
552 switch (context->getInputType(kInputTensor)) { in executeTanh()
554 return tanhFloat16(context->getInputBuffer<_Float16>(kInputTensor), in executeTanh()
555 context->getInputShape(kInputTensor), in executeTanh()
559 return tanhFloat32(context->getInputBuffer<float>(kInputTensor), in executeTanh()
560 context->getInputShape(kInputTensor), in executeTanh()
564 return tanhQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeTanh()
565 context->getInputShape(kInputTensor), in executeTanh()
569 return tanhQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeTanh()
570 context->getInputShape(kInputTensor), in executeTanh()
581 switch (context->getInputType(kInputTensor)) { in executeHardSwish()
583 const Shape& inputShape = context->getInputShape(kInputTensor); in executeHardSwish()
587 convertFloat16ToFloat32(context->getInputBuffer<_Float16>(kInputTensor), &inputFloat); in executeHardSwish()
596 convertShapeToTflshape(context->getInputShape(kInputTensor)), in executeHardSwish()
597 context->getInputBuffer<float>(kInputTensor), in executeHardSwish()
603 return hardSwishQuant(context->getInputBuffer<uint8_t>(kInputTensor), in executeHardSwish()
604 context->getInputShape(kInputTensor), in executeHardSwish()
608 return hardSwishQuant(context->getInputBuffer<int8_t>(kInputTensor), in executeHardSwish()
609 context->getInputShape(kInputTensor), in executeHardSwish()