Home
last modified time | relevance | path

Searched defs:partitioner (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util_test.py61 partitioner=None): argument
100 def _create_linear_model(self, feature_cols, partitioner): argument
540 partitioner = lambda shape, dtype: [1] * len(shape) function
569 partitioner = lambda shape, dtype: [1] * len(shape) function
603 partitioner = lambda shape, dtype: [1] * len(shape) function
639 partitioner = lambda shape, dtype: [1] * len(shape) function
684 partitioner = lambda shape, dtype: [1] * len(shape) function
727 partitioner = lambda shape, dtype: [1] * len(shape) function
798 partitioner = lambda shape, dtype: [1] * len(shape) function
Dsaver_test.py1010 def _save(partitioner=None): argument
1040 def _restore(partitioner=None): argument
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py293 partitioner=None, argument
445 partitioner=None, argument
571 partitioner, argument
1049 partitioner=None, argument
1107 def partitioner(self): member in VariableScope
1148 def set_partitioner(self, partitioner): argument
1184 partitioner=None, argument
1255 partitioner=None, argument
1479 partitioner=None, argument
1625 partitioner=None, argument
[all …]
Dpartitioned_variables.py298 partitioner = lambda **unused_kwargs: slicing function
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_optimizer.py78 partitioner=None): argument
115 def partitioner(self): member in SDCAOptimizer
Dsdca_estimator.py220 partitioner=None): argument
347 partitioner=None): argument
478 partitioner=None): argument
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables.py212 partitioner=None, argument
290 partitioner=None, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dcheckpoint_ops_test.py303 def _test_loading_variable_with_max_rows(self, np_value, partitioner, argument
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dlayers.py47 partitioner=None, argument
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py310 def add_variable_wrapped(name, shape, initializer, index, partitioner): argument
/external/tensorflow/tensorflow/python/layers/
Dbase.py310 partitioner=None, argument
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py75 partitioner=None): # pylint: disable=unused-argument argument
Dbase_layer.py268 partitioner=None, argument
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py1716 partitioner=None, argument