Searched defs:c_loss (Results 1 – 1 of 1) sorted by relevance
339 const auto c_loss = static_cast<T_ACC>(dY_i[l+k]); in compute_gI() local347 const auto c_loss = static_cast<T_ACC>(dY_i[l]); in compute_gI() local443 const auto c_loss = static_cast<T_ACC>(dY_i_vec_reg.val[k]); in layer_norm_grad_input_kernel_vectorized() local453 const auto c_loss = static_cast<T_ACC>(dY_i[l]); in layer_norm_grad_input_kernel_vectorized() local1054 const T_ACC c_loss = static_cast<T_ACC>((idx<N) ? k_dout[idx] : T(0)); in cuComputeGradInput() local1062 const T_ACC c_loss = static_cast<T_ACC>((idx<N) ? k_dout[idx] : T(0)); in cuComputeGradInput() local1109 const T_ACC c_loss = static_cast<T_ACC>(k_dout[l]); in cuComputeGradInput() local1119 const T_ACC c_loss = static_cast<T_ACC>(k_dout[l]); in cuComputeGradInput() local