HomeSort by relevance Sort by last modified time
    Searched refs:_diag (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/ops/linalg/
linear_operator_diag.py 145 self._diag = ops.convert_to_tensor(diag, name="diag")
146 self._check_diag(self._diag)
149 if not self._diag.dtype.is_complex:
160 dtype=self._diag.dtype,
161 graph_parents=[self._diag],
190 d_shape = self._diag.get_shape()
194 d_shape = array_ops.shape(self._diag)
200 self._diag,
214 math_ops.real(self._diag),
219 self._diag,
    [all...]
linear_operator_lower_triangular.py 151 self._diag = array_ops.matrix_diag_part(self._tril)
190 self._diag,
198 return math_ops.reduce_prod(self._diag, axis=[-1])
202 math_ops.log(math_ops.abs(self._diag)), axis=[-1])

Completed in 101 milliseconds