Searched refs:stream_executor_ (Results 1 – 7 of 7) sorted by relevance
38 stream_executor_(stream_executor) {} in GpuLayoutAssignment()54 se::StreamExecutor* stream_executor_; variable
134 HeuristicLayoutAssignment(instr, stream_executor_); in AddBackendConstraintsToDnnConvCustomCall()
35 : stream_executor_(stream_executor) {} in XlaDeviceAllocator()57 stream_executor_->GetAllocatorStats(); in GetStats()
48 se::StreamExecutor* stream_executor_;
126 se::StreamExecutor* stream_executor_; variable
124 stream_executor_ = PlatformUtil::GetStreamExecutors(local_client_->platform()) in LocalClientTestBase()
48 ->BorrowStream(stream_executor_)