stream_gpu-inl.h File Reference

implementation of GPU code More...

#include <memory>
#include "./base.h"
#include "./tensor.h"
#include "./logging.h"

Include dependency graph for stream_gpu-inl.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	mshadow::Stream< gpu >

Namespaces
	mshadow
	overloaded + operator between half_t and bf16_t

Functions
template<>
void	mshadow::DeleteStream< gpu > (Stream< gpu > *stream)
template<>
Stream< gpu > *	mshadow::NewStream< gpu > (bool create_blas_handle, bool create_dnn_handle, int dev_id)

Detailed Description

implementation of GPU code

Copyright (c) 2014 by Contributors

Author

Bing Xu, Tianqi Chen