Searched refs:MultiOutputLayout (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | mkldnn_ir.py | 20 MultiOutputLayout, 1850 MultiOutputLayout(x.get_device()),
|
H A D | ir.py | 3071 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 D | graph.py | 717 from .ir import MultiOutputLayout 723 if isinstance(getattr(buf, "layout", None), MultiOutputLayout):
|
H A D | scheduler.py | 48 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 D | simd.py | 1192 if not isinstance(buf.get_layout(), ir.MultiOutputLayout)
|
H A D | wrapper.py | 415 if isinstance(self.node.layout, ir.MultiOutputLayout):
|
H A D | cpp_wrapper_cpu.py | 1474 if isinstance(buffer.get_layout(), ir.MultiOutputLayout)
|