Searched refs:dot_instr (Results 1 – 3 of 3) sorted by relevance
36 const HloInstruction& dot_instr,42 const HloInstruction& dot_instr,
996 if (auto* dot_instr = DynCast<HloDotInstruction>(&instr)) { in IsBatchDot() local997 return dot_instr->dot_dimension_numbers().lhs_batch_dimensions_size() > 0; in IsBatchDot()1005 const HloInstruction& dot_instr, in DotImplementationCanHandleTranspose() argument1008 GetDotImplementationStrategy(dot_instr.parent()->parent()->config(), in DotImplementationCanHandleTranspose()1009 DotInfo(dot_instr), target_machine_features); in DotImplementationCanHandleTranspose()1020 const HloInstruction& dot_instr, in DotOperandsAndResultMustHaveRowMajorLayout() argument1023 GetDotImplementationStrategy(dot_instr.parent()->parent()->config(), in DotOperandsAndResultMustHaveRowMajorLayout()1024 DotInfo(dot_instr), target_machine_features); in DotOperandsAndResultMustHaveRowMajorLayout()
101 for (HloInstruction* dot_instr : dot_instrs) { in Run()103 ElideDegenerateBatchDimensionFromBatchDot(dot_instr)); in Run()