// To prevent this, we add control dependencies to make the int32 input edges
// into the PartitionedCall dead.  With this change the D2H copy only happens if
// the PartitionedCall is actually executed.
Status PredicateInt32Inputs(const Scope& root, Node* n,
                            Operation predicate_as_control) {
  std::vector<Output> int32_inputs;
  std::vector<int> int32_inputs_input_idxs;
  for (const Edge* e : n->in_edges()) {