experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TF Internal API: tf_export __internal__.CompositeTensor symbol.

Browse Source 
PiperOrigin-RevId: 333813269
Change-Id: Ifbac79f0479d0d084190445929dcc501d30851d9
This commit is contained in:
Rick Chao 2020-09-25 14:47:56 -07:00 committed by TensorFlower Gardener
parent 90123281d1
commit 2a594cda1b
5 changed files with 18 additions and 4 deletions

2
tensorflow/python/framework/composite_tensor.py
View File

@ -25,8 +25,10 @@ import six
from tensorflow.python import _pywrap_utils
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
from tensorflow.python.util import nest
from tensorflow.python.util.tf_export import tf_export
@tf_export("__internal__.CompositeTensor", v1=[])
@six.add_metaclass(abc.ABCMeta)
class CompositeTensor(object):
"""Abstract base class for Tensor-like objects that are composed from Tensors.

1
tensorflow/python/modules_with_exports.py
View File

@ -55,6 +55,7 @@ from tensorflow.python.util.tf_export import tf_export
# _internal APIs
from tensorflow.python.framework.combinations import *
from tensorflow.python.framework.composite_tensor import *
from tensorflow.python.framework.test_combinations import *
from tensorflow.python.util.tf_decorator import make_decorator
from tensorflow.python.util.tf_decorator import unwrap

7
tensorflow/python/tools/api/generator/BUILD
View File

@ -2,10 +2,8 @@
# Scripts used to generate TensorFlow Python API.
load("//tensorflow:tensorflow.bzl", "py_test")
load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "TENSORFLOW_API_INIT_FILES")
load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "TENSORFLOW_API_INIT_FILES_V1")
load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "KERAS_API_INIT_FILES")
load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "KERAS_API_INIT_FILES_V1")
load("//tensorflow/python/tools/api/generator:api_init_files.bzl", "KERAS_API_INIT_FILES", "TENSORFLOW_API_INIT_FILES")
load("//tensorflow/python/tools/api/generator:api_init_files_v1.bzl", "KERAS_API_INIT_FILES_V1", "TENSORFLOW_API_INIT_FILES_V1")
package(
licenses = ["notice"], # Apache 2.0
@ -85,6 +83,7 @@ py_test(
],
deps = [
"//tensorflow/python:client_testlib",
"//tensorflow/python:composite_tensor",
"//tensorflow/python:framework_combinations",
"//tensorflow/python:modules_with_exports",
"//tensorflow/python:no_contrib",

8
tensorflow/tools/api/golden/v2/tensorflow.__internal__.-composite-tensor.pbtxt Normal file
View File

@ -0,0 +1,8 @@
path: "tensorflow.__internal__.CompositeTensor"
tf_class {
is_instance: "<class \'tensorflow.python.framework.composite_tensor.CompositeTensor\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
}
}

4
tensorflow/tools/api/golden/v2/tensorflow.__internal__.pbtxt
View File

@ -1,5 +1,9 @@
path: "tensorflow.__internal__"
tf_module {
member {
name: "CompositeTensor"
mtype: "<type \'type\'>"
}
member {
name: "decorator"
mtype: "<type \'module\'>"