Commit 26378882 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

In TPU embedding mid-level API tpu_embedding.py, create embedding variables,... 

In TPU embedding mid-level API tpu_embedding.py, create embedding variables, slot variables and related ops outside of constructor.

This will enable TPUEstimator to provide specific names for the embedding and slot variables. (TPUEstimator only know these names after TPUEmbedding constructor is called.)

PiperOrigin-RevId: 231651332
parent f562ead4
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