File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ limitations under the License.
22
22
extern "C" {
23
23
#endif
24
24
25
- class TF_Status ;
25
+ struct TF_Status ;
26
26
27
27
extern const char kIllegalArgumentException [];
28
28
extern const char kIllegalStateException [];
Original file line number Diff line number Diff line change 15
15
16
16
package org .tensorflow ;
17
17
18
+ import static org .junit .Assert .assertArrayEquals ;
18
19
import static org .junit .Assert .assertEquals ;
19
20
import static org .junit .Assert .assertTrue ;
20
21
import static org .junit .Assert .fail ;
@@ -39,7 +40,7 @@ public void run() {
39
40
new AutoCloseableList <Tensor >(s .runner ().feed ("X" , x ).fetch ("Y" ).run ())) {
40
41
assertEquals (1 , outputs .size ());
41
42
final int [][] expected = {{31 }};
42
- assertEquals (expected , outputs .get (0 ).copyTo (new int [1 ][1 ]));
43
+ assertArrayEquals (expected , outputs .get (0 ).copyTo (new int [1 ][1 ]));
43
44
}
44
45
}
45
46
}
@@ -60,7 +61,7 @@ public void runWithMetadata() {
60
61
AutoCloseableList <Tensor > outputs = new AutoCloseableList <Tensor >(result .outputs );
61
62
assertEquals (1 , outputs .size ());
62
63
final int [][] expected = {{31 }};
63
- assertEquals (expected , outputs .get (0 ).copyTo (new int [1 ][1 ]));
64
+ assertArrayEquals (expected , outputs .get (0 ).copyTo (new int [1 ][1 ]));
64
65
// Sanity check on metadatar
65
66
// See comments in fullTraceRunOptions() for an explanation about
66
67
// why this check is really silly. Ideally, this would be:
You can’t perform that action at this time.
0 commit comments