Searched defs:saved_v (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | WeightNormKernel.cpp | 180 const TensorBase& saved_v, 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()
317 const TensorBase& saved_v, in weight_norm_backward_last_dim_kernel()
417 const TensorBase& saved_v, in weight_norm_backward_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | WeightNorm.cu | 220 const scalar_t* __restrict__ saved_v, in weight_norm_bwd_first_dim_kernel()
281 const scalar_t* __restrict__ saved_v, in weight_norm_bwd_last_dim_kernel()
430 const Tensor & saved_v, in weight_norm_backward_cuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | WeightNorm.cpp | 67 const Tensor& saved_v, in weight_norm_backward_cpu()
118 const Tensor & saved_v, in _weight_norm_differentiable_backward()
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | modules.cpp | 3535 torch::Tensor saved_v; in _multihead_attn_test_helper() local
|