Searched defs:sum_ptr (Results 1 – 2 of 2) sorted by relevance
102 const auto sum_ptr = reinterpret_cast<const T *>(sum_it.ptr()); in l2_normalize_yz() local
588 scalar_t* sum_ptr = sum_data + tid * n_channel; in batch_norm_cpu_backward_channels_last_impl() local1173 opmath_t* sum_ptr = sum_data + tid * n_channel; in batch_norm_cpu_backward_channels_last_internal() local