Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc165 const TensorShape& cached_shape) { in AreShapesCompatible() argument
167 if (actual_shape.dims() != cached_shape.dims()) return false; in AreShapesCompatible()
170 if (actual_shape.dim_size(0) > cached_shape.dim_size(0)) return false; in AreShapesCompatible()
173 if (actual_shape.dim_size(i) != cached_shape.dim_size(i)) return false; in AreShapesCompatible()