Searched refs:GetTestDims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 132 nvinfer1::Dims GetTestDims(const std::vector<int>& d) { in GetTestDims() function
195 return TrtDimsEquals(GetTestDims(lhs), rhs); in TrtDimsEqualsArray()
203 << "expected: " << DebugString(GetTestDims(lhs)) << "\n" in ExpectTrtDimsEqualsArray()
335 : dims_(GetTestDims(dims)), dynamic_range_(0.0f) {} in FakeITensor()
435 store.GetTempWeights(nvinfer1::DataType::kFLOAT, GetTestDims({2, 5})); in TEST()
806 "my_input", nvinfer1::DataType::kFLOAT, GetTestDims({123}), 1)); in TEST_F()
838 GetTestDims({1}), 1)); in TEST_F()
840 GetTestDims({2, 3}), 1)); in TEST_F()
842 "weird_input:2:3:4", nvinfer1::DataType::kHALF, GetTestDims({5, 3}), 1)); in TEST_F()
889 "my_input", nvinfer1::DataType::kFLOAT, GetTestDims({1, 2}), 1)); in TEST_F()
[all …]
|