Caffe2 - C++ API
A deep learning, cross platform ML framework
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
To register your own kernel for an operator, do in one (!) cpp file: C10_REGISTER_KERNEL(OperatorHand...
Definition: alias_info.h:7