| Passed | GatherTest.testBadAxis | | |
| Passed | GatherTest.testBadIndicesCPU | | |
| Passed | GatherTest.testBadIndicesType | | |
| Passed | GatherTest.testBatchDims(batch_dims=0, params=[6, 7, 8, 9], indices=[[2, 1], [0, 3]], expected=[[8, 7], [6, 9]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=0, params=[6, 7, 8, 9], indices=[[[3, 1], [2, 0]], [[0, 3], [2, 2]]], expected=[[[9, 7], [8, 6]], [[6, 9], [8, 8]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=1, params=[[6, 7], [8, 9]], indices=[[[[0, 1], [1, 0]], [[0, 0], [1, 1]]], [[[1, 1], [0, 0]], [[0, 1], [1, 0]]]], expected=[[[[6, 7], [7, 6]], [[6, 6], [7, 7]]], [[[9, 9], [8, 8]], [[8, 9], [9, 8]]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=2, params=[[[2, 3], [4, 5]], [[6, 7], [8, 9]]], indices=[[[[0, 1], [1, 0]], [[0, 0], [1, 1]]], [[[1, 1], [0, 0]], [[0, 1], [1, 0]]]], expected=[[[[2, 3], [3, 2]], [[4, 4], [5, 5]]], [[[7, 7], [6, 6]], [[8, 9], [9, 8]]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=1, axis=2, params=[[[10, 11, 12], [13, 14, 15]], [[20, 21, 22], [23, 24, 25]]], indices=[[[0, 1, 2, 1, 0]], [[0, 1, 2, 1, 0]]], expected=[[[[10, 11, 12, 11, 10]], [[13, 14, 15, 14, 13]]], [[[20, 21, 22, 21, 20]], [[23, 24, 25, 24, 23]]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=None, axis=1, params=[[10, 11, 12], [13, 14, 15]], indices=[1, 0], expected=[[11, 10], [14, 13]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=-3, axis=1, params=[[0, 1, 2], [3, 4, 5]], indices=[[[0, 1], [1, 0]]], expected=[[[[0, 1], [1, 0]]], [[[3, 4], [4, 3]]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=0, params=[8, 9], indices=[[[[0, 1], [1, 0]], [[0, 0], [1, 1]]], [[[1, 1], [0, 0]], [[0, 1], [1, 0]]]], expected=[[[[8, 9], [9, 8]], [[8, 8], [9, 9]]], [[[9, 9], [8, 8]], [[8, 9], [9, 8]]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=1, params=[[10, 11, 12, 13], [20, 21, 22, 23]], indices=[[2, 1], [0, 3]], expected=[[12, 11], [20, 23]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=2, params=[[[100, 101], [110, 111]], [[200, 201], [210, 211]]], indices=[[[0, 1], [1, 0]], [[0, 0], [1, 1]]], expected=[[[100, 101], [111, 110]], [[200, 200], [211, 211]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=-1, params=[[10, 11, 12, 13], [20, 21, 22, 23]], indices=[[2, 1], [0, 3]], expected=[[12, 11], [20, 23]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=-1, params=[[[100, 101], [110, 111]], [[200, 201], [210, 211]]], indices=[[[0, 1], [1, 0]], [[0, 0], [1, 1]]], expected=[[[100, 101], [111, 110]], [[200, 200], [211, 211]]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=1, params=[[10, 11, 12, 13], [20, 21, 22, 23]], indices=[2, 1], expected=[12, 21]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=2, params=[[[100, 101, 102, 103], [110, 111, 112, 113]], [[200, 201, 202, 203], [210, 211, 212, 213]]], indices=[[2, 1], [0, 3]], expected=[[102, 111], [200, 213]]) | | |
| Passed | GatherTest.testBatchDims(batch_dims=1, params=[[10, 11, 12, 13], [20, 21, 22, 23]], indices=[[[3, 1], [2, 0]], [[0, 3], [2, 2]]], expected=[[[13, 11], [12, 10]], [[20, 23], [22, 22]]]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=2, output_shape=[2, 3, 8, 9, 10, 5, 6, 7]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=3, output_shape=[2, 3, 4, 8, 9, 10, 6, 7]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=4, output_shape=[2, 3, 4, 5, 8, 9, 10, 7]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=5, output_shape=[2, 3, 4, 5, 6, 8, 9, 10]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=-4, output_shape=[2, 3, 8, 9, 10, 5, 6, 7]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=-3, output_shape=[2, 3, 4, 8, 9, 10, 6, 7]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=-2, output_shape=[2, 3, 4, 5, 8, 9, 10, 7]) | | |
| Passed | GatherTest.testBatchDimsMatchesPythonBatching(params_shape=[2, 3, 4, 5, 6, 7], indices_shape=[2, 3, 8, 9, 10], batch_dims=2, axis=-1, output_shape=[2, 3, 4, 5, 6, 8, 9, 10]) | | |
| Passed | GatherTest.testEmptySlices | | |
| Passed | GatherTest.testGatherResourceVariable | | |
| Passed | GatherTest.testHigherRank | | |
| Passed | GatherTest.testHigherRankGradientTape | | |
| Passed | GatherTest.testScalar1D | | |
| Passed | GatherTest.testScalar2D | | |
| Passed | GatherTest.testSimpleTwoD32 | | |
| Passed | GatherTest.testString | | |
| Passed | GatherTest.testUInt32AndUInt64 | | |
| Passed | GatherTest.testUnknownAxis | | |
| Passed | GatherTest.testUnknownIndices | | |
| Passed | GatherTest.testBadAxis (bad_axis=1) | | |
| Passed | GatherTest.testBadAxis (bad_axis=2) | | |
| Passed | GatherTest.testBadAxis (bad_axis=-2) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.int64, itype=<class 'numpy.int32'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.int64, itype=<class 'numpy.int64'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.float32, itype=<class 'numpy.int32'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.float32, itype=<class 'numpy.int64'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.complex64, itype=<class 'numpy.int32'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.complex64, itype=<class 'numpy.int64'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.complex128, itype=<class 'numpy.int32'>) | | |
| Passed | GatherTest.testEmptySlices (dtype=tf.complex128, itype=<class 'numpy.int64'>) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.int64, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.int64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.int64, indices=array(2), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.int64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.float32, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.float32, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.float32, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.float32, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex64, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex64, indices=array(2), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex128, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex128, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex128, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex128, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.int64, indices=array([[0, 1, 0],
[1, 1, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.int64, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.int64, indices=array([[1, 2, 1],
[1, 2, 2]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.int64, indices=array([[0, 0, 0],
[1, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.float32, indices=array([[0, 1, 1],
[1, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.float32, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.float32, indices=array([[2, 1, 0],
[1, 2, 2]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.float32, indices=array([[1, 0, 0],
[1, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex64, indices=array([[0, 0, 0],
[0, 1, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex64, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex64, indices=array([[0, 1, 2],
[1, 0, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex64, indices=array([[1, 0, 0],
[0, 0, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=0, dtype=tf.complex128, indices=array([[0, 0, 0],
[1, 1, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=1, dtype=tf.complex128, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=2, dtype=tf.complex128, indices=array([[1, 1, 1],
[2, 2, 2]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRank (axis=3, dtype=tf.complex128, indices=array([[1, 1, 1],
[1, 0, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.int64, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.int64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.int64, indices=array(2), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.int64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.float32, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.float32, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.float32, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.float32, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex64, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex64, indices=array(2), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex64, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex128, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex128, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex128, indices=array(0), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex128, indices=array(1), indices_shape=()) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.int64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.float32, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex64, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex128, indices=array([], dtype=int64), indices_shape=(0,)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.int64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.float32, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex64, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex128, indices=array([], shape=(2, 0), dtype=int64), indices_shape=(2, 0)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.int64, indices=array([[0, 1, 0],
[1, 1, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.int64, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.int64, indices=array([[1, 2, 1],
[1, 2, 2]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.int64, indices=array([[0, 0, 0],
[1, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.float32, indices=array([[0, 1, 1],
[1, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.float32, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.float32, indices=array([[2, 1, 0],
[1, 2, 2]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.float32, indices=array([[1, 0, 0],
[1, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex64, indices=array([[0, 0, 0],
[0, 1, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex64, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex64, indices=array([[0, 1, 2],
[1, 0, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex64, indices=array([[1, 0, 0],
[0, 0, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=0, dtype=tf.complex128, indices=array([[0, 0, 0],
[1, 1, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=1, dtype=tf.complex128, indices=array([[0, 0, 0],
[0, 0, 0]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=2, dtype=tf.complex128, indices=array([[1, 1, 1],
[2, 2, 2]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testHigherRankGradientTape (axis=3, dtype=tf.complex128, indices=array([[1, 1, 1],
[1, 0, 1]]), indices_shape=(2, 3)) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.int64, indices=4) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.int64, indices=[1, 2, 2, 4, 5]) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.float32, indices=4) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.float32, indices=[1, 2, 2, 4, 5]) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.complex64, indices=4) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.complex64, indices=[1, 2, 2, 4, 5]) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.complex128, indices=4) | | |
| Passed | GatherTest.testScalar1D (dtype=tf.complex128, indices=[1, 2, 2, 4, 5]) | | |
| Passed | GatherTest.testScalar2D (axis=0, dtype=tf.int64) | | |
| Passed | GatherTest.testScalar2D (axis=1, dtype=tf.int64) | | |
| Passed | GatherTest.testScalar2D (axis=0, dtype=tf.float32) | | |
| Passed | GatherTest.testScalar2D (axis=1, dtype=tf.float32) | | |
| Passed | GatherTest.testScalar2D (axis=0, dtype=tf.complex64) | | |
| Passed | GatherTest.testScalar2D (axis=1, dtype=tf.complex64) | | |
| Passed | GatherTest.testScalar2D (axis=0, dtype=tf.complex128) | | |
| Passed | GatherTest.testScalar2D (axis=1, dtype=tf.complex128) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=0, dtype=tf.int64) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=1, dtype=tf.int64) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=0, dtype=tf.float32) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=1, dtype=tf.float32) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=0, dtype=tf.complex64) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=1, dtype=tf.complex64) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=0, dtype=tf.complex128) | | |
| Passed | GatherTest.testSimpleTwoD32 (axis=1, dtype=tf.complex128) | | |
| Passed | GatherTest.testUInt32AndUInt64 (unsigned_type=tf.uint32) | | |
| Passed | GatherTest.testUInt32AndUInt64 (unsigned_type=tf.uint64) | | |
| Suppressed | GatherTest.testGatherRefVariable | | |
| Suppressed | GatherTest.test_session | | |