xla::anonymous_namespace{transpose_folding.cc} Namespace Reference

Functions
TransposeFolding::OperandIndices	CanFoldOperandsIntoDot (const HloInstruction &dot, const TransposeFolding::TransposableGemmOperandsFn &transposable_gemm_operands)
bool	FoldTransposeIntoDot (InstructionOperandsPair pair)

Function Documentation

CanFoldOperandsIntoDot()

TransposeFolding::OperandIndices xla::anonymous_namespace{transpose_folding.cc}::CanFoldOperandsIntoDot	(	const HloInstruction &	dot,
		const TransposeFolding::TransposableGemmOperandsFn &	transposable_gemm_operands
	)

Decide if the operands can be folded into dot.

1. The operand must be HloOpcode::kDot
2. Return the operand set that IsRank2Transpose()

Here is the caller graph for this function:

FoldTransposeIntoDot()

bool xla::anonymous_namespace{transpose_folding.cc}::FoldTransposeIntoDot

(

InstructionOperandsPair

pair

)

Fuse instruction of kTrasposeDot

Google docs:

Folds the operands of dot that are foldable transposes. computation is the parent HLO computation of dot. Returns whether the module is changed.