Searched refs:set_outer_dimension_partitions (Results 1 – 3 of 3) sorted by relevance
225 new_root->set_outer_dimension_partitions(dim_partition_counts); in AssignParallelTasksHelper()
1371 void set_outer_dimension_partitions(
3208 void HloInstruction::set_outer_dimension_partitions( in set_outer_dimension_partitions() function in xla::HloInstruction