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

Method set_mge_tensor_compnode

lite/src/mge/tensor_impl.cpp:128–134  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

126}
127
128void TensorImplDft::set_mge_tensor_compnode(const mgb::CompNode& comp_node) {
129 if (is_host()) {
130 m_host_tensor->comp_node(comp_node, true);
131 } else {
132 m_dev_tensor->comp_node(comp_node, true);
133 }
134}
135
136Layout TensorImplDft::get_layout() const {
137 if (is_host()) {

Callers 2

copy_from_continueMethod · 0.80
copy_from_fixlayoutMethod · 0.80

Calls 1

comp_nodeMethod · 0.45

Tested by

no test coverage detected