提交 f69d0d38 authored 作者: affanv14's avatar affanv14

use one naming convention

上级 27007886
...@@ -2889,7 +2889,7 @@ def local_abstractconv_cudnn(node): ...@@ -2889,7 +2889,7 @@ def local_abstractconv_cudnn(node):
@local_optimizer([AbstractConv2d, AbstractConv2d_gradWeights, AbstractConv2d_gradInputs]) @local_optimizer([AbstractConv2d, AbstractConv2d_gradWeights, AbstractConv2d_gradInputs])
def local_abstractconv_cudnn_alternative(node): def local_abstractconv_cudnn_alt(node):
if(not isinstance(node.op, (AbstractConv2d, AbstractConv2d_gradWeights, if(not isinstance(node.op, (AbstractConv2d, AbstractConv2d_gradWeights,
AbstractConv2d_gradInputs))): AbstractConv2d_gradInputs))):
return return
...@@ -2994,7 +2994,7 @@ def local_abstractconv_cudnn_alternative(node): ...@@ -2994,7 +2994,7 @@ def local_abstractconv_cudnn_alternative(node):
@local_optimizer([AbstractConv3d, AbstractConv3d_gradWeights, AbstractConv3d_gradInputs]) @local_optimizer([AbstractConv3d, AbstractConv3d_gradWeights, AbstractConv3d_gradInputs])
def local_abstractconv3d_cudnn_alternative(node): def local_abstractconv3d_cudnn_alt(node):
if(not isinstance(node.op, (AbstractConv3d, if(not isinstance(node.op, (AbstractConv3d,
AbstractConv3d_gradWeights, AbstractConv3d_gradWeights,
AbstractConv3d_gradInputs))): AbstractConv3d_gradInputs))):
......
...@@ -1629,7 +1629,7 @@ def local_abstractconv_gemm(node): ...@@ -1629,7 +1629,7 @@ def local_abstractconv_gemm(node):
@local_optimizer([AbstractConv2d]) @local_optimizer([AbstractConv2d])
def local_abstractconv_gemm_alternative(node): def local_abstractconv_gemm_alt(node):
if not isinstance(node.op, AbstractConv2d): if not isinstance(node.op, AbstractConv2d):
return None return None
img, kern = node.inputs img, kern = node.inputs
...@@ -2667,8 +2667,8 @@ register_opt('fast_compile')(abstractconv_groupopt) ...@@ -2667,8 +2667,8 @@ register_opt('fast_compile')(abstractconv_groupopt)
from .dnn import (local_abstractconv_cudnn, from .dnn import (local_abstractconv_cudnn,
local_abstractconv_gw_cudnn, local_abstractconv_gw_cudnn,
local_abstractconv_gi_cudnn, # noqa: 402 local_abstractconv_gi_cudnn, # noqa: 402
local_abstractconv_cudnn_alternative, local_abstractconv_cudnn_alt,
local_abstractconv3d_cudnn_alternative) local_abstractconv3d_cudnn_alt)
abstractconv_groupopt.register('local_abstractconv_dnn', abstractconv_groupopt.register('local_abstractconv_dnn',
local_abstractconv_cudnn, 20, local_abstractconv_cudnn, 20,
...@@ -2727,15 +2727,15 @@ conv_metaopt.register(local_abstractconv_gradinputs_gemm, ...@@ -2727,15 +2727,15 @@ conv_metaopt.register(local_abstractconv_gradinputs_gemm,
['default', 'conv_gemm']) ['default', 'conv_gemm'])
conv_metaopt.register(local_abstractconv3d_gradinputs_gemm, conv_metaopt.register(local_abstractconv3d_gradinputs_gemm,
['default', 'conv_gemm']) ['default', 'conv_gemm'])
conv_metaopt.register(local_abstractconv_gemm_alternative, conv_metaopt.register(local_abstractconv_gemm_alt,
['default', 'alternative', 'conv_gemm']) ['default', 'alternative', 'conv_gemm'])
conv_metaopt.register(local_abstractconv_gemm_gradweights_alt, conv_metaopt.register(local_abstractconv_gemm_gradweights_alt,
['default', 'alternative', 'conv_gemm']) ['default', 'alternative', 'conv_gemm'])
conv_metaopt.register(local_abstractconv_gradinputs_gemm_alt, conv_metaopt.register(local_abstractconv_gradinputs_gemm_alt,
['default', 'alternative', 'conv_gemm']) ['default', 'alternative', 'conv_gemm'])
conv_metaopt.register(local_abstractconv_cudnn_alternative, conv_metaopt.register(local_abstractconv_cudnn_alt,
['default', 'alternative', 'cudnn', 'conv_dnn']) ['default', 'alternative', 'cudnn', 'conv_dnn'])
conv_metaopt.register(local_abstractconv3d_cudnn_alternative, conv_metaopt.register(local_abstractconv3d_cudnn_alt,
['default', 'alternative', 'cudnn', 'conv_dnn']) ['default', 'alternative', 'cudnn', 'conv_dnn'])
conv_metaopt.register(local_abstractconv3d_alt, conv_metaopt.register(local_abstractconv3d_alt,
['default', 'alternative', 'conv_gemm']) ['default', 'alternative', 'conv_gemm'])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论