Skip to content

Commit 11d5e94

Browse files
committed
embedding_layer: pr fixes
1 parent e6189fe commit 11d5e94

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/nf/nf_embedding_layer_submodule.f90

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ module function get_params(self) result(params)
8888
real, pointer :: w_(:) => null()
8989

9090
w_(1: product(shape(self % weights))) => self % weights
91-
params = [w_]
91+
params = w_
9292
end function get_params
9393

9494
module function get_gradients(self) result(gradients)
@@ -97,7 +97,7 @@ module function get_gradients(self) result(gradients)
9797
real, pointer :: dw_(:) => null()
9898

9999
dw_(1: product(shape(self % dw))) => self % dw
100-
gradients = [dw_]
100+
gradients = dw_
101101
end function get_gradients
102102

103103
module subroutine set_params(self, params)

test/test_embedding_layer.f90

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ program test_embedding_layer
1212
print '(a)', 'test_embedding_layer: All tests passed.'
1313
else
1414
write(stderr, '(a)') 'test_embedding_layer: One or more tests failed.'
15-
stop 1
15+
error stop 1
1616
end if
1717

1818
contains

0 commit comments

Comments
 (0)