Searched defs:one_hot (Results 1 – 7 of 7) sorted by relevance
77 def extract_labels(f, one_hot=False, num_classes=10): argument111 one_hot=False, argument215 one_hot=False, argument
76 def _one_hot_to_embedding(one_hot, embedding_size): argument
65 xla::ComputationDataHandle one_hot; in Compile() local
70 def one_hot(text, function
232 xla::ComputationDataHandle* one_hot) { in OneHot()
2300 def one_hot(indices, function
2551 def one_hot(indices, num_classes): function