Searched defs:checkDeviceType (Results 1 – 1 of 1) sorted by relevance
247 static void checkDeviceType(CheckedFrom c, const Tensor& t, DeviceType device_type) { in checkDeviceType() function255 void checkDeviceType(CheckedFrom c, at::ArrayRef<Tensor> tensors, at::DeviceType device_type) { in checkDeviceType() function