Searched refs:SoftMaxInferType (Results 1 – 1 of 1) sorted by relevance
71 TypePtr SoftMaxInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in SoftMaxInferType() function82 …return abstract::MakeAbstract(SoftMaxInferShape(primitive, input_args), SoftMaxInferType(primitive… in SoftmaxInfer()