Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc131 {prim::kPrimSplit, {InferImplSplit, nullptr, true}}, in GetPrimitiveToEvalImplMap()
Dinfer_functions.h254 AbstractBasePtr InferImplSplit(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
Dprim_arrays.cc827 AbstractBasePtr InferImplSplit(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplSplit() function