Summary: TSIA

Test Plan: Used tensorflow in subsequent diffs.

Reviewers: zasgar, #engineering

Reviewed By: zasgar, #engineering

Subscribers: yzhao

Differential Revision: https://phab.corp.pixielabs.ai/D6290

GitOrigin-RevId: fbde26db522ca0d3fa623ce3f897f147292fdb14