Home
last modified time | relevance | path

Searched refs:GetWrappedKernelInvoke (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A Dlogging_op_resolver.h47 KernelEvalFuncPtr GetWrappedKernelInvoke(BuiltinOperator op,
51 KernelEvalFuncPtr GetWrappedKernelInvoke(const char* op, int version) const;
H A Dlogging_op_resolver.cc95 KernelEvalFuncPtr LoggingOpResolver::GetWrappedKernelInvoke(BuiltinOperator op, in GetWrappedKernelInvoke() function in tflite::optimize::calibration::LoggingOpResolver
111 KernelEvalFuncPtr LoggingOpResolver::GetWrappedKernelInvoke(const char* op, in GetWrappedKernelInvoke() function in tflite::optimize::calibration::LoggingOpResolver
H A Dlogging_op_resolver_test.cc111 resolver.GetWrappedKernelInvoke(BuiltinOperator_CONV_2D, 1); in TEST()
113 kernel_invoke = resolver.GetWrappedKernelInvoke(BuiltinOperator_ADD, 1); in TEST()
H A Dcalibrator.cc107 return logging_op_resolver_->GetWrappedKernelInvoke(op_info.name.c_str(), in GetKernelInvoke()
110 return logging_op_resolver_->GetWrappedKernelInvoke(op_info.builtin_op_code, in GetKernelInvoke()