Home
last modified time | relevance | path

Searched refs:variable_axis_size_partitioner (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py80 partitioner = partitioned_variables.variable_axis_size_partitioner(
175 …partitioner_axis3_str = partitioned_variables.variable_axis_size_partitioner( # pylint: disable=l…
564 partitioner=partitioned_variables.variable_axis_size_partitioner(4))
585 partitioner=partitioned_variables.variable_axis_size_partitioner(4))
Dinit_ops_test.py1349 partitioner = partitioned_variables.variable_axis_size_partitioner(1)
/external/tensorflow/tensorflow/python/training/
Dlocalhost_cluster_performance_test.py101 partitioner=partitioned_variables.variable_axis_size_partitioner(
/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py72 def variable_axis_size_partitioner( function
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py220 return partitioned_variables.variable_axis_size_partitioner(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2496 name: "variable_axis_size_partitioner"