Searched defs:AdaptiveAvgPool2D (Results 1 – 4 of 4) sorted by relevance
32 AdaptiveAvgPool2D() : BaseOperator(kNameAdaptiveAvgPool2D) { InitIOName({"x"}, {"y"}); } in AdaptiveAvgPool2D() function33 …explicit AdaptiveAvgPool2D(const std::string &kName) : BaseOperator(kName) { InitIOName({"x"}, {"y… in AdaptiveAvgPool2D() function
19 CUST_IMPLEMT_INFERFUNC(AdaptiveAvgPool2D, AdaptiveAvgPool2dInferShape) { in CUST_IMPLEMT_INFERFUNC() argument59 CUST_IMPLEMT_VERIFIER(AdaptiveAvgPool2D, AdaptiveAvgPool2dVerify) { return GRAPH_SUCCESS; } in CUST_IMPLEMT_VERIFIER() argument
56 [AdaptiveAvgPool2D] section57 opInfo.subTypeOfInferShape=158 opInfo.opsFlag=OPS_FLAG_CLOSE59 opInfo.engine=DNN_VM_AICPU60 opInfo.flagPartial=False61 opInfo.computeCost=10062 opInfo.flagAsync=False63 opInfo.opKernelLib=CUSTAICPUKernel64 opInfo.formatAgnostic=False65 opInfo.kernelSo=libcust_cpu_kernels.so[all …]
168 class AdaptiveAvgPool2D(Primitive): class