Searched defs:batch_dimensions (Results 1 – 2 of 2) sorted by relevance
371 std::vector<int64> batch_dimensions; in BuildTriangularSolve() local
297 HloInstruction* dot_operand, absl::Span<const int64> batch_dimensions, in NormalizeDotOperandToBatchMajorAndContractingMinor()1133 absl::Span<const int64> batch_dimensions) -> int64 { in HandleDotStrengthReduction()