Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc557 inline bool HasStaticShape(const nvinfer1::Dims& dims) { in HasStaticShape() function
2142 if (!HasStaticShape(input_dims) || in ConvertReshape()
2149 } else if (HasStaticShape(input_dims)) { in ConvertReshape()
2150 if (!HasStaticShape(reshape_dims) || in ConvertReshape()
Dconvert_nodes_test.cc1185 bool HasStaticShape(const nvinfer1::Dims& dims) const { in HasStaticShape() function in tensorflow::tensorrt::convert::OpConverterTest
1206 if (HasStaticShape(trt_dims)) { in AddTestTensor()