Searched refs:preserve_cpu_scalar_tensors (Results 1 – 1 of 1) sorted by relevance
420 def _maybe_broadcast(*args, preserve_cpu_scalar_tensors=True): argument432 if preserve_cpu_scalar_tensors and utils.is_cpu_scalar_tensor(x):2701 return list(_maybe_broadcast(*tensors, preserve_cpu_scalar_tensors=False))