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

Method add_workspace_output

src/core/impl/graph/helper.cpp:375–380  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

373}
374
375void cg::add_workspace_output(OperatorNodeBase* opr) {
376 opr->add_output("workspace")
377 ->add_flag(VarNode::Flag::VOLATILE_CONTENT)
378 .add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE)
379 .dtype(dtype::Byte());
380}
381
382void cg::copy_shape_to_tensor_value(DeviceTensorND& dest, const TensorShape& shp) {
383 dest.comp_node(CompNode::default_cpu())

Callers

nothing calls this directly

Calls 2

dtypeMethod · 0.45
add_outputMethod · 0.45

Tested by

no test coverage detected