Searched refs:OperatorKey (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export.h | 85 class OperatorKey {
87 OperatorKey() {} in OperatorKey() function
90 OperatorKey(
98 OperatorKey(::tflite::BuiltinOperator type, const std::string& custom_code, in OperatorKey() function
121 bool operator<(const OperatorKey& other) const {
134 bool operator==(const OperatorKey& other) const {
140 size_t operator()(const OperatorKey& key) const { in operator()
162 using OperatorsMap = std::unordered_map<OperatorKey, int, OperatorKey::Hash>;
|
| D | export_test.cc | 207 0, operators[details::OperatorKey(::tflite::BuiltinOperator_ADD, "", 1)]); in TEST_F()
208 EXPECT_EQ(1, operators[details::OperatorKey(::tflite::BuiltinOperator_CONV_2D, in TEST_F()
210 EXPECT_EQ(2, operators[details::OperatorKey(::tflite::BuiltinOperator_CUSTOM, in TEST_F()
213 3, operators[details::OperatorKey(::tflite::BuiltinOperator_SUB, "", 1)]); in TEST_F()
435 EXPECT_EQ(0, operators.at(details::OperatorKey( in TEST_F()
447 EXPECT_EQ(0, operators.at(details::OperatorKey( in TEST_F()
460 EXPECT_EQ(0, operators.at(details::OperatorKey( in TEST_F()
462 EXPECT_EQ(1, operators.at(details::OperatorKey( in TEST_F()
511 const auto key = details::OperatorKey(op_signature, ops_by_type, false); in TEST()
533 const auto key = details::OperatorKey(op_signature, ops_by_type, false); in TEST()
[all …]
|
| D | export.cc | 108 OperatorKey::OperatorKey( in OperatorKey() function in toco::tflite::details::OperatorKey
192 std::set<OperatorKey> keys; in LoadOperatorsMap()
195 keys.insert(OperatorKey(op_signature, ops_by_type, enable_select_tf_ops)); in LoadOperatorsMap()
308 const details::OperatorKey operator_key = details::OperatorKey( in ExportOperatorCodes()
357 const auto key = details::OperatorKey(op_signature, ops_by_type, in ExportOperators()
471 const details::OperatorKey& key = it.first; in Export()
|