Home
last modified time | relevance | path

Searched refs:TwoTensor_foo (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_custom_ops.py2670 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)