Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc497 const Node* matching_node = nullptr; in SetNodeShardingFromNeighbors() local
510 matching_node = possible_match; in SetNodeShardingFromNeighbors()
514 if (matching_node != nullptr) { in SetNodeShardingFromNeighbors()
515 n->set_assigned_device_name(matching_node->assigned_device_name()); in SetNodeShardingFromNeighbors()
516 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()