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