Home
last modified time | relevance | path

Searched defs:allowed_batch_sizes_ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.h91 std::vector<int32> allowed_batch_sizes_; variable
H A Dbatch_kernels.cc612 std::vector<int32> allowed_batch_sizes_; member in tensorflow::BatchKernel
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.h263 std::vector<int32> allowed_batch_sizes_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc252 std::vector<int32> allowed_batch_sizes_; member in tensorflow::tfd::__anonb95ad7eb0111::BatchFunctionFallbackKernel