Home
last modified time | relevance | path

Searched refs:DimsEqual (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles_test.cc58 bool DimsEqual(const nvinfer1::Dims& a, const nvinfer1::Dims& b) { in DimsEqual() function
212 EXPECT_TRUE(DimsEqual(dimvec[j], opt)); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc551 inline bool DimsEqual(const nvinfer1::Dims& dim_l, in DimsEqual() function
1734 if (DimsEqual(input_dims, dims)) { in PrepareTensorForShape()