| 69 | return *this; |
| 70 | } |
| 71 | TensorLayoutArray make_layouts(const TensorShapeArray& shapes) { |
| 72 | TensorLayoutArray layouts(shapes.size()); |
| 73 | for (size_t i = 0; i < shapes.size(); ++i) { |
| 74 | DType dt = |
| 75 | (m_dtype.find(i) != m_dtype.end() ? m_dtype[i] : dtype::Float32()); |
| 76 | if (m_fmt.find(i) == m_fmt.end()) { |
| 77 | layouts[i] = TensorLayout(shapes[i], dt); |
| 78 | layouts[i].init_contiguous_stride(); |
| 79 | } else |
| 80 | layouts[i] = TensorLayout(shapes[i], dt, m_fmt[i]); |
| 81 | } |
| 82 | return layouts; |
| 83 | } |
| 84 | BenchmarkerBase& set_proxy(std::unique_ptr<Proxy>& proxy) { |
| 85 | m_proxy.reset(nullptr); |
| 86 | m_proxy = std::move(proxy); |
no test coverage detected