Searched defs:saved_g (Results 1 – 3 of 3) sorted by relevance
221 const scalar_t* __restrict__ saved_g, in weight_norm_bwd_first_dim_kernel()282 const scalar_t* __restrict__ saved_g, in weight_norm_bwd_last_dim_kernel()431 const Tensor & saved_g, in weight_norm_backward_cuda()
68 const Tensor& saved_g, in weight_norm_backward_cpu()119 const Tensor & saved_g, in _weight_norm_differentiable_backward()
181 const TensorBase& saved_g, in weight_norm_backward_first_dim_kernel()318 const TensorBase& saved_g, in weight_norm_backward_last_dim_kernel()418 const TensorBase& saved_g, in weight_norm_backward_kernel()