Searched refs:cached_input_shapes (Results 1 – 1 of 1) sorted by relevance
104 const std::vector<TensorShape>& cached_input_shapes = pair.first; in GetEngineContext() local106 if (input_shapes.size() != cached_input_shapes.size()) { in GetEngineContext()108 << ", cached size: " << cached_input_shapes.size() in GetEngineContext()111 if (AreShapesCompatible(input_shapes, cached_input_shapes)) { in GetEngineContext()112 const int cached_batch_size = cached_input_shapes[0].dim_size(0); in GetEngineContext()