Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DFullyConnected.cpp40 constexpr uint32_t kBiasTensor = 2; variable
284 Shape bias = context->getInputShape(kBiasTensor); in validate()
295 Shape bias = context->getInputShape(kBiasTensor); in prepare()
310 context->getInputBuffer<float>(kBiasTensor), in execute()
311 context->getInputShape(kBiasTensor), in execute()
320 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
321 context->getInputShape(kBiasTensor), in execute()
330 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
331 context->getInputShape(kBiasTensor), in execute()
340 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
[all …]
DTransposeConv2D.cpp40 constexpr uint32_t kBiasTensor = 2; variable
486 Shape bias = context->getInputShape(kBiasTensor); in prepare()
552 context->getInputBuffer<float>(kBiasTensor), in execute()
553 context->getInputShape(kBiasTensor), param, in execute()
561 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
562 context->getInputShape(kBiasTensor), param, in execute()
574 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
575 context->getInputShape(kBiasTensor), param, in execute()
583 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
584 context->getInputShape(kBiasTensor), param, in execute()
[all …]
DDepthwiseConv2D.cpp38 constexpr uint32_t kBiasTensor = 2; variable
514 Shape bias = context->getInputShape(kBiasTensor); in prepare()
581 context->getInputBuffer<float>(kBiasTensor), in execute()
582 context->getInputShape(kBiasTensor), param.padding_left, in execute()
594 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
595 context->getInputShape(kBiasTensor), param.padding_left, in execute()
611 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
612 context->getInputShape(kBiasTensor), param.padding_left, in execute()
623 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
624 context->getInputShape(kBiasTensor), param.padding_left, in execute()
[all …]
DConv2D.cpp45 constexpr uint32_t kBiasTensor = 2; variable
632 Shape bias = context->getInputShape(kBiasTensor); in prepare()
703 context->getInputBuffer<float>(kBiasTensor), in execute()
704 context->getInputShape(kBiasTensor), param.padding_left, in execute()
715 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
716 context->getInputShape(kBiasTensor), param.padding_left, in execute()
731 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
732 context->getInputShape(kBiasTensor), param.padding_left, in execute()
743 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
744 context->getInputShape(kBiasTensor), param.padding_left, in execute()
[all …]
DUnidirectionalSequenceRNN.cpp35 constexpr uint32_t kBiasTensor = 3; variable
73 const T* bias = context->getInputBuffer<T>(kBiasTensor); in executeTyped()
157 Shape bias = context->getInputShape(kBiasTensor); in prepare()
DSVDF.h50 static constexpr int kBiasTensor = 3; // Optional variable
DRNN.h42 static constexpr int kBiasTensor = 3; variable
DSVDF.cpp38 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
91 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
DRNN.cpp40 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
61 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()