Searched refs:dot_operand (Results 1 – 1 of 1) sorted by relevance
297 HloInstruction* dot_operand, absl::Span<const int64> batch_dimensions, in NormalizeDotOperandToBatchMajorAndContractingMinor() argument301 for (int64 i = 0; i < dot_operand->shape().rank(); ++i) { in NormalizeDotOperandToBatchMajorAndContractingMinor()310 return MakeTransposeHlo(dot_operand, transpose_dimensions); in NormalizeDotOperandToBatchMajorAndContractingMinor()