Searched refs:partitioned_variable (Results 1 – 3 of 3) sorted by relevance
214 def determine_partitioned_axis(partitioned_variable): argument216 concatenated_variable_shape = partitioned_variable.get_shape()217 for partition in partitioned_variable:
662 partitioned_variable = variables.PartitionedVariable(669 concatenated = ops.convert_to_tensor(partitioned_variable)670 num_partitions = len(partitioned_variable)671 iterated_partitions = list(partitioned_variable)674 self.assertEqual([2], partitioned_variable.get_shape())675 self.assertEqual([2], partitioned_variable.shape)
177 partitioned_variable = variable_scope.get_variable(184 return p, partitioned_variable, params, feed_dict