Searched refs:multi_head (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
D | multi_head.py | 26 from tensorflow_estimator.contrib.estimator.python.estimator import multi_head 30 multi_head.__all__ = [s for s in dir(multi_head) if not s.startswith('__')] 32 from tensorflow_estimator.contrib.estimator.python.estimator.multi_head import *
|
/external/tensorflow/tensorflow/contrib/estimator/ |
D | BUILD | 28 ":multi_head", 148 name = "multi_head", 150 "python/estimator/multi_head.py",
|
/external/tensorflow/tensorflow/contrib/learn/ |
D | README.md | 88 `tf.contrib.estimator.multi_class_head` and `tf.contrib.estimator.multi_head`. 106 * `label_name`: Not needed in `tf.estimator`. If you don’t use `multi_head`, 107 drop this argument. If you use `multi_head`, refer to 108 `tf.contrib.estimator.multi_head` documentation.
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
D | head_test.py | 1684 head_lib.multi_head((named_head, unnamed_head)) 1691 head = head_lib.multi_head((head1, head2)) 1711 head = head_lib.multi_head((head1, head2, head3)) 1737 head = head_lib.multi_head((head1, head2), (1, .5)) 1762 head = head_lib.multi_head((head1, head2)) 1789 head = head_lib.multi_head((head1, head2), (1, .5)) 1837 head = head_lib.multi_head((head1, head2), (1, .5))
|
D | __init__.py | 319 from tensorflow.contrib.learn.python.learn.estimators.head import multi_head
|
D | head.py | 467 def multi_head(heads, loss_weights=None): function 2141 _multi_head = multi_head
|