Home
last modified time | relevance | path

Searched defs:foo_op (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/custom_operator/
H A Dtest_infer_schema_annotation.py17 def foo_op(x: torch.Tensor) -> torch.Tensor: function
30 def foo_op(x: int) -> int: function
117 def foo_op(x: Tensor) -> Tensor: function
189 def foo_op(x: D) -> Tensor: # noqa: F821 function
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_fx_passes.py64 def foo_op(x: torch.Tensor) -> torch.Tensor: function