p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix warnings

This commit is contained in:
gaopan12 2022-10-09 14:48:50 +08:00
parent 44379cd465
commit 072d08a995
2 changed files with 0 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mindspore/core/ops/batch_matmul.cc
View File

@ -157,24 +157,6 @@ TypePtr BatchMatmulInferType(const PrimitivePtr &prim, const std::vector<Abstrac
} // namespace
MIND_API_OPERATOR_IMPL(BatchMatMul, MatMul);
void BatchMatMul::Init(bool transpose_a, bool transpose_b) {
set_transpose_a(transpose_a);
set_transpose_b(transpose_b);
}
void BatchMatMul::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, api::MakeValue(transpose_a)); }
void BatchMatMul::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, api::MakeValue(transpose_b)); }
bool BatchMatMul::get_transpose_a() const {
auto value_ptr = GetAttr(kTransposeA);
return GetValue<bool>(value_ptr);
}
bool BatchMatMul::get_transpose_b() const {
auto value_ptr = GetAttr(kTransposeB);
return GetValue<bool>(value_ptr);
}
AbstractBasePtr BatchMatmulInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<AbstractBasePtr> &input_args) {

14
mindspore/core/ops/batch_matmul.h
View File

@ -33,20 +33,6 @@ class MIND_API BatchMatMul : public MatMul {
MIND_API_BASE_MEMBER(BatchMatMul);
/// \brief Constructor.
BatchMatMul() : MatMul(kNameBatchMatMul) { InitIOName({"x1", "x2"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.BatchMatmul for the inputs.
void Init(bool transpose_a = false, bool transpose_b = false);
/// \brief Set transpose_a.
void set_transpose_a(bool transpose_a);
/// \brief Set transpose_b.
void set_transpose_b(bool transpose_b);
/// \brief Get transpose_a.
///
/// \return transpose_a.
bool get_transpose_a() const;
/// \brief Get transpose_b.
///
/// \return transpose_b.
bool get_transpose_b() const;
};
abstract::AbstractBasePtr BatchMatmulInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);