Home
last modified time | relevance | path

Searched refs:StreamExecutorFactory (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.cc23 StreamExecutorFactory* MakeCUDAExecutorImplementation() { in MakeCUDAExecutorImplementation()
24 static StreamExecutorFactory instance; in MakeCUDAExecutorImplementation()
30 StreamExecutorFactory* MakeROCMExecutorImplementation() { in MakeROCMExecutorImplementation()
31 static StreamExecutorFactory instance; in MakeROCMExecutorImplementation()
37 StreamExecutorFactory* MakeOpenCLExecutorImplementation() { in MakeOpenCLExecutorImplementation()
38 static StreamExecutorFactory instance; in MakeOpenCLExecutorImplementation()
44 StreamExecutorFactory MakeHostExecutorImplementation;
Dstream_executor_internal.h381 using StreamExecutorFactory = variable
388 StreamExecutorFactory *MakeCUDAExecutorImplementation();
390 StreamExecutorFactory *MakeROCMExecutorImplementation();
392 StreamExecutorFactory *MakeOpenCLExecutorImplementation();
394 extern StreamExecutorFactory MakeHostExecutorImplementation;
Dstream_executor_pimpl.cc69 internal::StreamExecutorFactory factory; in StreamExecutorImplementationFromPlatformKind()