cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 521327449
This commit is contained in:
MediaPipe Team 2023-04-02 17:41:50 -07:00 committed by Copybara-Service
parent 1fa9b2c985
commit 696bedcaa1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mediapipe/tasks/cc/vision/utils/image_tensor_specs.cc
View File

@ -191,8 +191,9 @@ absl::StatusOr<ImageTensorSpecs> 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) {