Searched defs:output_rhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 436 int64_t output_rhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()
770 int64_t output_rhs_non_contracting_partitions, in PartitionBaseCase()
2462 int64_t output_rhs_non_contracting_partitions, in GetDotGroupPartitionContractingOutputShardings()
2581 int64_t output_rhs_non_contracting_partitions, in PartitionDotGroupOnContracting()
2850 int64_t output_rhs_non_contracting_partitions, int64_t lhs_batch_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2994 int64_t output_rhs_non_contracting_partitions, int64_t lhs_batch_partitions, in PrioritizeContractingDimensionsPartitioning()
3223 int64_t output_rhs_non_contracting_partitions, int64_t lhs_batch_partitions, in LhsIsBestMatchForNonContractingPartitioning()
3401 const int64_t output_rhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local
|