Searched refs:se_stream (Results 1 – 4 of 4) sorted by relevance
104 stream_executor::Stream *se_stream = ctx->op_device_context()->stream(); in tfKernelGenLaunchKernel() local106 se_stream->implementation()->GpuStreamHack()); in tfKernelGenLaunchKernel()
103 stream_executor::Stream *se_stream = ctx->op_device_context()->stream(); in tfKernelGenLaunchKernel() local105 reinterpret_cast<CUstream>(se_stream->implementation()->GpuStreamHack()); in tfKernelGenLaunchKernel()
76 SE_Stream* se_stream() const { return stream_; } in se_stream() function
49 ->se_stream(); in ToC()77 static_cast<tensorflow::tpu::TpuStream*>(impl)->se_stream(); in ToC()