Table Of Contents
Table Of Contents

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.