Searched refs:allowed_batch_sizes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/batching/python/ops/ |
| D | batch_ops.py | 61 allowed_batch_sizes=None, argument
124 allowed_batch_sizes=allowed_batch_sizes,
140 allowed_batch_sizes=None, argument
182 allowed_batch_sizes=allowed_batch_sizes,
|
| D | batch_ops_test.py | 91 allowed_batch_sizes=[5, 10],
174 allowed_batch_sizes=[3, 10],
307 allowed_batch_sizes=[3, 10],
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 49 name: "allowed_batch_sizes"
110 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/core/kernels/ |
| D | batch_kernels.cc | 212 const std::vector<int32>& allowed_batch_sizes, in Create() argument
228 new_resource->allowed_batch_sizes_ = allowed_batch_sizes; in Create()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 2981 name: "allowed_batch_sizes"
3215 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"
|
| D | ops_history.v1.pbtxt | 8799 name: "allowed_batch_sizes"
8876 name: "allowed_batch_sizes"
9133 name: "allowed_batch_sizes"
|