unpack_expanded_weight_or_tensor (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:unpack_expanded_weight_or_tensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/
H A D	layer_norm_expanded_weights.py	`13 unpack_expanded_weight_or_tensor, 59 weight_ = unpack_expanded_weight_or_tensor(ctx.weight) 60 bias_ = unpack_expanded_weight_or_tensor(ctx.bias)`
H A D	linear_expanded_weights.py	`12 unpack_expanded_weight_or_tensor, 46 results.append(grad_output.matmul(unpack_expanded_weight_or_tensor(weight)))`
H A D	instance_norm_expanded_weights.py	`13 unpack_expanded_weight_or_tensor, 51 weight_ = unpack_expanded_weight_or_tensor(`
H A D	group_norm_expanded_weights.py	`14 unpack_expanded_weight_or_tensor, 59 weight_c = unpack_expanded_weight_or_tensor(`
H A D	expanded_weights_utils.py	`126 unpacked = unpack_expanded_weight_or_tensor(maybe_expanded_weight) 149 def unpack_expanded_weight_or_tensor(maybe_expanded_weight, func=lambda x: x): function`
H A D	conv_utils.py	`11 unpack_expanded_weight_or_tensor, 172 weight_ = unpack_expanded_weight_or_tensor(ctx.weight)`
/aosp_15_r20/external/pytorch/test/
H A D	test_expanded_weights.py	17 unpack_expanded_weight_or_tensor, 165 unpack_expanded_weight_or_tensor( 171 self.assertEqual(input, unpack_expanded_weight_or_tensor(input)) 172 self.assertTrue(unpack_expanded_weight_or_tensor(4) is None) 177 unpack_expanded_weight_or_tensor( 183 self.assertTrue(unpack_expanded_weight_or_tensor(input, lambda x: x is input)) 185 unpack_expanded_weight_or_tensor(4, lambda x: x is input) is None 194 unpack_expanded_weight_or_tensor(input) 200 unpack_expanded_weight_or_tensor(input, lambda x: x is input)