Lines Matching refs:expected_shapes
1383 def _norm_backwards_test_helper(self, op, args, output_mask, expected_shapes): argument
1394 assertEqualShapes(grads[0], expected_shapes[0])
1395 assertEqualShapes(grads[1], expected_shapes[1])
1396 assertEqualShapes(grads[2], expected_shapes[2])
1409 assertEqualShapes(out_kwargs["out0"], expected_shapes[0])
1410 assertEqualShapes(out_kwargs["out1"], expected_shapes[1])
1411 assertEqualShapes(out_kwargs["out2"], expected_shapes[2])
1436 expected_shapes = (
1444 args, output_mask, expected_shapes)
1468 expected_shapes = (
1475 args, output_mask, expected_shapes)
1502 expected_shapes = (
1508 args, output_mask, expected_shapes)