| /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,
162 VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
187 registry.RegisterDeviceCopyFn(VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()
190 VariantDeviceCopyDirection::HOST_TO_DEVICE, kTypeIndex, f), in TEST()
|
| D | variant_op_registry.cc | 148 T, VariantDeviceCopyDirection::HOST_TO_DEVICE, \
|
| D | variant_op_copy_test.cc | 92 StoredTensorValue, VariantDeviceCopyDirection::HOST_TO_DEVICE,
|
| D | variant_op_registry.h | 58 HOST_TO_DEVICE = 1, enumerator
|
| D | dataset.cc | 202 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | event_span.cc | 180 return HOST_TO_DEVICE; in ClassifyGpuEvent()
195 return HOST_TO_DEVICE; in ClassifyCpuEvent()
219 case HOST_TO_DEVICE: in PrintEventType()
373 case HOST_TO_DEVICE: in AddDeviceMemoryTransferEvent()
|
| D | event_span.h | 54 HOST_TO_DEVICE = 80, enumerator
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.cc | 37 REGISTER_CSR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_variant.cc | 81 REGISTER_RAGGED_TENSOR_VARIANT_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| D | tensor_map.cc | 63 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| D | tensor_list.cc | 74 REGISTER_LIST_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_stats_to_pod_stats.cc | 59 add_event(kInput, {HOST_WAIT_INPUT, HOST_TO_DEVICE, DEVICE_WAIT_HOST}); in CreatePodStatsRecord()
|
| D | op_stats_to_pod_stats_test.cc | 65 type_ps[HOST_TO_DEVICE] = kHostToDevicePs; in CreateOpStats()
|
| D | op_stats_to_pod_viewer_test.cc | 66 type_ps[HOST_TO_DEVICE] = kHostToDevicePs; in CreateOpStats()
|
| D | xplane_to_step_events.cc | 185 case HOST_TO_DEVICE: in ConvertDeviceTraceXLineToStepEvents()
|
| D | op_stats_to_input_pipeline_analysis.cc | 224 details.set_host_to_device_ms(GetTimeInMs(type_ps, HOST_TO_DEVICE) + in ComputeGenericInputPipelineAnalysisResult()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 101 VariantDeviceCopyDirection::HOST_TO_DEVICE, v[i], &v_out[i], copier); in CopyHostToDevice()
326 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 60 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::HOST_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_executable_info.proto | 71 HOST_TO_DEVICE = 2; enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 509 return CreateChannelHandleByType(ChannelHandle::HOST_TO_DEVICE); in CreateHostToDeviceChannelHandle()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 573 .value("HOST_TO_DEVICE", ChannelHandle::HOST_TO_DEVICE); in BuildXlaCompilerSubmodule()
|
| /external/llvm-project/polly/lib/CodeGen/ |
| D | PPCGCodeGeneration.cpp | 415 enum DataDirection { HOST_TO_DEVICE, DEVICE_TO_HOST }; enumerator
1202 if (Direction == HOST_TO_DEVICE) in createDataTransfer()
1242 createDataTransfer(UserStmt, HOST_TO_DEVICE); in createUser()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 369 HOST_TO_DEVICE = 3; enumerator
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 725 s = VariantDeviceCopy(VariantDeviceCopyDirection::HOST_TO_DEVICE, in MakeTensorFromProto()
|