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

Method apply_on_device_tensornd

imperative/src/impl/op_def.cpp:36–41  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

34 def, std::move(inputs), output_descs, validated);
35}
36void OpDef::apply_on_device_tensornd(
37 const OpDef& def, const SmallVector<DeviceTensorND>& inputs,
38 SmallVector<DeviceTensorND>* outputs) {
39 def.trait()->apply_on_device_tensornd(def, inputs, outputs);
40 return;
41}
42
43VarNodeArray OpDef::apply_on_var_node(const OpDef& def, const VarNodeArray& inputs) {
44 return def.trait()->apply_on_var_node(def, inputs);

Callers 2

elemwise.cppFile · 0.80
tensor_manip.cppFile · 0.80

Calls 1

traitMethod · 0.45

Tested by

no test coverage detected