Home
last modified time | relevance | path

Searched defs:gpu_src_ptr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc160 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in SetProtoFromGPU() local
214 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in DeviceToDeviceCopy() local
281 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToCPU() local
440 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToSameGPU() local
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc420 se::DeviceMemoryBase gpu_src_ptr(DMAHelper::base(ctx->mutable_output(0)), in DoAddChain() local