Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp42 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer() function
297 GetOptionalBuffer<const float>(aux_input_), in Eval()
298 GetOptionalBuffer<const float>(fw_aux_input_to_input_weights_), in Eval()
299 GetOptionalBuffer<const float>(fw_aux_input_to_forget_weights_), in Eval()
300 GetOptionalBuffer<const float>(fw_aux_input_to_cell_weights_), in Eval()
301 GetOptionalBuffer<const float>(fw_aux_input_to_output_weights_), in Eval()
310 GetOptionalBuffer<const float>(fw_input_layer_norm_weights_), in Eval()
311 GetOptionalBuffer<const float>(fw_forget_layer_norm_weights_), in Eval()
312 GetOptionalBuffer<const float>(fw_cell_layer_norm_weights_), in Eval()
313 GetOptionalBuffer<const float>(fw_output_layer_norm_weights_), in Eval()
[all …]
DLSTM.cpp43 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer() function
991 GetOptionalBuffer<const _Float16>(input_to_input_weights_), in Eval()
996 GetOptionalBuffer<const _Float16>(recurrent_to_input_weights_), in Eval()
1001 GetOptionalBuffer<const _Float16>(cell_to_input_weights_), in Eval()
1002 GetOptionalBuffer<const _Float16>(cell_to_forget_weights_), in Eval()
1003 GetOptionalBuffer<const _Float16>(cell_to_output_weights_), in Eval()
1009 GetOptionalBuffer<const _Float16>(input_gate_bias_), in Eval()
1013 GetOptionalBuffer<const _Float16>(projection_weights_), in Eval()
1014 GetOptionalBuffer<const _Float16>(projection_bias_), in Eval()
1017 GetOptionalBuffer<const _Float16>(input_layer_norm_weights_), in Eval()
[all …]