File: common_types.h

package info (click to toggle)
halide 21.0.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 55,752 kB
  • sloc: cpp: 289,334; ansic: 22,751; python: 7,486; makefile: 4,299; sh: 2,508; java: 1,549; javascript: 282; pascal: 207; xml: 127; asm: 9
file content (16 lines) | stat: -rw-r--r-- 381 bytes parent folder | download | duplicates (4) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
 
#ifndef COMMON_TYPES_H_
#define COMMON_TYPES_H_

#include "Halide.h"
#include "onnx_converter.h"

struct HalideModel {
    std::shared_ptr<Model> model;
    std::shared_ptr<Halide::Pipeline> rep;
    std::vector<std::string> input_names;
    std::unordered_map<std::string, int> input_types;
    std::vector<std::string> output_names;
    std::vector<int> output_types;
};

#endif