tensorflow::TFHandler Class Reference

Wrapper for Tensorflow which handles construction and prediction. More...

#include "/cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N21-01-18/TensorFlowHandler/TFHandler.h"

Public Member Functions
	TFHandler (std::string model, int CPUlimit=1)
	Basic constructor, takes path to model pb. More...
	~TFHandler ()
void	Initialize (int CPUlimit)
std::vector< Tensor >	Predict (std::vector< std::pair< std::string, Tensor >> inputs, std::vector< std::string > outputLabels)

Private Attributes
Session *	fSession
std::string	fModelPath

Detailed Description

Wrapper for Tensorflow which handles construction and prediction.

Definition at line 19 of file TFHandler.h.

Constructor & Destructor Documentation

tensorflow::TFHandler::TFHandler	(	std::string	model,
		int	CPUlimit = 1
	)

Basic constructor, takes path to model pb.

Definition at line 18 of file TFHandler.cxx.

References Initialize().

                                                       :
    fSession(0),
    fModelPath(ModelPath)
  {
    Initialize(CPUlimit);
  }

tensorflow::TFHandler::~TFHandler

(

)

Definition at line 25 of file TFHandler.cxx.

References fSession.

  {
    if(fSession) delete fSession;
  }

Member Function Documentation

void tensorflow::TFHandler::Initialize

(

int

CPUlimit

)

Definition at line 30 of file TFHandler.cxx.

References om::cout, allTimeWatchdog::endl, fModelPath, fSession, fabricate::options, and fabricate::status.

Referenced by TFHandler().

  {
    // Initalize a session and create a frozen tensorflow graph
    GraphDef graph_def;
    // Force tf to only use a specified number of cores
    tensorflow::SessionOptions options;
    tensorflow::ConfigProto &config = options.config;
    if (CPUlimit > 0) {
        config.set_inter_op_parallelism_threads(CPUlimit);
        config.set_intra_op_parallelism_threads(CPUlimit);
        config.set_use_per_session_threads(false);
    }
                             
    Status status = NewSession(options, &fSession);
    if (!status.ok()) {
      std::cout<<"Error when making tensorflow session: "<< status.ToString() << std::endl;
      return;
    }

    status = ReadBinaryProto(Env::Default(),fModelPath, &graph_def);
    if (!status.ok()){
      std::cout<<"Error when reading model file: "<< status.ToString() << std::endl;
      return;
    }

    status = fSession->Create(graph_def);
    if (!status.ok()){
      std::cout<<"Error when creating tensorflow graph: "<< status.ToString() << std::endl;
      return;
    }

    std::cout<<"Successfully loaded tensorflow graph."<<std::endl;
  }

std::vector< Tensor > tensorflow::TFHandler::Predict	(	std::vector< std::pair< std::string, Tensor >>	inputs,
		std::vector< std::string >	outputLabels
	)

Definition at line 64 of file TFHandler.cxx.

References om::cout, allTimeWatchdog::endl, fSession, dumpEventsToText::inputs, dumpEventsToText::outputs, and fabricate::status.

Referenced by cvnneutronprongtf::CVNNeutronProngTF::CalcResult(), cvntf::CVNTF::produce(), cvntf::CVNCosmicTF::produce(), regcvntf::RegCVNTF::produce(), nuonecvntf::NuonECVNTF::produce(), cvnprongtf::CVNProngTF::produce(), cvneventtf::CVNEventTF::produce(), cvntf::CVNProngEvaluatorTF::produce(), and nerd::NERDEval::run_graph().

  {
    std::vector<Tensor> outputs;

    // Go go go!
    Status status = fSession->Run(inputs,outputLabels,{}, &outputs);

    if (!status.ok()){
      std::cout<<"Error when running session: "<< status.ToString() << std::endl;
    }

    return outputs;
  }

Member Data Documentation

std::string tensorflow::TFHandler::fModelPath

private

Definition at line 32 of file TFHandler.h.

Referenced by Initialize().

Session* tensorflow::TFHandler::fSession

private

Definition at line 31 of file TFHandler.h.

Referenced by Initialize(), Predict(), and ~TFHandler().

---

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

• /cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N21-01-18/TensorFlowHandler/TFHandler.h
• /cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N21-01-18/TensorFlowHandler/TFHandler.cxx