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.