Searched refs:CombineShapePair (Results 1 – 1 of 1) sorted by relevance
219 XlaOp CombineShapePair(absl::Span<const XlaOp> pair, in CombineShapePair() function249 XlaOp result = CombineShapePair(outputs, shape_pair, shape); in ThreeFryRngBit32()638 XlaOp normal = CombineShapePair({bits_0, bits_1}, shape_pair, shape); in NormalFloatingPointDistribution()