Searched refs:kBiasTensor (Results 1 – 8 of 8) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | FullyConnected.cpp | 34 constexpr uint32_t kBiasTensor = 2; variable
196 Shape bias = context->getInputShape(kBiasTensor); in prepare()
235 context->getInputBuffer<float>(kBiasTensor), in execute()
236 context->getInputShape(kBiasTensor), in execute()
245 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
246 context->getInputShape(kBiasTensor), in execute()
255 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
256 context->getInputShape(kBiasTensor), in execute()
|
| D | TransposeConv2D.cpp | 34 constexpr uint32_t kBiasTensor = 2; variable
468 Shape bias = context->getInputShape(kBiasTensor); in prepare()
532 context->getInputBuffer<float>(kBiasTensor), in execute()
533 context->getInputShape(kBiasTensor), param, in execute()
541 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
542 context->getInputShape(kBiasTensor), param, in execute()
554 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
555 context->getInputShape(kBiasTensor), param, in execute()
563 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
564 context->getInputShape(kBiasTensor), param, in execute()
|
| D | Conv2D.cpp | 34 constexpr uint32_t kBiasTensor = 2; variable
492 Shape bias = context->getInputShape(kBiasTensor); in prepare()
561 context->getInputBuffer<float>(kBiasTensor), in execute()
562 context->getInputShape(kBiasTensor), param.padding_left, in execute()
573 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
574 context->getInputShape(kBiasTensor), param.padding_left, in execute()
589 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
590 context->getInputShape(kBiasTensor), param.padding_left, in execute()
601 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
602 context->getInputShape(kBiasTensor), param.padding_left, in execute()
|
| D | UnidirectionalSequenceRNN.cpp | 30 constexpr uint32_t kBiasTensor = 3; variable
64 const T* bias = context->getInputBuffer<T>(kBiasTensor); in executeTyped()
135 Shape bias = context->getInputShape(kBiasTensor); in prepare()
|
| D | SVDF.h | 48 static constexpr int kBiasTensor = 3; // Optional variable
|
| D | RNN.h | 40 static constexpr int kBiasTensor = 3; variable
|
| D | SVDF.cpp | 34 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
75 GetInput(operation, operands, kBiasTensor); in Prepare()
|
| D | RNN.cpp | 35 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
61 GetInput(operation, operands, kBiasTensor); in Prepare()
|