mlinference.h

Go to the documentation of this file.

/*
    This software is Copyright by the Board of Trustees of Michigan
    State University (c) Copyright 2017.
 
    You may use this software under the terms of the GNU public license
    (GPL).  The terms of this license are described at:
 
     http://www.gnu.org/licenses/gpl.txt
 
     Authors:
             Aaron Chester
         Bashir Sadeghi
         FRIB
         Michigan State University
         East Lansing, MI 48824-1321
*/
 
#ifndef MLINFERENCE_H
#define MLINFERENCE_H
 
#include <cstdint>
#include <vector>
 
#include <torch/script.h> // Can forward-declare?
 
namespace ddastoys {
 
    struct FitInfo;
    
    namespace mlinference {
    
    void performInference(
        FitInfo* pResult,
        std::vector<uint16_t>& trace, unsigned saturation,
        torch::jit::script::Module& module
        );
    
    }
}
 
#endif