Searched refs:host_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | tuple_thunk.cc | 37 auto host_size = size * sizeof(void*); in ExecuteOnStream() local
41 tuple_element_buffer_addresses.get(), host_size) in ExecuteOnStream()
46 host_size); in ExecuteOnStream()
|
| D | infeed_thunk.cc | 88 auto host_size = inner_tuple_element_addresses.size() * sizeof(void*); in ExecuteOnStream() local
93 inner_tuple_element_addresses.data(), host_size); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 281 auto host_size = host_src.size() * sizeof(T); in SynchronousMemcpyH2D() local
282 CHECK(device_dst->size() == 0 || device_dst->size() >= host_size); in SynchronousMemcpyH2D()
283 return SynchronousMemcpyH2D(host_src.begin(), host_size, device_dst); in SynchronousMemcpyH2D()
296 auto host_size = host_dst.size() * sizeof(T); in SynchronousMemcpyD2H() local
297 CHECK(device_src.size() == 0 || host_size >= device_src.size()); in SynchronousMemcpyD2H()
298 return SynchronousMemcpyD2H(device_src, host_size, host_dst.begin()); in SynchronousMemcpyD2H()
|
| D | stream.h | 1696 auto host_size = host_dst.size() * sizeof(T); in ThenMemcpyD2H() local
1697 CHECK(gpu_src.size() == 0 || host_size >= gpu_src.size()); in ThenMemcpyD2H()
1698 return ThenMemcpy(host_dst.begin(), gpu_src, host_size); in ThenMemcpyD2H()
1707 auto host_size = host_src.size() * sizeof(T); in ThenMemcpyH2D() local
1708 CHECK(gpu_dst->size() == 0 || gpu_dst->size() >= host_size); in ThenMemcpyH2D()
1709 return ThenMemcpy(gpu_dst, host_src.begin(), host_size); in ThenMemcpyH2D()
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_descriptor_set.c | 433 uint64_t host_size = pCreateInfo->maxSets * sizeof(struct radv_descriptor_set); in radv_CreateDescriptorPool() local
434 host_size += sizeof(struct radeon_winsys_bo*) * bo_count; in radv_CreateDescriptorPool()
435 host_size += sizeof(struct radv_descriptor_range) * range_count; in radv_CreateDescriptorPool()
436 size += host_size; in radv_CreateDescriptorPool()
|