forked from nudt_dsp/netrans
15 lines
376 B
Python
15 lines
376 B
Python
from netranslib.layer.customlayer import CustomLayer
|
|
from netranslib.layer.netranslayer import IoMap
|
|
|
|
class SyncHost(CustomLayer):
|
|
|
|
op = 'sync_host'
|
|
|
|
# label, description
|
|
def_input = [IoMap('in0', 'in', 'input port')]
|
|
def_output = [IoMap('out0', 'out', 'output port')]
|
|
|
|
def setup(self, inputs, outputs):
|
|
outputs[0].shape = inputs[0].shape.copy()
|
|
|