Searched refs:TwoTensor_foo (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_custom_ops.py | 2670 def TwoTensor_foo(cls, func, types, args, kwargs): function
2676 m._register_torch_dispatch_rule("foo", TwoTensor, TwoTensor_foo)
2696 def TwoTensor_foo(mode, func, types, args, kwargs): function
2701 m._register_torch_dispatch_rule("foo", TwoTensorMode, TwoTensor_foo)
|