Searched refs:conv_transpose_ref (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 3854 def conv_transpose_ref(input, weight, bias, stride=1, padding=0, function
14859 ref=partial(conv_transpose_ref, fn=torch.nn.functional.conv_transpose1d),
14905 ref=partial(conv_transpose_ref, fn=torch.nn.functional.conv_transpose2d),
14953 ref=partial(conv_transpose_ref, fn=torch.nn.functional.conv_transpose3d),
|