Searched defs:VkMemoryLayout (Results 1 – 4 of 4) sorted by relevance
36 node: torch.fx.Node, storage: VkStorageType, layout: VkMemoryLayout47 layout: VkMemoryLayout,92 default_memory_layout: VkMemoryLayout = VkMemoryLayout.TENSOR_WIDTH_PACKED,
44 class VkMemoryLayout(IntEnum): class
133 def required_image_extents(sizes: torch.Size, layout: VkMemoryLayout) -> ImageExtents:
107 optimal_layout: Optional[VkMemoryLayout] = None,