Searched refs:OpConverterTest (Results 1 – 2 of 2) sorted by relevance
1067 class OpConverterTest : public ::testing::Test { class1069 OpConverterTest() : scope_(Scope::NewRootScope()) { in OpConverterTest() function in tensorflow::tensorrt::convert::OpConverterTest1074 ~OpConverterTest() override { QCHECK_EQ(0, cudaStreamDestroy(stream_)); } in ~OpConverterTest()1329 void TestConvertConst(OpConverterTest* test) { in TestConvertConst()1406 TEST_F(OpConverterTest, ConvertConst) { in TEST_F() argument1427 TEST_F(OpConverterTest, ConvertTranspose) { in TEST_F() argument1489 TEST_F(OpConverterTest, ConvertReshape) { in TEST_F() argument1576 TEST_F(OpConverterTest, ConvertMatMul) { in TEST_F() argument1644 void TestConvertBiasAdd(OpConverterTest* test) { in TestConvertBiasAdd()1720 TEST_F(OpConverterTest, ConvertBiasAdd) { in TEST_F() argument[all …]
416 friend class OpConverterTest; variable582 friend class OpConverterTest; variable