Lines Matching refs:gputools
47 BackendOptions& set_platform(perftools::gputools::Platform* platform);
48 perftools::gputools::Platform* platform() const;
57 perftools::gputools::Platform* platform_ = nullptr;
69 using StreamPtr = Pool<perftools::gputools::Stream>::SmartPtr;
82 perftools::gputools::Platform* platform() const { return platform_; }
99 const std::vector<perftools::gputools::StreamExecutor*>& stream_executors()
105 StatusOr<perftools::gputools::StreamExecutor*> stream_executor(
111 perftools::gputools::StreamExecutor* default_stream_executor() const {
121 perftools::gputools::StreamExecutor* executor);
160 Backend(perftools::gputools::Platform* platform, Compiler* compiler,
161 tensorflow::gtl::ArraySlice<perftools::gputools::StreamExecutor*>
169 perftools::gputools::Platform* platform_;
175 std::vector<perftools::gputools::StreamExecutor*> stream_executors_;
180 std::map<perftools::gputools::StreamExecutor*,
181 Pool<perftools::gputools::Stream>>
