`neural_compressor.profiling.profiler.onnxrt_profiler.profiler`

Onnxrt profiler.

Module Contents

Classes

Tensorflow profiler class.

class neural_compressor.profiling.profiler.onnxrt_profiler.profiler.Profiler(model: neural_compressor.model.onnx_model.ONNXModel, dataloader: neural_compressor.data.dataloaders.onnxrt_dataloader.ONNXRTDataLoader, log_file: str | None = None)[source]: Tensorflow profiler class.