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)