Skip to content
Snippets Groups Projects
Commit 6f4cda12 authored by Juncheng's avatar Juncheng Committed by Li Xinqi
Browse files

fix autovar split_axis (#2125)

parent 39b47855
No related branches found
Tags v3.19-rc1
No related merge requests found
......@@ -12,6 +12,7 @@ OperatorConf GenerateVariableOpConf(const BlobDesc& blob_desc, const std::string
blob_desc.shape().ToProto(var_op_conf->mutable_shape());
var_op_conf->set_data_type(blob_desc.data_type());
var_op_conf->set_model_name(model_name);
var_op_conf->mutable_split_axis()->Clear();
return var_op;
}
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment