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.cc99 s_copy_init = VariantDeviceCopy( in CopyHostToDevice()
164 s_copy_init = VariantDeviceCopy( in CopyDeviceToHost()
229 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc113 Status VariantDeviceCopy( in VariantDeviceCopy() function
Dvariant_op_registry.h251 Status VariantDeviceCopy(
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc721 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()