Searched refs:LoggingOpResolver (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| H A D | logging_op_resolver.cc | 31 LoggingOpResolver::LoggingOpResolver( in LoggingOpResolver() function in tflite::optimize::calibration::LoggingOpResolver
84 const TfLiteRegistration* LoggingOpResolver::FindOp(BuiltinOperator op, in FindOp()
95 KernelEvalFuncPtr LoggingOpResolver::GetWrappedKernelInvoke(BuiltinOperator op, in GetWrappedKernelInvoke()
100 const TfLiteRegistration* LoggingOpResolver::FindOp(const char* op, in FindOp()
111 KernelEvalFuncPtr LoggingOpResolver::GetWrappedKernelInvoke(const char* op, in GetWrappedKernelInvoke()
|
| H A D | logging_op_resolver_test.cc | 74 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
108 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
135 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
159 LoggingOpResolver resolver(BuiltinOpsSet(), ops_to_replace, base_resolver, in TEST()
181 LoggingOpResolver(BuiltinOpsSet(), ops_to_replace, base_resolver, in TEST()
195 LoggingOpResolver resolver(ops_to_replace, CustomOpsSet(), base_resolver, in TEST()
209 LoggingOpResolver resolver(BuiltinOpsSet(), ops_to_replace, base_resolver, in TEST()
|
| H A D | calibrator.cc | 65 std::unique_ptr<LoggingOpResolver> logging_op_resolver, in Calibrator()
98 std::unique_ptr<LoggingOpResolver> logging_op_resolver_;
167 std::unique_ptr<LoggingOpResolver> logging_op_resolver, in CreateCalibrator()
459 auto logging_op_resolver = std::make_unique<LoggingOpResolver>( in BuildLoggingInterpreter()
|
| H A D | logging_op_resolver.h | 32 class LoggingOpResolver : public OpResolver {
39 LoggingOpResolver(const BuiltinOpsSet& builtin_ops_to_replace,
|