Searched defs:enable_large_batch_splitting_ (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | batch_kernels.h | 95 bool enable_large_batch_splitting_; variable |
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ | ||
H A D | runtime_fallback_batch_tf_opkernels.cc | 253 bool enable_large_batch_splitting_; member in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel |