forked from mindspore-Ecosystem/mindspore
!46661 [MS][Win]fix win compile bug
Merge pull request !46661 from mengyuanli/fix_win_bug
This commit is contained in:
commit
dcaca99b04
|
@ -33,13 +33,10 @@ class MIND_API FloorDiv : public BaseOperator {
|
|||
FloorDiv() : BaseOperator(kNameFloorDiv) { InitIOName({"x", "y"}, {"output"}); }
|
||||
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.FloorDiv for the inputs.
|
||||
void Init() const {}
|
||||
MIND_API abstract::AbstractBasePtr FloorDivInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
||||
const std::vector<abstract::AbstractBasePtr> &input_args);
|
||||
|
||||
using PrimFloorDivPtr = std::shared_ptr<FloorDiv>;
|
||||
};
|
||||
MIND_API abstract::AbstractBasePtr FloorDivInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
||||
const std::vector<abstract::AbstractBasePtr> &input_args);
|
||||
using PrimFloorDivPtr = std::shared_ptr<FloorDiv>;
|
||||
} // namespace ops
|
||||
} // namespace mindspore
|
||||
|
||||
|
|
|
@ -58,10 +58,9 @@ class MIND_API LRNGrad : public BaseOperator {
|
|||
///
|
||||
/// \return beta.
|
||||
float get_beta() const;
|
||||
|
||||
MIND_API abstract::AbstractBasePtr LrnGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
||||
const std::vector<abstract::AbstractBasePtr> &input_args);
|
||||
};
|
||||
MIND_API abstract::AbstractBasePtr LrnGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
||||
const std::vector<abstract::AbstractBasePtr> &input_args);
|
||||
} // namespace ops
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CORE_OPS_LRN_GRAD_H_
|
||||
|
|
|
@ -38,14 +38,11 @@ class MIND_API ResizeBicubicGrad : public BaseOperator {
|
|||
void set_half_pixel_centers(const bool half_pixel_centers);
|
||||
bool get_align_corners() const;
|
||||
bool get_half_pixel_centers() const;
|
||||
MIND_API abstract::AbstractBasePtr ResizeBicubicGradInfer(const abstract::AnalysisEnginePtr &,
|
||||
const PrimitivePtr &primitive,
|
||||
const std::vector<AbstractBasePtr> &input_args);
|
||||
using PrimResizeBicubicGrad = std::shared_ptr<ResizeBicubicGrad>;
|
||||
};
|
||||
MIND_API abstract::AbstractBasePtr ResizeBicubicGradInfer(const abstract::AnalysisEnginePtr &,
|
||||
const PrimitivePtr &primitive,
|
||||
const std::vector<AbstractBasePtr> &input_args);
|
||||
using PrimResizeBicubicGrad = std::shared_ptr<ResizeBicubicGrad>;
|
||||
} // namespace ops
|
||||
} // namespace mindspore
|
||||
|
||||
|
|
|
@ -38,14 +38,11 @@ class MIND_API ResizeBicubic : public BaseOperator {
|
|||
void set_half_pixel_centers(const bool half_pixel_centers);
|
||||
bool get_align_corners() const;
|
||||
bool get_half_pixel_centers() const;
|
||||
MIND_API abstract::AbstractBasePtr ResizeBicubicInfer(const abstract::AnalysisEnginePtr &,
|
||||
const PrimitivePtr &primitive,
|
||||
const std::vector<AbstractBasePtr> &input_args);
|
||||
using PrimResizeBicubic = std::shared_ptr<ResizeBicubic>;
|
||||
};
|
||||
MIND_API abstract::AbstractBasePtr ResizeBicubicInfer(const abstract::AnalysisEnginePtr &,
|
||||
const PrimitivePtr &primitive,
|
||||
const std::vector<AbstractBasePtr> &input_args);
|
||||
using PrimResizeBicubic = std::shared_ptr<ResizeBicubic>;
|
||||
} // namespace ops
|
||||
} // namespace mindspore
|
||||
|
||||
|
|
|
@ -68,9 +68,9 @@ class MIND_API Splice : public BaseOperator {
|
|||
///
|
||||
/// \param[in] output_dim Define the output_dim.
|
||||
int64_t get_output_dim() const;
|
||||
MIND_API abstract::AbstractBasePtr SpliceInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
||||
const std::vector<abstract::AbstractBasePtr> &input_args);
|
||||
};
|
||||
MIND_API abstract::AbstractBasePtr SpliceInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
||||
const std::vector<abstract::AbstractBasePtr> &input_args);
|
||||
} // namespace ops
|
||||
} // namespace mindspore
|
||||
|
||||
|
|
Loading…
Reference in New Issue