Searched refs:custom_op_name (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver_test.cc | 150 std::string custom_op_name = "custom"; in TEST() local
151 base_resolver.AddCustom(custom_op_name.c_str(), &custom_registration); in TEST()
154 {custom_op_name, /*version*/ 1}, in TEST()
160 auto reg = resolver.FindOp(custom_op_name.c_str(), 1); in TEST()
163 EXPECT_EQ(reg->custom_name, custom_op_name.c_str()); in TEST()
172 std::string custom_op_name = "unresolved_custom_op"; in TEST() local
175 {custom_op_name, /*version*/ 1}, in TEST()
201 std::string custom_op_name = "FlexAdd"; in TEST() local
204 {custom_op_name, /*version*/ 1}, in TEST()
210 auto reg = resolver.FindOp(custom_op_name.c_str(), 1); in TEST()
|
| /external/tensorflow/tensorflow/lite/ |
| D | util.cc | 148 TfLiteRegistration CreateUnresolvedCustomOp(const char* custom_op_name) { in CreateUnresolvedCustomOp() argument
155 custom_op_name, in CreateUnresolvedCustomOp()
|
| D | util.h | 87 TfLiteRegistration CreateUnresolvedCustomOp(const char* custom_op_name);
|