export_onnx.hpp File Reference

#include "lbann/callbacks/callback.hpp"
#include <lbann/base.hpp>
#include <onnx/onnx_pb.h>
#include <google/protobuf/message.h>
#include <iostream>
#include <memory>
#include <vector>

Include dependency graph for export_onnx.hpp:

Go to the source code of this file.

Classes
class	lbann::callback::export_onnx
	Callback to export a trained model to onnx format. More...

Namespaces
	lbann
	lbann::callback

Functions
std::unique_ptr< callback_base >	lbann::callback::build_export_onnx_callback_from_pbuf (const google::protobuf::Message &proto_msg, const std::shared_ptr< lbann_summary > &)