Searched refs:get_layer_norm_output (Results 1 – 4 of 4) sorted by relevance
121 CLTensor *out = &get_layer_norm_output(g); in configure_layer_norm()339 forget_activation_input = &get_layer_norm_output(LayerNormGate::Forget); in configure()375 cell_activation_input = &get_layer_norm_output(LayerNormGate::Cell); in configure()433 input_activation_input = &get_layer_norm_output(LayerNormGate::Input); in configure()500 output_activation_input = &get_layer_norm_output(LayerNormGate::Output); in configure()
69 Tensor &out = get_layer_norm_output(g); in configure_layer_norm()415 forget_activation_input = &get_layer_norm_output(LayerNormGate::Forget); in configure()450 cell_activation_input = &get_layer_norm_output(LayerNormGate::Cell); in configure()508 input_activation_input = &get_layer_norm_output(LayerNormGate::Input); in configure()573 output_activation_input = &get_layer_norm_output(LayerNormGate::Output); in configure()
447 inline Tensor &get_layer_norm_output(LayerNormGate g) in get_layer_norm_output() function
485 inline CLTensor &get_layer_norm_output(LayerNormGate g) in get_layer_norm_output() function