Home
last modified time | relevance | path

Searched refs:cudaStream_t (Results 1 – 25 of 52) sorted by relevance

123

/external/skqp/src/compute/hs/cuda/sm_35/u64/
Dhs_cuda.h60 cudaStream_t stream0, // primary stream
61 cudaStream_t stream1, // auxilary streams
62 cudaStream_t stream2); // for concurrency
/external/skqp/src/compute/hs/cuda/sm_35/u32/
Dhs_cuda.h60 cudaStream_t stream0, // primary stream
61 cudaStream_t stream1, // auxilary streams
62 cudaStream_t stream2); // for concurrency
/external/skia/src/compute/hs/cuda/sm_35/u64/
Dhs_cuda.h60 cudaStream_t stream0, // primary stream
61 cudaStream_t stream1, // auxilary streams
62 cudaStream_t stream2); // for concurrency
/external/skia/src/compute/hs/cuda/sm_35/u32/
Dhs_cuda.h60 cudaStream_t stream0, // primary stream
61 cudaStream_t stream1, // auxilary streams
62 cudaStream_t stream2); // for concurrency
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h23 virtual const cudaStream_t& stream() const = 0;
97 static const cudaStream_t default_stream = cudaStreamDefault; in initializeDeviceProp()
114 CudaStreamDevice(const cudaStream_t* stream, int device = -1) in initializeDeviceProp()
135 const cudaStream_t& stream() const { return *stream_; } in initializeDeviceProp()
177 const cudaStream_t* stream_; in initializeDeviceProp()
193 EIGEN_STRONG_INLINE const cudaStream_t& stream() const { in initializeDeviceProp()
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
Dinc_op_kernel.cu.cc38 int count, cudaStream_t stream) { in IncrementKernel()
66 const cudaStream_t* stream = CHECK_NOTNULL( in Compute()
67 reinterpret_cast<const cudaStream_t*>(context->op_device_context() in Compute()
Dinc_op_kernel.h27 int count, cudaStream_t stream);
Dinc_op_plugin.cc70 void** outputs, void*, cudaStream_t stream) { in enqueue()
Dinc_op_plugin.h72 void* workspace, cudaStream_t stream) override;
/external/tensorflow/tensorflow/core/kernels/
Dwhere_op_gpu.cu.h74 cudaStream_t stream = 0, in operator()
89 cudaStream_t stream = 0,
106 cudaStream_t stream = 0,
125 cudaStream_t stream = 0,
142 const cudaStream_t& cu_stream = GetCudaStream(ctx);
269 const cudaStream_t& cu_stream = GetCudaStream(ctx);
Dsearchsorted_op_gpu.cu.cc67 const cudaStream_t& stream = GetCudaStream(context); in Compute()
87 const cudaStream_t& stream = GetCudaStream(context); in Compute()
Dbincount_op_gpu.cu.cc58 const cudaStream_t& stream = GetCudaStream(context); in Compute()
/external/skia/src/compute/hs/cuda/bench/
Dmain.c59 cudaStream_t stream0,
60 cudaStream_t stream1,
61 cudaStream_t stream2);
253 cudaStream_t stream0, in hs_bench()
254 cudaStream_t stream1, in hs_bench()
255 cudaStream_t stream2, in hs_bench()
515 cudaStream_t stream0,stream1,stream2; in main()
/external/skqp/src/compute/hs/cuda/bench/
Dmain.c59 cudaStream_t stream0,
60 cudaStream_t stream1,
61 cudaStream_t stream2);
253 cudaStream_t stream0, in hs_bench()
254 cudaStream_t stream1, in hs_bench()
255 cudaStream_t stream2, in hs_bench()
515 cudaStream_t stream0,stream1,stream2; in main()
/external/skqp/src/compute/hs/cuda/
Dhs_cuda.inl421 cudaStream_t streams[3];
498 hs_barrier_enqueue(cudaStream_t to, cudaStream_t from)
539 cudaStream_t to = state->streams[idx];
611 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
663 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
759 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
772 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
807 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
818 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
840 cudaStream_t stream = state->streams[hs_state_acquire(state,fm)];
[all …]
/external/skia/src/compute/hs/cuda/
Dhs_cuda.inl421 cudaStream_t streams[3];
498 hs_barrier_enqueue(cudaStream_t to, cudaStream_t from)
539 cudaStream_t to = state->streams[idx];
611 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
663 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
759 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
772 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
807 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
818 cudaStream_t stream = state->streams[hs_indices_acquire(&indices)];
840 cudaStream_t stream = state->streams[hs_state_acquire(state,fm)];
[all …]
/external/clang/test/SemaCUDA/Inputs/
Dcuda.h20 typedef struct cudaStream *cudaStream_t; typedef
23 cudaStream_t stream = 0);
/external/clang/test/PCH/Inputs/
Dcuda.h17 typedef struct cudaStream *cudaStream_t; typedef
20 cudaStream_t stream = 0);
/external/clang/test/CodeGenCUDA/Inputs/
Dcuda.h17 typedef struct cudaStream *cudaStream_t; typedef
20 cudaStream_t stream = 0);
/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h293 inline const cudaStream_t& GetCudaStream(OpKernelContext* context) { in GetCudaStream()
294 const cudaStream_t* ptr = CHECK_NOTNULL( in GetCudaStream()
295 reinterpret_cast<const cudaStream_t*>(context->op_device_context() in GetCudaStream()
332 size_t shared_memory_size_bytes, cudaStream_t stream, in CudaLaunchKernel()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudart_stub.cc87 cudaError_t CUDART_CB cudaStreamAddCallback(cudaStream_t stream, in cudaStreamAddCallback()
92 cudaError_t (*)(cudaStream_t stream, cudaStreamCallback_t callback, in cudaStreamAddCallback()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.cc95 const cudaStream_t* cuda_stream = in Create()
96 CHECK_NOTNULL(reinterpret_cast<const cudaStream_t*>( in Create()
Dnccl_all_reduce_thunk.cc286 cudaStream_t* cu_stream = reinterpret_cast<cudaStream_t*>( in DoAllReduce()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_int8_calibrator.h61 const cudaStream_t stream);
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc323 const cudaStream_t* stream = CHECK_NOTNULL( in ExecuteCalibration()
324 reinterpret_cast<const cudaStream_t*>(ctx->op_device_context() in ExecuteCalibration()
499 const cudaStream_t* stream = CHECK_NOTNULL( in ExecuteTrtEngine()
500 reinterpret_cast<const cudaStream_t*>(ctx->op_device_context() in ExecuteTrtEngine()

123