fix fc parser bug. modify pooling parser and add its ut. fix gatherND parser bug.pull/4294/head
parent
2dc4dae41c
commit
754ef22831
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -1,34 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserAbs : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserAbs() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./abs.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserAbs, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Abs) << "wrong Op Type";
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -0,0 +1,105 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
|
||||
class TestTfliteParserRelu : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserRelu() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./relu.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserRelu, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Activation) << "wrong Op Type";
|
||||
}
|
||||
|
||||
class TestTfliteParserRelu6 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserRelu6() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./relu6.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserRelu6, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Activation) << "wrong Op Type";
|
||||
}
|
||||
|
||||
class TestTfliteParserTanh : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserTanh() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./tanh.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserTanh, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Activation) << "wrong Op Type";
|
||||
}
|
||||
|
||||
// logistic
|
||||
|
||||
class TestTfliteParserPrelu : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserPrelu() = default;
|
||||
void SetUp() override {
|
||||
meta_graph = LoadAndConvert("./prelu.tflite");
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserPrelu, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Prelu) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserPrelu, AttrValue) {
|
||||
std::vector<float> slope(20, 0);
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive->value.AsPrelu(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.AsPrelu()->slope, slope);
|
||||
}
|
||||
|
||||
class TestTfliteParserLeakyRelu : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserLeakyRelu() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./leaky_relu.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserLeakyRelu, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_LeakyReLU) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserLeakyRelu, AttrValue) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
|
||||
auto val = meta_graph->nodes.front()->primitive->value.AsLeakyReLU();
|
||||
ASSERT_NE(val, nullptr);
|
||||
ASSERT_EQ(val->negativeSlope, 0.20000000298023224);
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,78 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserAdd1 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserAdd1() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./add1.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserAdd1, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Add) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserAdd1, Tensor) {
|
||||
ASSERT_GT(meta_graph->allTensors.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(0)->data.size(), 0);
|
||||
ASSERT_GT(meta_graph->allTensors.at(1)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(2)->data.size(), 0);
|
||||
}
|
||||
|
||||
class TestTfliteParserAdd2 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserAdd2() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./add2.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserAdd2, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Add) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserAdd2, Tensor) {
|
||||
ASSERT_GT(meta_graph->allTensors.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(0)->data.size(), 0);
|
||||
ASSERT_GT(meta_graph->allTensors.at(1)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(2)->data.size(), 0);
|
||||
}
|
||||
|
||||
class TestTfliteParserAdd3 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserAdd3() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./add3.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserAdd3, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Add) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserAdd3, Tensor) {
|
||||
ASSERT_GT(meta_graph->allTensors.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(0)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(1)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(2)->data.size(), 0);
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
File diff suppressed because it is too large
Load Diff
@ -1,33 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserCeil : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserCeil() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./ceil.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserCeil, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Ceil) << "wrong Op Type";
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,34 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserCos : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserCos() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./cos.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserCos, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Cos) << "wrong Op Type";
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,78 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserDiv1 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserDiv1() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./div1.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserDiv1, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Div) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserDiv1, Tensor) {
|
||||
ASSERT_GT(meta_graph->allTensors.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(0)->data.size(), 0);
|
||||
ASSERT_GT(meta_graph->allTensors.at(1)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(2)->data.size(), 0);
|
||||
}
|
||||
|
||||
class TestTfliteParserDiv2 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserDiv2() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./div2.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserDiv2, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Div) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserDiv2, Tensor) {
|
||||
ASSERT_GT(meta_graph->allTensors.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(0)->data.size(), 0);
|
||||
ASSERT_GT(meta_graph->allTensors.at(1)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(2)->data.size(), 0);
|
||||
}
|
||||
|
||||
class TestTfliteParserDiv3 : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserDiv3() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./div3.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserDiv3, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Div) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserDiv3, Tensor) {
|
||||
ASSERT_GT(meta_graph->allTensors.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(0)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(1)->data.size(), 0);
|
||||
ASSERT_EQ(meta_graph->allTensors.at(2)->data.size(), 0);
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,36 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserEqual : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserEqual() = default;
|
||||
void SetUp() override {
|
||||
meta_graph = LoadAndConvert("./equal.tflite");
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserEqual, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Equal) << "wrong Op Type";
|
||||
}
|
||||
} // namespace mindspore
|
@ -1,33 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserFloorDiv : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserFloorDiv() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./floor_div.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserFloorDiv, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_FloorDiv) << "wrong Op Type";
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,33 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserFloorMod : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserFloorMod() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./floor_mod.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserFloorMod, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_FloorMod) << "wrong Op Type";
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,33 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserFloor : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserFloor() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./floor.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserFloor, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Floor) << "wrong Op Type";
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,36 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserGreaterEqual : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserGreaterEqual() = default;
|
||||
void SetUp() override {
|
||||
meta_graph = LoadAndConvert("./greater_equal.tflite");
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserGreaterEqual, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_GreaterEqual) << "wrong Op Type";
|
||||
}
|
||||
} // namespace mindspore
|
@ -1,42 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserLeakyRelu : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserLeakyRelu() = default;
|
||||
void SetUp() override { meta_graph = LoadAndConvert("./leaky_relu.tflite", ""); }
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserLeakyRelu, OpType) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_LeakyReLU) << "wrong Op Type";
|
||||
}
|
||||
|
||||
TEST_F(TestTfliteParserLeakyRelu, AttrValue) {
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
|
||||
auto val = meta_graph->nodes.front()->primitive->value.AsLeakyReLU();
|
||||
ASSERT_NE(val, nullptr);
|
||||
ASSERT_EQ(val->negativeSlope, 0.20000000298023224);
|
||||
}
|
||||
|
||||
} // namespace mindspore
|
@ -1,36 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserLessEqual : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserLessEqual() = default;
|
||||
void SetUp() override {
|
||||
meta_graph = LoadAndConvert("./less_equal.tflite");
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserLessEqual, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_LessEqual) << "wrong Op Type";
|
||||
}
|
||||
} // namespace mindspore
|
@ -1,36 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "ut/tools/converter/parser/tflite/tflite_parsers_test_utils.h"
|
||||
#include <iostream>
|
||||
#include "common/common_test.h"
|
||||
|
||||
namespace mindspore {
|
||||
class TestTfliteParserLess : public TestTfliteParser {
|
||||
public:
|
||||
TestTfliteParserLess() = default;
|
||||
void SetUp() override {
|
||||
meta_graph = LoadAndConvert("./less.tflite");
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(TestTfliteParserLess, OpType) {
|
||||
ASSERT_NE(meta_graph, nullptr);
|
||||
ASSERT_GT(meta_graph->nodes.size(), 0);
|
||||
ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr);
|
||||
ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Less) << "wrong Op Type";
|
||||
}
|
||||
} // namespace mindspore
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue