Searched refs:tpDeviceType (Results 1 – 1 of 1) sorted by relevance
96 c10::DeviceType convertDeviceType(const std::string& tpDeviceType) { in convertDeviceType() argument97 if (tpDeviceType == tensorpipe::kCpuDeviceType) { in convertDeviceType()99 } else if (tpDeviceType == tensorpipe::kCudaDeviceType) { in convertDeviceType()