Searched refs:GetTestDims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 77 nvinfer1::Dims GetTestDims(const std::vector<int>& d) { in GetTestDims() function
155 return TrtDimsEquals(GetTestDims(lhs), rhs); in TrtDimsEqualsArray()
163 << "expected: " << DebugString(GetTestDims(lhs)) << "\n" in ExpectTrtDimsEqualsArray()
213 : dims_(GetTestDims(dims)), dynamic_range_(0.0f) {} in FakeITensor()
303 store.GetTempWeights(DT_FLOAT, GetTestDims({2, 5})); in TEST()
592 "my_input", nvinfer1::DataType::kFLOAT, GetTestDims({123}), 1)); in TEST_F()
622 GetTestDims({1}), 1)); in TEST_F()
624 GetTestDims({2, 3}), 1)); in TEST_F()
626 "weird_input:2:3:4", nvinfer1::DataType::kHALF, GetTestDims({5, 3}), 1)); in TEST_F()
672 "my_input", nvinfer1::DataType::kFLOAT, GetTestDims({1, 2}), 1)); in TEST_F()
[all …]
|