Searched defs:grad_weight_ (Results 1 – 3 of 3) sorted by relevance
294 const Tensor &grad_weight_, in slow_conv2d_grad_weight()
610 Tensor grad_weight_; in batch_norm_backward_cuda_template() local786 Tensor grad_weight_; in batch_norm_backward_reduce_cuda_template() local
1070 Tensor grad_weight_ = at::empty_like(grad_weight); in raw_cudnn_convolution_backward_weight_out_v7() local