| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_kronecker.py | 505 diag_part = self.operators[0].diag_part()
507 diag_part = diag_part[..., :, array_ops.newaxis]
508 op_diag_part = operator.diag_part()[..., array_ops.newaxis, :]
509 diag_part *= op_diag_part
510 diag_part = array_ops.reshape(
511 diag_part,
513 [array_ops.shape(diag_part)[:-2], [-1]], axis=0))
518 diag_part.set_shape(
520 return diag_part
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | matrix_inverse_tril.py | 150 diag_part = array_ops.matrix_diag_part(x)
152 diag_part, array_ops.zeros_like(diag_part),
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | vector_laplace_linear_operator.py | 276 return 2. * array_ops.matrix_diag(math_ops.square(self.scale.diag_part()))
282 return 2. * math_ops.square(self.scale.diag_part())
293 return np.sqrt(2) * math_ops.abs(self.scale.diag_part())
|
| D | vector_exponential_linear_operator.py | 252 return array_ops.matrix_diag(math_ops.square(self.scale.diag_part()))
258 return math_ops.square(self.scale.diag_part())
269 return math_ops.abs(self.scale.diag_part())
|
| D | mvn_linear_operator.py | 247 return array_ops.matrix_diag(math_ops.square(self.scale.diag_part()))
253 return math_ops.square(self.scale.diag_part())
264 return math_ops.abs(self.scale.diag_part())
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixDiagPart.pbtxt | 4 name: "linalg.diag_part"
|
| D | api_def_DiagPart.pbtxt | 7 name: "diag_part"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 300 tf_ans_inv = array_ops.diag_part(expected_ans)
447 tf_ans_inv = array_ops.diag_part(tensor)
471 tf_ans = array_ops.diag_part(t)
510 array_ops.diag_part(0.0)
553 y = array_ops.diag_part(x1)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.-linear-operator-scaled-identity.pbtxt | 100 name: "diag_part"
101 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-full-matrix.pbtxt | 95 name: "diag_part"
96 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-zeros.pbtxt | 95 name: "diag_part"
96 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-block-diag.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator.pbtxt | 94 name: "diag_part"
95 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-kronecker.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-inversion.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-diag.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.-linear-operator-diag.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-kronecker.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-identity.pbtxt | 96 name: "diag_part"
97 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-block-diag.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-composition.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-lower-triangular.pbtxt | 95 name: "diag_part"
96 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-full-matrix.pbtxt | 95 name: "diag_part"
96 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-inversion.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|
| D | tensorflow.linalg.-linear-operator-adjoint.pbtxt | 99 name: "diag_part"
100 argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'diag_part\'], "
|