Function mindspore::ops::DiagPartInfer
Defined in File diag_part.h
Function Documentation
-
abstract::AbstractBasePtr mindspore::ops::DiagPartInfer(const abstract::AnalysisEnginePtr&, const PrimitivePtr &primitive, const std::vector<abstract::AbstractBasePtr> &input_args)