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)
|