Skip to content

Commit 2ee908a

Browse files
Update local.py
Possible solution to tensorflow#48486
1 parent 987d4be commit 2ee908a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/python/keras/layers/local.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -734,7 +734,7 @@ def local_conv_matmul(inputs, kernel, kernel_mask, output_shape):
734734
kernel = kernel_mask * kernel
735735
kernel = make_2d(kernel, split_dim=backend.ndim(kernel) // 2)
736736

737-
output_flat = math_ops.sparse_matmul(inputs_flat, kernel, b_is_sparse=True)
737+
output_flat = math_ops.matmul(inputs_flat, kernel, b_is_sparse=True)
738738
output = backend.reshape(output_flat, [
739739
backend.shape(output_flat)[0],
740740
] + output_shape.as_list()[1:])

0 commit comments

Comments
 (0)