Searched refs:original_indices (Results 1 – 5 of 5) sorted by relevance
81 auto original_indices = Reshape( in DynamicPartitionGrad() local90 DynamicPartition(scope, original_indices, partitions, num_partitions); in DynamicPartitionGrad()
38 original_indices = array_ops.reshape(41 original_indices, indices, num_partitions)
148 original_indices = math_ops.range(array_ops.size(flat_ids))197 pindices = data_flow_ops.dynamic_partition(original_indices,
145 original_indices = math_ops.range(num_keys)146 partitioned_indices = data_flow_ops.dynamic_partition(original_indices,
773 original_indices = math_ops.range(array_ops.size(flat_ids))821 pindices = data_flow_ops.dynamic_partition(original_indices,