Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning_utils.py214 def determine_partitioned_axis(partitioned_variable): function
245 axis = 0 if len(partitioned_var) == 1 else determine_partitioned_axis(