func::FuncOp func() {
      return SymbolTable::lookupNearestSymbolFrom<func::FuncOp>(*this, getF());
    }

    // Erases variable arguments from `batch_func_op`. `erase_indices` contains
    // the indices of the arguments to erase.
    void eraseArguments(const BitVector& erase_indices);

    // Gets the argument operands to the called function.
    operand_range getArgOperands() {

            instead of relying on `Model.inputs` and `Model.input_names`, which
            are no longer set for subclass Models. This attr is set in eager,
            `tf.function`, and graph modes. This gets rid of the need for users
            to manually call `Model._set_inputs` when using Custom Training
            Loops(CTLs).
        *   Dynamic shapes are supported for generators by calling the Model on