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

Function get_input_layout_constraint

imperative/src/impl/ops/broadcast.cpp:12–15  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

10namespace imperative {
11namespace meshgrid {
12SmallVector<VarNode::LayoutConstraintCallback> get_input_layout_constraint(
13 const OpDef& def, const SmallVector<TensorPtr>& inputs) {
14 return SmallVector<VarNode::LayoutConstraintCallback>(inputs.size());
15}
16
17std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
18 const OpDef& def, const SmallVector<LogicalTensorDesc>& inputs) {

Callers

nothing calls this directly

Calls 5

try_reshapeMethod · 0.80
sizeMethod · 0.45
proxy_to_default_cpuMethod · 0.45
get_valueMethod · 0.45
total_nr_elemsMethod · 0.45

Tested by

no test coverage detected