| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 58 partitioner=None): argument
97 def _create_linear_model(self, feature_cols, partitioner): argument
564 partitioner = lambda shape, dtype: [1] * len(shape) function
593 partitioner = lambda shape, dtype: [1] * len(shape) function
627 partitioner = lambda shape, dtype: [1] * len(shape) function
663 partitioner = lambda shape, dtype: [1] * len(shape) function
708 partitioner = lambda shape, dtype: [1] * len(shape) function
751 partitioner = lambda shape, dtype: [1] * len(shape) function
822 partitioner = lambda shape, dtype: [1] * len(shape) function
|
| D | saver_test.py | 1075 def _save(partitioner=None): argument
1106 def _restore(partitioner=None): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 326 partitioner=None, argument
475 partitioner=None, argument
601 partitioner, argument
1148 partitioner=None, argument
1206 def partitioner(self): member in VariableScope
1247 def set_partitioner(self, partitioner): argument
1283 partitioner=None, argument
1354 partitioner=None, argument
1609 partitioner=None, argument
1786 partitioner=None, argument
[all …]
|
| D | partitioned_variables.py | 333 partitioner = lambda **unused_kwargs: slicing function
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | stateful_rng_spmd_partitioner.h | 35 spmd::SpmdPartitionerOptions options, spmd::SpmdPartitioner* partitioner) in StatefulRngSpmdPartitioningVisitor()
|
| D | spmd_partitioner.h | 331 SpmdPartitioner* partitioner; member
|
| D | spmd_partitioner.cc | 1992 SpmdPartitionerOptions options, SpmdPartitioner* partitioner) in SpmdPartitioningVisitor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_spmd_partitioner.h | 34 spmd::SpmdPartitionerOptions options, spmd::SpmdPartitioner* partitioner) in GpuSpmdPartitioningVisitor()
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | variable_scope_shim.py | 164 partitioner=None, argument
307 partitioner=None, argument
|
| D | base.py | 334 partitioner=None, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/io_ops/ |
| D | checkpoint_ops_test.py | 326 def _test_loading_variable_with_max_rows(self, np_value, partitioner, argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_preprocessing_layer.py | 372 partitioner=None, argument
|
| D | base_layer_utils.py | 63 partitioner=None): # pylint: disable=unused-argument argument
|
| D | base_layer_v1.py | 325 partitioner=None, argument
|