Searched refs:arg_node_name (Results 1 – 1 of 1) sorted by relevance
1299 const int num_cores_per_replica, const std::string& arg_node_name, in ParseAndValidateShardingFromNeighbors() argument1305 << arg_node_name << " has " << TPU_FAST_MEM_ATTR << " attribute"; in ParseAndValidateShardingFromNeighbors()1328 VLOG(2) << "place " << arg_node_name << " on fast memory because " in ParseAndValidateShardingFromNeighbors()