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
assert_op.cc
1
#include "caffe2/operators/assert_op.h"
2
3
namespace
caffe2
{
4
5
REGISTER_CPU_OPERATOR(Assert, AssertOp<CPUContext>);
6
7
OPERATOR_SCHEMA(Assert)
8
.NumInputs(1)
9
.NumOutputs(0)
10
.SetDoc(R
"DOC(
11
Assertion op. Takes in a tensor of bools, ints, longs, or long longs and checks
12
if all values are true when coerced into a boolean. In other words, for non-bool
13
types this asserts that all values in the tensor are non-zero.
14
)DOC")
15
.Arg(
16
"error_msg"
,
17
"An error message to print when the assert fails."
,
18
false
);
19
20
}
// namespace caffe2
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
convert_encoded_to_raw_leveldb.cc:47
Generated on Thu Apr 19 2018 13:03:54 for Caffe2 - C++ API by
1.8.11