Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp47 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer() function
460 const float* auxInput = GetOptionalBuffer<const float>(aux_input_); in Eval()
499 GetOptionalBuffer<const float>(fw_aux_input_to_input_weights_), in Eval()
500 GetOptionalBuffer<const float>(fw_aux_input_to_forget_weights_), in Eval()
501 GetOptionalBuffer<const float>(fw_aux_input_to_cell_weights_), in Eval()
502 GetOptionalBuffer<const float>(fw_aux_input_to_output_weights_), in Eval()
511 GetOptionalBuffer<const float>(fw_input_layer_norm_weights_), in Eval()
512 GetOptionalBuffer<const float>(fw_forget_layer_norm_weights_), in Eval()
513 GetOptionalBuffer<const float>(fw_cell_layer_norm_weights_), in Eval()
514 GetOptionalBuffer<const float>(fw_output_layer_norm_weights_), in Eval()
[all …]
DLSTM.cpp48 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer() function
1031 GetOptionalBuffer<const _Float16>(input_to_input_weights_), in Eval()
1036 GetOptionalBuffer<const _Float16>(recurrent_to_input_weights_), in Eval()
1041 GetOptionalBuffer<const _Float16>(cell_to_input_weights_), in Eval()
1042 GetOptionalBuffer<const _Float16>(cell_to_forget_weights_), in Eval()
1043 GetOptionalBuffer<const _Float16>(cell_to_output_weights_), in Eval()
1049 GetOptionalBuffer<const _Float16>(input_gate_bias_), in Eval()
1053 GetOptionalBuffer<const _Float16>(projection_weights_), in Eval()
1054 GetOptionalBuffer<const _Float16>(projection_bias_), in Eval()
1057 GetOptionalBuffer<const _Float16>(input_layer_norm_weights_), in Eval()
[all …]