Searched refs:module_inputs_torch_nn_InstanceNormNd (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_modules.py | 1856 def module_inputs_torch_nn_InstanceNormNd(module_info, device, dtype, requires_grad, training, **kw… function
4078 module_inputs_func=partial(module_inputs_torch_nn_InstanceNormNd, N=1),
4085 module_inputs_func=partial(module_inputs_torch_nn_InstanceNormNd, N=2),
4092 module_inputs_func=partial(module_inputs_torch_nn_InstanceNormNd, N=3),
|