forked from mindspore-Ecosystem/mindspore
!31644 code clean
Merge pull request !31644 from marui/master_code_clean
This commit is contained in:
commit
49a1251cc8
|
@ -49,6 +49,7 @@ class AscendMemAdapter {
|
||||||
std::string DevMemDetailInfo() const;
|
std::string DevMemDetailInfo() const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
AscendMemAdapter() = default;
|
||||||
struct MemoryBlock {
|
struct MemoryBlock {
|
||||||
MemoryBlock(void *ptr, const size_t size, const std::string &tag) {
|
MemoryBlock(void *ptr, const size_t size, const std::string &tag) {
|
||||||
mem_ptr = ptr;
|
mem_ptr = ptr;
|
||||||
|
|
|
@ -29,7 +29,7 @@ namespace mindspore {
|
||||||
namespace opt {
|
namespace opt {
|
||||||
class SegmentEltwiseFusionPass : public FusionBasePass {
|
class SegmentEltwiseFusionPass : public FusionBasePass {
|
||||||
public:
|
public:
|
||||||
explicit SegmentEltwiseFusionPass(FusionIdAllocatorPtr idAllocator)
|
explicit SegmentEltwiseFusionPass(const FusionIdAllocatorPtr &idAllocator)
|
||||||
: FusionBasePass("SegmentEltwiseFusionPass", idAllocator) {}
|
: FusionBasePass("SegmentEltwiseFusionPass", idAllocator) {}
|
||||||
~SegmentEltwiseFusionPass() override = default;
|
~SegmentEltwiseFusionPass() override = default;
|
||||||
|
|
||||||
|
|
|
@ -98,7 +98,7 @@ const AnfNodePtr SliceGradUnifyMindIR::Process(const FuncGraphPtr &graph, const
|
||||||
}
|
}
|
||||||
std::vector<std::vector<int64_t>> paddings;
|
std::vector<std::vector<int64_t>> paddings;
|
||||||
for (size_t i = 0; i < x_shape.size(); ++i) {
|
for (size_t i = 0; i < x_shape.size(); ++i) {
|
||||||
paddings.emplace_back(std::vector<int64_t>{begins[i], x_shape[i] - begins[i] - sizes[i]});
|
paddings.emplace_back(std::vector<int64_t>{begins[i], (x_shape[i] - begins[i] - sizes[i])});
|
||||||
}
|
}
|
||||||
common::AnfAlgo::SetNodeAttr(kAttrPaddings, MakeValue(paddings), pad);
|
common::AnfAlgo::SetNodeAttr(kAttrPaddings, MakeValue(paddings), pad);
|
||||||
common::AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(std::vector<std::string>{"x"}), pad);
|
common::AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(std::vector<std::string>{"x"}), pad);
|
||||||
|
|
Loading…
Reference in New Issue