Ejemplos de reset_global_vector_size en Python

Ejemplo n.º 1

Archivo: library.py Proyecto: 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

Ejemplo n.º 2

Archivo: library.py Proyecto: 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