hmsc-r / hmsc-hpc

GNU General Public License v3.0
13 stars 4 forks source link

Fix truncated normal call with empty tensor #3

Closed trossi closed 11 months ago

trossi commented 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.