MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / make_layouts

Method make_layouts

dnn/test/common/benchmarker.h:71–83  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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);

Callers 1

algo_benchmarkFunction · 0.45

Calls 5

TensorLayoutClass · 0.85
sizeMethod · 0.45
findMethod · 0.45
endMethod · 0.45

Tested by

no test coverage detected