Searched defs:grad_g (Results 1 – 3 of 3) sorted by relevance
76 auto grad_g = at::empty_like(saved_g, at::MemoryFormat::Contiguous); in weight_norm_backward_cpu() local151 auto grad_g = per_dim_sums/norms; in _weight_norm_differentiable_backward() local157 auto grad_g = per_dim_sums/norms; in _weight_norm_differentiable_backward() local
218 scalar_t* __restrict__ grad_g, in weight_norm_bwd_first_dim_kernel()279 scalar_t* __restrict__ grad_g, in weight_norm_bwd_last_dim_kernel()443 auto grad_g = at::empty_like(saved_g, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in weight_norm_backward_cuda() local
178 TensorBase& grad_g, in weight_norm_backward_first_dim_kernel()315 TensorBase& grad_g, in weight_norm_backward_last_dim_kernel()415 TensorBase& grad_g, in weight_norm_backward_kernel()