Commit 44c19190 authored by Lukas Geiger's avatar Lukas Geiger Committed by Gunhan Gulsoy
Browse files

Support session config in tf.contrib.predictor (#19542) 

* Support session config in tf.contrib.predictor

This PR allows users to supply a custom session config uses by the predictor.

This can be essential for some GPU setups in order to play nicely with other processes running on the same GPU.

* Test passing session config to tf.contrib.predictor
parent 96788111
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment