| H A D | convert_nodes_test.cc | 1383 void AddTestTensor( in AddTestTensor() function in tensorflow::tensorrt::convert::OpConverterTest
1811 void AddTestTensor(const string& name, const std::vector<int32>& dims, in AddTestTensor() function in tensorflow::tensorrt::convert::ParameterizedOpConverterTestBase
1869 void AddTestTensor(const string& name, const std::vector<int32>& dims, in AddTestTensor() function in tensorflow::tensorrt::convert::ParameterizedOpConverterTestBase
1872 AddTestTensor<T>(name, dims, tf_type_, values, partial_input_shape_dims);
2005 AddTestTensor("input", p.input_dims, input_tf_type, input_values); in RunTests()
2036 AddTestTensor("input", dims); in runExpectedToFailTest()
2099 AddTestTensor("input1", {2, 1, 2}, data[0]); in RunTests()
2104 AddTestTensor("input2", {2, 2, 1}, data[2]); in RunTests()
2652 AddTestTensor(node_input[i].name, node_input[i].dims, tf_type_, in TEST_P()
2714 AddTestTensor("input", p.input_dims, {1, 2, 3, 4, 5, 6}, in TEST_P()
[all …]
|