Home
last modified time | relevance | path

Searched refs:CanFoldTransposeOperandIntoDot (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.h86 StatusOr<bool> CanFoldTransposeOperandIntoDot(const HloInstruction& dot,
H A Dgpu_compiler.cc535 pipeline.AddPass<TransposeFolding>(CanFoldTransposeOperandIntoDot); in OptimizeHloModule()
770 pipeline.AddPass<TransposeFolding>(CanFoldTransposeOperandIntoDot, in OptimizeHloPostLayoutAssignment()
H A Dmatmul_utils.cc197 StatusOr<bool> CanFoldTransposeOperandIntoDot(const HloInstruction& dot, in CanFoldTransposeOperandIntoDot() function