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

Method Broadcast

src/opr/impl/tensor_manip.cpp:405–408  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

403MGB_DYN_TYPE_OBJ_FINAL_IMPL(Broadcast);
404
405Broadcast::Broadcast(VarNode* inp, VarNode* tshp, const OperatorNodeConfig& config)
406 : Super{inp->owner_graph(), config, "broadcast", {inp}} {
407 reshapebrdcast_init(inp, tshp);
408}
409
410SymbolVar Broadcast::make(
411 SymbolVar inp, SymbolVar tshp, const OperatorNodeConfig& config) {

Callers 5

make_full_if_noneFunction · 0.80
syncbn_concat_statsFunction · 0.80
_make_full_if_noneFunction · 0.80
einsumFunction · 0.80
broadcastFunction · 0.80

Calls 1

owner_graphMethod · 0.45

Tested by 1

broadcastFunction · 0.64