Searched defs:saved_norm (Results 1 – 2 of 2) sorted by relevance
182 const TensorBase& saved_norm, in weight_norm_backward_first_dim_kernel()319 const TensorBase& saved_norm, in weight_norm_backward_last_dim_kernel()419 const TensorBase& saved_norm, in weight_norm_backward_kernel()
69 const Tensor& saved_norm, in weight_norm_backward_cpu()