Home
last modified time | relevance | path

Searched refs:sharding_it (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc824 auto sharding_it = split_dimension_map.begin(); in CreateOrGetSplitNodesForInputSharding() local
827 int split_dimension = sharding_it->first; in CreateOrGetSplitNodesForInputSharding()
828 int num_split = sharding_it->second; in CreateOrGetSplitNodesForInputSharding()
843 sharding_it++; in CreateOrGetSplitNodesForInputSharding()
848 while (sharding_it != split_dimension_map.end()) { in CreateOrGetSplitNodesForInputSharding()
849 split_dimension = sharding_it->first; in CreateOrGetSplitNodesForInputSharding()
850 num_split = sharding_it->second; in CreateOrGetSplitNodesForInputSharding()
871 sharding_it++; in CreateOrGetSplitNodesForInputSharding()