Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dmkldnn_ir.py20 MultiOutputLayout,
1850 MultiOutputLayout(x.get_device()),
H A Dir.py3071 if not isinstance(self.layout, (MultiOutputLayout, NonOwningLayout)):
5655 MultiOutputLayout(device),
5733 class MultiOutputLayout(IRNode): class
6596 MultiOutputLayout(x.get_device()),
7627 layout: MultiOutputLayout, argument
7706 layout=MultiOutputLayout(device),
7747 layout: MultiOutputLayout, argument
7825 layout=MultiOutputLayout(device),
8312 MultiOutputLayout(device),
H A Dgraph.py717 from .ir import MultiOutputLayout
723 if isinstance(getattr(buf, "layout", None), MultiOutputLayout):
H A Dscheduler.py48 from .ir import ComputedBuffer, MultiOutput, MultiOutputLayout
428 ir.MultiOutputLayout,
581 if isinstance(buf.layout, MultiOutputLayout):
3586 and not isinstance(buffer.layout, MultiOutputLayout)
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dsimd.py1192 if not isinstance(buf.get_layout(), ir.MultiOutputLayout)
H A Dwrapper.py415 if isinstance(self.node.layout, ir.MultiOutputLayout):
H A Dcpp_wrapper_cpu.py1474 if isinstance(buffer.get_layout(), ir.MultiOutputLayout)