"DeepTree" network with binary classifiers at each node
You can try the SVM with your decision function :

https://scikit-learn.org/stable/modules/generated/sklearn.svm.SVC.html#sklearn.svm.SVC

it can handle muilt classfication problem with default ovr decision function

if you stick with tensorflow you can only try linver svc : https://www.tensorflow.org/api_docs/python/tf/contrib/learn/SVM