Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp159 bw_aux_input_to_cell_weights_ = GetInput(operation, operands, kBwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
318 !IsNullInput(bw_aux_input_to_cell_weights_) && in Prepare()
322 IsNullInput(bw_aux_input_to_cell_weights_) && in Prepare()
373 NN_RET_CHECK_EQ(NumDimensions(bw_aux_input_to_cell_weights_), 2); in Prepare()
374 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_cell_weights_, 0), n_bw_cell); in Prepare()
375 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
554 GetOptionalBuffer<const float>(bw_aux_input_to_cell_weights_), in Eval()
679 GetOptionalBuffer<const _Float16>(bw_aux_input_to_cell_weights_), in Eval()
DBidirectionalSequenceLSTM.h172 const RunTimeOperandInfo* bw_aux_input_to_cell_weights_; variable