Home
last modified time | relevance | path

Searched refs:max_partitions (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py200 def _testMinMaxVariablePartitioner(self, max_partitions, axis, min_slice_size, argument
204 max_partitions=max_partitions, axis=axis, min_slice_size=min_slice_size)
217 max_partitions=100,
228 max_partitions=100,
238 max_partitions=16,
246 max_partitions=1,
257 max_partitions=100,
265 max_partitions=100,
275 max_partitions=100,
283 max_partitions=100,
[all …]
Dembedding_ops_test.py183 max_partitions=num_shards, min_slice_size=1),
/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py158 def min_max_variable_partitioner(max_partitions=1, axis=0, argument
215 max_partitions,
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils_test.py72 max_partitions=5, axis=0, min_slice_size=8 << 10))
279 max_partitions=5, axis=0, min_slice_size=8 << 10))
289 max_partitions=5, axis=0, min_slice_size=16 << 10))
317 max_partitions=5, axis=0, min_slice_size=8 << 10))
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.h319 int max_partitions,
Dnnapi_delegate.cc5004 int max_partitions, in LimitDelegatedPartitions() argument
5008 if (max_partitions <= 0 || num_partitions <= max_partitions) { in LimitDelegatedPartitions()
5020 if (number_delegated_partitions > max_partitions) { in LimitDelegatedPartitions()
5031 for (int i = 0; i < max_partitions; i++) { in LimitDelegatedPartitions()
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py161 max_partitions=self._max_shards,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1693 …argspec: "args=[\'max_partitions\', \'axis\', \'min_slice_size\', \'bytes_per_string_element\'], v…