Searched refs:maybe_node_with_sharding_info (Results 1 – 1 of 1) sorted by relevance
1277 for (const Node* maybe_node_with_sharding_info : in ParseInputShardingFromAdjacentNode() local1279 if (maybe_node_with_sharding_info->type_string() != "XlaSharding") continue; in ParseInputShardingFromAdjacentNode()1283 ParseShardingFromDevice(*maybe_node_with_sharding_info, in ParseInputShardingFromAdjacentNode()1287 NodeAndSharding(maybe_node_with_sharding_info, *sharding_config)); in ParseInputShardingFromAdjacentNode()