Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
operators
experimental
c10
schemas
enforce_finite.cc
1
#include "caffe2/operators/experimental/c10/schemas/enforce_finite.h"
2
#include <ATen/core/dispatch/OpSchemaRegistration.h>
3
#include "caffe2/core/operator_c10wrapper.h"
4
5
using
caffe2::CPUContext
;
6
7
namespace
caffe2
{
8
namespace
ops {
9
// TODO Parse schema string instead of creating FunctionSchema manually
10
C10_DEFINE_OP_SCHEMA(
11
EnforceFinite,
12
FunctionSchema(
13
"_c10_experimental::EnforceFinite"
,
14
""
,
15
(std::vector<c10::Argument>{
c10::Argument
(
"input"
)}),
16
(std::vector<c10::Argument>{})));
17
}
18
}
19
20
namespace
caffe2
{
21
REGISTER_C10_OPERATOR_FOR_CAFFE2_DISPATCH_CPU(
22
ops::EnforceFinite(),
23
C10EnforceFinite_DontUseThisOpYet)
24
}
caffe2::CPUContext
The CPU Context, representing the bare minimum of what a Context class in Caffe2 should implement...
Definition:
context.h:40
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
blob.h:13
c10::Argument
Definition:
function_schema.h:14
Generated on Thu Mar 21 2019 13:06:18 for Caffe2 - C++ API by
1.8.11