If somebody was looking for answer here it is:
@tf.function
def outerprodflatten(x, y, channel_dims):
    """
    According to StackOverflow:
    https://stackoverflow.com/questions/68361071/multiply-outputs-from-two-conv2d-layers-in-tensorflow-2
    """
    return tf.repeat(x,channel_dims,-1)*tf.tile(y,[1,1,1,channel_dims])