def main()

in tt_embeddings_benchmark.py [0:0]


def main(
    batch_size,
    iters,
    long_index,
    pooling_factor,
    p_shapes,
    q_shapes,
    ranks,
    sparse,
    optimizer,
    run_baseline,