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

Method Linspace

src/opr/impl/tensor_gen.cpp:47–57  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

45MGB_DYN_TYPE_OBJ_FINAL_IMPL(Linspace);
46
47Linspace::Linspace(
48 VarNode* start, VarNode* stop, VarNode* num, const Param& param,
49 const OperatorNodeConfig& config)
50 : Super{start->owner_graph(), config, "linspce", {start, stop}},
51 m_param{param} {
52 add_input({start, stop, num});
53 add_output(None)
54 ->dtype(dtype::Float32())
55 .add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE);
56 add_equivalence_component<PODHash<Param>>(&m_param);
57}
58
59SymbolVar Linspace::make(
60 SymbolVar start, SymbolVar stop, SymbolVar num, const Param& param,

Callers 2

diag_plane_subgraphFunction · 0.80
linspaceFunction · 0.80

Calls 2

owner_graphMethod · 0.45
dtypeMethod · 0.45

Tested by

no test coverage detected