blob: f3c4a5f81d55a58ac7f1b4d359276be71685c568 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/*
* Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
*
* 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 "FeatureBiasAddConverter.h"
#include "GraphBlock.h"
#include "Dialect/IR/TFLNodes.h"
#include "TestGraph.h"
#include "TestHelper.h"
#include <loco.h>
#include <gtest/gtest.h>
TEST(FeatureBiasAddConverterTest, basic_test)
{
exo::test::ExampleGraph<exo::test::ExampleGraphType::FeatureBiasAdd> g;
{ // attrib setting
// pull
g.pull->dtype(loco::DataType::FLOAT32);
g.pull->shape({1, 2, 2, 3});
// bias value
g.constgen->dtype(loco::DataType::FLOAT32);
g.constgen->shape({3});
g.constgen->size<loco::DataType::FLOAT32>(3);
g.constgen->at<loco::DataType::FLOAT32>(0) = 0.5;
g.constgen->at<loco::DataType::FLOAT32>(1) = 1;
g.constgen->at<loco::DataType::FLOAT32>(2) = 1.5;
}
EXO_TEST_ASSERT_NODE_COUNT({g.push}, 7); // sanity check
// let's convert!!
{
exo::test::TypeShapeReadyPhase test_phase;
test_phase.add_pass<exo::FeatureBiasAddConverter>();
test_phase.run(g.graph());
/*
Expected:
Pull - FeatureEncoder - FeatureDecode - TFLAdd - FeatureEncode - FeatureDecode - Push
|
ConstGen - BiasEncode - BiasDecode ---+
*/
}
// check surroundings
auto tfl_add = exo::test::find_first_node_bytype<locoex::TFLAdd>(g.graph());
{
ASSERT_TRUE(tfl_add != nullptr);
// input x and its pred
{
auto actual_fea_dec = dynamic_cast<loco::FeatureDecode *>(tfl_add->x());
ASSERT_TRUE(actual_fea_dec != nullptr);
auto actual_fea_enc = dynamic_cast<loco::FeatureEncode *>(actual_fea_dec->input());
ASSERT_TRUE(actual_fea_enc != nullptr);
ASSERT_TRUE(actual_fea_enc == g.fea_enc);
}
// input y and its pred
{
auto actual_bias_dec = dynamic_cast<loco::BiasDecode *>(tfl_add->y());
ASSERT_TRUE(actual_bias_dec != nullptr);
auto actual_bias_enc = dynamic_cast<loco::BiasEncode *>(actual_bias_dec->input());
ASSERT_TRUE(actual_bias_enc != nullptr);
ASSERT_TRUE(actual_bias_enc == g.bias_enc);
}
// output check
{
auto actual_fea_enc = exo::test::get_only_succ<loco::FeatureEncode>(tfl_add);
ASSERT_TRUE(actual_fea_enc != nullptr);
auto actual_fea_dec = exo::test::get_only_succ<loco::FeatureDecode>(actual_fea_enc);
ASSERT_TRUE(actual_fea_dec != nullptr);
ASSERT_TRUE(actual_fea_dec == g.fea_dec);
}
}
}
|