Closed trossi closed 11 months ago
This PR will fix the error ./tensorflow/core/util/gpu_launch_config.h:129] Check failed: work_element_count > 0 (0 vs. 0) that comes when parameterized_truncated_normal() is called with empty tensors.
./tensorflow/core/util/gpu_launch_config.h:129] Check failed: work_element_count > 0 (0 vs. 0)
parameterized_truncated_normal()
This PR will fix the error
./tensorflow/core/util/gpu_launch_config.h:129] Check failed: work_element_count > 0 (0 vs. 0)
that comes whenparameterized_truncated_normal()
is called with empty tensors.