Searched refs:partitioned_var (Results 1 – 4 of 4) sorted by relevance
233 def partitioned_variable_assign(partitioned_var, new_value): argument245 axis = 0 if len(partitioned_var) == 1 else determine_partitioned_axis(246 partitioned_var)249 [partition.get_shape()[axis] for partition in partitioned_var])255 for partition in partitioned_var:259 *op_list, name=partitioned_var.name + '_group_assign')
48 partitioned_var = list(variable_scope.get_variable(55 save = saver.Saver(partitioned_var)
303 partitioned_var = variable_scope._get_partitioned_variable(311 return list(partitioned_var)
791 partitioned_var = variables.PartitionedVariable(name=name,797 self._partitioned_vars[name] = partitioned_var798 return partitioned_var