Searched refs:TransposeFolding (Results 1 – 7 of 7) sorted by relevance
34 TransposeFolding::OperandIndices CanFoldOperandsIntoDot( in CanFoldOperandsIntoDot()36 const TransposeFolding::TransposableGemmOperandsFn& in CanFoldOperandsIntoDot()43 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoDot()56 TransposeFolding::OperandIndices CanFoldOperandsIntoConvolution( in CanFoldOperandsIntoConvolution()58 const TransposeFolding::TransposableConvOperandsFn& in CanFoldOperandsIntoConvolution()64 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoConvolution()76 std::pair<HloInstruction*, TransposeFolding::OperandIndices>;189 TransposeFolding::TransposeFolding( in TransposeFolding() function in xla::TransposeFolding195 StatusOr<bool> TransposeFolding::Run(HloModule* module) { in Run()
47 TransposeFolding transpose_folding( in FoldTranspose()49 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()53 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()96 TransposeFolding transpose_folding( in TEST_F()98 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()102 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()124 TransposeFolding transpose_folding( in TEST_F()126 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()130 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
26 class TransposeFolding : public HloModulePass {49 explicit TransposeFolding(
184 TransposeFolding transpose_folding( in TEST_F()186 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()189 TransposeFolding::NeverFoldTranspose); in TEST_F()214 TransposeFolding transpose_folding( in TEST_F()216 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()219 TransposeFolding::NeverFoldTranspose); in TEST_F()245 TransposeFolding transpose_folding( in TEST_F()247 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()250 TransposeFolding::NeverFoldTranspose); in TEST_F()
312 pipeline.AddPass<TransposeFolding>( in RunHloPassesThroughLayoutAssn()314 const TransposeFolding::OperandIndices& candidate_operands) { in RunHloPassesThroughLayoutAssn()318 : TransposeFolding::OperandIndices{}; in RunHloPassesThroughLayoutAssn()320 TransposeFolding::NeverFoldTranspose); in RunHloPassesThroughLayoutAssn()
244 pipeline.AddPass<TransposeFolding>( in OptimizeHloModule()246 const TransposeFolding::OperandIndices& candidate_operands) { in OptimizeHloModule()248 : TransposeFolding::OperandIndices{}; in OptimizeHloModule()250 TransposeFolding::NeverFoldTranspose); in OptimizeHloModule()
753 XLA_TYPED_TEST(DotOperationTest_F16F32F64CF64, TransposeFolding) { in XLA_TYPED_TEST() argument