Searched refs:set_outer_dimension_partitions (Results 1 – 4 of 4) sorted by relevance
237 new_root->set_outer_dimension_partitions(dim_partition_counts); in AssignParallelTasksHelper()
1641 void set_outer_dimension_partitions(
1826 clone->set_outer_dimension_partitions(outer_dimension_partitions_); in CloneWithNewOperands()3888 void HloInstruction::set_outer_dimension_partitions( in set_outer_dimension_partitions() function in xla::HloInstruction
2646 instruction->set_outer_dimension_partitions(*outer_dimension_partitions); in ParseInstructionRhs()