• Home
  • Raw
  • Download

Lines Matching refs:kInputTensor

44 constexpr uint32_t kInputTensor = 0;  variable
365 auto inputType = context->getInputType(kInputTensor); in validate()
382 const Shape& input = context->getInputShape(kInputTensor); in validate()
394 auto inputType = context->getInputType(kInputTensor); in validateHardSwish()
410 Shape input = context->getInputShape(kInputTensor); in prepare()
446 switch (context->getInputType(kInputTensor)) { in executeRelu()
448 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
449 context->getInputShape(kInputTensor), in executeRelu()
453 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
454 context->getInputShape(kInputTensor), in executeRelu()
458 return reluQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu()
459 context->getInputShape(kInputTensor), in executeRelu()
463 return reluQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu()
464 context->getInputShape(kInputTensor), in executeRelu()
475 switch (context->getInputType(kInputTensor)) { in executeRelu1()
477 return relu1Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu1()
478 context->getInputShape(kInputTensor), in executeRelu1()
482 return relu1Float(context->getInputBuffer<float>(kInputTensor), in executeRelu1()
483 context->getInputShape(kInputTensor), in executeRelu1()
487 return relu1Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu1()
488 context->getInputShape(kInputTensor), in executeRelu1()
492 return relu1Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu1()
493 context->getInputShape(kInputTensor), in executeRelu1()
504 switch (context->getInputType(kInputTensor)) { in executeRelu6()
506 return relu6Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu6()
507 context->getInputShape(kInputTensor), in executeRelu6()
511 return relu6Float(context->getInputBuffer<float>(kInputTensor), in executeRelu6()
512 context->getInputShape(kInputTensor), in executeRelu6()
516 return relu6Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu6()
517 context->getInputShape(kInputTensor), in executeRelu6()
521 return relu6Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu6()
522 context->getInputShape(kInputTensor), in executeRelu6()
533 switch (context->getInputType(kInputTensor)) { in executeLogistic()
535 return logisticFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeLogistic()
536 context->getInputShape(kInputTensor), in executeLogistic()
540 return logisticFloat(context->getInputBuffer<float>(kInputTensor), in executeLogistic()
541 context->getInputShape(kInputTensor), in executeLogistic()
545 return logisticQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeLogistic()
546 context->getInputShape(kInputTensor), in executeLogistic()
550 return logisticQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeLogistic()
551 context->getInputShape(kInputTensor), in executeLogistic()
562 switch (context->getInputType(kInputTensor)) { in executeTanh()
564 return tanhFloat16(context->getInputBuffer<_Float16>(kInputTensor), in executeTanh()
565 context->getInputShape(kInputTensor), in executeTanh()
569 return tanhFloat32(context->getInputBuffer<float>(kInputTensor), in executeTanh()
570 context->getInputShape(kInputTensor), in executeTanh()
574 return tanhQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeTanh()
575 context->getInputShape(kInputTensor), in executeTanh()
579 return tanhQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeTanh()
580 context->getInputShape(kInputTensor), in executeTanh()
591 switch (context->getInputType(kInputTensor)) { in executeHardSwish()
593 const Shape& inputShape = context->getInputShape(kInputTensor); in executeHardSwish()
597 convertFloat16ToFloat32(context->getInputBuffer<_Float16>(kInputTensor), &inputFloat); in executeHardSwish()
606 convertShapeToTflshape(context->getInputShape(kInputTensor)), in executeHardSwish()
607 context->getInputBuffer<float>(kInputTensor), in executeHardSwish()
613 return hardSwishQuant(context->getInputBuffer<uint8_t>(kInputTensor), in executeHardSwish()
614 context->getInputShape(kInputTensor), in executeHardSwish()
618 return hardSwishQuant(context->getInputBuffer<int8_t>(kInputTensor), in executeHardSwish()
619 context->getInputShape(kInputTensor), in executeHardSwish()