Searched refs:bcast2 (Results 1 – 2 of 2) sorted by relevance
309 bcast2 = bcast_shape.broadcast_dimension(axis, row_length)314 self.assertShapeEq(bcast2, bcast_shape)
597 auto bcast2 = Broadcast(z, {3}); in XLA_TEST_F() local599 ConcatInDim(&builder, {concat, bcast2}, /*dimension=*/0); in XLA_TEST_F()