Log which InferenceCalculator impl is used at runtime.
PiperOrigin-RevId: 506343086
This commit is contained in:
parent
3ee377f671
commit
0919a6c0a3
|
@ -63,6 +63,10 @@ class InferenceCalculatorSelectorImpl
|
||||||
for (const auto& suffix : impls) {
|
for (const auto& suffix : impls) {
|
||||||
const auto impl = absl::StrCat("InferenceCalculator", suffix);
|
const auto impl = absl::StrCat("InferenceCalculator", suffix);
|
||||||
if (!mediapipe::CalculatorBaseRegistry::IsRegistered(impl)) continue;
|
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;
|
CalculatorGraphConfig::Node impl_node = subgraph_node;
|
||||||
impl_node.set_calculator(impl);
|
impl_node.set_calculator(impl);
|
||||||
return tool::MakeSingleNodeGraph(std::move(impl_node));
|
return tool::MakeSingleNodeGraph(std::move(impl_node));
|
||||||
|
|
Loading…
Reference in New Issue
Block a user