Home
last modified time | relevance | path

Searched refs:DEVICE_TO_DEVICE (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dchannel_tracker.cc36 if (type != ChannelHandle::DEVICE_TO_DEVICE && in NewChannel()
/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc229 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
384 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc59 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc149 T, VariantDeviceCopyDirection::DEVICE_TO_DEVICE, \
Dvariant_op_registry_test.cc140 VariantDeviceCopyDirection::DEVICE_TO_DEVICE, in TEST()
Dvariant_op_copy_test.cc100 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
Ddataset.cc181 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
Dvariant_op_registry.h56 DEVICE_TO_DEVICE = 3, enumerator
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc505 return CreateChannelHandleByType(ChannelHandle::DEVICE_TO_DEVICE); in CreateChannelHandle()
Dxla_builder.cc2328 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in SendWithToken()
2381 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in RecvWithToken()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc96 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto326 DEVICE_TO_DEVICE = 1; enumerator