Searched refs:kWeightsTensor (Results 1 – 7 of 7) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | FullyConnected.cpp | 184 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
198 context->getInputBuffer<float>(kWeightsTensor), in execute()
199 context->getInputShape(kWeightsTensor), in execute()
208 context->getInputBuffer<_Float16>(kWeightsTensor), in execute()
209 context->getInputShape(kWeightsTensor), in execute()
218 context->getInputBuffer<uint8_t>(kWeightsTensor), in execute()
219 context->getInputShape(kWeightsTensor), in execute()
228 context->getInputBuffer<int8_t>(kWeightsTensor), in execute()
229 context->getInputShape(kWeightsTensor), in execute()
|
| D | UnidirectionalSequenceRNN.cpp | 56 const T* weights = context->getInputBuffer<T>(kWeightsTensor); in executeTyped()
57 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
120 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
|
| D | RNN.cpp | 33 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
54 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | FullyConnected.cpp | 92 const float weightsScale = context->getInputShape(kWeightsTensor).scale; in validate()
124 Shape weights = context->getInputShape(kWeightsTensor); in validate()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | FullyConnected.h | 28 constexpr uint32_t kWeightsTensor = 1; variable
|
| D | UnidirectionalSequenceRNN.h | 26 constexpr uint32_t kWeightsTensor = 1; variable
|
| D | RNN.h | 46 static constexpr int kWeightsTensor = 1; variable
|