TensorShape x_shape({5, 2});
  auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape));
  auto y = Diag(scope_, x);
  TensorShape y_shape({5, 2, 5, 2});
  RunTest(x, x_shape, y, y_shape);
}

TEST_F(ArrayGradTest, DiagPartGrad) {
  TensorShape x_shape({5, 2, 5, 2});
  auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape));
  auto y = DiagPart(scope_, x);
  TensorShape y_shape({5, 2});

//go:build linux && (mips64 || mips64le)

package syscall

import (
	"unsafe"
)

const (
	_SYS_setgroups  = SYS_SETGROUPS
	_SYS_clone3     = 5435
	_SYS_faccessat2 = 5439
	_SYS_fchmodat2  = 5452
)

//sys	Dup2(oldfd int, newfd int) (err error)
//sys	Fchown(fd int, uid int, gid int) (err error)
//sys	Fstatfs(fd int, buf *Statfs_t) (err error)

	/* x^162880 mod p(x), x^162816 mod p(x) */
DATA ·IEEEConst+1536(SB)/8,$0x0000000190ef812c
DATA ·IEEEConst+1544(SB)/8,$0x0000000090db8c44

	/* x^161856 mod p(x), x^161792 mod p(x) */
DATA ·IEEEConst+1552(SB)/8,$0x0000000067a2c786
DATA ·IEEEConst+1560(SB)/8,$0x000000010010a4ce

	/* x^160832 mod p(x), x^160768 mod p(x) */
DATA ·IEEEConst+1568(SB)/8,$0x0000000048b9496c
DATA ·IEEEConst+1576(SB)/8,$0x00000001c8f4c72c

000002d0  00 ac 0c 00 00 a8 03 00  1d 20 03 05 56 dd 40 c0  |......... ..V.@.|
000002e0  b6 4d 5a df 26 e7 4a f1  a8 47 ef af 12 6e 5f 01  |.MZ.&.J..G...n_.|
000002f0  2a 7a 19 1a 45 52 8c 47  d2 53 08 04 00 80 8a 6a  |*z..ER.G.S.....j|
00000300  9d 8b 38 73 da 92 bc f6  05 79 90 af 7a 43 59 62  |..8s.....y..zCYb|
00000310  bc 97 b6 af ef ce 5f 59  07 81 93 bc c5 3c 5f f9  |......_Y.....<_.|

00000010  69 4b 53 4a f6 fc e0 c8  0e 05 ad 19 d8 b3 37 67  |iKSJ..........7g|
00000020  f2 1f 3b 33 62 17 03 01  00 20 c2 e7 d3 d2 38 ba  |..;3b.... ....8.|
00000030  27 13 ed d1 73 fc 9b f5  4f 61 f0 35 27 8e 54 52  |'...s...Oa.5'.TR|
00000040  b7 ce bd 6a 05 c6 61 95  1e 8a 15 03 01 00 20 6e  |...j..a....... n|
00000050  07 21 86 31 f8 74 6b 45  4d 10 2f 5d e7 2b 64 77  |.!.1.tkEM./].+dw|
00000060  a9 7f 1e 49 ed d5 d4 1e  2e f2 b1 c5 4d 02 46

00000330  c3 fb 96 05 bf b5 bd bf  e2 28 07 7e 51 a6 84 90  |.........(.~Q...|
00000340  bf 9e 2e f6 b5 04 8e 06  7a 63 c8 00 84 a1 a3 2c  |........zc.....,|
00000350  f3 6f 52 52 c4 ce 4a 59  31 1f d4 ab 2e f4 75 90  |.oRR..JY1.....u.|
00000360  a5 3b ff ab 20 be 51 92  c5 f4 4d 8b f2 2a a7 ff  |.;.. .Q...M..*..|
00000370  90 07 40 3e d6 9c cf 23  54 d1 65 d3 74 79 af 51  |..@>...#T.e.ty.Q|

import org.apache.lucene.util.BytesRef;
import org.apache.lucene.util.CharsRef;
import org.apache.lucene.util.UnicodeUtil;
import org.apache.lucene.util.fst.FST;

// https://issues.apache.org/jira/browse/LUCENE-5252
public final class NGramSynonymTokenizer extends Tokenizer {

    public static final int DEFAULT_N_SIZE = 2;

    public static final String DEFAULT_DELIMITERS = " 　\t\n\r";

    static final int BUFFER_SIZE = 4096;

00000000  16 03 03 00 5d 02 00 00  59 03 03 6d b7 f7 cf 1d  |....]...Y..m....|
00000010  f9 c0 02 cb ee 90 23 93  41 8e 26 24 3e 74 31 ce  |......#.A.&$>t1.|
00000020  4f 53 f8 9d 0f 94 02 b2  66 c9 87 20 d6 5c 35 52  |OS......f.. .\5R|
00000030  4b b1 f2 bb 2e 1d 95 ff  7d 83 f0 58 a8 0a ed b1  |K.......}..X....|
00000040  54 25 03 ca ea 7b 8d 1a  8f 9f 43 51 c0 09 00 00  |T%...{....CQ....|

                )
                Assignment [indexes: 41..50, line/column: 5/1..5/10, file: test] (
                    lhs = PropertyAccess [indexes: 41..42, line/column: 5/1..5/2, file: test] (
                        name = d
                    )
                    rhs = BooleanLiteral [indexes: 45..50, line/column: 5/5..5/10, file: test] (false)
                )""".trimIndent()

	a[541] = 0
	a[542] = 0
	a[543] = 0
	a[544] = 0
	a[545] = 0
	a[546] = 0
	a[547] = 0
	a[548] = 0
	a[549] = 0
	a[550] = 0
	a[551] = 0
	a[552] = 0
	a[553] = 0
	a[554] = 0
	a[555] = 0
	a[556] = 0
	a[557] = 0
	a[558] = 0
	a[559] = 0
	a[560] = 0
	a[561] = 0
	a[562] = 0
	a[563] = 0
	a[564] = 0
	a[565] = 0
	a[566] = 0
	a[567] = 0
	a[568] = 0
	a[569] = 0