Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc103 s_copy_init = VariantDeviceCopy( in CopyHostToDevice()
179 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
386 s_copy_init = VariantDeviceCopy( in CopyDeviceToHost()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc128 Status VariantDeviceCopy( in VariantDeviceCopy() function
Dvariant_op_registry.h299 Status VariantDeviceCopy(
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device.cc399 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc864 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()