Home
last modified time | relevance | path

Searched refs:kBiasTensor (Results 1 – 8 of 8) 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 …]
DConv2D.cpp45 constexpr uint32_t kBiasTensor = 2; variable
607 Shape bias = context->getInputShape(kBiasTensor); in prepare()
678 context->getInputBuffer<float>(kBiasTensor), in execute()
679 context->getInputShape(kBiasTensor), param.padding_left, in execute()
690 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
691 context->getInputShape(kBiasTensor), param.padding_left, in execute()
706 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
707 context->getInputShape(kBiasTensor), param.padding_left, in execute()
718 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
719 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()