Home
last modified time | relevance | path

Searched refs:fixed_size_partitioner (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dcheckpoint_ops_test.py156 partitioner=partitioned_variables.fixed_size_partitioner(2))
189 partitioner=partitioned_variables.fixed_size_partitioner(2))
226 partitioner=partitioned_variables.fixed_size_partitioner(2))
261 partitioner=partitioned_variables.fixed_size_partitioner(2))
294 partitioner=partitioned_variables.fixed_size_partitioner(2))
340 partitioner=partitioned_variables.fixed_size_partitioner(2))
378 partitioner=partitioned_variables.fixed_size_partitioner(2))
Dsaver_large_partitioned_variable_test.py51 partitioner=partitioned_variables.fixed_size_partitioner(4),
Dslot_creator_test.py147 partitioner=partitioned_variables.fixed_size_partitioner(2))
169 partitioner=partitioned_variables.fixed_size_partitioner(2))
Dsaver_test.py172 partitioner = partitioned_variables.fixed_size_partitioner(num_shards=2)
439 partitioner=partitioned_variables.fixed_size_partitioner(
444 partitioner=partitioned_variables.fixed_size_partitioner(
1076 partitioner=partitioned_variables.fixed_size_partitioner(
1083 partitioner=partitioned_variables.fixed_size_partitioner(
1089 partitioner=partitioned_variables.fixed_size_partitioner(
1096 partitioner=partitioned_variables.fixed_size_partitioner(
/external/tensorflow/tensorflow/python/kernel_tests/
Dcheckpoint_ops_test.py419 partitioner=partitioned_variables.fixed_size_partitioner(3),
429 partitioner=partitioned_variables.fixed_size_partitioner(3),
442 partitioner=partitioned_variables.fixed_size_partitioner(3),
450 partitioner=partitioned_variables.fixed_size_partitioner(3),
Dpartitioned_variables_test.py44 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
55 partitioner = partitioned_variables.fixed_size_partitioner(4, axis=0)
63 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
610 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
Dembedding_ops_test.py808 partitioner=partitioned_variables.fixed_size_partitioner(num_shards),
/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py222 def fixed_size_partitioner(num_shards, axis=0): function
Dnn_test.py626 partitioner=partitioned_variables.fixed_size_partitioner(num_shards),
630 partitioner=partitioned_variables.fixed_size_partitioner(num_shards),
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py316 partitioned_variables.fixed_size_partitioner(self._num_unit_shards)
387 partitioned_variables.fixed_size_partitioner(self._num_proj_shards)
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dsaved_model_test.py603 self.cluster_resolver, tf1.fixed_size_partitioner(2))
629 self.cluster_resolver, tf1.fixed_size_partitioner(2))
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph_test.py288 partitioner = partitioned_variables.fixed_size_partitioner(1)
/external/tensorflow/tensorflow/python/keras/feature_column/
Ddense_features_test.py139 partitioner=partitioned_variables.fixed_size_partitioner(2))
774 partitioner = partitioned_variables.fixed_size_partitioner(2, axis=0)
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py977 partitioned_variables.fixed_size_partitioner(self._num_unit_shards)
1002 partitioned_variables.fixed_size_partitioner(self._num_proj_shards)
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dbase_test.py173 partitioner = partitioned_variables.fixed_size_partitioner(3)
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2_test.py627 partitioner=partitioned_variables.fixed_size_partitioner(2),
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_test.py226 partitioner = partitioned_variables.fixed_size_partitioner(num_shards)
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py991 partitioner=partitioned_variables.fixed_size_partitioner(3, axis=0),
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py1648 partitioner=partitioned_variables.fixed_size_partitioner(2, axis=0)):
2315 partitioner=partitioned_variables.fixed_size_partitioner(2, axis=0)):
2899 partitioner=partitioned_variables.fixed_size_partitioner(3, axis=0)):
4976 partitioner = partitioned_variables.fixed_size_partitioner(2, axis=0)
5844 partitioner = partitioned_variables.fixed_size_partitioner(2, axis=0)
Dfeature_column_v2_test.py1787 partitioner=partitioned_variables.fixed_size_partitioner(2, axis=0)):
2602 partitioner=partitioned_variables.fixed_size_partitioner(3, axis=0)):
5771 partitioner = partitioned_variables.fixed_size_partitioner(2, axis=0)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py2871 partitioner=partitioned_variables.fixed_size_partitioner(num_slices),
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1316 name: "fixed_size_partitioner"