Home
last modified time | relevance | path

Searched refs:NonOwningLayout (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dtriton_utils.py102 if isinstance(layout, torch._inductor.ir.NonOwningLayout):
Dwrapper.py1870 if isinstance(layout, ir.NonOwningLayout):
/external/pytorch/torch/_inductor/fx_passes/
Dpost_grad.py536 src.layout = ir.NonOwningLayout(dst)
/external/pytorch/torch/_inductor/
Dir.py2895 class NonOwningLayout(Layout): class
3071 if not isinstance(self.layout, (MultiOutputLayout, NonOwningLayout)):
3110 if isinstance(self.layout, NonOwningLayout):
3910 src.data.layout = NonOwningLayout(dst)
4310 while isinstance(x.get_layout(), NonOwningLayout):