Home
last modified time | relevance | path

Searched refs:TensorUtilTest (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/runtime/core/exec_aten/util/test/
H A Dtensor_util_test.cpp23 class TensorUtilTest : public ::testing::Test { class
40 TEST_F(TensorUtilTest, IdentityChecks) { in TEST_F() argument
56 TEST_F(TensorUtilTest, SameShapesDifferentDtypes) { in TEST_F() argument
81 TEST_F(TensorUtilTest, DifferentShapesSameDtypes) { in TEST_F() argument
113 TEST_F(TensorUtilTest, ZeroDimensionalTensor) { in TEST_F() argument
129 TEST_F(TensorUtilTest, EmptyTensor) { in TEST_F() argument
146 TEST_F(TensorUtilTest, GetLeadingDimsSmokeTest) { in TEST_F() argument
160 TEST_F(TensorUtilTest, GetLeadingDimsInputOutOfBoundDies) { in TEST_F() argument
170 TEST_F(TensorUtilTest, GetTrailingDimsSmokeTest) { in TEST_F() argument
184 TEST_F(TensorUtilTest, GetTrailingDimsInputOutOfBoundDies) { in TEST_F() argument
[all …]
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/test/
H A Dtensor_util_test.cpp224 TEST(TensorUtilTest, DifferentDtypesAreNotCloseOrEqual) { in TEST() argument
235 TEST(TensorUtilTest, DifferentSizesAreNotCloseOrEqual) { in TEST() argument
245 TEST(TensorUtilTest, DifferentLayoutsDies) { in TEST() argument
264 TEST(TensorUtilTest, IntTensorIsCloseAndEqualToItself) { in TEST() argument
272 TEST(TensorUtilTest, IdenticalIntTensorsAreCloseAndEqual) { in TEST() argument
282 TEST(TensorUtilTest, NonIdenticalIntTensorsAreNotCloseOrEqual) { in TEST() argument
292 TEST(TensorUtilTest, EmptyTensorsAreCloseAndEqual) { in TEST() argument
308 TEST(TensorUtilTest, FloatTensorIsCloseAndEqualToItself) { in TEST() argument
316 TEST(TensorUtilTest, IdenticalFloatTensorsAreCloseAndEqual) { in TEST() argument
326 TEST(TensorUtilTest, NearlyIdenticalFloatTensorsAreCloseButNotEqual) { in TEST() argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dtensor_util_test.cc41 TEST(TensorUtilTest, MoveHostBufferToTfTensorOk) { in TEST() argument
60 TEST(TensorUtilTest, CopyShtToTfTensorOk) { in TEST() argument
71 TEST(TensorUtilTest, GetTfShapeOk) { in TEST() argument
79 TEST(TensorUtilTest, GetTensorMetadataOk) { in TEST() argument
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtensor_util_test.py41 class TensorUtilTest(test.TestCase, parameterized.TestCase): class