Searched refs:gt_r2 (Results 1 – 1 of 1) sorted by relevance
86 xla::XlaOp gt_r2 = xla::Gt(predictions_r2, targets_values_r1, {0}); in Compile() local91 xla::XlaOp one_hot_r2 = xla::Select(gt_r2, one_r2, zero_r2); in Compile()