Searched defs:save_invstd_a (Results 1 – 2 of 2) sorted by relevance
41 auto save_invstd_a = conditional_accessor_1d<const param_t>(save_invstd); in batch_norm_cpu_collect_linear_and_constant_terms() local431 auto save_invstd_a = conditional_accessor_1d<const scalar_t>(save_invstd); in batch_norm_cpu_backward_contiguous_impl() local1021 auto save_invstd_a = conditional_accessor_1d<const param_t>(save_invstd); in batch_norm_cpu_backward_contiguous_internal() local1140 auto save_invstd_a = conditional_accessor_1d<const param_t>(save_invstd); in batch_norm_cpu_backward_channels_last_internal() local
349 auto save_invstd_a = conditional_accessor_1d<const param_t>(save_invstd); in batch_norm_backward_cpu_template() local