Searched defs:common_device (Results 1 – 3 of 3) sorted by relevance
48 inline void check_and_update_common_device(std::optional<Device>& common_device, const at::Tensor& … in check_and_update_common_device()65 inline void check_and_update_common_device(std::optional<Device>& common_device, const std::optiona… in check_and_update_common_device()71 inline void check_and_update_common_device(std::optional<Device>& common_device, at::ITensorListRef… in check_and_update_common_device()77 inline void check_and_update_common_device(std::optional<Device>& common_device, const List<std::op… in check_and_update_common_device()
6 void common_device_check_failure(Device common_device, const at::Tensor& tensor, at::CheckedFrom me… in common_device_check_failure()
353 Device common_device = kCPU; in compute_types() local