mxnet
|
implementation of GPU code More...
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) |
implementation of GPU code
Copyright (c) 2014 by Contributors