this is a TPU compatible version of tf.matrix_band_part(tf.ones([nd, ns]), -1, ns-nd) where the lower right triangle contains 1s
(nd, ns, *, dtype)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected