furiosa.optimizer.frontend.onnx package

Subpackages

• furiosa.optimizer.frontend.onnx.transformer package
  • Subpackages
    • furiosa.optimizer.frontend.onnx.transformer.experimental package
      • Submodules
      • furiosa.optimizer.frontend.onnx.transformer.experimental.eliminate_detection_postprocess module
      • furiosa.optimizer.frontend.onnx.transformer.experimental.embedding_bag_porting module
      • furiosa.optimizer.frontend.onnx.transformer.experimental.fuse_div_for_bert module
      • furiosa.optimizer.frontend.onnx.transformer.experimental.reify_conv_for_bert module
      • Module contents
  • Submodules
  • furiosa.optimizer.frontend.onnx.transformer.convert_negative_pads_to_slice module
  • furiosa.optimizer.frontend.onnx.transformer.convert_prelu_to_relu module
  • furiosa.optimizer.frontend.onnx.transformer.fuse_batchnorm module
  • furiosa.optimizer.frontend.onnx.transformer.fuse_gather_matmul module
  • furiosa.optimizer.frontend.onnx.transformer.infer_squeeze_axes module
  • furiosa.optimizer.frontend.onnx.transformer.polish_model module
  • furiosa.optimizer.frontend.onnx.transformer.utils module
  • Module contents
• furiosa.optimizer.frontend.onnx.utils package
  • Submodules
  • furiosa.optimizer.frontend.onnx.utils.check_model module
  • furiosa.optimizer.frontend.onnx.utils.inference_shape module
  • furiosa.optimizer.frontend.onnx.utils.version_checker module
  • Module contents

Module contents

furiosa.optimizer.frontend.onnx.optimize_model(model: onnx.onnx_ml_pb2.ModelProto, input_shapes: Optional[Dict[str, List[int]]] = None, opset_version: int = 13) → onnx.onnx_ml_pb2.ModelProto