Home
last modified time | relevance | path

Searched refs:kWeightsTensor (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/common/operations/
DFullyConnected.cpp39 constexpr uint32_t kWeightsTensor = 1; variable
250 const float weightsScale = context->getInputShape(kWeightsTensor).scale; in validate()
283 Shape weights = context->getInputShape(kWeightsTensor); in validate()
294 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
308 context->getInputBuffer<float>(kWeightsTensor), in execute()
309 context->getInputShape(kWeightsTensor), in execute()
318 context->getInputBuffer<_Float16>(kWeightsTensor), in execute()
319 context->getInputShape(kWeightsTensor), in execute()
328 context->getInputBuffer<uint8_t>(kWeightsTensor), in execute()
329 context->getInputShape(kWeightsTensor), in execute()
[all …]
DUnidirectionalSequenceRNN.cpp33 constexpr uint32_t kWeightsTensor = 1; variable
69 const T* weights = context->getInputBuffer<T>(kWeightsTensor); in executeTyped()
70 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped()
155 Shape weights = context->getInputShape(kWeightsTensor); in prepare()
DRNN.h40 static constexpr int kWeightsTensor = 1; variable
DRNN.cpp37 weights_ = GetInput(operation, operands, kWeightsTensor); in RNN()
58 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()