Home
last modified time | relevance | path

Searched refs:AddTfOp (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc57 AddTfOp(testing::kUnpack, {0}, {1, 2}); in TEST_F()
58 AddTfOp(testing::kUnpack, {3}, {4, 5}); in TEST_F()
59 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
60 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
61 AddTfOp(testing::kMul, {6, 7}, {8}); in TEST_F()
82 AddTfOp(testing::kAdd, {0, 1}, {2}); in TEST_F()
101 AddTfOp(testing::kAdd, {0, 1}, {2}); in TEST_F()
120 AddTfOp(testing::kUnpack, {0}, {1, 2}); in TEST_F()
121 AddTfOp(testing::kUnpack, {3}, {4, 5}); in TEST_F()
122 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
[all …]
Dkernel_test.cc64 AddTfOp(testing::kUnpack, {0}, {1, 2}); in TEST_F()
65 AddTfOp(testing::kUnpack, {3}, {4, 5}); in TEST_F()
66 AddTfOp(testing::kAdd, {1, 4}, {6}); in TEST_F()
67 AddTfOp(testing::kAdd, {2, 5}, {7}); in TEST_F()
68 AddTfOp(testing::kMul, {6, 7}, {8}); in TEST_F()
97 AddTfOp(testing::kNonExistent, {0}, {1}); in TEST_F()
108 AddTfOp(testing::kIdentity, {0}, {1, 2}); in TEST_F()
124 AddTfOp(testing::kIncompatibleNodeDef, {0}, {1}); in TEST_F()
138 AddTfOp(testing::kUnpack, {0}, {1, 2}); in TEST_F()
154 AddTfOp(testing::kUnpack, {0}, {1, 2}); in TEST_F()
[all …]
Dtest_util.cc133 void FlexModelTest::AddTfOp(TfOpType op, const std::vector<int>& inputs, in AddTfOp() function in tflite::flex::testing::FlexModelTest
163 AddTfOp("FlexUnpack", "Unpack", attributes, inputs, outputs); in AddTfOp()
166 AddTfOp("FlexIdentity", "Identity", attributes, inputs, outputs); in AddTfOp()
169 AddTfOp("FlexAdd", "Add", attributes, inputs, outputs); in AddTfOp()
172 AddTfOp("FlexMul", "Mul", attributes, inputs, outputs); in AddTfOp()
174 AddTfOp("FlexRFFT", "RFFT", "", inputs, outputs); in AddTfOp()
176 AddTfOp("FlexImag", "Imag", "", inputs, outputs); in AddTfOp()
178 AddTfOp("NonExistentOp", "NonExistentOp", "", inputs, outputs); in AddTfOp()
181 AddTfOp("FlexCast", "Cast", "", inputs, outputs); in AddTfOp()
185 void FlexModelTest::AddTfOp(const char* tflite_name, const string& tf_name, in AddTfOp() function in tflite::flex::testing::FlexModelTest
Dtest_util.h110 void AddTfOp(TfOpType op, const std::vector<int>& inputs,
121 void AddTfOp(const char* tflite_name, const string& tf_name,