1 #ifndef CAFFE2_OPERATORS_LOCALLY_CONNECTED_OP_UTIL_H_ 2 #define CAFFE2_OPERATORS_LOCALLY_CONNECTED_OP_UTIL_H_ 6 #include "caffe2/core/types.h" 16 int output_image_size;
18 std::vector<int> input_image_dims;
19 std::vector<int> column_dims;
20 std::vector<int> column_transposed_dims;
21 std::vector<int> Y_transposed_dims;
34 void SetColumnBufferShapeImpl(
37 int output_image_size,
39 std::vector<int>* column_dims,
40 std::vector<int>* column_transposed_dims,
41 std::vector<int>* column_axes,
42 std::vector<int>* column_transposed_axes);
44 void SetYBufferShapeImpl(
47 int output_image_size,
49 std::vector<int>* Y_dims,
50 std::vector<int>* Y_transposed_dims,
51 std::vector<int>* Y_axes,
52 std::vector<int>* Y_transposed_axes);
57 #endif // CAFFE2_OPERATORS_LOCALLY_CONNECTED_OP_UTIL_H_ A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...