Home
last modified time | relevance | path

Searched refs:kFilterTensor (Results 1 – 9 of 9) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DTransposeConv2D.cpp70 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
438 Shape filter = context->getInputShape(kFilterTensor); in prepare()
503 context->getInputBuffer<float>(kFilterTensor), in execute()
504 context->getInputShape(kFilterTensor), in execute()
512 context->getInputBuffer<_Float16>(kFilterTensor), in execute()
513 context->getInputShape(kFilterTensor), in execute()
519 if (context->getInputType(kFilterTensor) == in execute()
524 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
525 context->getInputShape(kFilterTensor), in execute()
527 context->getInputExtraParams(kFilterTensor)) in execute()
[all …]
DDepthwiseConv2D.cpp93 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
418 Shape filter = context->getInputShape(kFilterTensor); in prepare()
484 context->getInputBuffer<float>(kFilterTensor), in execute()
485 context->getInputShape(kFilterTensor), in execute()
497 context->getInputBuffer<_Float16>(kFilterTensor), in execute()
498 context->getInputShape(kFilterTensor), in execute()
508 if (context->getInputType(kFilterTensor) == in execute()
513 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
514 context->getInputShape(kFilterTensor), in execute()
516 context->getInputExtraParams(kFilterTensor)) in execute()
[all …]
DConv2D.cpp106 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
532 Shape filter = context->getInputShape(kFilterTensor); in prepare()
602 context->getInputBuffer<float>(kFilterTensor), in execute()
603 context->getInputShape(kFilterTensor), in execute()
614 context->getInputBuffer<_Float16>(kFilterTensor), in execute()
615 context->getInputShape(kFilterTensor), in execute()
624 if (context->getInputType(kFilterTensor) == in execute()
629 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
630 context->getInputShape(kFilterTensor), in execute()
632 context->getInputExtraParams(kFilterTensor)) in execute()
[all …]
/packages/modules/NeuralNetworks/common/types/operations/src/
DConv2D.cpp32 const auto filterRank = getNumberOfDimensions(context->getInputShape(kFilterTensor)); in validate()
41 auto filterType = context->getInputType(kFilterTensor); in validate()
64 context->getInputExtraParams(kFilterTensor)) in validate()
82 const float filterScale = context->getInputShape(kFilterTensor).scale; in validate()
DDepthwiseConv2D.cpp32 auto filterType = context->getInputType(kFilterTensor); in validate()
55 context->getInputExtraParams(kFilterTensor)) in validate()
78 const float filterScale = context->getInputShape(kFilterTensor).scale; in validate()
DTransposeConv2D.cpp31 const auto filterType = context->getInputType(kFilterTensor); in validate()
43 context->getInputExtraParams(kFilterTensor)) in validate()
/packages/modules/NeuralNetworks/common/types/operations/include/
DDepthwiseConv2D.h28 constexpr uint32_t kFilterTensor = 1; variable
DConv2D.h28 constexpr uint32_t kFilterTensor = 1; variable
DTransposeConv2D.h27 constexpr uint32_t kFilterTensor = 1; variable