diff --git a/tensorflow/python/keras/backend.py b/tensorflow/python/keras/backend.py index fe97f746b079fe..7ea7b11ced8e1a 100644 --- a/tensorflow/python/keras/backend.py +++ b/tensorflow/python/keras/backend.py @@ -4834,6 +4834,7 @@ def local_conv(inputs, return permute_dimensions(output, permutation) +@keras_export('keras.backend.local_conv1d') def local_conv1d(inputs, kernel, kernel_size, strides, data_format=None): """Apply 1D conv with un-shared weights. @@ -4868,6 +4869,7 @@ def local_conv1d(inputs, kernel, kernel_size, strides, data_format=None): data_format) +@keras_export('keras.backend.local_conv2d') def local_conv2d(inputs, kernel, kernel_size, diff --git a/tensorflow/tools/api/golden/v1/tensorflow.keras.backend.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.keras.backend.pbtxt index c64a90890f13e4..45c224734b8347 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.keras.backend.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.keras.backend.pbtxt @@ -268,6 +268,14 @@ tf_module { name: "less_equal" argspec: "args=[\'x\', \'y\'], varargs=None, keywords=None, defaults=None" } + member_method { + name: "local_conv1d" + argspec: "args=[\'inputs\', \'kernel\', \'kernel_size\', \'strides\', \'data_format\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "local_conv2d" + argspec: "args=[\'inputs\', \'kernel\', \'kernel_size\', \'strides\', \'output_shape\', \'data_format\'], varargs=None, keywords=None, defaults=[\'None\'], " + } member_method { name: "log" argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None" diff --git a/tensorflow/tools/api/golden/v2/tensorflow.keras.backend.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.keras.backend.pbtxt index 49e3d1155c3cb7..81844f6f9186f6 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.keras.backend.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.keras.backend.pbtxt @@ -264,6 +264,14 @@ tf_module { name: "less_equal" argspec: "args=[\'x\', \'y\'], varargs=None, keywords=None, defaults=None" } + member_method { + name: "local_conv1d" + argspec: "args=[\'inputs\', \'kernel\', \'kernel_size\', \'strides\', \'data_format\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "local_conv2d" + argspec: "args=[\'inputs\', \'kernel\', \'kernel_size\', \'strides\', \'output_shape\', \'data_format\'], varargs=None, keywords=None, defaults=[\'None\'], " + } member_method { name: "log" argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"