Home
last modified time | relevance | path

Searched refs:SE_CHECK_OK (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/lib/
Dstatus.h31 #define SE_CHECK_OK(val) TF_CHECK_OK(val) macro
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_platform.cc94 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeHostPlatform()
Dhost_gpu_executor.cc167 [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); in CreateStreamDependency()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dplatform.cc99 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeXlaInterpreterPlatform()
Dexecutor.cc95 [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); in CreateStreamDependency()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_platform.cc168 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeROCmPlatform()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_platform.cc201 SE_CHECK_OK(MultiPlatformManager::RegisterPlatform(std::move(platform))); in InitializeCudaPlatform()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc175 SE_CHECK_OK(tensorflow::ReadInt64FromEnvVar("TF_PER_DEVICE_MEMORY_LIMIT_MB", in GetMemoryLimitBytes()