Home
last modified time | relevance | path

Searched refs:base_resolver (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dlogging_op_resolver_test.cc55 MutableOpResolver base_resolver; in TEST() local
60 base_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &conv_registration); in TEST()
66 base_resolver.AddBuiltin(BuiltinOperator_ADD, &add_registration); in TEST()
72 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
89 MutableOpResolver base_resolver; in TEST() local
94 base_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &conv_registration); in TEST()
100 base_resolver.AddBuiltin(BuiltinOperator_ADD, &add_registration); in TEST()
106 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
116 MutableOpResolver base_resolver; in TEST() local
121 base_resolver.AddBuiltin(BuiltinOperator_CONV_2D, &conv_registration); in TEST()
[all …]
Dlogging_op_resolver.cc29 const CustomOpsSet& custom_ops_to_replace, const OpResolver& base_resolver, in LoggingOpResolver() argument
36 base_resolver.FindOp(op_and_version.first, op_and_version.second); in LoggingOpResolver()
50 const TfLiteRegistration* base_registration = base_resolver.FindOp( in LoggingOpResolver()
Dlogging_op_resolver.h41 const OpResolver& base_resolver,