Browse Source

fix ascend bug

tags/v1.1.0
xiaoyisd 5 years ago
parent
commit
9a07c607e6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/quant.py

+ 1
- 1
mindspore/nn/layer/quant.py View File

@@ -423,7 +423,7 @@ class Conv2dBnFoldQuantOneConv(Cell):
data_format = 'NCHW'
self.format = Validator.check_string(data_format, ['NCHW', 'NHWC'], 'format', self.cls_name)
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
if context.get_context("enable_ge"):
self.is_ge_backend = True


Loading…
Cancel
Save