GPU broadcast resulted in non-concrete element type Any when running several NODEs in a single training

GPU broadcast resulted in non-concrete element type Any when running several NODEs in a single training #7831