integrate with dist tensor layer
integrate with dist tensor layer