Searched defs:gpu_buffer (Results 1 – 5 of 5) sorted by relevance
68 tfrt::AsyncValueRef<tfrt::gpu::GpuBuffer> gpu_buffer, tfrt::DType dtype, in MoveGpuBufferToTFTensor()71 auto* gpu_buffer = reinterpret_cast<tfrt::AsyncValue*>(arg); in MoveGpuBufferToTFTensor() local
49 tfrt::AsyncValueRef<tfrt::gpu::GpuBuffer> gpu_buffer = in CopyRefGpuTensorToRuntimeFallbackTensor() local139 llvm::Expected<tfrt::gpu::GpuBuffer> gpu_buffer = in ConvertRuntimeFallbackTensorToDenseGpuTensor() local
32 api::ParamsBuffer gpu_buffer; member
106 Buffer gpu_buffer; in CreateCLObject() local
67 Buffer gpu_buffer; in CreateMetalObject() local