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

!48558 fix acl BatchToSpace and SpaceToBatch ops 

Merge pull request !48558 from xulei/batchtospace
This commit is contained in:
i-robot 2023-02-09 07:08:36 +00:00 committed by Gitee
parent bb8d835b26 18d013458f
commit d6d132e3ae
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mindspore/ccsrc/transform/graph_ir/op_adapter.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2019-2022 Huawei Technologies Co., Ltd
* Copyright 2019-2023 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -419,17 +419,17 @@ class OpAdapter : public BaseOpAdapter {
const AnyTraits<std::vector<int64_t>>) {
MS_EXCEPTION_IF_NULL(value);
MS_LOG(DEBUG) << "Value: " << value->type_name();
if (!value->isa<ValueList>()) {
MS_LOG(EXCEPTION) << "Value should be ValueList, but got " << value->type_name();
if (!value->isa<ValueSequence>()) {
MS_LOG(EXCEPTION) << "Value should be ValueSequence, but got " << value->type_name();
}
auto vec = value->cast<ValueListPtr>();
auto vec = value->cast<ValueSequencePtr>();
std::vector<int64_t> list;
for (auto &it : vec->value()) {
MS_EXCEPTION_IF_NULL(it);
if (!it->isa<ValueList>()) {
MS_LOG(EXCEPTION) << "It should be ValueList, but got " << it->type_name();
if (!it->isa<ValueSequence>()) {
MS_LOG(EXCEPTION) << "It should be ValueSequence, but got " << it->type_name();
}
auto sub_vector = it->cast<ValueListPtr>();
auto sub_vector = it->cast<ValueSequencePtr>();
for (auto &item : sub_vector->value()) {
list.push_back(static_cast<int64_t>(GetValue<int64_t>(item)));
}