Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
c10
core
Stream.cpp
1
#include <c10/core/Stream.h>
2
3
namespace
c10
{
4
5
// Not very parseable, but I don't know a good compact syntax for streams.
6
// Feel free to change this into something more compact if needed.
7
std::ostream& operator<<(std::ostream& stream,
const
Stream& s) {
8
stream <<
"stream "
<< s.id() <<
" on device "
<< s.device();
9
return
stream;
10
}
11
12
}
// namespace c10
c10
To register your own kernel for an operator, do in one (!) cpp file: C10_REGISTER_KERNEL(OperatorHand...
Definition:
alias_info.h:7
Generated on Thu Mar 21 2019 13:06:10 for Caffe2 - C++ API by
1.8.11