Searched refs:mean_std_normalization_layer (Results 1 – 4 of 4) sorted by relevance
37 SimpleTensor<T> mean_std_normalization_layer(const SimpleTensor<T> &src, float epsilon) in mean_std_normalization_layer() function67 SimpleTensor<uint8_t> mean_std_normalization_layer(const SimpleTensor<uint8_t> &src, float epsilon) in mean_std_normalization_layer() function70 SimpleTensor<float> dst_tmp = mean_std_normalization_layer<float>(src_tmp, epsilon); in mean_std_normalization_layer()75 template SimpleTensor<float> mean_std_normalization_layer(const SimpleTensor<float> &src, float eps…76 template SimpleTensor<half> mean_std_normalization_layer(const SimpleTensor<half> &src, float epsil…
39 SimpleTensor<T> mean_std_normalization_layer(const SimpleTensor<T> &src, float epsilon = 1e-8f);
424 forget_gate = reference::mean_std_normalization_layer(forget_gate); in compute_reference()454 input_gate = reference::mean_std_normalization_layer(input_gate); in compute_reference()471 cell_state_out = reference::mean_std_normalization_layer(cell_state_out); in compute_reference()499 output = reference::mean_std_normalization_layer(output); in compute_reference()
121 return reference::mean_std_normalization_layer(ref_src, epsilon); in compute_reference()