Searched defs:memory_layout (Results 1 – 9 of 9) sorted by relevance
61 const api::GPUMemoryLayout memory_layout, in calc_strides()94 const api::GPUMemoryLayout memory_layout, in calc_gpu_sizes()156 const api::GPUMemoryLayout memory_layout) { in create_image_extents()224 const api::GPUMemoryLayout memory_layout, in vTensor()256 const api::GPUMemoryLayout memory_layout) in vTensor()
82 const utils::GPUMemoryLayout memory_layout) in TmpTensor()241 const utils::GPUMemoryLayout memory_layout, in add_tensor()272 const utils::GPUMemoryLayout memory_layout, in add_tensor()281 const utils::GPUMemoryLayout memory_layout) { in add_tensor_like()287 const utils::GPUMemoryLayout memory_layout) { in add_tensor_like()
74 utils::GPUMemoryLayout memory_layout) { in set_memory_layout_override()
1968 utils::GPUMemoryLayout memory_layout = in TEST() local2050 utils::GPUMemoryLayout memory_layout = in TEST() local2113 utils::GPUMemoryLayout memory_layout = in TEST() local2236 utils::GPUMemoryLayout memory_layout = in TEST() local2318 utils::GPUMemoryLayout memory_layout = in TEST() local2437 utils::GPUMemoryLayout memory_layout = in run_from_gpu_test()2492 utils::GPUMemoryLayout memory_layout = in round_trip_test()2544 utils::GPUMemoryLayout memory_layout = in compute_graph_round_trip_test()2659 utils::GPUMemoryLayout memory_layout) { in test_binary_op()2746 utils::GPUMemoryLayout memory_layout, in test_mm()
110 def run_test(memory_layout): argument
16 const utils::GPUMemoryLayout memory_layout) { in calculate_sizes()440 const utils::GPUMemoryLayout memory_layout, in vTensor()487 const utils::GPUMemoryLayout memory_layout) in vTensor()
138 utils::GPUMemoryLayout memory_layout = in get_graph_config() local195 utils::GPUMemoryLayout memory_layout = in add_tensor_to_graph() local
61 memory_layout: VkMemoryLayout = VkMemoryLayout.DEFAULT_LAYOUT variable in VkTensor
92 const utils::GPUMemoryLayout memory_layout) { in prepack_biases()