SparseTensor is not supported. The return value of the decorated function
must be a Tensor or a list/tuple of Tensors.
  }];

  let arguments = (ins
    Arg<Variadic<TF_Tensor>, [{The tensors to be batched.}]>:$in_tensors,
    Arg<Variadic<TF_Tensor>, [{The tensors which are captured in the function, and don't need
to be batched.}]>:$captured_tensors,

    SymbolRefAttr:$f,
    I64Attr:$num_batch_threads,

*  `tf.Tensor`