projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: baf490b) | patch
[TF:XLA] Implement MatrixSetDiag and MatrixBandPart.
authorPeter Hawkins <phawkins@google.com>
Thu, 1 Feb 2018 14:47:06 +0000 (06:47 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 2 Feb 2018 01:03:04 +0000 (17:03 -0800)
commit47fcca75bc8ec9e3c9d484e055c94facef280e21
tree263bbebd7ced8527197f19204ea9e26c49a500e0tree | snapshot
parentbaf490ba79acaacb458078370e4bad1c3fd17563commit | diff
[TF:XLA] Implement MatrixSetDiag and MatrixBandPart.
Add support for int32 indices to the MatrixBandPart operator.

PiperOrigin-RevId: 184133343
tensorflow/compiler/tests/BUILD diff | blob | history
tensorflow/compiler/tests/binary_ops_test.py diff | blob | history
tensorflow/compiler/tests/matrix_band_part_test.py [new file with mode: 0644] blob
tensorflow/compiler/tf2xla/kernels/BUILD diff | blob | history
tensorflow/compiler/tf2xla/kernels/matrix_band_part_op.cc [new file with mode: 0644] blob
tensorflow/compiler/tf2xla/kernels/matrix_set_diag_op.cc [new file with mode: 0644] blob
tensorflow/core/kernels/matrix_band_part_op.cc diff | blob | history
tensorflow/core/ops/array_ops.cc diff | blob | history
tensorflow/python/kernel_tests/matrix_band_part_op_test.py diff | blob | history
Domain: Machine Learning / ML Framework;