Home
last modified time | relevance | path

Searched defs:maximum_batch_size (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
H A Dsegment.h45 std::optional<int> maximum_batch_size = std::nullopt; member
H A Dsegment_test.cc111 void EnableImplicitBatchModeForStaticEngine(int maximum_batch_size = 1000) { in EnableImplicitBatchModeForStaticEngine()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph_test.cc144 int maximum_batch_size = 1000) { in RunConvertGraph()