Searched defs:CopyTensorInSameDevice (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/gpu/ | ||
D | gpu_util_platform_specific.cc | 40 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::GPUDeviceContext |
/external/tensorflow/tensorflow/compiler/jit/ | ||
D | xla_device_context.cc | 99 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext |
/external/tensorflow/tensorflow/core/framework/ | ||
D | device_base.h | 86 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function |
D | rendezvous_test.cc | 281 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device, in CopyTensorInSameDevice() function in tensorflow::__anon2f0f779b0111::DummyDeviceContext |