inference_benchmark.py File Reference

Go to the source code of this file.

Namespaces
namespace	inference_benchmark

Functions
float	inference_benchmark.get_memory_usage_mb ()
List[List[int]]	inference_benchmark.parse_shapes (List[str] shape_strs)
List[tensorinfo]	inference_benchmark.get_validated_input_tensorinfos (infer.session sess, List[List[int]] static_shapes)
	inference_benchmark.benchmark_inference (str nnpackage_path, str backends, List[List[int]] input_shapes, int repeat)
	inference_benchmark.main ()