Home
last modified time | relevance | path

Searched defs:iota_broadcast (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddiag_op.cc47 xla::XlaOp iota_broadcast = xla::Broadcast(iota, {last_dim_size}); in CreateDiagonal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc1296 auto iota_broadcast = in MatchIotaBroadCastInDim() local
1350 auto iota_broadcast = in MatchConstIotaBroadCastInDim() local