Searched refs:position_b (Results 1 – 2 of 2) sorted by relevance
159 for (const HloPosition& position_b : positions) { in SetPositionsAndComputeUses() local160 if (&position_a != &position_b) { in SetPositionsAndComputeUses()161 DCHECK_NE(position_a, position_b); in SetPositionsAndComputeUses()
75 position_b, momentum_b, accept_prob_b = self.transition_kernel(86 backward_mask[:, None] * position_b)