Caffe2 - C++ API
A deep learning, cross platform ML framework
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
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13