let summary = [{
An op that enqueues a list of input batch tensors to TPUEmbedding.
  }];

  let description = [{
An op that enqueues a list of input batch tensors to TPUEmbedding.
  }];

  let arguments = (ins