Searched refs:MakeGlobalAvgPool2D (Results 1 – 3 of 3) sorted by relevance
340 Expr MakeGlobalAvgPool2D(Expr data, in MakeGlobalAvgPool2D() function350 .set_body_typed(MakeGlobalAvgPool2D);
298 Expr MakeGlobalAvgPool2D(Expr data, String layout) { in MakeGlobalAvgPool2D() function305 TVM_REGISTER_GLOBAL("relay.op.nn._make.global_avg_pool2d").set_body_typed(MakeGlobalAvgPool2D);