Searched defs:rhs_batch_partitions (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 438 int64_t rhs_batch_partitions, int64_t lhs_contracting_partitions, in GetWindowedEinsumConfiguration()
765 int64_t lhs_batch_partitions, int64_t rhs_batch_partitions, in PartitionBaseCase()
2851 int64_t rhs_batch_partitions, const Window& conv_window) { in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2995 int64_t rhs_batch_partitions, int64_t output_batch_partitions, in PrioritizeContractingDimensionsPartitioning()
3224 int64_t rhs_batch_partitions, SpmdBuilder* b, const Window& conv_window, in LhsIsBestMatchForNonContractingPartitioning()
3387 const int64_t rhs_batch_partitions = in PartitionDot() local
|