Searched refs:arg_index_mapping (Results 1 – 1 of 1) sorted by relevance
827 std::map<int, int> arg_index_mapping; in RemoveUnusedXlaInput() local833 arg_index_mapping[i] = new_arg_index; in RemoveUnusedXlaInput()841 n->AddAttr("index", arg_index_mapping[index]); in RemoveUnusedXlaInput()856 auto iter = arg_index_mapping.find(j); in RemoveUnusedXlaInput()857 if (iter != arg_index_mapping.end()) { in RemoveUnusedXlaInput()877 auto iter = arg_index_mapping.find(arg_index); in RemoveUnusedXlaInput()878 if (iter != arg_index_mapping.end()) { in RemoveUnusedXlaInput()897 auto iter = arg_index_mapping.find(j); in RemoveUnusedXlaInput()898 if (iter != arg_index_mapping.end()) { in RemoveUnusedXlaInput()904 auto iter = arg_index_mapping.find(i + num_per_replica_inputs); in RemoveUnusedXlaInput()[all …]