Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1067 class OpConverterTest : public ::testing::Test { class
1069 OpConverterTest() : scope_(Scope::NewRootScope()) { in OpConverterTest() function in tensorflow::tensorrt::convert::OpConverterTest
1074 ~OpConverterTest() override { QCHECK_EQ(0, cudaStreamDestroy(stream_)); } in ~OpConverterTest()
1329 void TestConvertConst(OpConverterTest* test) { in TestConvertConst()
1406 TEST_F(OpConverterTest, ConvertConst) { in TEST_F() argument
1427 TEST_F(OpConverterTest, ConvertTranspose) { in TEST_F() argument
1489 TEST_F(OpConverterTest, ConvertReshape) { in TEST_F() argument
1576 TEST_F(OpConverterTest, ConvertMatMul) { in TEST_F() argument
1644 void TestConvertBiasAdd(OpConverterTest* test) { in TestConvertBiasAdd()
1720 TEST_F(OpConverterTest, ConvertBiasAdd) { in TEST_F() argument
[all …]
Dconvert_nodes.h416 friend class OpConverterTest; variable
582 friend class OpConverterTest; variable