Commit 355fb5e1 authored by Yifei Feng's avatar Yifei Feng Committed by TensorFlower Gardener
Browse files

Fix the issue where gpu_option is not respected for keras estimator. 

Set keras backend session with the given config before any get_session call creates a new session.
Fix #14776.

PiperOrigin-RevId: 187916300
parent 36b3c94a
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