Home
last modified time | relevance | path

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

/external/pytorch/torch/fx/passes/backends/
Dcudagraphs.py4 from torch.fx.passes.operator_support import OperatorSupport
11 class CudaGraphsSupport(OperatorSupport):
/external/pytorch/test/fx/
Dtest_partitioner_order.py8 from torch.fx.passes.operator_support import OperatorSupport
12 class DummyDevOperatorSupport(OperatorSupport):
/external/pytorch/torch/fx/passes/
Doperator_support.py39 class OperatorSupport(OperatorSupportBase): class
/external/pytorch/test/export/
Dtest_passes.py47 from torch.fx.passes.operator_support import OperatorSupport
68 class _AddOperatorSupport(OperatorSupport):
73 class _AtenAddOperatorSupport(OperatorSupport):
/external/pytorch/test/
Dtest_fx_passes.py12 from torch.fx.passes.operator_support import OperatorSupport
243 class MockOperatorSupport(OperatorSupport):
/external/pytorch/torch/onnx/_internal/
Donnxruntime.py30 from torch.fx.passes.operator_support import OperatorSupport
179 class OrtOperatorSupport(OperatorSupport):