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

Method SizeArgs

dnn/src/rocm/pooling/algo.cpp:19–24  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

17MEGDNN_DEF_GET_ALGO_FROM_DESC(PoolingForwardImpl)
18
19PoolingForwardImpl::AlgoBase::SizeArgs::SizeArgs(
20 PoolingForwardImpl* o, const TensorLayout& src, const TensorLayout& dst)
21 : handle{concrete_handle(o->handle())},
22 opr{o},
23 layout_src{&src},
24 layout_dst{&dst} {}
25
26PoolingForwardImpl::AlgoBase::ExecArgs::ExecArgs(
27 PoolingForwardImpl* opr, _megdnn_tensor_in src, _megdnn_tensor_out dst,

Callers

nothing calls this directly

Calls 2

concrete_handleFunction · 0.50
handleMethod · 0.45

Tested by

no test coverage detected