Searched defs:iota_broadcast (Results 1 – 2 of 2) sorted by relevance
47 xla::XlaOp iota_broadcast = xla::Broadcast(iota, {last_dim_size}); in CreateDiagonal() local
1296 auto iota_broadcast = in MatchIotaBroadCastInDim() local1350 auto iota_broadcast = in MatchConstIotaBroadCastInDim() local