Searched refs:TensorShapeTestHelper (Results 1 – 2 of 2) sorted by relevance
27 class TensorShapeTestHelper { class177 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 …]
126 friend class TensorShapeTestHelper; variable