| /external/clang/test/CodeGenCUDA/Inputs/ |
| cuda.h | 17 typedef struct cudaStream *cudaStream_t;
20 cudaStream_t stream = 0);
|
| /external/clang/test/PCH/Inputs/ |
| cuda.h | 17 typedef struct cudaStream *cudaStream_t;
20 cudaStream_t stream = 0);
|
| /external/clang/test/SemaCUDA/Inputs/ |
| cuda.h | 20 typedef struct cudaStream *cudaStream_t;
23 cudaStream_t stream = 0);
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorDeviceCuda.h | 23 virtual const cudaStream_t& stream() const = 0;
97 static const cudaStream_t default_stream = cudaStreamDefault;
114 CudaStreamDevice(const cudaStream_t* stream, int device = -1)
135 const cudaStream_t& stream() const { return *stream_; }
177 const cudaStream_t* stream_;
193 EIGEN_STRONG_INLINE const cudaStream_t& stream() const {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| where_op_gpu.cu.h | 71 cudaStream_t stream = 0,
86 cudaStream_t stream = 0,
103 cudaStream_t stream = 0,
122 cudaStream_t stream = 0,
139 const cudaStream_t& cu_stream = GetCudaStream(ctx);
266 const cudaStream_t& cu_stream = GetCudaStream(ctx);
|
| bincount_op_gpu.cu.cc | 58 const cudaStream_t& stream = GetCudaStream(context);
|
| histogram_op_gpu.cu.cc | 69 const cudaStream_t& stream = GetCudaStream(context);
|
| reduction_gpu_kernels.cu.h | 451 const cudaStream_t& cu_stream) {
515 const cudaStream_t& cu_stream) {
556 const cudaStream_t& cu_stream) {
596 T init, const cudaStream_t& cu_stream) {
633 const cudaStream_t& cu_stream) {
652 const cudaStream_t& cu_stream) {
666 const cudaStream_t& cu_stream) {
787 const cudaStream_t& cu_stream = GetCudaStream(ctx);
[all...] |
| softmax_op_gpu.cu.cc | 104 const cudaStream_t& cu_stream = GetCudaStream(context);
|
| dynamic_partition_op_gpu.cu.cc | 328 const cudaStream_t& cu_stream = GetCudaStream(c);
361 const cudaStream_t& cu_stream = GetCudaStream(c);
|
| cuda_solvers.cc | 93 explicit CudaSolverHandles(cudaStream_t stream) {
118 std::unordered_map<cudaStream_t, std::unique_ptr<CudaSolverHandles>>;
150 const cudaStream_t* cu_stream_ptr = CHECK_NOTNULL(
151 reinterpret_cast<const cudaStream_t*>(context->op_device_context()
[all...] |
| topk_op_gpu.cu.cc | 370 cudaError LaunchTopKKernel(const cudaStream_t& stream, int num_shards,
435 const cudaStream_t& cu_stream = GetCudaStream(ctx);
549 const cudaStream_t& cu_stream = GetCudaStream(context);
|
| cuda_solvers.h | 314 cudaStream_t cuda_stream_;
|
| /external/tensorflow/tensorflow/contrib/tensorrt/kernels/ |
| trt_engine_op.cc | 123 const cudaStream_t* stream = CHECK_NOTNULL(
124 reinterpret_cast<const cudaStream_t*>(context->op_device_context()
|
| /external/tensorflow/tensorflow/core/util/ |
| cuda_launch_config.h | 293 inline const cudaStream_t& GetCudaStream(OpKernelContext* context) {
294 const cudaStream_t* ptr = CHECK_NOTNULL(
295 reinterpret_cast<const cudaStream_t*>(context->op_device_context()
|
| /external/tensorflow/tensorflow/contrib/tensorrt/ |
| tensorrt_test.cc | 111 cudaStream_t stream;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_device.cc | 94 void Reinitialize(OpKernelContext* context, const cudaStream_t* cuda_stream,
111 const cudaStream_t& stream() const override { return *stream_; }
167 static void CUDART_CB asyncFree(cudaStream_t stream, cudaError_t status,
180 const cudaStream_t* stream_; // Not owned.
650 void Reinitialize(OpKernelContext* context, const cudaStream_t* cuda_stream,
[all...] |
| /external/clang/unittests/ASTMatchers/ |
| ASTMatchersTest.h | 173 "typedef struct cudaStream *cudaStream_t;"
176 " cudaStream_t stream = 0);";
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_manager.cc | 447 const cudaStream_t* cu_stream = reinterpret_cast<const cudaStream_t*>(
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops_gpu.cu.cc | 189 const cudaStream_t& cu_stream = GetCudaStream(ctx);
324 const cudaStream_t& cu_stream = GetCudaStream(ctx);
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| ring.h | 73 cudaStream_t CudaStreamForMPI();
|
| mpi_ops.cc | 246 cudaStream_t stream;
[all...] |
