Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc827 std::map<int, int> arg_index_mapping; in RemoveUnusedXlaInput() local
833 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 …]