def testRewriteArgumentDocstring(self): docs = """Add `a` and `b` Args: a: first arg b: second arg """ new_docs = deprecation.rewrite_argument_docstring( deprecation.rewrite_argument_docstring(docs, "a", "left"), "b", "right") new_docs_ref = """Add `left` and `right` Args: left: first arg right: second arg """ self.assertEqual(new_docs, new_docs_ref)
@tf_export("reduce_join") def reduce_join(inputs, axis=None, keep_dims=False, separator="", name=None, reduction_indices=None): inputs_t = ops.convert_to_tensor(inputs) reduction_indices = _reduce_join_reduction_dims(inputs_t, axis, reduction_indices) return gen_string_ops.reduce_join(inputs=inputs_t, reduction_indices=reduction_indices, keep_dims=keep_dims, separator=separator, name=name) reduce_join.__doc__ = deprecation.rewrite_argument_docstring( gen_string_ops.reduce_join.__doc__, "reduction_indices", "axis") ops.NotDifferentiable("RegexReplace") ops.NotDifferentiable("StringToHashBucket") ops.NotDifferentiable("StringToHashBucketFast") ops.NotDifferentiable("StringToHashBucketStrong") ops.NotDifferentiable("ReduceJoin") ops.NotDifferentiable("StringJoin") ops.NotDifferentiable("StringSplit") ops.NotDifferentiable("AsString") ops.NotDifferentiable("EncodeBase64") ops.NotDifferentiable("DecodeBase64")
keep_dims=False, separator="", name=None, reduction_indices=None): inputs_t = ops.convert_to_tensor(inputs) reduction_indices = _reduce_join_reduction_dims( inputs_t, axis, reduction_indices) return gen_string_ops.reduce_join( inputs=inputs_t, reduction_indices=reduction_indices, keep_dims=keep_dims, separator=separator, name=name) reduce_join.__doc__ = deprecation.rewrite_argument_docstring( gen_string_ops.reduce_join.__doc__, "reduction_indices", "axis") reduce_join.__doc__ = reduce_join.__doc__.replace("tf.reduce_join(", "tf.strings.reduce_join(") # This wrapper provides backwards compatibility for code that predates the # unit argument and that passed 'name' as a positional argument. @tf_export(v1=["strings.length"]) def string_length(input, name=None, unit="BYTE"): return gen_string_ops.string_length(input, unit=unit, name=name) @tf_export("strings.length", v1=[]) def string_length_v2(input, unit="BYTE", name=None): return string_length(input, name, unit)