Searched defs:one_hot (Results 1 – 6 of 6) sorted by relevance
65 xla::XlaOp one_hot; in Compile() local
26 namespace one_hot { namespace
93 const xla::XlaOp& off_value, xla::XlaOp* one_hot) { in OneHot()
962 const int axis, mlir::TF::OneHotOp& one_hot) { in RelayoutOneHotInput()1088 auto one_hot = mlir::dyn_cast<mlir::TF::OneHotOp>(op); in ComputeLayoutForward() local1113 auto one_hot = mlir::dyn_cast<mlir::TF::OneHotOp>(op); in ComputeLayoutBackward() local
4304 def one_hot(indices, function
3633 def one_hot(indices, num_classes): function