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

Method get_input_layout_constraint

imperative/src/impl/op_def.cpp:52–55  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

50}
51
52SmallVector<VarNode::LayoutConstraintCallback> OpDef::get_input_layout_constraint(
53 const OpDef& def, const SmallVector<TensorPtr>& inputs) {
54 return def.trait()->get_input_layout_constraint(def, inputs);
55}
56
57EncodedSubgraph OpDef::make_backward_graph(
58 const OpDef& def, const SmallVector<LogicalTensorDesc>& inputs,

Callers 12

broadcast.cppFile · 0.80
vision.cppFile · 0.80
indexing.cppFile · 0.80
lamb.cppFile · 0.80
fill.cppFile · 0.80
subtensor.cppFile · 0.80
utility.cppFile · 0.80
pixel_shuffle.cppFile · 0.80
io_remote.cppFile · 0.80
reduce.cppFile · 0.80
matmul.cppFile · 0.80
tensor_manip.cppFile · 0.80

Calls 1

traitMethod · 0.45

Tested by

no test coverage detected