Searched refs:ExpectTrtDimsEqualsArray (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 200 void ExpectTrtDimsEqualsArray(const std::vector<int>& lhs, in ExpectTrtDimsEqualsArray() function
295 ExpectTrtDimsEqualsArray(expected_dims, weights.shape_); in ValidateWeights()
489 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()
508 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()
522 ExpectTrtDimsEqualsArray({}, ptr->GetTrtDims()); in TEST()
613 ExpectTrtDimsEqualsArray({non_batch_dim}, output.GetTrtDims()); in TEST_F()
853 ExpectTrtDimsEqualsArray({1}, inputs[0].tensor()->getDimensions()); in TEST_F()
854 ExpectTrtDimsEqualsArray({2, 3}, inputs[2].tensor()->getDimensions()); in TEST_F()
855 ExpectTrtDimsEqualsArray({5, 3}, inputs[3].tensor()->getDimensions()); in TEST_F()
902 ExpectTrtDimsEqualsArray({2, 1}, output_tensor->getDimensions()); in TEST_F()
[all …]
|