| 99 | } |
| 100 | |
| 101 | VARP createVar(const std::string name, INTS shape, halide_type_t dtype) { |
| 102 | int size = 1; |
| 103 | for (int dim : shape) { |
| 104 | size *= dim; |
| 105 | } |
| 106 | std::unique_ptr<char[]> data(new char[size * dtype.bytes()]); |
| 107 | loadData(name, data.get(), size, dtype); |
| 108 | return _Const(data.get(), shape, NHWC, dtype); |
| 109 | } |
| 110 | |
| 111 | template <typename T> |
| 112 | bool compareVar(VARP var, std::string name) { |