Searched refs:TransposeFolding (Results 1 – 7 of 7) sorted by relevance
34 TransposeFolding::OperandIndices CanFoldOperandsIntoDot( in CanFoldOperandsIntoDot()36 const TransposeFolding::TransposableGemmOperandsFn& in CanFoldOperandsIntoDot()72 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoDot()85 TransposeFolding::OperandIndices CanFoldOperandsIntoConvolution( in CanFoldOperandsIntoConvolution()87 const TransposeFolding::TransposableConvOperandsFn& in CanFoldOperandsIntoConvolution()93 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoConvolution()105 std::pair<HloInstruction*, TransposeFolding::OperandIndices>;218 TransposeFolding::TransposeFolding( in TransposeFolding() function in xla::TransposeFolding224 StatusOr<bool> TransposeFolding::Run(HloModule* module) { in Run()
46 TransposeFolding transpose_folding( in FoldTranspose()48 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()52 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()97 TransposeFolding transpose_folding( in TEST_F()99 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()103 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()125 TransposeFolding transpose_folding( in TEST_F()127 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()131 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
26 class TransposeFolding : public HloModulePass {56 explicit TransposeFolding(
186 TransposeFolding transpose_folding( in TEST_F()188 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()191 TransposeFolding::NeverFoldTranspose); in TEST_F()216 TransposeFolding transpose_folding( in TEST_F()218 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()221 TransposeFolding::NeverFoldTranspose); in TEST_F()247 TransposeFolding transpose_folding( in TEST_F()249 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()252 TransposeFolding::NeverFoldTranspose); in TEST_F()
378 pipeline.AddPass<TransposeFolding>( in RunHloPassesThroughLayoutAssn()380 const TransposeFolding::OperandIndices& candidate_operands) { in RunHloPassesThroughLayoutAssn()384 : TransposeFolding::OperandIndices{}; in RunHloPassesThroughLayoutAssn()386 TransposeFolding::NeverFoldTranspose); in RunHloPassesThroughLayoutAssn()
260 pipeline.AddPass<TransposeFolding>( in OptimizeHloModule()262 const TransposeFolding::OperandIndices& candidate_operands) { in OptimizeHloModule()265 : TransposeFolding::OperandIndices{}; in OptimizeHloModule()443 pipeline.AddPass<TransposeFolding>( in OptimizeHloPostLayoutAssignment()445 const TransposeFolding::OperandIndices& candidate_operands) { in OptimizeHloPostLayoutAssignment()447 : TransposeFolding::OperandIndices{}; in OptimizeHloPostLayoutAssignment()449 TransposeFolding::NeverFoldTranspose); in OptimizeHloPostLayoutAssignment()
793 XLA_TYPED_TEST(DotOperationTest_F16F32F64CF64, TransposeFolding) { in XLA_TYPED_TEST() argument