Searched refs:HOST_TO_DEVICE (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | channel_tracker.cc | 37 type != ChannelHandle::HOST_TO_DEVICE && in NewChannel()
79 if (channel.type == ChannelHandle::HOST_TO_DEVICE) { in RegisterSendInternal()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry_test.cc | 88 VariantValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
145 VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
170 registry.RegisterDeviceCopyFn(VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
173 VariantDeviceCopyDirection::HOST_TO_DEVICE, kTypeIndex, f), in TEST()
|
| D | variant_op_registry.cc | 143 T, VariantDeviceCopyDirection::HOST_TO_DEVICE, \
|
| D | variant_op_copy_test.cc | 92 StoredTensorValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
|
| D | dataset.cc | 179 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| D | variant_op_registry.h | 54 HOST_TO_DEVICE = 1, enumerator
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 100 VariantDeviceCopyDirection::HOST_TO_DEVICE, v[i], &v_out[i], copier); in CopyHostToDevice()
382 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 57 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 509 return CreateChannelHandleByType(ChannelHandle::HOST_TO_DEVICE); in CreateHostToDeviceChannelHandle()
|
| D | xla_builder.cc | 2467 if (handle.type() != ChannelHandle::HOST_TO_DEVICE) { in RecvFromHost()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cc | 94 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 721 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 334 HOST_TO_DEVICE = 3; enumerator
|