Searched defs:is_cpu_scalar (Results 1 – 2 of 2) sorted by relevance
188 def is_cpu_scalar(x: TensorLikeType) -> bool: function
825 bool TensorIteratorBase::is_cpu_scalar(int64_t arg) const { in is_cpu_scalar() function in at::TensorIteratorBase