Searched refs:TensorShapeTestHelper (Results 1 – 2 of 2) sorted by relevance
28 class TensorShapeTestHelper { class178 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INVALID); in TEST()179 TensorShapeTestHelper::set_data_type(&s, DT_INT32); in TEST()181 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INT32); in TEST()183 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_INT32); in TEST()184 TensorShapeTestHelper::set_data_type(&s, DT_UINT16_REF); in TEST()186 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_UINT16_REF); in TEST()188 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_UINT16_REF); in TEST()190 EXPECT_EQ(TensorShapeTestHelper::data_type(&s), DT_UINT16_REF); in TEST()193 EXPECT_EQ(TensorShapeTestHelper::data_type(&s2), DT_UINT16_REF); in TEST()[all …]
126 friend class TensorShapeTestHelper; variable