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

Function reduce_to

imperative/python/src/grad_override.cpp:63–66  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

61}
62
63ValueRef reduce_to(ValueRef x, ValueRef s) {
64 static auto op = Reduce::make();
65 return imperative::apply(*op, x, s)[0];
66}
67
68ValueRef reshape_to(ValueRef x, ValueRef s) {
69 static auto op = Reshape::make();

Callers 3

elemwise_grad_ruleFunction · 0.85
broadcast_grad_ruleFunction · 0.85

Calls 2

makeFunction · 0.50
applyFunction · 0.50

Tested by

no test coverage detected