Session.h

Go to the documentation of this file.

/*
 * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef _LOCOMOTIV_SESSION_H_
#define _LOCOMOTIV_SESSION_H_
 
#include "locomotiv/NodeData.h"
 
#include <loco.h>
 
#include <memory>
#include <vector>
 
namespace locomotiv
{
 
class Session final
{
public:
  Session() = delete;
 
  Session(loco::Graph *g) : Session(g, loco::output_nodes(g))
  {
    // DO NOTHING
  }
 
  Session(loco::Graph *g, const std::vector<loco::Node *> &custom_outputs)
    : _graph(g), _outputs(custom_outputs)
  {
    // DO NOTHING
  }
 
  template <typename InputIt>
  Session(loco::Graph *g, InputIt begin, InputIt end) : _graph(g), _outputs(begin, end)
  {
    // DO NOTHING
  }
 
  ~Session();
 
  uint32_t input_size() const { return _graph->inputs()->size(); }
 
  void set_input(uint32_t index, std::unique_ptr<NodeData> &&data);
 
  void infer();
 
  uint32_t output_size() const { return _outputs.size(); }
 
  const NodeData *get_output(uint32_t index);
 
  const loco::Node *get_output_node(uint32_t index) { return _outputs.at(index); }
 
private:
  loco::Graph *_graph;
  std::vector<loco::Node *> _outputs;
};
 
} // namespace locomotiv
 
#endif // _LOCOMOTIV_SESSION_H_