Searched refs:SE_CHECK_OK (Results 1 – 7 of 7) sorted by relevance
31 #define SE_CHECK_OK(val) TF_CHECK_OK(val) macro
99 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeHostPlatform()
104 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeXlaInterpreterPlatform()
97 [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); in CreateStreamDependency()
169 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeROCmPlatform()
205 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeCudaPlatform()
133 SE_CHECK_OK(tensorflow::ReadInt64FromEnvVar("TF_PER_DEVICE_MEMORY_LIMIT_MB", in GetMemoryLimitBytes()