| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | FullyConnected.cpp | 185 Shape bias = context->getInputShape(kBiasTensor); in prepare()
200 context->getInputBuffer<float>(kBiasTensor), in execute()
201 context->getInputShape(kBiasTensor), in execute()
210 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
211 context->getInputShape(kBiasTensor), in execute()
220 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
221 context->getInputShape(kBiasTensor), in execute()
230 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
231 context->getInputShape(kBiasTensor), in execute()
|
| D | TransposeConv2D.cpp | 439 Shape bias = context->getInputShape(kBiasTensor); in prepare()
505 context->getInputBuffer<float>(kBiasTensor), in execute()
506 context->getInputShape(kBiasTensor), param, in execute()
514 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
515 context->getInputShape(kBiasTensor), param, in execute()
529 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
530 context->getInputShape(kBiasTensor), param, in execute()
538 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
539 context->getInputShape(kBiasTensor), param, in execute()
556 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
[all …]
|
| D | DepthwiseConv2D.cpp | 419 Shape bias = context->getInputShape(kBiasTensor); in prepare()
486 context->getInputBuffer<float>(kBiasTensor), in execute()
487 context->getInputShape(kBiasTensor), param.padding_left, in execute()
499 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
500 context->getInputShape(kBiasTensor), param.padding_left, in execute()
518 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
519 context->getInputShape(kBiasTensor), param.padding_left, in execute()
530 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
531 context->getInputShape(kBiasTensor), param.padding_left, in execute()
552 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
[all …]
|
| D | Conv2D.cpp | 533 Shape bias = context->getInputShape(kBiasTensor); in prepare()
604 context->getInputBuffer<float>(kBiasTensor), in execute()
605 context->getInputShape(kBiasTensor), param.padding_left, in execute()
616 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
617 context->getInputShape(kBiasTensor), param.padding_left, in execute()
634 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
635 context->getInputShape(kBiasTensor), param.padding_left, in execute()
646 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
647 context->getInputShape(kBiasTensor), param.padding_left, in execute()
667 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
[all …]
|
| D | UnidirectionalSequenceRNN.cpp | 60 const T* bias = context->getInputBuffer<T>(kBiasTensor); in executeTyped()
122 Shape bias = context->getInputShape(kBiasTensor); in prepare()
|
| D | SVDF.cpp | 38 bias_ = GetInput(operation, operands, kBiasTensor); in SVDF()
91 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
|
| D | RNN.cpp | 36 bias_ = GetInput(operation, operands, kBiasTensor); in RNN()
57 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | DepthwiseConv2D.h | 29 constexpr uint32_t kBiasTensor = 2; variable
|
| D | Conv2D.h | 29 constexpr uint32_t kBiasTensor = 2; variable
|
| D | TransposeConv2D.h | 28 constexpr uint32_t kBiasTensor = 2; variable
|
| D | FullyConnected.h | 29 constexpr uint32_t kBiasTensor = 2; variable
|
| D | UnidirectionalSequenceRNN.h | 28 constexpr uint32_t kBiasTensor = 3; variable
|
| D | SVDF.h | 50 static constexpr int kBiasTensor = 3; // Optional variable
|
| D | RNN.h | 48 static constexpr int kBiasTensor = 3; variable
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | FullyConnected.cpp | 125 Shape bias = context->getInputShape(kBiasTensor); in validate()
|