batch_norm_update_stats_and_invert (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A D	Normalization.cu	`380 void batch_norm_update_stats_and_invert( in batch_norm_update_stats_and_invert() function 444 batch_norm_update_stats_and_invert( in batch_norm_cuda_out()`