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.cc502 const Node* matching_node = nullptr; in SetNodeShardingFromNeighbors() local
517 matching_node = possible_match; in SetNodeShardingFromNeighbors()
521 if (matching_node != nullptr) { in SetNodeShardingFromNeighbors()
522 n->set_assigned_device_name(matching_node->assigned_device_name()); in SetNodeShardingFromNeighbors()
523 n->set_requested_device(matching_node->requested_device()); in SetNodeShardingFromNeighbors()