Arg<TF_Float32Tensor, [{Value of velocities used in the ADAM optimization algorithm.}]>:$velocities,

    DefaultValuedOptionalAttr<I64Attr, "-1">:$table_id,
    DefaultValuedOptionalAttr<StrAttr, "\"\"">:$table_name,
    I64Attr:$num_shards,
    I64Attr:$shard_id,
    DefaultValuedOptionalAttr<StrAttr, "\"\"">:$config
  );

  let results = (outs);
}