def _GenModelInitOpConfAndRetLbi(var_op_conf):
variable_op_conf = op_conf_util.VariableOpConf()
variable_op_conf.CopyFrom(var_op_conf.variable_conf)
op_conf = op_conf_util.OperatorConf()
op_conf.name = "model_init"
op_conf.device_tag = "cpu"
op_conf.model_init_conf.out.append("out_0")
op_conf.model_init_conf.variable_op_name.append(var_op_conf.name)
op_conf.model_init_conf.original_variable_conf.append(variable_op_conf)
lbi = logical_blob_id_util.LogicalBlobId()
lbi.op_name = op_conf.name
lbi.blob_name = op_conf.model_init_conf.out[0]
return op_conf, lbi
def _GenModelLoadOpConfAndRetLbi(var_op_conf, path_lbi):
variable_op_conf = op_conf_util.VariableOpConf()
variable_op_conf.CopyFrom(var_op_conf.variable_conf)
op_conf = op_conf_util.OperatorConf()
op_conf.name = "model_load"
op_conf.device_tag = "cpu"
op_conf.model_load_conf.path = "{}/{}".format(path_lbi.op_name,
path_lbi.blob_name)
op_conf.model_load_conf.out.append("out_0")
op_conf.model_load_conf.variable_op_name.append(var_op_conf.name)
op_conf.model_load_conf.original_variable_conf.append(variable_op_conf)
lbi = logical_blob_id_util.LogicalBlobId()
lbi.op_name = op_conf.name
lbi.blob_name = op_conf.model_load_conf.out[0]
return (op_conf, lbi)
