experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tf.matrix_set_diag is now tf.linalg.set_diag

Browse Source 
PiperOrigin-RevId: 276386518
Change-Id: I5d02f2d5af68d733fb863223291ddd883cdfa901
This commit is contained in:
Brian Atkinson 2019-10-23 17:34:48 -07:00 committed by TensorFlower Gardener
parent 01cba40984
commit 23da884d00
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/testing/op_tests/matrix_set_diag.py
View File

@ -43,7 +43,7 @@ def make_matrix_set_diag_tests(options):
dtype=parameters["input_dtype"], name="input", shape=input_shape)
diag_tensor = tf.compat.v1.placeholder(
dtype=parameters["input_dtype"], name="diagonal", shape=diag_shape)
outs = tf.matrix_set_diag(input_tensor, diag_tensor)
outs = tf.linalg.set_diag(input_tensor, diag_tensor)
return [input_tensor, diag_tensor], [outs]
def build_inputs(parameters, sess, inputs, outputs):