def MakeScopeSymbol(job_conf_str, parallel_conf_str, is_mirrored):
job_conf = text_format.Parse(job_conf_str, job_conf_pb.JobConfigProto())
parallel_conf = text_format.Parse(parallel_conf_str,
placement_pb.ParallelConf())
return scope_util.MakeInitialScope(job_conf, parallel_conf.device_tag,
list(parallel_conf.device_name),
is_mirrored).symbol_id
def InitScopeStack():
job_conf = job_conf_pb.JobConfigProto()
job_conf.predict_conf.SetInParent()
job_conf.job_name = ""
scope = MakeInitialScope(job_conf, "cpu", ["0:0"], is_mirrored=False)
global scope_stack_
scope_stack_ = [scope]
def InitNormalModeScope(self):
job_conf = job_conf_pb.JobConfigProto()
job_conf.predict_conf.SetInParent()
job_conf.job_name = ""
scope = compiler.MakeInitialScope(job_conf,
"cpu", ["0:0"],
is_mirrored=False)
self.job_name2current_scope_[""] = scope
def _get_job_conf(self, job_name):
if job_name in self.job_name2job_conf_:
return self.job_name2job_conf_[job_name]
else:
job_conf = job_conf_proto.JobConfigProto()
job_conf.job_name = job_name
job_conf.predict_conf.SetInParent()
self.job_name2job_conf_[job_name] = job_conf
return job_conf
def __init__(self, job_func=None, job_config_proto=None, function_attribute=None):
if job_config_proto is None:
job_config_proto = job_conf_pb.JobConfigProto()
if function_attribute is None:
function_attribute = FunctionAttribute()
self.job_func = job_func
self.job_config_proto = job_config_proto
self.job_config_proto.predict_conf.SetInParent()
self.function_attribute = function_attribute
