Searched refs:GetOperator (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 34 const BaseOperator& GetOperator(const string& name, OperatorType type) { in GetOperator() function in toco::tflite::__anonfbd96e950111::OperatorTest
93 SerializeAndDeserialize(GetOperator(name, type), T(), &options); in CheckSimpleOperator()
108 auto output_toco_op = SerializeAndDeserialize(GetOperator(name, type), op); in CheckReducerOperator()
165 SerializeAndDeserialize(GetOperator("ADD", OperatorType::kAdd), op); in TEST_F()
173 SerializeAndDeserialize(GetOperator("ADD_N", OperatorType::kAddN), op); in TEST_F()
194 SerializeAndDeserialize(GetOperator("CAST", OperatorType::kCast), op); in TEST_F()
203 GetOperator("CONCATENATION", OperatorType::kConcatenation), op); in TEST_F()
211 GetOperator("DEPTH_TO_SPACE", OperatorType::kDepthToSpace), op); in TEST_F()
223 GetOperator("FAKE_QUANT", OperatorType::kFakeQuant), op); in TEST_F()
233 GetOperator("FULLY_CONNECTED", OperatorType::kFullyConnected), op); in TEST_F()
[all …]
|