Searched defs:allowed_batch_sizes_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | batch_kernels.h | 91 std::vector<int32> allowed_batch_sizes_; variable |
| D | batch_kernels.cc | 612 std::vector<int32> allowed_batch_sizes_; member in tensorflow::BatchKernel |
| /external/tensorflow/tensorflow/core/kernels/batching_util/ | ||
| D | batch_resource_base.h | 263 std::vector<int32> allowed_batch_sizes_; variable |
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ | ||
| D | runtime_fallback_batch_tf_opkernels.cc | 252 std::vector<int32> allowed_batch_sizes_; member in tensorflow::tfd::__anon6e2be26b0111::BatchFunctionFallbackKernel |