Searched refs:cached_shape (Results 1 – 1 of 1) sorted by relevance
165 const TensorShape& cached_shape) { in AreShapesCompatible() argument167 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()