Home
last modified time | relevance | path

Searched refs:host_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h269 auto host_size = host_src.size() * sizeof(T); in SynchronousMemcpyH2D() local
270 CHECK(device_dst->size() == 0 || device_dst->size() >= host_size); in SynchronousMemcpyH2D()
271 return SynchronousMemcpyH2D(host_src.begin(), host_size, device_dst); in SynchronousMemcpyH2D()
284 auto host_size = host_dst.size() * sizeof(T); in SynchronousMemcpyD2H() local
285 CHECK(device_src.size() == 0 || host_size >= device_src.size()); in SynchronousMemcpyD2H()
286 return SynchronousMemcpyD2H(device_src, host_size, host_dst.begin()); in SynchronousMemcpyD2H()
Dstream.h1738 auto host_size = host_dst.size() * sizeof(T); in ThenMemcpyD2H() local
1739 CHECK(gpu_src.size() == 0 || host_size >= gpu_src.size()); in ThenMemcpyD2H()
1740 return ThenMemcpy(host_dst.begin(), gpu_src, host_size); in ThenMemcpyD2H()
1749 auto host_size = host_src.size() * sizeof(T); in ThenMemcpyH2D() local
1750 CHECK(gpu_dst->size() == 0 || gpu_dst->size() >= host_size); in ThenMemcpyH2D()
1751 return ThenMemcpy(gpu_dst, host_src.begin(), host_size); in ThenMemcpyH2D()
/external/mesa3d/src/freedreno/vulkan/
Dtu_descriptor_set.c565 uint64_t host_size = pCreateInfo->maxSets * sizeof(struct tu_descriptor_set); in tu_CreateDescriptorPool() local
566 host_size += sizeof(struct tu_bo*) * bo_count; in tu_CreateDescriptorPool()
567 host_size += A6XX_TEX_CONST_DWORDS * 4 * dynamic_count; in tu_CreateDescriptorPool()
568 size += host_size; in tu_CreateDescriptorPool()
/external/mesa3d/src/broadcom/vulkan/
Dv3dv_descriptor_set.c438 uint64_t host_size = in v3dv_CreateDescriptorPool() local
440 host_size += sizeof(struct v3dv_descriptor) * descriptor_count; in v3dv_CreateDescriptorPool()
441 size += host_size; in v3dv_CreateDescriptorPool()
/external/mesa3d/src/amd/vulkan/
Dradv_descriptor_set.c721 uint64_t host_size = pCreateInfo->maxSets * sizeof(struct radv_descriptor_set); in radv_CreateDescriptorPool() local
722 host_size += sizeof(struct radeon_winsys_bo*) * bo_count; in radv_CreateDescriptorPool()
723 host_size += sizeof(struct radv_descriptor_range) * range_count; in radv_CreateDescriptorPool()
724 size += host_size; in radv_CreateDescriptorPool()