Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc163 const std::vector<TensorShape>& cached_shapes) { in AreShapesCompatible() argument
178 if (!match_shape(actual_shapes[i], cached_shapes[i])) { in AreShapesCompatible()
Dutils.h176 const std::vector<TensorShape>& cached_shapes);