Browse Source

Fix the problems of the ad optimization

tags/v0.5.0-beta
Kang 5 years ago
parent
commit
c613be36a9
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/optimizer/ad/dfunctor.h

+ 1
- 1
mindspore/ccsrc/optimizer/ad/dfunctor.h View File

@@ -61,7 +61,7 @@ struct PrimitiveTotalEqual {
} }
}; };


using Registry = std::unordered_map<PrimitivePtr, FuncGraphPtr, PrimitiveHasher, PrimitiveTotalEqual>;
using Registry = std::unordered_map<PrimitivePtr, FuncGraphPtr, PrimitiveHasher>;
class KPrim; class KPrim;
extern KPrim g_k_prims; extern KPrim g_k_prims;
class DFunctor; class DFunctor;


Loading…
Cancel
Save