Searched defs:select_mask (Results 1 – 1 of 1) sorted by relevance
199 XlaOp select_mask = Select(Eq(input, reduced_input, broadcast_dims), in ArgMinMaxTwoPass() local208 XlaOp select_mask = Select(Eq(input, reduced_input, broadcast_dims), in ArgMinMaxTwoPass() local