Searched refs:num_replicated_outputs (Results 1 – 1 of 1) sorted by relevance
1377 int num_replicated_outputs = output_types.size() / num_replicas; in RemoveUnusedXlaOutput() local1437 num_replicated_outputs); in RemoveUnusedXlaOutput()1440 e->src_output() < num_replicas * num_replicated_outputs) { in RemoveUnusedXlaOutput()1445 for (int j = 0; j < num_replicated_outputs; j++) { in RemoveUnusedXlaOutput()1448 for (const Edge* e : output_edges[i * num_replicated_outputs + j]) { in RemoveUnusedXlaOutput()1452 i * (num_replicated_outputs - ret_indices_to_remove.size()) + in RemoveUnusedXlaOutput()1458 TF_RET_CHECK(output_edges[i * num_replicated_outputs + j].empty()) in RemoveUnusedXlaOutput()1460 << output_edges[i * num_replicated_outputs + j][0]->DebugString(); in RemoveUnusedXlaOutput()