Home
last modified time | relevance | path

Searched refs:MakeGlobalAvgPool2D (Results 1 – 3 of 3) sorted by relevance

/dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/op/nn/
H A Dpooling.cc340 Expr MakeGlobalAvgPool2D(Expr data, in MakeGlobalAvgPool2D() function
350 .set_body_typed(MakeGlobalAvgPool2D);
/dports/misc/tvm/incubator-tvm-0.6.1/src/relay/op/nn/
H A Dpooling.cc340 Expr MakeGlobalAvgPool2D(Expr data, in MakeGlobalAvgPool2D() function
350 .set_body_typed(MakeGlobalAvgPool2D);
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/op/nn/
H A Dpooling.cc298 Expr MakeGlobalAvgPool2D(Expr data, String layout) { in MakeGlobalAvgPool2D() function
305 TVM_REGISTER_GLOBAL("relay.op.nn._make.global_avg_pool2d").set_body_typed(MakeGlobalAvgPool2D);