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