Commit d725e36e authored by Saurabh Saxena's avatar Saurabh Saxena Committed by TensorFlower Gardener
Browse files

Adds a mechanism for whitelisting ops for use in map_fns using a... 

Adds a mechanism for whitelisting ops for use in map_fns using a WHITELIST_OP_FOR_DATASET_FUNCTIONS macro.
Changes criteria for whitelisting stateful dataset ops. We used to check for the presence of the output_shapes attr. Now we just match the name and the output type.

PiperOrigin-RevId: 183014929
parent 5dfcb10d
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