Searched refs:is_cpu_scalar_tensor (Results 1 – 3 of 3) sorted by relevance
434 if isinstance(a, TensorLike) and not is_cpu_scalar_tensor(a)526 a for a in tensors if isinstance(a, TensorLike) and not is_cpu_scalar_tensor(a)706 def is_cpu_scalar_tensor(a: Any) -> bool: function728 if allow_cpu_scalar_tensors and is_cpu_scalar_tensor(arg):775 if allow_cpu_scalar_tensors and is_cpu_scalar_tensor(arg):801 if allow_cpu_scalar_tensors and is_cpu_scalar_tensor(arg):
413 if not utils.is_cpu_scalar_tensor(arg):429 if utils.is_cpu_scalar_tensor(arg):
432 if preserve_cpu_scalar_tensors and utils.is_cpu_scalar_tensor(x):