Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/overrides_benchmark/
H A Dcommon.py12 class WithTorchFunction: class
25 return WithTorchFunction(args[0]._tensor + args[1]._tensor)
H A Dbench.py4 from common import SubTensor, SubWithTorchFunction, WithTorchFunction
53 types = torch.tensor, SubTensor, WithTorchFunction, SubWithTorchFunction
H A Dpyspybench.py3 from common import SubTensor, SubWithTorchFunction, WithTorchFunction # noqa: F401
H A DREADME.md25 py-spy record -o overridden.svg --native -- python pyspybench.py WithTorchFunction
/aosp_15_r20/external/pytorch/.ci/pytorch/
H A Dtest.sh1029 python pyspybench.py WithTorchFunction -n 1