Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dcheckpoint_utils_test.py68 partitioner=partitioned_variables.min_max_variable_partitioner(
295 partitioner=partitioned_variables.min_max_variable_partitioner(
305 partitioner=partitioned_variables.min_max_variable_partitioner(
333 partitioner=partitioned_variables.min_max_variable_partitioner(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dpartitioned_variables.py156 def min_max_variable_partitioner(max_partitions=1, axis=0, function
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/
H A Dpartitioned_variables_test.py198 partitioner = partitioned_variables.min_max_variable_partitioner(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dsharded_variable.py162 return partitioned_variables.min_max_variable_partitioner(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dembedding_ops_test.py177 partitioner=partitioned_variables.min_max_variable_partitioner(
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.pbtxt1700 name: "min_max_variable_partitioner"