Searched refs:SetDefaultDeviceType (Results 1 – 3 of 3) sorted by relevance
63 ->SetDefaultDeviceType(static_cast<int8_t>(c10::kCUDA)); in TEST()73 ->SetDefaultDeviceType(default_type->type); in TEST()
104 virtual void SetDefaultDeviceType(int8_t type) = 0;
154 void SetDefaultDeviceType(int8_t type) override { in SetDefaultDeviceType() function in torch::lazy::TSBackendImpl