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

Method do_make_node_prop

src/jit/impl/executor_opr.cpp:408–419  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

406}
407
408JITExecutor::NodeProp* JITExecutor::do_make_node_prop() const {
409 auto ret = Super::do_make_node_prop();
410 using DepType = NodeProp::DepType;
411 SmallVector<DepType> dt(input().size());
412 auto&& placeholders = internal_graph().placeholders();
413 for (size_t i = 0; i < dt.size(); ++i) {
414 dt[i] = placeholders[i]->is_host_value_shape_input() ? DepType::HOST_VALUE
415 : DepType::DEV_VALUE;
416 }
417 ret->reset_dep_type(input(), dt);
418 return ret;
419}
420
421megdnn::TensorShape JITExecutor::broadcasted_input_shape() const {
422 megdnn::TensorShapeArray inp_shps;

Callers

nothing calls this directly

Calls 3

reset_dep_typeMethod · 0.80
do_make_node_propFunction · 0.50
sizeMethod · 0.45

Tested by

no test coverage detected