Beispiel #1
0
 def SoftFuseDWConvToPad(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups,
                               NNDCT_OP.DEPTHWISE_CONV2D, NNDCT_OP.PAD)
Beispiel #2
0
 def SoftFuseChunk(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.CHUNK)
Beispiel #3
0
 def SoftFuseConvToSpaceBatchNd(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.CONV2D,
                                 NNDCT_OP.SPACE_TO_BATCH_ND)
Beispiel #4
0
 def SoftFuseReduceSum(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.SUM)
Beispiel #5
0
 def SoftFuseRepeat(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.REPEAT)
Beispiel #6
0
 def SoftFuseClamp(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.CLAMP)
Beispiel #7
0
 def SoftFuseFlatten(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.FLATTEN)
Beispiel #8
0
 def SoftFuseExpand(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.EXPAND)
Beispiel #9
0
 def SoftFuseLeakyRelu(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.LEAKY_RELU)
Beispiel #10
0
 def SoftFuseChannelScale(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.CHANNEL_SCALE)
Beispiel #11
0
 def SoftFuseContiguous(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.CONTIGUOUS)
Beispiel #12
0
 def SoftFuseDropout(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.DROPOUT)
Beispiel #13
0
 def SoftFuseUpSampling(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.UP_SAMPLING)
Beispiel #14
0
 def SoftFusePermute(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.PERMUTE)
Beispiel #15
0
 def SoftFuseSqueeze(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.SQUEEZE)
Beispiel #16
0
 def SoftFuseDivide(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.DIV)
Beispiel #17
0
 def SoftFusePixelShuffle(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.PIXEL_SHUFFLE)
Beispiel #18
0
 def SoftFuseInplaceCopy(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups,
                                 NNDCT_OP.INPLACE_COPY)
Beispiel #19
0
 def SoftFuseSplit(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.SPLIT)
Beispiel #20
0
 def SoftFuseSelect(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.SELECT)
Beispiel #21
0
 def SoftFuseStrideSlice(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.STRIDED_SLICE)
Beispiel #22
0
 def SoftFuseBatchSpaceNdToConv(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups,
                                 NNDCT_OP.BATCH_TO_SPACE_ND,
                                 NNDCT_OP.CONV2D)
Beispiel #23
0
 def SoftFuseTranspose(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.TRANSPOSE)
Beispiel #24
0
 def SoftFuseHardtanh(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.HARDTANH)
Beispiel #25
0
 def SoftFuseTile(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.TILE)
Beispiel #26
0
 def SoftFuseReluk(graph, quant_groups):
     return graph_utils.group_up(graph, quant_groups, NNDCT_OP.RELUK)
Beispiel #27
0
 def SoftFuseConvToPad(graph, quant_groups):
   return graph_utils.group_up(graph, quant_groups, NNDCT_OP.CONV2D,
                               NNDCT_OP.PAD)