Searched refs:conv_output_batch_partitions (Results 1 – 1 of 1) sorted by relevance
2561 const int64 conv_output_batch_partitions = get_partitions_for_dims( in PartitionDot() local2563 if ((conv_lhs_batch_partitions == conv_output_batch_partitions || in PartitionDot()2564 conv_rhs_batch_partitions == conv_output_batch_partitions) && in PartitionDot()2565 conv_output_batch_partitions > 1) { in PartitionDot()