Searched defs:one_hot (Results 1 – 8 of 8) sorted by relevance
86 def extract_labels(f, one_hot=False, num_classes=10): argument128 one_hot=False, argument234 one_hot=False, argument
27 one_hot = text.one_hot variable
79 def _one_hot_to_embedding(one_hot, embedding_size): argument
65 xla::XlaOp one_hot; in Compile() local
87 const xla::XlaOp& off_value, xla::XlaOp* one_hot) { in OneHot()
24 namespace one_hot { namespace
2903 def one_hot(indices, function
2760 def one_hot(indices, num_classes): function