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

Method dnn_tensor

imperative/src/impl/physical_tensor.cpp:705–709  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

703}
704
705DnnTensorND Tensor::dnn_tensor() {
706 mgb_assert(m_blob, "uninitialized tensor.");
707 mgb_assert(m_layout.ndim, "dnn don't support scalar");
708 return DnnTensorND{m_layout, m_blob->storage(), m_offset};
709}
710
711DnnTensorND Tensor::dnn_tensor(TensorShape new_shape) {
712 mgb_assert(m_blob, "uninitialized tensor.");

Callers 7

create_workspaceMethod · 0.80
call_dnnMethod · 0.80
alloc_outputMethod · 0.80
megray_recv_tensorFunction · 0.80
apply_on_physical_tensorFunction · 0.80
apply_on_physical_tensorFunction · 0.80

Calls 2

storageMethod · 0.80
reshapeMethod · 0.45

Tested by

no test coverage detected