Searched refs:dot_instr (Results 1 – 3 of 3) sorted by relevance
37 const HloInstruction& dot_instr,43 const HloInstruction& dot_instr,
1192 if (auto* dot_instr = DynCast<HloDotInstruction>(&instr)) { in IsBatchDot() local1193 return dot_instr->dot_dimension_numbers().lhs_batch_dimensions_size() > 0; in IsBatchDot()1201 const HloInstruction& dot_instr, in DotImplementationCanHandleTranspose() argument1204 GetDotImplementationStrategy(dot_instr.parent()->parent()->config(), in DotImplementationCanHandleTranspose()1205 DotInfo(dot_instr), target_machine_features); in DotImplementationCanHandleTranspose()1213 const HloInstruction& dot_instr, in DotOperandsAndResultMustHaveRowMajorLayout() argument1218 if (IsBatchDot(dot_instr)) { in DotOperandsAndResultMustHaveRowMajorLayout()1223 GetDotImplementationStrategy(dot_instr.parent()->parent()->config(), in DotOperandsAndResultMustHaveRowMajorLayout()1224 DotInfo(dot_instr), target_machine_features); in DotOperandsAndResultMustHaveRowMajorLayout()
121 for (HloInstruction* dot_instr : dot_instrs) { in Run()123 ElideDegenerateBatchDimensionFromBatchDot(dot_instr)); in Run()