Searched defs:weight_output_dim (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xnnpack_integration.py | 42 def test_linear(self, batch_size, data_shape, weight_output_dim, use_bias): argument
64 def test_linear_1d_input(self, input_size, weight_output_dim, use_bias): argument
250 def test_linear(self, batch_size, data_shape, weight_output_dim, use_bias): argument
|