Home
last modified time | relevance | path

Searched defs:CopyTensorInSameDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc40 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::GPUDeviceContext
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc99 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h86 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
Drendezvous_test.cc281 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device, in CopyTensorInSameDevice() function in tensorflow::__anon2f0f779b0111::DummyDeviceContext