Fixed a bug where compute_gradient_v2 didn't get added to python/__init__.py .

PiperOrigin-RevId: 233986266
This commit is contained in:
Peng Wang 2019-02-14 10:46:15 -08:00 committed by TensorFlower Gardener
parent 8aa71253c7
commit 3d89145866
2 changed files with 5 additions and 0 deletions

View File

@ -84,6 +84,7 @@ from tensorflow.python.feature_column import feature_column_lib as feature_colum
from tensorflow.python.layers import layers
from tensorflow.python.module import module
from tensorflow.python.ops import bitwise_ops as bitwise
from tensorflow.python.ops import gradient_checker_v2
from tensorflow.python.ops import image_ops as image
from tensorflow.python.ops import manip_ops as manip
from tensorflow.python.ops import metrics

View File

@ -16,6 +16,10 @@ tf_module {
name: "benchmark_config"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "compute_gradient"
argspec: "args=[\'f\', \'x\', \'delta\'], varargs=None, keywords=None, defaults=[\'0.001\'], "
}
member_method {
name: "create_local_cluster"
argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], varargs=None, keywords=None, defaults=[\'grpc\', \'None\', \'None\'], "