Closed yangninghua closed 5 years ago
elif parent_type == 'ExpandBackward':
aa = 1
elif parent_type == 'ThnnConv2DBackward':
aa = 1
elif parent_type == 'ThnnBatchNormBackward':
aa = 1
elif parent_type == 'ThMulBackward':
aa = 1
elif parent_type == 'ConstantPadNdBackward':
aa = 1
elif parent_type == 'SliceBackward':
aa = 1
elif parent_type == 'CloneBackward':
aa = 1
elif parent_type == 'CatBackward':
aa = 1
elif parent_type == 'AdaptiveAvgPool2DBackward':
aa = 1
elif parent_type == 'ThAddBackward':
aa = 1
elif parent_type == 'SqueezeBackward1':
aa = 1
elif parent_type == 'TBackward':
aa = 1
elif parent_type == 'ThAddmmBackward':
aa = 1
pytorch >= 0.4.1
@yangninghua,it real work?
layer_dict = {'ConvNdBackward': 'Convolution', 'ThresholdBackward': 'ReLU', 'MaxPool2dBackward': 'Pooling', 'AvgPool2dBackward': 'Pooling', 'DropoutBackward': 'Dropout', 'AddmmBackward': 'InnerProduct', 'BatchNormBackward': 'BatchNorm', 'AddBackward': 'Eltwise', 'ViewBackward': 'Reshape', 'ConcatBackward': 'Concat', 'UpsamplingNearest2d': 'Deconvolution', 'UpsamplingBilinear2d': 'Deconvolution', 'SigmoidBackward': 'Sigmoid', 'LeakyReLUBackward': 'ReLU', 'NegateBackward': 'Power', 'MulBackward': 'Eltwise', 'SpatialCrossMapLRNFunc': 'LRN',