Searched defs:one_hot (Results 1 – 7 of 7) sorted by relevance
89 def _extract_labels(f, one_hot=False, num_classes=10): argument129 one_hot=False, argument269 one_hot=False, argument
61 def one_hot(input_text, function
65 xla::XlaOp one_hot; in Compile() local
26 namespace one_hot { namespace
85 const xla::XlaOp& off_value, xla::XlaOp* one_hot) { in OneHot()
4178 def one_hot(indices, function
3643 def one_hot(indices, num_classes): function