Searched defs:fuse_type (Results 1 – 2 of 2) sorted by relevance
101 …explicit FuseElemwiseFwd(FuseType fuse_type) : FusePattern("elemwise_fwd"), fuse_type_(fuse_type) { in FuseElemwiseFwd()117 …explicit FuseElemwiseBroadcastFwd(FuseType fuse_type) : FusePattern("elemwise_broadcast_fwd"), fus… in FuseElemwiseBroadcastFwd()133 explicit FuseDynElemwiseBroadcastFwd(FuseType fuse_type) in FuseDynElemwiseBroadcastFwd()150 FuseReduceFwd(FuseType fuse_type, size_t size_limit) in FuseReduceFwd()172 FuseDynReduceFwd(FuseType fuse_type, size_t size_limit) in FuseDynReduceFwd()194 FuseElemwiseBroadcastBwd(FuseType fuse_type, size_t size_limit) in FuseElemwiseBroadcastBwd()
139 FuseReduceFwd(FuseType fuse_type, size_t size_limit) in FuseReduceFwd()