| 585 | } |
| 586 | |
| 587 | Status SavedModelOptimizer::Optimize() { |
| 588 | if (option_.native_tf_mode) { |
| 589 | return RunNativeTFGraphPass(); |
| 590 | } |
| 591 | |
| 592 | return RunODLGraphPass(); |
| 593 | } |
| 594 | |
| 595 | Status SavedModelOptimizer::RunNativeTFGraphPass() { |
| 596 | if (option_.shard_embedding) { |