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
find_op.cc
1
#include "caffe2/operators/find_op.h"
2
3
namespace
caffe2
{
4
5
OPERATOR_SCHEMA(Find)
6
.NumInputs(2)
7
.NumOutputs(1)
8
.IdenticalTypeAndShapeOfInput(1)
9
.Input(0,
"index"
,
"Index (integers)"
)
10
.Input(1,
"query"
,
"Needles / query"
)
11
.Output(
12
0,
13
"query_indices"
,
14
"Indices of the needles in index or 'missing value'"
)
15
.Arg(
"missing_value"
,
"Placeholder for items that are not found"
)
16
.SetDoc(R
"DOC(
17
Finds elements of second input from first input,
18
outputting the last (max) index for each query.
19
If query not find, inserts missing_value.
20
See IndexGet() for a version that modifies the index when
21
values are not found.
22
)DOC");
23
24
REGISTER_CPU_OPERATOR(Find, FindOp<CPUContext>)
25
26
}
// 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:55 for Caffe2 - C++ API by
1.8.11