Limpe as operações de remoção do gráfico TensorFlow

Eu tenho um gráfico congelado treinado que estou tentando executar em um dispositivo ARM. Basicamente, estou usando contrib / pi_examples / label_image, mas com minha rede em vez de Inception. Minha rede foi treinada com abandono, o que agora me causa problemas:

Invalid argument: No OpKernel was registered to support Op 'Switch' with these attrs.  Registered kernels:
  device='CPU'; T in [DT_FLOAT]
  device='CPU'; T in [DT_INT32]
  device='GPU'; T in [DT_STRING]
  device='GPU'; T in [DT_BOOL]
  device='GPU'; T in [DT_INT32]
  device='GPU'; T in [DT_FLOAT]

 [[Node: l_fc1_dropout/cond/Switch = Switch[T=DT_BOOL](is_training_pl, is_training_pl)]]

Uma solução que posso ver é criar uma biblioteca estática TF que inclua a operação correspondente. Por outro lado, pode ser uma idéia melhor eliminar as operações de abandono da rede para torná-la mais simples e rápida. Existe uma maneira de fazer isso?

Obrigado.