Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp152 fw_aux_input_to_cell_weights_ = GetInput(operation, operands, kFwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
308 !IsNullInput(fw_aux_input_to_cell_weights_) && in Prepare()
312 IsNullInput(fw_aux_input_to_cell_weights_) && in Prepare()
353 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_cell_weights_), 2u); in Prepare()
354 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 0), n_fw_cell); in Prepare()
355 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
501 GetOptionalBuffer<const float>(fw_aux_input_to_cell_weights_), in Eval()
625 GetOptionalBuffer<const _Float16>(fw_aux_input_to_cell_weights_), in Eval()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceLSTM.h165 const RunTimeOperandInfo* fw_aux_input_to_cell_weights_; variable