Searched refs:terminal_input_pos (Results 1 – 1 of 1) sorted by relevance
313 absl::flat_hash_map<int, int> terminal_input_pos; in RewriteIdentityNAndInputsOutputs() local318 terminal_input_pos[i] = new_idx++; in RewriteIdentityNAndInputsOutputs()349 int new_pos = terminal_input_pos[input_tensor.index()]; in RewriteIdentityNAndInputsOutputs()371 if (terminal_input_pos.find(i) != terminal_input_pos.end()) { in RewriteIdentityNAndInputsOutputs()