Table Of Contents
Table Of Contents

mxnet.test_utils.check_speed

mxnet.test_utils.check_speed(sym, location=None, ctx=None, N=20, grad_req=None, typ='whole', **kwargs)[source]

Check the running speed of a symbol.

Parameters
  • sym (Symbol) – Symbol to run the speed test.

  • location (none or dict of str to np.ndarray) – Location to evaluate the inner executor.

  • ctx (Context) – Running context.

  • N (int, optional) – Repeat times.

  • grad_req (None or str or list of str or dict of str to str, optional) – Gradient requirements.

  • typ (str, optional) –

    “whole” or “forward”

    • ”whole”

      Test the forward_backward speed.

    • ”forward”

      Only test the forward speed.