Searched refs:BuildGemmThunk (Results 1 – 2 of 2) sorted by relevance
331 std::unique_ptr<Thunk> BuildGemmThunk(const HloInstruction* inst);
354 AddThunkToThunkSequence(BuildGemmThunk(dot)); in HandleDot()676 AddThunkToThunkSequence(BuildGemmThunk(fusion)); in HandleFusion()1774 std::unique_ptr<Thunk> IrEmitterUnnested::BuildGemmThunk( in BuildGemmThunk() function in xla::gpu::IrEmitterUnnested