Home
last modified time | relevance | path

Searched refs:partial_device (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h472 const DeviceNameUtils::ParsedName& partial_device,
Ddistributed_tpu_rewrite_pass.cc3465 const DeviceNameUtils::ParsedName& partial_device, in CopyOutsideCompilationNodes() argument
3494 if (DeviceNameUtils::IsSpecification(device, partial_device)) { in CopyOutsideCompilationNodes()
3496 DeviceNameUtils::MergeDevNames(&device, partial_device)); in CopyOutsideCompilationNodes()
3528 DeviceNameUtils::ParsedName partial_device = tpu_device; in ReplicateOutsideCompilationNodes() local
3529 partial_device.has_type = false; in ReplicateOutsideCompilationNodes()
3530 partial_device.has_id = false; in ReplicateOutsideCompilationNodes()
3550 partial_device, node_images, graph)); in ReplicateOutsideCompilationNodes()