From 0919a6c0a363115f7fad783e8eab09790e6f61a7 Mon Sep 17 00:00:00 2001 From: Juhyun Lee Date: Wed, 1 Feb 2023 09:43:28 -0800 Subject: [PATCH] Log which InferenceCalculator impl is used at runtime. PiperOrigin-RevId: 506343086 --- mediapipe/calculators/tensor/inference_calculator.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mediapipe/calculators/tensor/inference_calculator.cc b/mediapipe/calculators/tensor/inference_calculator.cc index 4ccdc07e1..2a6936eba 100644 --- a/mediapipe/calculators/tensor/inference_calculator.cc +++ b/mediapipe/calculators/tensor/inference_calculator.cc @@ -63,6 +63,10 @@ class InferenceCalculatorSelectorImpl for (const auto& suffix : impls) { const auto impl = absl::StrCat("InferenceCalculator", suffix); if (!mediapipe::CalculatorBaseRegistry::IsRegistered(impl)) continue; + VLOG(1) << "Using " << suffix << " for InferenceCalculator with " + << (options.has_model_path() + ? "model " + options.model_path() + : "output_stream " + subgraph_node.output_stream(0)); CalculatorGraphConfig::Node impl_node = subgraph_node; impl_node.set_calculator(impl); return tool::MakeSingleNodeGraph(std::move(impl_node));