Searched defs:only_cuda (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 197 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True, significant_only=True argument
215 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True argument
221 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True argument
|