Searched defs:output_batch_partitions (Results 1 – 1 of 1) sorted by relevance
766 int64_t output_batch_partitions, int64_t lhs_contracting_partitions, in PartitionBaseCase()2463 int64_t output_batch_partitions, in GetDotGroupPartitionContractingOutputShardings()2579 int64_t output_batch_partitions, in PartitionDotGroupOnContracting()2995 int64_t rhs_batch_partitions, int64_t output_batch_partitions, in PrioritizeContractingDimensionsPartitioning()3389 const int64_t output_batch_partitions = in PartitionDot() local