Searched refs:iota_broadcast (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddiag_op.cc44 xla::ComputationDataHandle iota_broadcast = local
46 xla::ComputationDataHandle mask = builder->Eq(iota_broadcast, iota, {0});

Completed in 177 milliseconds