Searched refs:module_inputs_torch_nn_ConvNd (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_modules.py | 956 def module_inputs_torch_nn_ConvNd(module_info, device, dtype, requires_grad, training, **kwargs): function
3498 module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=1, lazy=False),
3517 module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=2, lazy=False),
3543 module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=3, lazy=False),
3561 … module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=1, lazy=False, transposed=True),
3584 … module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=2, lazy=False, transposed=True),
3618 … module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=3, lazy=False, transposed=True),
3688 module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=1, lazy=True),
3710 module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=2, lazy=True),
3739 module_inputs_func=partial(module_inputs_torch_nn_ConvNd, N=3, lazy=True),
[all …]
|