neural_compressor.profiling.profiler.tensorflow_profiler.profiler
Tensorflow profiler.
Module Contents
Classes
Tensorflow profiler class. |
- class neural_compressor.profiling.profiler.tensorflow_profiler.profiler.Profiler(model: neural_compressor.model.tensorflow_model.TensorflowBaseModel, dataloader: neural_compressor.data.dataloaders.tensorflow_dataloader.TensorflowDataLoader, log_file: str | None = None)[source]
Tensorflow profiler class.