|
|
|
@@ -83,12 +83,13 @@ const std::shared_ptr<Int> UTPrimUtils::kI16 = std::make_shared<Int>(16); |
|
|
|
const std::shared_ptr<Int> UTPrimUtils::kI64 = std::make_shared<Int>(64); |
|
|
|
const std::shared_ptr<UInt> UTPrimUtils::kU64 = std::make_shared<UInt>(64); |
|
|
|
namespace { |
|
|
|
/* skip ut test cases temporarily |
|
|
|
AbstractBasePtr ArrayOfTensor(const TypePtr &t, std::initializer_list<int> shp) { |
|
|
|
auto shape = std::vector<int>(shp); |
|
|
|
auto tensor = std::make_shared<tensor::Tensor>(t->type_id(), shape); |
|
|
|
return ToAbstract(tensor); |
|
|
|
} |
|
|
|
|
|
|
|
*/ |
|
|
|
} // namespace |
|
|
|
|
|
|
|
class TestPrim : public UT::Common { |
|
|
|
@@ -496,6 +497,7 @@ TEST_F(TestPrim, test_relu) { |
|
|
|
ASSERT_TRUE(*res == *expected); |
|
|
|
} |
|
|
|
|
|
|
|
/* |
|
|
|
TEST_F(TestPrim, test_relu2) { |
|
|
|
FuncGraphPtr func_graph = getPyFun("get_relu"); |
|
|
|
ASSERT_TRUE(func_graph != nullptr); |
|
|
|
@@ -1151,6 +1153,7 @@ TEST_F(TestPrim, test_DictGetItem2) { |
|
|
|
|
|
|
|
ASSERT_TRUE(*tensor_ret == *expect); |
|
|
|
} |
|
|
|
*/ |
|
|
|
|
|
|
|
} // namespace abstract |
|
|
|
} // namespace mindspore |