Home
last modified time | relevance | path

Searched refs:bw_input_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp121 bw_input_to_output_weights_ = GetInput(operation, operands, kBwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
281 const uint32_t n_bw_cell = SizeOfDimension(bw_input_to_output_weights_, 0); in Prepare()
398 bw_input_to_cell_weights_, bw_input_to_output_weights_, in Prepare()
542 GetBuffer<const float>(bw_input_to_output_weights_), in Eval()
543 bw_input_to_output_weights_->shape(), in Eval()
667 GetBuffer<const _Float16>(bw_input_to_output_weights_), in Eval()
668 bw_input_to_output_weights_->shape(), in Eval()
DBidirectionalSequenceLSTM.h209 const RunTimeOperandInfo* bw_input_to_output_weights_; variable