Tensor Comprehensions
|
#include <functional>
#include "caffe2/core/logging.h"
#include "caffe2/core/registry.h"
#include "caffe2/proto/caffe2.pb.h"
Go to the source code of this file.
Classes | |
class | caffe2::ReferenceImplementationRegistry |
class | caffe2::ReferenceImplementationRegistry::Register |
Namespaces | |
caffe2 | |
Macros | |
#define | TC_REFERENCE_IMPLEMENTATION(name, func) |
Typedefs | |
using | caffe2::ReferenceImplementation = std::function< void(NetDef *, const OperatorDef &)> |
#define TC_REFERENCE_IMPLEMENTATION | ( | name, | |
func | |||
) |