Searched refs:cast_single_tensor (Results 1 – 1 of 1) sorted by relevance
941 def cast_single_tensor(x): function964 return nest.map_structure(cast_single_tensor, x)