| 1365 | } |
| 1366 | |
| 1367 | bool StaticInferManagerImpl::SubgraphStaticInferHelperImpl::call_register_for_shape( |
| 1368 | Tag dest, const ShapeInferDesc& desc, RegisterHelperPtr helper) { |
| 1369 | mgb_assert(desc.src_type == SourceType::DEP); |
| 1370 | |
| 1371 | auto callback = [&](StaticInferManagerImpl& mgr, SourceType src_type, |
| 1372 | const TagTraitArray& deps) -> TagTraitBase* { |
| 1373 | using namespace std::placeholders; |
| 1374 | auto f = std::bind( |
| 1375 | &SubgraphStaticInferHelperImpl::infer_shape_raw, deps, desc.infer_func, |
| 1376 | _1, _2); |
| 1377 | |
| 1378 | mgr.register_shape_infer(dest, {src_type, {}, f}); |
| 1379 | return mgr.get_tag_trait_container(dest).shape; |
| 1380 | }; |
| 1381 | |
| 1382 | return (this->*helper)(dest, desc.deps, callback); |
| 1383 | } |
| 1384 | |
| 1385 | bool StaticInferManagerImpl::SubgraphStaticInferHelperImpl::call_register_for_value( |
| 1386 | Tag dest, const ValueInferDesc& desc, RegisterHelperPtr helper) { |
nothing calls this directly
no test coverage detected