Searched refs:gpuStream_t (Results 1 – 13 of 13) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDeviceGpu.h | 28 virtual const gpuStream_t& stream() const = 0;
125 static const gpuStream_t default_stream = gpuStreamDefault;
139 GpuStreamDevice(const gpuStream_t* stream, int device = -1)
159 const gpuStream_t& stream() const { return *stream_; } in stream()
201 const gpuStream_t* stream_;
217 EIGEN_STRONG_INLINE const gpuStream_t& stream() const { in stream()
|
| D | TensorGpuHipCudaDefines.h | 27 #define gpuStream_t hipStream_t macro
57 #define gpuStream_t cudaStream_t macro
|
| D | TensorGpuHipCudaUndefines.h | 15 #undef gpuStream_t
|
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper.h | 50 using gpuStream_t = cudaStream_t; variable
54 using gpuStream_t = hipStream_t; variable
88 inline const gpuStream_t& GetGpuStream(OpKernelContext* context) { in GetGpuStream()
89 const gpuStream_t* ptr = CHECK_NOTNULL( in GetGpuStream()
90 reinterpret_cast<const gpuStream_t*>(context->op_device_context() in GetGpuStream()
103 size_t shared_memory_size_bytes, gpuStream_t stream, in GpuLaunchKernel()
|
| D | cuda_sparse.h | 37 using gpuStream_t = cudaStream_t; variable
56 using gpuStream_t = hipStream_t; variable
458 gpuStream_t gpu_stream_;
|
| D | gpu_device_functions.h | 45 using gpuStream_t = cudaStream_t; variable
56 using gpuStream_t = hipStream_t; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op_gpu.cu.h | 71 gpuStream_t stream = 0, in operator()
86 gpuStream_t stream = 0,
104 gpuStream_t stream = 0,
123 gpuStream_t stream = 0,
|
| D | histogram_op_gpu.cu.cc | 69 const gpuStream_t& stream = GetGpuStream(context); in Compute()
|
| D | reduction_gpu_kernels.cu.h | 651 const gpuStream_t& cu_stream) {
719 const gpuStream_t& cu_stream) {
763 const gpuStream_t& cu_stream) {
811 T init, const gpuStream_t& cu_stream) {
854 const gpuStream_t& cu_stream) {
874 T init, const gpuStream_t& cu_stream) {
889 const gpuStream_t& cu_stream) {
951 const gpuStream_t& cu_stream) {
1077 const gpuStream_t& cu_stream = GetGpuStream(ctx);
|
| D | bincount_op_gpu.cu.cc | 69 const gpuStream_t& stream = GetGpuStream(context); in Compute()
|
| D | topk_op_gpu.h | 374 cudaError LaunchTopKKernel(const gpuStream_t& stream, int num_shards, in LaunchTopKKernel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_context.cc | 44 using gpuStream_t = cudaStream_t; typedef
66 using gpuStream_t = hipStream_t; typedef
302 const gpuStream_t* gpu_stream = in Create()
303 CHECK_NOTNULL(reinterpret_cast<const gpuStream_t*>( in Create()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 93 typedef cudaStream_t gpuStream_t; typedef
100 typedef hipStream_t gpuStream_t;
128 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
144 const gpuStream_t& stream() const override { return *stream_; } in stream()
208 static void CUDART_CB asyncFree(gpuStream_t stream, cudaError_t status, in asyncFree()
211 static void asyncFree(gpuStream_t stream, hipError_t status, void* userData) in asyncFree()
225 const gpuStream_t* stream_; // Not owned.
911 ->Reinitialize(context, static_cast<const gpuStream_t*>(gpu_stream), in Reinitialize()
921 ->Reinitialize(context, static_cast<const gpuStream_t*>(gpu_stream), in Reinitialize()
1165 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice()
|