handle all restart policies

[abin-thomas-by]

What this PR does / why we need it:
Handle all restart policies mentioned in the docs. Pod level restarts won't work when failure happens for a system reason.
Similar to PR :- kubeflow/common#189

Which issue(s) this PR fixes (optional, in Fixes #<issue number>, #<issue number>, ... format, will close the issue(s) when PR gets merged):
Fixes #

Checklist:

• Docs included if any changes are user facing

[abin-thomas-by]

/assign @Jeffwan

[johnugeorge]

@abin-thomas-by Can you rebase?

[abin-thomas-by]

@johnugeorge done!

[johnugeorge]

/lgtm

/assign @terrytangyuan

[coveralls]

Pull Request Test Coverage Report for Build 3806422891

• 12 of 16 (75.0%) changed or added relevant lines in 1 file are covered.
• 3 unchanged lines in 2 files lost coverage.
• Overall coverage decreased (-0.1%) to 39.231%

---

Changes Missing Coverage	Covered Lines	Changed/Added Lines	%
pkg/controller.v1/tensorflow/tfjob_controller.go	12	16	75.0%

Files with Coverage Reduction	New Missed Lines	%
pkg/controller.v1/tensorflow/tfjob_controller.go	1	70.83%
pkg/controller.v1/mpi/mpijob_controller.go	2	76.99%

Totals
Change from base Build 3782940340:	-0.1%
Covered Lines:	2663
Relevant Lines:	6788

---

💛 - Coveralls

[johnugeorge]

Please fix formatting issues

[abin-thomas-by]

@johnugeorge fixed

[johnugeorge]

/lgtm

[terrytangyuan]

Thanks!

/lgtm

[google-oss-prow]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: abin-thomas-by, terrytangyuan

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• OWNERS [terrytangyuan]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment