diff options
Diffstat (limited to 'runtimes/neurun/test/graph/operand/Set.cc')
-rw-r--r-- | runtimes/neurun/test/graph/operand/Set.cc | 29 |
1 files changed, 13 insertions, 16 deletions
diff --git a/runtimes/neurun/test/graph/operand/Set.cc b/runtimes/neurun/test/graph/operand/Set.cc index 00b6a7222..fce422398 100644 --- a/runtimes/neurun/test/graph/operand/Set.cc +++ b/runtimes/neurun/test/graph/operand/Set.cc @@ -16,33 +16,30 @@ #include <gtest/gtest.h> -#include "model/operand/Set.h" +#include "model/Operands.h" TEST(graph_operand_Set, set_test) { - neurun::model::operand::Set set; + neurun::model::Operands set; - ::neurun::model::operand::Shape shape0{3}; - shape0.dim(0) = 1; - shape0.dim(1) = 2; - shape0.dim(2) = 3; + ::neurun::model::Shape shape0{1, 2, 3}; - ::neurun::model::operand::Shape shape1{4}; + ::neurun::model::Shape shape1(4); shape1.dim(0) = 10; shape1.dim(1) = 20; shape1.dim(2) = 30; shape1.dim(3) = 40; - ::neurun::model::operand::TypeInfo type{ANEURALNETWORKS_TENSOR_INT32, 0, 0}; + ::neurun::model::TypeInfo type{neurun::model::DataType::INT32}; - set.append(shape0, type); - set.append(shape1, type); + set.emplace(shape0, type); + set.emplace(shape1, type); - ASSERT_EQ(set.exist(neurun::model::operand::Index{0u}), true); - ASSERT_EQ(set.exist(neurun::model::operand::Index{1u}), true); - ASSERT_EQ(set.exist(neurun::model::operand::Index{2u}), false); + ASSERT_EQ(set.exist(neurun::model::OperandIndex{0u}), true); + ASSERT_EQ(set.exist(neurun::model::OperandIndex{1u}), true); + ASSERT_EQ(set.exist(neurun::model::OperandIndex{2u}), false); - ASSERT_EQ(set.at(neurun::model::operand::Index{0u}).shape().dim(0), 1); - ASSERT_EQ(set.at(neurun::model::operand::Index{0u}).shape().dim(1), 2); - ASSERT_EQ(set.at(neurun::model::operand::Index{0u}).shape().dim(2), 3); + ASSERT_EQ(set.at(neurun::model::OperandIndex{0u}).shape().dim(0), 1); + ASSERT_EQ(set.at(neurun::model::OperandIndex{0u}).shape().dim(1), 2); + ASSERT_EQ(set.at(neurun::model::OperandIndex{0u}).shape().dim(2), 3); } |