Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_lru_cache.cc112 const int cached_batch_size = cached_input_shapes[0].dim_size(0); in GetEngineContext() local
113 if (min_matched_batch_size > cached_batch_size) { in GetEngineContext()
114 min_matched_batch_size = cached_batch_size; in GetEngineContext()