Searched defs:ValidateAllowedBatchSizes (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ | ||
| H A D | batch_kernels.cc | 439 Status BatchFunctionKernel::ValidateAllowedBatchSizes() const { in ValidateAllowedBatchSizes() function in tensorflow::BatchFunctionKernel |
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ | ||
| H A D | runtime_fallback_batch_tf_opkernels.cc | 404 Status BatchFunctionFallbackKernel::ValidateAllowedBatchSizes() const { in ValidateAllowedBatchSizes() function in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel |