Searched refs:kWeightsTensor (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | FullyConnected.cpp | 33 constexpr uint32_t kWeightsTensor = 1; variable
168 const float weightsScale = context->getInputShape(kWeightsTensor).scale; in validate()
195 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
233 context->getInputBuffer<float>(kWeightsTensor), in execute()
234 context->getInputShape(kWeightsTensor), in execute()
243 context->getInputBuffer<_Float16>(kWeightsTensor), in execute()
244 context->getInputShape(kWeightsTensor), in execute()
253 context->getInputBuffer<uint8_t>(kWeightsTensor), in execute()
254 context->getInputShape(kWeightsTensor), in execute()
|
| D | UnidirectionalSequenceRNN.cpp | 28 constexpr uint32_t kWeightsTensor = 1; variable
60 const T* weights = context->getInputBuffer<T>(kWeightsTensor); in executeTyped()
61 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
133 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
|
| D | RNN.h | 38 static constexpr int kWeightsTensor = 1; // Optional variable
|
| D | RNN.cpp | 32 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
57 GetInput(operation, operands, kWeightsTensor); in Prepare()
|