diff --git a/mediapipe/tasks/cc/vision/utils/image_tensor_specs.cc b/mediapipe/tasks/cc/vision/utils/image_tensor_specs.cc index 4d3d2cb9..3f0425a6 100644 --- a/mediapipe/tasks/cc/vision/utils/image_tensor_specs.cc +++ b/mediapipe/tasks/cc/vision/utils/image_tensor_specs.cc @@ -191,8 +191,9 @@ absl::StatusOr BuildInputImageTensorSpecs( MediaPipeTasksStatus::kInvalidInputTensorDimensionsError); } - size_t byte_depth = - tensor_type == tflite::TensorType_FLOAT32 ? sizeof(float) : sizeof(uint8); + size_t byte_depth = tensor_type == tflite::TensorType_FLOAT32 + ? sizeof(float) + : sizeof(uint8_t); int bytes_size = byte_depth * batch * height * width * depth; // Sanity checks. if (tensor_type == tflite::TensorType_FLOAT32) {