Commit 20179793 authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Applying kernel priorities to datasets that have both CPU and GPU... 

Applying kernel priorities to datasets that have both CPU and GPU implementations. This makes sure that placer would prefer the CPU kernel over the GPU kernel if no with tf.device(..) is specified.

This is necessary for creating dataset kernels as we go (i.e. in the __init__ methods of the Dataset class).

PiperOrigin-RevId: 223388100
parent ed168349
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