Searched refs:DEVICE_TO_DEVICE (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.cc | 39 REGISTER_CSR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_variant.cc | 84 VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| D | tensor_map.cc | 65 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| D | tensor_list.cc | 76 REGISTER_LIST_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | event_span.cc | 184 return DEVICE_TO_DEVICE; in ClassifyGpuEvent()
227 case DEVICE_TO_DEVICE: in PrintEventType()
379 case DEVICE_TO_DEVICE: in AddDeviceMemoryTransferEvent()
|
| D | event_span.h | 65 DEVICE_TO_DEVICE = 120, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | channel_tracker.cc | 36 if (type != ChannelHandle::DEVICE_TO_DEVICE && in NewChannel()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_stats_to_pod_stats.cc | 55 add_event(kDeviceToDevice, {DEVICE_TO_DEVICE, DEVICE_WAIT_DEVICE}); in CreatePodStatsRecord()
|
| D | op_stats_to_pod_stats_test.cc | 69 type_ps[DEVICE_TO_DEVICE] = kDeviceToDevicePs; in CreateOpStats()
|
| D | op_stats_to_pod_viewer_test.cc | 70 type_ps[DEVICE_TO_DEVICE] = kDeviceToDevicePs; in CreateOpStats()
|
| D | xplane_to_step_events.cc | 186 case DEVICE_TO_DEVICE: in ConvertDeviceTraceXLineToStepEvents()
|
| D | op_stats_to_input_pipeline_analysis.cc | 229 details.set_device_to_device_ms(GetTimeInMs(type_ps, DEVICE_TO_DEVICE) + in ComputeGenericInputPipelineAnalysisResult()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.cc | 154 T, VariantDeviceCopyDirection::DEVICE_TO_DEVICE, \
|
| D | variant_op_registry_test.cc | 157 VariantDeviceCopyDirection::DEVICE_TO_DEVICE, in TEST()
|
| D | variant_op_copy_test.cc | 100 StoredTensorValue, VariantDeviceCopyDirection::DEVICE_TO_DEVICE,
|
| D | variant_op_registry.h | 60 DEVICE_TO_DEVICE = 3, enumerator
|
| D | dataset.cc | 204 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 177 VariantDeviceCopy(VariantDeviceCopyDirection::DEVICE_TO_DEVICE, v[i], in CopyDeviceToDevice()
328 REGISTER_WRAPPED_TENSOR_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 62 REGISTER_OPTIONAL_COPY(VariantDeviceCopyDirection::DEVICE_TO_DEVICE);
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 505 return CreateChannelHandleByType(ChannelHandle::DEVICE_TO_DEVICE); in CreateChannelHandle()
|
| D | xla_builder.cc | 3160 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in SendWithToken()
3213 if (handle.type() != ChannelHandle::DEVICE_TO_DEVICE) { in RecvWithToken()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 571 .value("DEVICE_TO_DEVICE", ChannelHandle::DEVICE_TO_DEVICE) in BuildXlaCompilerSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 361 DEVICE_TO_DEVICE = 1; enumerator
|