inference/compute/pt/pytorch_linear.py [265:285]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        QPS = batch_size / elap

        print(
            "{0:6},  {1:6},  {2:6},  {3:6},  {4:6},  {5:10.6f},  {6:8.1f}, {7:10.1f}".format(
                layer_num,
                input_size,
                hidden_size,
                output_size,
                batch_size,
                elap,
                QPS,
                flops / elap / 1.0e12,
            )
        )


if __name__ == "__main__":

    import argparse

    parser = argparse.ArgumentParser(description="Measure the performance of MLP")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train/compute/pt/pytorch_linear.py [264:284]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        QPS = batch_size / elap

        print(
            "{0:6},  {1:6},  {2:6},  {3:6},  {4:6},  {5:10.6f},  {6:8.1f}, {7:10.1f}".format(
                layer_num,
                input_size,
                hidden_size,
                output_size,
                batch_size,
                elap,
                QPS,
                flops / elap / 1.0e12,
            )
        )


if __name__ == "__main__":

    import argparse

    parser = argparse.ArgumentParser(description="Measure the performance of MLP")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -