Home
last modified time | relevance | path

Searched refs:_partitioner (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py111 def _partitioner(shape, dtype): function
154 return _partitioner
181 def _partitioner(shape, dtype): function
218 return _partitioner
233 def _partitioner(shape, **unused_args): function
237 return _partitioner
Dvariable_scope.py1154 self._partitioner = partitioner
1201 return self._partitioner
1243 self._partitioner = partitioner
1290 partitioner = self._partitioner
1364 partitioner = self._partitioner
1915 self._partitioner = partitioner
1949 if self._partitioner is not None:
1950 variable_scope_object.set_partitioner(self._partitioner)
2008 if self._partitioner is not None:
2009 variable_scope_object.set_partitioner(self._partitioner)
[all …]
/external/tensorflow/tensorflow/python/keras/feature_column/
Dbase_feature_layer.py64 self._partitioner = partitioner
76 self.name, partitioner=self._partitioner):
128 self._partitioner)
/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util_test.py872 def _partitioner(shape, dtype): # pylint:disable=unused-argument function
881 cols_to_vars = self._create_linear_model(all_linear_cols, _partitioner)
998 def _partitioner(shape, dtype): # pylint:disable=unused-argument function
1007 cols_to_vars = self._create_linear_model(all_linear_cols, _partitioner)
1058 def _partitioner(shape, dtype): # pylint:disable=unused-argument function
1075 with variable_scope.variable_scope("", partitioner=_partitioner):
1128 def _partitioner(shape, dtype): # pylint:disable=unused-argument function
1145 with variable_scope.variable_scope("", partitioner=_partitioner):