Searched defs:OneHotOp (Results 1 – 2 of 2) sorted by relevance
30 explicit OneHotOp(int num_classes) : num_classes_(num_classes) {} in OneHotOp() function
89 class OneHotOp(PyTensorOperation): class