Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.h260 std::vector<int32> allowed_batch_sizes_; variable
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_batch_tf_opkernels.cc252 std::vector<int32> allowed_batch_sizes_; member in tensorflow::tfd::__anon546964210111::BatchFunctionFallbackKernel
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc499 std::vector<int32> allowed_batch_sizes_; member in tensorflow::BatchFunctionKernel
605 std::vector<int32> allowed_batch_sizes_; member in tensorflow::BatchKernel