This fixes a problem where the input tensor dimensions were printed instead of the subgraph input dimensions.
PiperOrigin-RevId: 457438874
diff --git a/tensorflow/lite/experimental/acceleration/mini_benchmark/call.cc b/tensorflow/lite/experimental/acceleration/mini_benchmark/call.cc
index 7729d62..9794753 100644
--- a/tensorflow/lite/experimental/acceleration/mini_benchmark/call.cc
+++ b/tensorflow/lite/experimental/acceleration/mini_benchmark/call.cc
@@ -88,7 +88,7 @@
"node and the subgraph to invoke (input tensor %s[ %s], subgraph "
"tensor %s[ %s])",
__FILE__, __LINE__, node_input->name, node_input_dims.str().c_str(),
- subgraph_input->name, node_input_dims.str().c_str());
+ subgraph_input->name, subgraph_input_dims.str().c_str());
return kTfLiteError;
}
// Batch size of subgraph's input should be 1.