| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_timer.cc | 20 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
76 stream->cuda_stream())
82 stream->cuda_stream())
|
| cuda_event.cc | 19 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
42 stream->cuda_stream());
|
| cuda_stream.cc | 16 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
58 return AsCUDAStream(stream)->cuda_stream();
|
| cuda_event.h | 20 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
|
| cuda_stream.h | 69 CUstream cuda_stream() const { function in class:perftools::gputools::cuda::CUDAStream
|
| cuda_gpu_executor.cc | 31 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
609 AsCUDAStream(stream)->cuda_stream(),
629 CUDAStream *cuda_stream = AsCUDAStream(stream); local
630 if (!cuda_stream->IsIdle()) {
633 cuda_stream->Destroy();
[all...] |
| cuda_rng.cc | 22 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
|
| cuda_fft.cc | 24 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
|
| cuda_blas.cc | 44 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
[all...] |
| cuda_dnn.cc | 28 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_device.cc | 94 void Reinitialize(OpKernelContext* context, const cudaStream_t* cuda_stream,
105 stream_ = cuda_stream;
650 void Reinitialize(OpKernelContext* context, const cudaStream_t* cuda_stream,
653 stream_device_.Reinitialize(context, cuda_stream, tf_gpu_id, base_allocator,
817 const cudaStream_t* cuda_stream = reinterpret_cast<const cudaStream_t*>( local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| maxpooling_op_gpu.cu.cc | 62 // kThreadsPerBlock, 0, cuda_stream>>>(...);
|
