Searched defs:check_same_shape (Results 1 – 2 of 2) sorted by relevance
761 def check_same_shape(*args, allow_cpu_scalar_tensors: bool): function
460 C10_ALWAYS_INLINE void check_same_shape( in check_same_shape() function