Searched defs:grad_w (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | WeightNormKernel.cpp | 179 const TensorBase& grad_w, in weight_norm_backward_first_dim_kernel()
195 [](Vec grad_w, Vec saved_v) { return grad_w * saved_v; }, in weight_norm_backward_first_dim_kernel()
215 [a, b](Vec grad_w, Vec v) { return Vec(a) * grad_w - Vec(b) * v; }, in weight_norm_backward_first_dim_kernel()
232 [](Vec out, Vec grad_w, Vec v) { return out + grad_w * v; }, in sum_product_per_row()
276 [](Vec grad_w, Vec v, Vec a, Vec b) { return a * grad_w - b * v; }, in apply_per_row_backward()
316 const TensorBase& grad_w, in weight_norm_backward_last_dim_kernel()
416 const TensorBase& grad_w, in weight_norm_backward_kernel()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | WeightNorm.cpp | 66 const Tensor& grad_w, in weight_norm_backward_cpu()
117 (const Tensor & grad_w, in _weight_norm_differentiable_backward()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | WeightNorm.cu | 219 const scalar_t* __restrict__ grad_w, in weight_norm_bwd_first_dim_kernel()
280 const scalar_t* __restrict__ grad_w, in weight_norm_bwd_last_dim_kernel()
429 (const Tensor & grad_w, in weight_norm_backward_cuda()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 825 ideep::tensor grad_w, grad_b; in mkldnn_convolution_backward_weights() local
976 ideep::tensor grad_w, grad_b; in mkldnn_convolution_transpose_backward_weights() local
|
| /external/pytorch/test/cpp/api/ |
| D | nn_utils.cpp | 317 torch::Tensor grad_w = torch::arange(-50., 50).view({10, 10}).div_(5); in TEST_F() local
|