Home
last modified time | relevance | path

Searched refs:TensorShapeTestHelper (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc27 class TensorShapeTestHelper { class
177 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INVALID); in TEST()
178 TensorShapeTestHelper::set_data_type(&s, DT_INT32); in TEST()
180 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INT32); in TEST()
182 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INT32); in TEST()
183 TensorShapeTestHelper::set_data_type(&s, DT_UINT16_REF); in TEST()
185 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_UINT16_REF); in TEST()
187 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_UINT16_REF); in TEST()
189 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_UINT16_REF); in TEST()
192 EXPECT_EQ(TensorShapeTestHelper::data_type(&s2), DT_UINT16_REF); in TEST()
[all …]
Dtensor_shape.h126 friend class TensorShapeTestHelper; variable