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

fix ascend bug

This commit is contained in:
xiaoyisd 2020-11-27 10:27:24 +08:00
parent 3fa7233f68
commit 9a07c607e6
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