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
|
#include "caffe2/core/common.h"
#include "caffe2/core/test_utils.h"
#include "caffe2/opt/converter.h"
#include "caffe2/opt/custom/concat_elim.h"
#include "caffe2/predictor/emulator/data_filler.h"
#include "caffe2/utils/proto_utils.h"
#include <gtest/gtest.h>
using namespace caffe2::testing;
using namespace caffe2::emulator;
using caffe2::OperatorDef;
using std::vector;
TEST(Converter, ClipRangesGatherSigridHashConverter) {
OperatorDef op;
op.set_type("ClipRangesGatherSigridHash");
op.add_arg()->CopyFrom(caffe2::MakeArgument<bool>("hash_into_int32", true));
auto nnDef = convertToNeuralNetOperator(op);
auto* pNNDef =
static_cast<nom::repr::ClipRangesGatherSigridHash*>(nnDef.get());
EXPECT_TRUE(pNNDef);
EXPECT_TRUE(pNNDef->getHashIntoInt32());
OperatorDef op2;
op2.set_type("ClipRangesGatherSigridHash");
op2.add_arg()->CopyFrom(caffe2::MakeArgument<bool>("hash_into_int32", false));
auto nnDef2 = convertToNeuralNetOperator(op2);
auto* pNNDef2 =
static_cast<nom::repr::ClipRangesGatherSigridHash*>(nnDef2.get());
EXPECT_TRUE(pNNDef2);
EXPECT_FALSE(pNNDef2->getHashIntoInt32());
}
TEST(Converter, ClipRangesGatherSigridHashV2Converter) {
OperatorDef op;
op.set_type("ClipRangesGatherSigridHashV2");
op.add_arg()->CopyFrom(caffe2::MakeArgument<bool>("hash_into_int32", true));
auto nnDef = convertToNeuralNetOperator(op);
auto* pNNDef =
static_cast<nom::repr::ClipRangesGatherSigridHashV2*>(nnDef.get());
EXPECT_TRUE(pNNDef);
EXPECT_TRUE(pNNDef->getHashIntoInt32());
OperatorDef op2;
op2.set_type("ClipRangesGatherSigridHashV2");
op2.add_arg()->CopyFrom(caffe2::MakeArgument<bool>("hash_into_int32", false));
auto nnDef2 = convertToNeuralNetOperator(op2);
auto* pNNDef2 =
static_cast<nom::repr::ClipRangesGatherSigridHashV2*>(nnDef2.get());
EXPECT_TRUE(pNNDef2);
EXPECT_FALSE(pNNDef2->getHashIntoInt32());
}
|
