If i were you i would first try the TCC (Torque Converter Clutch) solenoid, also check the wiring and make sure that it's not your problem.
If that doesn't work then it might be the TCM (Transmission Control Module.)
If that doesn't work then it might be the TCM (Transmission Control Module.)