Home
last modified time | relevance | path

Searched refs:SetProtoFromGPU (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.h38 static void SetProtoFromGPU(const Tensor& tensor, Device* dev,
Dgpu_util.cc115 void GPUUtil::SetProtoFromGPU(const Tensor& tensor, Device* dev, in SetProtoFromGPU() function in tensorflow::GPUUtil
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc244 GPUUtil::SetProtoFromGPU( in AddRequest()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1099 GPUUtil::SetProtoFromGPU( in RecvHandler()