Cannot use optimizer on GPU device.

[stepochkin]

Running on GPU device I get the following error:

Cannot assign a device for operation 'apply_updates/exDeepFm/embedding/embedding_layer/YellowFin': Cou
ld not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
Colocation Debug Info:                                                                                                                
Colocation group had the following types and devices:                                                                                 
SparseApplyMomentum: CPU                                                                                                   
Shape: GPU CPU                                                                                                                        
Square: GPU CPU                                                                                                                       
Unique: GPU CPU                                                                                                          
Cast: GPU CPU                                                                                                              
UnsortedSegmentSum: GPU CPU                                                                                                           
Identity: GPU CPU                                                                                                                     
Assign: GPU CPU                                                                                                                       
StridedSlice: GPU CPU                                                                                                                 
Const: GPU CPU                                                                                                                        
VariableV2: GPU CPU
TruncatedNormal: GPU CPU
Gather: GPU CPU
Fill: GPU CPU
Mul: GPU CPU
Add: GPU CPU