Searched refs:arg_index_to_sharded_input_map (Results 1 – 1 of 1) sorted by relevance
797 arg_index_to_sharded_input_map) { in CreateOrGetSplitNodesForInputSharding() argument799 auto iter = arg_index_to_sharded_input_map->find(input_index); in CreateOrGetSplitNodesForInputSharding()800 if (iter != arg_index_to_sharded_input_map->end()) { in CreateOrGetSplitNodesForInputSharding()900 (*arg_index_to_sharded_input_map)[input_index] = sharded_input_info; in CreateOrGetSplitNodesForInputSharding()