Commit 52a6a1e2 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by Amit Patankar
Browse files

Fix a bug where Dataset.map() will throw the following error when num_threads 

is specified but output_buffer_size is not:

ValueError: Tensor conversion requested dtype int64 for Tensor with dtype
int32: 'Tensor("num_threads:0", shape=(), dtype=int32)'
PiperOrigin-RevId: 162766317
parent af64d4c4
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