Searched refs:predictions_r2 (Results 1 – 1 of 1) sorted by relevance
67 xla::XlaOp predictions_r2 = context->Input(0); in Compile() local77 xla::XlaOp zero_r2_f32 = xla::ZerosLike(predictions_r2); in Compile()78 xla::XlaOp select_r2 = xla::Select(eq_r2, predictions_r2, zero_r2_f32); in Compile()86 xla::XlaOp gt_r2 = xla::Gt(predictions_r2, targets_values_r1, {0}); in Compile()