| #pragma once | |
| #include "caffe2/core/context.h" | |
| #include "caffe2/core/operator.h" | |
| #include "caffe2/utils/conversions.h" | |
| #include "caffe2/utils/math.h" | |
| namespace caffe2 { | |
| std::vector<TensorShape> FCShapeInference( | |
| const OperatorDef& def, | |
| const vector<TensorShape>& in, | |
| bool pretransposed_weight); | |
| OpSchema::Cost CostInferenceForFC( | |
| const OperatorDef& def, | |
| const vector<TensorShape>& in, | |
| bool pretransposed_weight = false); | |
| } // namespace caffe2 |