Home
last modified time | relevance | path

Searched refs:multi_head (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/contrib/estimator/python/estimator/
Dmulti_head.py26 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/
DBUILD28 ":multi_head",
148 name = "multi_head",
150 "python/estimator/multi_head.py",
/external/tensorflow/tensorflow/contrib/learn/
DREADME.md88 `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/
Dhead_test.py1684 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__.py319 from tensorflow.contrib.learn.python.learn.estimators.head import multi_head
Dhead.py467 def multi_head(heads, loss_weights=None): function
2141 _multi_head = multi_head