Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!9098 fix the bug that Conv2dBnFoldQuantoneconv ascend . 

From: @xiaoyisd
Reviewed-by: @sanjaychan
Signed-off-by: @sanjaychan
This commit is contained in:
mindspore-ci-bot 2020-11-27 19:40:53 +08:00 committed by Gitee
parent 3874160faf 9a07c607e6
commit 864288e10a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/quant.py
View File

@ -423,7 +423,7 @@ class Conv2dBnFoldQuantOneConv(Cell):
data_format = 'NCHW' data_format = 'NCHW'
self.format = Validator.check_string(data_format, ['NCHW', 'NHWC'], 'format', self.cls_name) self.format = Validator.check_string(data_format, ['NCHW', 'NHWC'], 'format', self.cls_name)
self.is_gpu = context.get_context('device_target') == "GPU" self.is_gpu = context.get_context('device_target') == "GPU"
self.is_Ascend = context.get_context('device_target') == "Ascend" self.is_ascend = context.get_context('device_target') == "Ascend"
self.is_graph_mode = context.get_context("mode") == context.GRAPH_MODE self.is_graph_mode = context.get_context("mode") == context.GRAPH_MODE
if context.get_context("enable_ge"): if context.get_context("enable_ge"):
self.is_ge_backend = True self.is_ge_backend = True