Caffe2 - C++ API
A deep learning, cross platform ML framework
recurrent_network_blob_fetcher_op.cc
1 
17 #include "caffe2/operators/recurrent_network_blob_fetcher_op.h"
18 
19 namespace caffe2 {
20 
21 REGISTER_CPU_OPERATOR(
22  RecurrentNetworkBlobFetcher,
23  RecurrentNetworkBlobFetcherOp<CPUContext>);
24 
25 OPERATOR_SCHEMA(RecurrentNetworkBlobFetcher)
26  .NumInputs(1)
27  .NumOutputs(1)
28  .SetDoc(R"DOC(
29 Retrieves blobs from scratch workspaces (which contain intermediate recurrent
30 network computation for each timestep) and puts them in the global
31 workspace under CPUContext.
32 )DOC")
33  .Arg("prefix", "Prefix string to prepend extracted blobs.")
34  .Input(
35  0,
36  "ScratchWorkspaceBlob",
37  "Name of scratch workspace blob returned by recurrent network.")
38  .Output(
39  0,
40  "blob_names",
41  "1D tensor of strings containing extracted blob names.");
42 
43 SHOULD_NOT_DO_GRADIENT(RecurrentNetworkBlobFetcher);
44 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.