Searched refs:ArgForPyTorch (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | test_case_gen.py | 268 class ArgForPyTorch: class
284 arg.cpp_arg if isinstance(arg, ArgForPyTorch) else arg
286 if not isinstance(arg, ArgForPyTorch) or arg.used_in_cpp()
289 argument_to_pytorch(arg) for arg in args if not isinstance(arg, ArgForPyTorch)
294 if isinstance(arg, ArgForPyTorch) and arg.used_in_torch()
400 ArgForPyTorch(
423 ArgForPyTorch(1, "dim", 1),
424 ArgForPyTorch(EnumArg("ScalarType::Float"), "dtype", torch.float),
466 ArgForPyTorch(
491 torch.ones(10, 10), ArgForPyTorch(StringArg("tanh"), None, None), torch_fn=m
[all …]
|