talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 477585110
This commit is contained in:
MediaPipe Team
2022-09-28 16:47:33 -07:00
committed by Copybara-Service
parent 8c8a9cda5a
commit dcc5587483
4 changed files with 180 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/formats/tensor_test.cc
+6
View File
@@ -1,5 +1,8 @@
#include "mediapipe/framework/formats/tensor.h"
#include <cstring>
#include <string>
#include "mediapipe/framework/port/gmock.h"
#include "mediapipe/framework/port/gtest.h"
#if !MEDIAPIPE_DISABLE_GPU
@@ -23,6 +26,9 @@ TEST(General, TestDataTypes) {
Tensor t2(Tensor::ElementType::kFloat16, Tensor::Shape{4, 3, 2, 3});
EXPECT_EQ(t2.bytes(), t2.shape().num_elements() * 2);
Tensor t_char(Tensor::ElementType::kChar, Tensor::Shape{4});
EXPECT_EQ(t_char.bytes(), t_char.shape().num_elements() * sizeof(char));
}
TEST(Cpu, TestMemoryAllocation) {