Searched refs:per_replica_inputs (Results 1 – 2 of 2) sorted by relevance
1021 per_replica_inputs = multi_worker_iterator.get_next()1027 select_replica, per_replica_inputs),))
608 std::vector<std::pair<Node*, int>> per_replica_inputs; in MoveHeadOutsideCompilationToHost() local612 per_replica_inputs.push_back(std::make_pair(e->src(), e->src_output())); in MoveHeadOutsideCompilationToHost()621 Node* src = per_replica_inputs[replica_id * old_num_per_replica_inputs + in MoveHeadOutsideCompilationToHost()625 per_replica_inputs[replica_id * old_num_per_replica_inputs + in MoveHeadOutsideCompilationToHost()