Searched refs:ExpectTrtDimsEqualsArray (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 160 void ExpectTrtDimsEqualsArray(const std::vector<int>& lhs, in ExpectTrtDimsEqualsArray() function
197 ExpectTrtDimsEqualsArray(expected_dims, weights.shape_); in ValidateWeights()
357 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()
376 ExpectTrtDimsEqualsArray({1}, ptr->GetTrtDims()); in TEST()
390 ExpectTrtDimsEqualsArray({}, ptr->GetTrtDims()); in TEST()
475 ExpectTrtDimsEqualsArray({non_batch_dim}, output.GetTrtDims()); in TEST_F()
637 ExpectTrtDimsEqualsArray({1}, inputs[0].tensor()->getDimensions()); in TEST_F()
638 ExpectTrtDimsEqualsArray({2, 3}, inputs[2].tensor()->getDimensions()); in TEST_F()
639 ExpectTrtDimsEqualsArray({5, 3}, inputs[3].tensor()->getDimensions()); in TEST_F()
685 ExpectTrtDimsEqualsArray({2, 1}, output_tensor->getDimensions()); in TEST_F()
[all …]
|