| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.h | 119 std::vector<int32> allowed_batch_sizes) in BatchResourceBase() argument
123 allowed_batch_sizes_(std::move(allowed_batch_sizes)) { in BatchResourceBase()
130 std::vector<int32> allowed_batch_sizes) in BatchResourceBase() argument
134 allowed_batch_sizes_(std::move(allowed_batch_sizes)) {} in BatchResourceBase()
138 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes,
144 const std::vector<int32>& allowed_batch_sizes);
|
| D | batch_resource_base.cc | 111 void RecordBatchParamAllowedBatchSizes(const string& allowed_batch_sizes, in RecordBatchParamAllowedBatchSizes() argument
118 cell->GetCell(model_name, op_name)->Set(allowed_batch_sizes); in RecordBatchParamAllowedBatchSizes()
229 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes, in GetBatcherQueueOptions() argument
246 if (allowed_batch_sizes.empty()) { in GetBatcherQueueOptions()
250 *allowed_batch_sizes.rbegin(); in GetBatcherQueueOptions()
261 const std::vector<int32>& allowed_batch_sizes) { in GetAdaptiveBatcherQueueOptions() argument
267 if (allowed_batch_sizes.empty()) { in GetAdaptiveBatcherQueueOptions()
270 batcher_queue_options.max_batch_size = *allowed_batch_sizes.rbegin(); in GetAdaptiveBatcherQueueOptions()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 35 allowed_batch_sizes=None, argument
111 allowed_batch_sizes=allowed_batch_sizes,
|
| D | batch_ops_test.py | 106 allowed_batch_sizes=[5, 10],
195 allowed_batch_sizes=[3, 10],
396 allowed_batch_sizes=[3, 10],
458 allowed_batch_sizes=[1, 2],
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 88 const std::vector<int32>& allowed_batch_sizes, in Create() argument
102 allowed_batch_sizes, in Create()
104 allowed_batch_sizes)); in Create()
111 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes, in Create() argument
122 true /* enable large batch split */, allowed_batch_sizes), in Create()
123 allowed_batch_sizes)); in Create()
133 std::vector<int32> allowed_batch_sizes) in BatchResource() argument
137 std::move(allowed_batch_sizes)), in BatchResource()
145 std::vector<int32> allowed_batch_sizes) in BatchResource() argument
149 std::move(allowed_batch_sizes)), in BatchResource()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Batch.pbtxt | 32 name: "allowed_batch_sizes"
109 name: "allowed_batch_sizes"
|
| D | BatchFunction.pbtxt | 39 name: "allowed_batch_sizes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Batch.pbtxt | 32 name: "allowed_batch_sizes"
109 name: "allowed_batch_sizes"
|
| D | BatchFunction.pbtxt | 39 name: "allowed_batch_sizes"
123 name: "allowed_batch_sizes"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 49 name: "allowed_batch_sizes"
94 `allowed_batch_sizes`) will be splitted into multiple batches with batch size.
118 allowed_batch_sizes=[3, 10],
|
| D | api_def_Batch.pbtxt | 28 allowed_batch_sizes: Optional list of allowed batch sizes. If left empty, does
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 837 …m_batch_threads\', \'max_batch_size\', \'batch_timeout_micros\', \'allowed_batch_sizes\', \'max_en…
|
| D | tensorflow.raw_ops.pbtxt | 317 …out_micros\', \'grad_timeout_micros\', \'max_enqueued_batches\', \'allowed_batch_sizes\', \'contai…
349 …', \'batch_timeout_micros\', \'Tout\', \'max_enqueued_batches\', \'allowed_batch_sizes\', \'contai…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1333 allowed_batch_sizes=[3, 10],
1362 DefaultValuedAttr<I64ArrayAttr, "{}">:$allowed_batch_sizes,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1737 …m_batch_threads\', \'max_batch_size\', \'batch_timeout_micros\', \'allowed_batch_sizes\', \'max_en…
|
| D | tensorflow.raw_ops.pbtxt | 317 …out_micros\', \'grad_timeout_micros\', \'max_enqueued_batches\', \'allowed_batch_sizes\', \'contai…
349 …', \'batch_timeout_micros\', \'Tout\', \'max_enqueued_batches\', \'allowed_batch_sizes\', \'contai…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3329 name: "allowed_batch_sizes"
3570 name: "allowed_batch_sizes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 8799 name: "allowed_batch_sizes"
8876 name: "allowed_batch_sizes"
9133 name: "allowed_batch_sizes"
|