Searched defs:contracting_dims (Results 1 – 10 of 10) sorted by relevance
54 std::vector<DimNums> contracting_dims; member
266 auto contracting_dims = (operand_idx == 0) in IsRowColumnTransposeDotOperand() local
980 int64_t rank, absl::Span<const int64_t> contracting_dims, in GetOnlyNonContractingNonBatchDim()1005 absl::Span<const int64_t> contracting_dims, in CanFoldDotIntoIndexedArray()
600 absl::Span<const int64_t> batch_dims) -> bool { in ValidateDotDimensionNumbers()625 absl::Span<const int64_t> batch_dims) -> bool { in ValidateDotDimensionNumbers()
396 std::vector<int64_t> contracting_dims; in InferDotShardingFromOperands() local
52 absl::Span<const int64_t> contracting_dims) { in GetNonContractingDims()218 auto contracting_dims = (operand_idx == 0) in CanFoldTransposeOperandIntoDot() local
456 std::vector<DimsMapping> contracting_dims; member
2114 -> std::optional<PartitionedHlo> { in PartitionDotGroupOnBatch()
362 C* batch_dims, C* contracting_dims) { in DeleteDimsFromContainer()
1263 std::vector<int64_t> contracting_dims = RandomDims(0, 2); in ChooseXlaDotArguments() local