Searched refs:cast_single_tensor (Results 1 – 1 of 1) sorted by relevance
1352 def cast_single_tensor(x, dtype=None): function1376 return cast_single_tensor(targets, dtype=outputs[0].dtype)1382 new_targets.append(cast_single_tensor(target, dtype=out.dtype))1399 return nest.map_structure(functools.partial(cast_single_tensor, dtype=dtype),