|
|
|
@@ -66,8 +66,8 @@ class GeGenerator { |
|
|
|
/// @param [in] model_file_name: name of model file. |
|
|
|
/// @return SUCCESS or FAILED |
|
|
|
/// |
|
|
|
Status BuildSingleOpModel(OpDescPtr &op_desc, const std::vector<GeTensor> &inputs, |
|
|
|
const std::vector<GeTensor> &outputs, const std::string &model_file_name); |
|
|
|
Status BuildSingleOpModel(OpDescPtr &op_desc, const std::vector<GeTensor> &inputs, const std::vector<GeTensor> &outputs, |
|
|
|
const std::string &model_file_name); |
|
|
|
/// |
|
|
|
/// @ingroup ge |
|
|
|
/// @brief: Build single Op into model buff. |
|
|
|
|