Searched refs:bits_1_starts (Results 1 – 1 of 1) sorted by relevance
630 std::vector<int64> bits_1_starts(shape_pair.half_shape.rank(), 0); in NormalFloatingPointDistribution() local631 bits_1_starts[shape_pair.new_concat_dim] = 1; in NormalFloatingPointDistribution()632 XlaOp bits_1 = Slice(bits_state.value, bits_1_starts, in NormalFloatingPointDistribution()