Searched defs:max_batch_size_ (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ | ||
| H A D | batch_kernels.h | 88 int32 max_batch_size_; variable |
| H A D | batch_kernels.cc | 609 int32 max_batch_size_; member in tensorflow::BatchKernel |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ | ||
| H A D | union_find.h | 114 std::optional<int> max_batch_size_; variable |
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ | ||
| H A D | runtime_fallback_batch_tf_opkernels.cc | 249 int32 max_batch_size_; member in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel |