Home
last modified time | relevance | path

Searched defs:AdaptiveBatchSchedulerOptions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.h104 struct AdaptiveBatchSchedulerOptions { struct
110 absl::optional<AdaptiveBatchSchedulerOptions> argument
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc265 struct AdaptiveBatchSchedulerOptions { struct in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel
266 int32 min_in_flight_batches_limit = kMinInflightBatches;
267 int32 initial_in_flight_batches_limit = kInitialInflightBatches;
268 int32 max_in_flight_batches_limit = kMaxInflightBatches;
269 int32 batches_to_average_over = kBatchesToAverageOver;