Searched refs:get_node_memory_layout (Results 1 – 2 of 2) sorted by relevance
184 layout = utils.get_node_memory_layout(arg)243 arg_layout = utils.get_node_memory_layout(arg)
231 def get_node_memory_layout(node: torch.fx.Node) -> Optional[VkMemoryLayout]: function