#include <Frontend.h>

Public Types
enum class	FileType { Text , Binary }

Public Member Functions
	Frontend ()

std::unique_ptr< loco::Graph >	load (const ModelSignature &, const char *, FileType) const

std::unique_ptr< loco::Graph >	load (const ModelSignature &, std::istream *, FileType) const

Detailed Description

Definition at line 31 of file Frontend.h.

Member Enumeration Documentation

◆ FileType

enum class moco::tf::Frontend::FileType

strong

Enumerator
Text
Binary

Definition at line 34 of file Frontend.h.

  {
    Text,
    Binary,
  };

Constructor & Destructor Documentation

◆ Frontend()

Frontend::Frontend ( )

Definition at line 188 of file Frontend.cpp.

{
  // DO NOTHING
}

Member Function Documentation

◆ load() [1/2]

std::unique_ptr< loco::Graph > Frontend::load	(	const ModelSignature &	signature,
		const char *	modelfile,
		FileType	type
	)		const

Definition at line 193 of file Frontend.cpp.

{
  // Using c++ standard library, rather than file descriptor, makes these lines portable
  std::ifstream ifs{modelfile, std::ios::in | std::ios::binary};
  return load(signature, &ifs, type);
}

References load().

Referenced by load().

◆ load() [2/2]

std::unique_ptr< loco::Graph > Frontend::load	(	const ModelSignature &	signature,
		std::istream *	stream,
		FileType	type
	)		const

Definition at line 201 of file Frontend.cpp.

{
  tensorflow::GraphDef tf_graph_def;
 
  load_tf(stream, type, tf_graph_def);
 
  transform_tf(signature, tf_graph_def);
 
  auto graph = import(signature, tf_graph_def);
 
  return std::move(graph);
}

The documentation for this class was generated from the following files:

compiler/moco-tf/include/moco/tf/Frontend.h
compiler/moco-tf/src/Frontend.cpp

Public Types

Public Member Functions