Lines Matching defs:partitions
52 // partitions = [0, 0, 1, 1, 0]
54 // dynamic_partition(data, partitions, num_partitions) = {
65 auto partitions = op.input(1);
70 // Note: the shape of the partitions is a prefix of the data shape.
71 // shape(partitions) = [5]
72 auto partitions_shape = Shape(scope, partitions);
73 // We now create a partitions-shaped tensor with integers from
74 // [0..size(partitions)) This will be dynamic_partitioned with the
90 DynamicPartition(scope, original_indices, partitions, num_partitions);
103 // Stop propagation along the partitions input
