Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp153 fw_aux_input_to_cell_weights_ = GetInput(operation, operands, kFwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
309 !IsNullInput(fw_aux_input_to_cell_weights_) && in Prepare()
313 IsNullInput(fw_aux_input_to_cell_weights_) && in Prepare()
354 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_cell_weights_), 2); in Prepare()
355 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 0), n_fw_cell); in Prepare()
356 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
502 GetOptionalBuffer<const float>(fw_aux_input_to_cell_weights_), in Eval()
626 GetOptionalBuffer<const _Float16>(fw_aux_input_to_cell_weights_), in Eval()
DBidirectionalSequenceLSTM.h168 const RunTimeOperandInfo* fw_aux_input_to_cell_weights_; variable