def diagPartOp(self, tensor, dtype, expected_ans):
    self._diagPartOp(tensor, dtype, expected_ans, False)
    self._diagPartOp(tensor, dtype, expected_ans, True)