Searched refs:neighbor_node (Results 1 – 2 of 2) sorted by relevance
1300 const Node& neighbor_node, int64* inferred_core_id, bool* is_fast_mem, in ParseAndValidateShardingFromNeighbors() argument1302 if (neighbor_node.attrs().Find(TPU_FAST_MEM_ATTR) != nullptr) { in ParseAndValidateShardingFromNeighbors()1304 VLOG(2) << "place " << neighbor_node.name() << " on fast memory because " in ParseAndValidateShardingFromNeighbors()1312 ParseInputShardingFromAdjacentNode(num_cores_per_replica, neighbor_node)); in ParseAndValidateShardingFromNeighbors()1322 if (neighbor_node.type_string() == "ReadVariableOp") { in ParseAndValidateShardingFromNeighbors()1323 for (const Edge* e : neighbor_node.out_edges()) { in ParseAndValidateShardingFromNeighbors()
3885 Node* neighbor_node = (*current_neighbor_iter)->dst(); in CheckForNodeFusion() local3888 if ((*node_checker)(neighbor_node)) { in CheckForNodeFusion()3890 nodes.push_back(neighbor_node); in CheckForNodeFusion()3891 current_neighbor_stack.push(neighbor_node->out_edges().begin()); in CheckForNodeFusion()