Searched refs:tensor_diag_part (Results 1 – 6 of 6) sorted by relevance
240 name: "tensor_diag_part"
252 name: "tensor_diag_part"
612 d = array_ops.tensor_diag_part(r)
2888 array_ops.tensor_diag_part(self.total_cm), dtype=self._dtype)
2713 def tensor_diag_part( function
3030 …ss`, `tf.linalg.tensor_diag` (corresponds to `tf.diag`), `tf.linalg.tensor_diag_part` (corresponds…