| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | channel_tracker.cc | 38 type != ChannelHandle::DEVICE_TO_HOST) { in NewChannel()
101 if (channel.type == ChannelHandle::DEVICE_TO_HOST) { in RegisterRecvInternal()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.cc | 38 REGISTER_CSR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_variant.cc | 82 REGISTER_RAGGED_TENSOR_VARIANT_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
|
| D | tensor_map.cc | 64 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
|
| D | tensor_list.cc | 75 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | event_span.cc | 182 return DEVICE_TO_HOST; in ClassifyGpuEvent()
229 case DEVICE_TO_HOST: in PrintEventType()
376 case DEVICE_TO_HOST: in AddDeviceMemoryTransferEvent()
|
| D | event_span.h | 67 DEVICE_TO_HOST = 130, enumerator
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_stats_to_pod_stats.cc | 60 add_event(kOutput, {DEVICE_TO_HOST}); in CreatePodStatsRecord()
|
| D | op_stats_to_pod_stats_test.cc | 70 type_ps[DEVICE_TO_HOST] = kDeviceToHostPs; in CreateOpStats()
|
| D | op_stats_to_pod_viewer_test.cc | 71 type_ps[DEVICE_TO_HOST] = kDeviceToHostPs; in CreateOpStats()
|
| D | xplane_to_step_events.cc | 187 case DEVICE_TO_HOST: { in ConvertDeviceTraceXLineToStepEvents()
|
| D | op_stats_to_input_pipeline_analysis.cc | 226 details.set_output_ms(GetTimeInMs(type_ps, DEVICE_TO_HOST)); in ComputeGenericInputPipelineAnalysisResult()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.cc | 151 T, VariantDeviceCopyDirection::DEVICE_TO_HOST, \
|
| D | variant_op_copy_test.cc | 96 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_HOST,
|
| D | variant_op_registry.h | 59 DEVICE_TO_HOST = 2, enumerator
|
| D | dataset.cc | 203 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 327 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
380 VariantDeviceCopyDirection::DEVICE_TO_HOST, v[i], &v_out[i], copier); in CopyDeviceToHost()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 61 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_HOST);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_executable_info.proto | 70 DEVICE_TO_HOST = 1; enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 513 return CreateChannelHandleByType(ChannelHandle::DEVICE_TO_HOST); in CreateDeviceToHostChannelHandle()
|
| D | xla_builder.cc | 3258 if (handle.type() != ChannelHandle::DEVICE_TO_HOST) { in SendToHost()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 572 .value("DEVICE_TO_HOST", ChannelHandle::DEVICE_TO_HOST) in BuildXlaCompilerSubmodule()
|
| /external/llvm-project/polly/lib/CodeGen/ |
| D | PPCGCodeGeneration.cpp | 415 enum DataDirection { HOST_TO_DEVICE, DEVICE_TO_HOST }; enumerator
1252 createDataTransfer(UserStmt, DEVICE_TO_HOST); in createUser()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 365 DEVICE_TO_HOST = 2; enumerator
|