• Home
  • Raw
  • Download

Lines Matching refs:kInputTensor

364     Shape input = context->getInputShape(kInputTensor);  in prepare()
400 switch (context->getInputType(kInputTensor)) { in executeRelu()
402 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
403 context->getInputShape(kInputTensor), in executeRelu()
407 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
408 context->getInputShape(kInputTensor), in executeRelu()
412 return reluQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu()
413 context->getInputShape(kInputTensor), in executeRelu()
417 return reluQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu()
418 context->getInputShape(kInputTensor), in executeRelu()
429 switch (context->getInputType(kInputTensor)) { in executeRelu1()
431 return relu1Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu1()
432 context->getInputShape(kInputTensor), in executeRelu1()
436 return relu1Float(context->getInputBuffer<float>(kInputTensor), in executeRelu1()
437 context->getInputShape(kInputTensor), in executeRelu1()
441 return relu1Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu1()
442 context->getInputShape(kInputTensor), in executeRelu1()
446 return relu1Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu1()
447 context->getInputShape(kInputTensor), in executeRelu1()
458 switch (context->getInputType(kInputTensor)) { in executeRelu6()
460 return relu6Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu6()
461 context->getInputShape(kInputTensor), in executeRelu6()
465 return relu6Float(context->getInputBuffer<float>(kInputTensor), in executeRelu6()
466 context->getInputShape(kInputTensor), in executeRelu6()
470 return relu6Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu6()
471 context->getInputShape(kInputTensor), in executeRelu6()
475 return relu6Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu6()
476 context->getInputShape(kInputTensor), in executeRelu6()
487 switch (context->getInputType(kInputTensor)) { in executeLogistic()
489 return logisticFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeLogistic()
490 context->getInputShape(kInputTensor), in executeLogistic()
494 return logisticFloat(context->getInputBuffer<float>(kInputTensor), in executeLogistic()
495 context->getInputShape(kInputTensor), in executeLogistic()
499 return logisticQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeLogistic()
500 context->getInputShape(kInputTensor), in executeLogistic()
504 return logisticQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeLogistic()
505 context->getInputShape(kInputTensor), in executeLogistic()
516 switch (context->getInputType(kInputTensor)) { in executeTanh()
518 return tanhFloat16(context->getInputBuffer<_Float16>(kInputTensor), in executeTanh()
519 context->getInputShape(kInputTensor), in executeTanh()
523 return tanhFloat32(context->getInputBuffer<float>(kInputTensor), in executeTanh()
524 context->getInputShape(kInputTensor), in executeTanh()
528 return tanhQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeTanh()
529 context->getInputShape(kInputTensor), in executeTanh()
533 return tanhQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeTanh()
534 context->getInputShape(kInputTensor), in executeTanh()
545 switch (context->getInputType(kInputTensor)) { in executeHardSwish()
547 const Shape& inputShape = context->getInputShape(kInputTensor); in executeHardSwish()
551 convertFloat16ToFloat32(context->getInputBuffer<_Float16>(kInputTensor), &inputFloat); in executeHardSwish()
560 convertShapeToTflshape(context->getInputShape(kInputTensor)), in executeHardSwish()
561 context->getInputBuffer<float>(kInputTensor), in executeHardSwish()
567 return hardSwishQuant(context->getInputBuffer<uint8_t>(kInputTensor), in executeHardSwish()
568 context->getInputShape(kInputTensor), in executeHardSwish()
572 return hardSwishQuant(context->getInputBuffer<int8_t>(kInputTensor), in executeHardSwish()
573 context->getInputShape(kInputTensor), in executeHardSwish()