Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_impls.py797 def make_fast_binary_impl(slow_ref): function
949 make_fast_binary_impl(torch._refs.add)
952 make_fast_binary_impl(torch._refs.sub)
954 …register_fast_op_impl(torch.ops.aten.mul.Tensor)(make_fast_binary_impl(torch._refs.mul)) # type: …
956 make_fast_binary_impl(torch._refs.div)