Exemplo n.º 1
0
 def __init__(self):
     self.id_ = oneflow_api.NewSessionId()
     self.job_name2function_desc_ = {}
     self.status_ = SessionStatus.OPEN
     self.cond_var_ = threading.Condition()
     self.running_job_cnt_ = 0
     self.inter_user_job_info_ = None
     self.uuid2watch_handler_ = {}
     self.config_proto_ = None
     self.resource_ = None
     self.is_mirrored_strategy_enabled_stack_ = []
     self.job_name2var_name2var_blob_ = {}
     self.job_name2module_name2module_ = {}
     self.existed_module_names_ = set()
     self.var_name2var_blob_ = {}
     # parallel desc symbol id in op attribute does not always correct
     # for lazy ops as parallel conf may be updated in some passes
     # (like optimizer_placement_optimization_pass)
     self.interface_op_name2op_attr_ = {}
     self.interface_op_name2job_name_ = {}
     self.lazy_interface_op_name2parallel_conf_ = {}
     self.op_name2lazy_blob_cache_ = {}
     self.job_name2name_scope_stack_ = {}
     self.eager_global_function_desc_stack_ = []
     self.function_flag_name2default_val_ = {}
     self._UpdateFunctionFlagName2DefaultVal()
     self.scope_attr_name2default_val_ = {}
     self._UpdateScopeAttrName2DefaultVal()
     self.instruction_list_ = instr_cfg.InstructionListProto()
     self.eager_symbol_list_ = eager_symbol_util.EagerSymbolList()
     self.backward_blob_register_ = blob_register_util.BlobRegister()
     self.snapshot_mgr_ = SnapshotManager()
     self.eager_config_proto_ctx_ = None
Exemplo n.º 2
0
 def __init__(self):
     self.id_ = oneflow_api.NewSessionId()
     self.job_name2function_desc_ = {}
     self.status_ = SessionStatus.OPEN
     self.cond_var_ = threading.Condition()
     self.running_job_cnt_ = 0
     self.inter_user_job_info_ = None
     self.uuid2watch_handler_ = {}
     self.config_proto_ = None
     self.resource_ = None
     self.is_mirrored_strategy_enabled_stack_ = []
     self.function_flag_name2default_val_ = {}
     self.job_name2var_name2var_blob_ = {}
     self.job_name2module_name2module_ = {}
     self.existed_module_names_ = set()
     self.var_name2var_blob_ = {}
     self.interface_op_name2op_attr_ = {}
     self.interface_op_name2job_name_ = {}
     self.job_name2name_scope_stack_ = {}
     self.eager_global_function_desc_stack_ = []
     self._UpdateFunctionFlagName2DefaultVal()
     self.instruction_list_ = instr_util.InstructionListProto()
     self.eager_symbol_list_ = eager_symbol_util.EagerSymbolList()
     self.backward_blob_register_ = blob_register_util.BlobRegister()
     self.snapshot_mgr_ = SnapshotManager()
     self.eager_config_proto_ctx_ = None