Searched refs:GPUDeviceContext (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
35 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()
42 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
50 Status GPUDeviceContext::ThenExecute(Device* device, se::Stream* stream, in ThenExecute()
|
| D | gpu_util.cc | 76 auto gs = static_cast<const GPUDeviceContext*>(ctx)->stream(); in PrepareCopy()
128 static_cast<const GPUDeviceContext*>(device_context) in SetProtoFromGPU()
201 static_cast<const GPUDeviceContext*>(send_dev_context) in DeviceToDeviceCopy()
218 static_cast<const GPUDeviceContext*>(recv_dev_context)->stream(); in DeviceToDeviceCopy()
269 static_cast<const GPUDeviceContext*>(device_context) in CopyGPUTensorToCPU()
314 static_cast<const GPUDeviceContext*>(device_context) in CopyCPUTensorToGPU()
|
| D | gpu_device.cc | 437 new GPUDeviceContext(0, stream_->compute, in Init()
545 GPUDeviceContext* gpu_device_context = device_context_; in Compute()
548 static_cast<GPUDeviceContext*>(context->op_device_context()); in Compute()
618 GPUDeviceContext* gpu_device_context = device_context_; in ComputeAsync()
621 static_cast<GPUDeviceContext*>(context->op_device_context()); in ComputeAsync()
1054 const GPUDeviceContext* gpu_dc = static_cast<GPUDeviceContext*>(dc); in ReinitializeGpuDevice()
|
| D | gpu_device.h | 146 GPUDeviceContext* device_context_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 29 class GPUDeviceContext : public DeviceContext {
32 GPUDeviceContext(int stream_id, se::Stream* stream, in GPUDeviceContext() function
49 ~GPUDeviceContext() override {} in ~GPUDeviceContext()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 76 context(static_cast<GPUDeviceContext*>(info->default_context)), in Participant()
107 GPUDeviceContext* const context;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 374 static_cast<GPUDeviceContext*>(ctx->op_device_context()); in SetOutputForConstant()
|