set_grad_sample_if_exists (reference) in projects: aosp_15_r20

Searched refs:set_grad_sample_if_exists (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/
H A D	linear_expanded_weights.py	`11 set_grad_sample_if_exists, 56 set_grad_sample_if_exists( 59 set_grad_sample_if_exists(`
H A D	layer_norm_expanded_weights.py	`10 set_grad_sample_if_exists, 81 set_grad_sample_if_exists(ctx.weight, weight_per_sample_grad) 83 set_grad_sample_if_exists(`
H A D	instance_norm_expanded_weights.py	`11 set_grad_sample_if_exists, 91 set_grad_sample_if_exists( 97 set_grad_sample_if_exists(`
H A D	group_norm_expanded_weights.py	`12 set_grad_sample_if_exists, 94 set_grad_sample_if_exists( 101 set_grad_sample_if_exists(`
H A D	conv_utils.py	`10 set_grad_sample_if_exists, 200 set_grad_sample_if_exists(ctx.weight, weight_grad_sample) 201 set_grad_sample_if_exists(`
H A D	embedding_expanded_weights.py	`10 set_grad_sample_if_exists, 82 set_grad_sample_if_exists(weight, weight_per_sample_grad)`
H A D	expanded_weights_utils.py	`125 def set_grad_sample_if_exists(maybe_expanded_weight, per_sample_grad_fn): function`
/aosp_15_r20/external/pytorch/test/
H A D	test_expanded_weights.py	`14 set_grad_sample_if_exists, 138 set_grad_sample_if_exists(expanded_weight, test_fn) 143 set_grad_sample_if_exists(basic_tensor, test_fn) 147 set_grad_sample_if_exists(non_tensor, test_fn) 159 set_grad_sample_if_exists(grad_tensor, test_fn)`