Home
last modified time | relevance | path

Searched defs:VkStorageType (Results 1 – 3 of 3) 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
46 storage: VkStorageType,
91 default_storage_type: VkStorageType = VkStorageType.TEXTURE_3D,
163 storage: VkStorageType,
/aosp_15_r20/external/executorch/backends/vulkan/
H A Dop_registry.py106 optimal_storage: Optional[VkStorageType] = None,
159 def propose_memory_layout(self, storage: VkStorageType) -> Optional[VkMemoryLayout]:
181 def supported_memory_layouts(self, storage: VkStorageType) -> Set[VkMemoryLayout]:
/aosp_15_r20/external/executorch/backends/vulkan/serialization/
H A Dvulkan_graph_schema.py34 class VkStorageType(IntEnum): class