Searched refs:neighbor_node (Results 1 – 2 of 2) sorted by relevance
3865 Node* neighbor_node = (*current_neighbor_iter)->dst(); in CheckForNodeFusion() local3868 if ((*node_checker)(neighbor_node)) { in CheckForNodeFusion()3870 nodes.push_back(neighbor_node); in CheckForNodeFusion()3871 current_neighbor_stack.push(neighbor_node->out_edges().begin()); in CheckForNodeFusion()
1509 const Node& neighbor_node, int64_t* inferred_core_id, bool* is_fast_mem, in ParseAndValidateShardingFromNeighbors() argument1511 if (neighbor_node.attrs().Find(TPU_FAST_MEM_ATTR) != nullptr) { in ParseAndValidateShardingFromNeighbors()1513 VLOG(2) << "place " << neighbor_node.name() << " on fast memory because " in ParseAndValidateShardingFromNeighbors()1521 ParseInputShardingFromAdjacentNode(num_cores_per_replica, neighbor_node)); in ParseAndValidateShardingFromNeighbors()1531 if (neighbor_node.type_string() == "ReadVariableOp") { in ParseAndValidateShardingFromNeighbors()1532 for (const Edge* e : neighbor_node.out_edges()) { in ParseAndValidateShardingFromNeighbors()