Searched refs:half_shape (Results 1 – 1 of 1) sorted by relevance
163 Shape half_shape; member173 pair.half_shape = ShapeUtil::MakeShape(shape.element_type(), {1}); in SplitShapeIntoHalves()212 pair.half_shape = ShapeUtil::MakeShape(shape.element_type(), half_shape_dims); in SplitShapeIntoHalves()246 GetThreeFryInputsAndUpdatedState(initial_state, shape_pair.half_shape); in ThreeFryRngBit32()627 std::vector<int64>(shape_pair.half_shape.rank(), 0), in NormalFloatingPointDistribution()628 shape_pair.half_shape.dimensions(), in NormalFloatingPointDistribution()629 std::vector<int64>(shape_pair.half_shape.rank(), 1)); in NormalFloatingPointDistribution()630 std::vector<int64> bits_1_starts(shape_pair.half_shape.rank(), 0); in NormalFloatingPointDistribution()634 std::vector<int64>(shape_pair.half_shape.rank(), 1)); in NormalFloatingPointDistribution()