Searched refs:NoneLayout (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_ir.py | 22 NoneLayout,
422 MutationOutput(NoneLayout(inputs[0].get_device()), inputs[0], self),
423 MutationOutput(NoneLayout(inputs[1].get_device()), inputs[1], self),
477 kernel_layout=NoneLayout(inputs[1].get_device()), # type: ignore[arg-type]
1084 layout=NoneLayout(qaccum.get_device()),
1752 layout=NoneLayout(other.get_device()),
|
| D | ir.py | 2920 class NoneLayout(IRNode): class
4731 NoneLayout(device), # type: ignore[arg-type]
4830 NoneLayout(x.get_device()), # type: ignore[arg-type]
4887 NoneLayout(dst.get_device()), # type: ignore[arg-type]
4927 NoneLayout(variable.get_device()), # type: ignore[arg-type]
5024 NoneLayout(x.get_device()), # type: ignore[arg-type]
5074 NoneLayout(x.get_device()), # type: ignore[arg-type]
5133 …super().__init__(None, NoneLayout(torch.device("cpu")), self.unwrap_storage([data])) # type: igno…
5159 NoneLayout(torch.device("cpu")), # type: ignore[arg-type]
5644 NoneLayout(device),
[all …]
|
| D | scheduler.py | 132 if isinstance(self.node.layout, ir.NoneLayout):
595 elif isinstance(buf.layout, ir.NoneLayout):
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | wrapper.py | 758 no_return = isinstance(extern_kernel.layout, ir.NoneLayout)
1868 if isinstance(layout, ir.NoneLayout):
|