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

Update java generator.

Browse Source 
- Add a "package" argument so it can be used for packages other than tensorflow.
- Use [-subpackages](https://docs.oracle.com/javase/7/docs/technotes/tools/solaris/javadoc.html) arg to make generaion recursive.
- Merge bazel generated op-source so ops docs are generated.

$> tree /tmp/gen_java

org/
└── tensorflow
    ├── DataType.html
    ├── EagerSession.DevicePlacementPolicy.html
    ├── EagerSession.html
    ├── EagerSession.Options.html
    ├── EagerSession.ResourceCleanupStrategy.html
    ├── examples
    │   ├── LabelImage.html
    │   └── package-summary.html
    ├── ExecutionEnvironment.html
    ├── Graph.html
    ├── GraphOperationBuilder.html
    ├── GraphOperation.html
    ├── Graph.WhileSubgraphBuilder.html
    ├── op
    │   ├── annotation
    │   │   ├── Operator.html
    │   │   └── package-summary.html
    │   ├── core
    │   │   ├── Abort.html
    │   │   ├── Abort.Options.html
    │   │   ├── All.html
    │   │   ├── All.Options.html
    │   │   ├── AllToAll.html
    │   │   ├── AnonymousIteratorV2.html
    │   │   ├── AnonymousMemoryCache.html
    │   │   ├── AnonymousMultiDeviceIterator.html
    │   │   ├── AnonymousRandomSeedGenerator.html
    │   │   ├── AnonymousSeedGenerator.html
    │   │   ├── Any.html
    │   │   ├── Any.Options.html
    │   │   ├── ApplyAdagradV2.html
    │   │   ├── ApplyAdagradV2.Options.html
    │   │   ├── AssertCardinalityDataset.html
    │   │   ├── AssertNextDataset.html
    │   │   ├── AssertThat.html
    │   │   ├── AssertThat.Options.html
   ...

PiperOrigin-RevId: 308898353
Change-Id: Idedd2283d205ae08d32c3dc48b3ec5adb6c15ac4
This commit is contained in:
Mark Daoust 2020-04-28 14:31:16 -07:00 committed by TensorFlower Gardener
parent b9f3811fbb
commit cd893cfb6a
1 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tensorflow/tools/docs/build_java_api_docs.py
View File

@ -19,6 +19,9 @@ from __future__ import division
from __future__ import print_function
import pathlib
import shutil
import subprocess
import tempfile
from absl import app
from absl import flags
@ -42,15 +45,31 @@ flags.DEFINE_bool(
'search_hints', True,
'[UNUSED] Include metadata search hints in the generated files')
# Use this flag to disable bazel generation if you're not setup for it.
flags.DEFINE_bool('gen_ops', True, 'enable/disable bazel-generated ops')
# __file__ is the path to this file
DOCS_TOOLS_DIR = pathlib.Path(__file__).resolve().parent
TENSORFLOW_ROOT = DOCS_TOOLS_DIR.parent.parent
SOURCE_PATH = TENSORFLOW_ROOT / 'java/src/main/java'
TENSORFLOW_ROOT = DOCS_TOOLS_DIR.parents[2]
SOURCE_PATH = TENSORFLOW_ROOT / 'tensorflow/java/src/main/java'
OP_SOURCE_PATH = (
TENSORFLOW_ROOT /
'bazel-bin/tensorflow/java/ops/src/main/java/org/tensorflow/op')
def main(unused_argv):
merged_source = pathlib.Path(tempfile.mkdtemp())
shutil.copytree(SOURCE_PATH, merged_source / 'java')
if FLAGS.gen_ops:
subprocess.check_call(
['bazel', 'build', '//tensorflow/java:java_op_gen_sources'],
cwd=TENSORFLOW_ROOT)
shutil.copytree(OP_SOURCE_PATH, merged_source / 'java/org/tensorflow/ops')
gen_java.gen_java_docs(
source_path=SOURCE_PATH,
package='org.tensorflow',
source_path=merged_source / 'java',
output_dir=pathlib.Path(FLAGS.output_dir),
site_path=pathlib.Path(FLAGS.site_path))