Searched refs:actual_input_shapes (Results 1 – 1 of 1) sorted by relevance
98 const std::vector<TensorShape>& actual_input_shapes,334 const std::vector<TensorShape>& actual_input_shapes, in GetCompatibleCachedEngine() argument336 const int batch_size = actual_input_shapes[0].dim_size(0); in GetCompatibleCachedEngine()340 *engine_input_shapes = actual_input_shapes; in GetCompatibleCachedEngine()