Python reset_global_vector_size 예제들

예제 #1

파일: library.py 프로젝트: ryandeng1/mc2-copy

 def vectorized_function(*args, **kwargs):
     if len(args) > 0 and isinstance(args[0], program.Tape.Register):
         instructions_base.set_global_vector_size(args[0].size)
         res = function(*args, **kwargs)
         instructions_base.reset_global_vector_size()
     elif 'size' in kwargs:
         instructions_base.set_global_vector_size(kwargs['size'])
         del kwargs['size']
         res = function(*args, **kwargs)
         instructions_base.reset_global_vector_size()
     else:
         res = function(*args, **kwargs)
     return res

예제 #2

파일: library.py 프로젝트: lance6716/SPDZ-2

 def vectorized_function(*args, **kwargs):
     if len(args) > 0 and isinstance(args[0], program.Tape.Register):
         instructions_base.set_global_vector_size(args[0].size)
         res = function(*args, **kwargs)
         instructions_base.reset_global_vector_size()
     elif 'size' in kwargs:
         instructions_base.set_global_vector_size(kwargs['size'])
         del kwargs['size']
         res = function(*args, **kwargs)
         instructions_base.reset_global_vector_size()
     else:
         res = function(*args, **kwargs)
     return res