Home
last modified time | relevance | path

Searched defs:VkMemoryLayout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/backends/vulkan/_passes/
H A Dtag_memory_meta_pass.py36 node: torch.fx.Node, storage: VkStorageType, layout: VkMemoryLayout
47 layout: VkMemoryLayout,
92 default_memory_layout: VkMemoryLayout = VkMemoryLayout.TENSOR_WIDTH_PACKED,
/aosp_15_r20/external/executorch/backends/vulkan/serialization/
H A Dvulkan_graph_schema.py44 class VkMemoryLayout(IntEnum): class
/aosp_15_r20/external/executorch/backends/vulkan/
H A Dutils.py133 def required_image_extents(sizes: torch.Size, layout: VkMemoryLayout) -> ImageExtents:
H A Dop_registry.py107 optimal_layout: Optional[VkMemoryLayout] = None,