Searched refs:DepthToSpaceNet (Results 1 – 3 of 3) sorted by relevance
26 class DepthToSpaceNet(nn.Cell): class28 super(DepthToSpaceNet, self).__init__()53 dts = DepthToSpaceNet(nptype, block_size, input_shape)
25 class DepthToSpaceNet(nn.Cell): class27 super(DepthToSpaceNet, self).__init__()54 dts = DepthToSpaceNet(nptype, block_size, input_shape)
286 class DepthToSpaceNet(Cell): class288 super(DepthToSpaceNet, self).__init__()359 'block': DepthToSpaceNet(),