Searched refs:new_concat_dim (Results 1 – 1 of 1) sorted by relevance
166 int64 new_concat_dim; member176 pair.new_concat_dim = 0; in SplitShapeIntoHalves()211 pair.new_concat_dim = pair.split_dim + 1; in SplitShapeIntoHalves()226 XlaOp result = ConcatInDim(builder, pair, shape_pair.new_concat_dim); in CombineShapePair()631 bits_1_starts[shape_pair.new_concat_dim] = 1; in NormalFloatingPointDistribution()