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