return func_arguments_;
  }
  const llvm::SmallVector<Value>& FuncOutputs() const { return func_outputs_; }

 private:
  // Compute once at construction and save as field.
  const llvm::SmallVector<Value> func_arguments_;
  const llvm::SmallVector<Value> func_outputs_;
};

// Helper data structure for output parameters to `ExtractSubgraphToFunc`.

  IRMapping values_in_scope;
  // Function arguments can appear as operands, so they clone should
  // be aware of them.
  assert(subgraph.FuncArguments().size() == new_func.getNumArguments());
  for (int i = 0; i < subgraph.FuncArguments().size(); ++i) {
    Value original_value = subgraph.FuncArguments()[i];
    Value new_func_arg = new_func.getArgument(i);
    values_in_scope.map(original_value, new_func_arg);
  }