Home
last modified time | relevance | path

Searched refs:InferImplFusedSparseAdam (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc250 {prim::kPrimFusedSparseAdam, {InferImplFusedSparseAdam, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
Dinfer_functions.h48 AbstractBasePtr InferImplFusedSparseAdam(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
Dprim_nn.cc185 AbstractBasePtr InferImplFusedSparseAdam(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplFusedSparseAdam() function