Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc100 s_copy_init = VariantDeviceCopy( in CopyHostToDevice()
177 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
379 s_copy_init = VariantDeviceCopy( in CopyDeviceToHost()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc118 Status VariantDeviceCopy( in VariantDeviceCopy() function
Dvariant_op_registry.h298 Status VariantDeviceCopy(
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc725 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()