diff --git a/tensorflow/opensource_only.files b/tensorflow/opensource_only.files index 6b9817035c15f7..eb5a116cdd36f4 100644 --- a/tensorflow/opensource_only.files +++ b/tensorflow/opensource_only.files @@ -5,69 +5,65 @@ tensorflow/compat_template.__init__.py tensorflow/compat_template_v1.__init__.py tensorflow/contrib/mpi/BUILD tensorflow/python/autograph/core/config.py -tensorflow/python/tpu/profiler/pip_package/BUILD -tensorflow/python/tpu/profiler/pip_package/build_pip_package.sh tensorflow/python/tpu/profiler/pip_package/README +tensorflow/python/tpu/profiler/pip_package/build_pip_package.sh tensorflow/python/tpu/profiler/pip_package/setup.py +tensorflow/python/tpu/profiler/pip_package/BUILD tensorflow/stream_executor/build_defs.bzl tensorflow/third_party/BUILD tensorflow/third_party/__init__.py tensorflow/third_party/android/BUILD -tensorflow/third_party/android/android.bzl.tpl tensorflow/third_party/android/android_configure.BUILD.tpl +tensorflow/third_party/android/android.bzl.tpl tensorflow/third_party/android/android_configure.bzl tensorflow/third_party/arm_neon_2_x86_sse.BUILD tensorflow/third_party/astor.BUILD tensorflow/third_party/backports_weakref.BUILD tensorflow/third_party/boringssl/BUILD -tensorflow/third_party/clang_toolchain/download_clang.bzl -tensorflow/third_party/clang_toolchain/cc_configure_clang.bzl tensorflow/third_party/clang_toolchain/BUILD -tensorflow/third_party/com_google_absl.BUILD +tensorflow/third_party/clang_toolchain/cc_configure_clang.bzl +tensorflow/third_party/clang_toolchain/download_clang.bzl tensorflow/third_party/codegen.BUILD -tensorflow/third_party/common.bzl -tensorflow/third_party/cub.BUILD -tensorflow/third_party/cython.BUILD -tensorflow/third_party/curl.BUILD -tensorflow/third_party/eigen.BUILD tensorflow/third_party/double_conversion.BUILD -tensorflow/third_party/eigen3/BUILD -tensorflow/third_party/eigen3/Eigen/Core +tensorflow/third_party/cython.BUILD tensorflow/third_party/eigen3/Eigen/Cholesky -tensorflow/third_party/eigen3/Eigen/Eigenvalues +tensorflow/third_party/eigen3/Eigen/Core tensorflow/third_party/eigen3/Eigen/LU -tensorflow/third_party/eigen3/Eigen/QR tensorflow/third_party/eigen3/Eigen/SVD -tensorflow/third_party/eigen3/LICENSE +tensorflow/third_party/eigen3/Eigen/Eigenvalues +tensorflow/third_party/eigen3/Eigen/QR tensorflow/third_party/eigen3/gpu_packet_math.patch -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/Tensor -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/ThreadPool +tensorflow/third_party/eigen3/BUILD +tensorflow/third_party/eigen3/LICENSE tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatMatProduct.h -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/FixedPointTypes.h -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatMatProductAVX2.h -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatMatProductNEON.h -tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/PacketMathAVX2.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/ThreadPool tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatVecProduct.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/PacketMathAVX2.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatMatProductAVX2.h tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/PacketMathAVX512.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/FixedPointTypes.h tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/TypeCastingAVX2.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatMatProductNEON.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/MatMatProduct.h tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/TypeCastingAVX512.h +tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/Tensor tensorflow/third_party/eigen3/unsupported/Eigen/MatrixFunctions tensorflow/third_party/eigen3/unsupported/Eigen/SpecialFunctions -tensorflow/third_party/enum34.BUILD -tensorflow/third_party/farmhash.BUILD -tensorflow/third_party/fft2d/BUILD -tensorflow/third_party/fft2d/LICENSE +tensorflow/third_party/curl.BUILD +tensorflow/third_party/cub.BUILD +tensorflow/third_party/eigen.BUILD +tensorflow/third_party/com_google_absl.BUILD +tensorflow/third_party/common.bzl tensorflow/third_party/fft2d/fft.h +tensorflow/third_party/fft2d/LICENSE +tensorflow/third_party/fft2d/BUILD tensorflow/third_party/fft2d/fft2d.BUILD tensorflow/third_party/fft2d/fft2d.h -tensorflow/third_party/functools32.BUILD -tensorflow/third_party/gast.BUILD -tensorflow/third_party/gif.BUILD +tensorflow/third_party/enum34.BUILD tensorflow/third_party/git/BUILD tensorflow/third_party/git/BUILD.tpl tensorflow/third_party/git/git_configure.bzl -tensorflow/third_party/googleapis.BUILD +tensorflow/third_party/gast.BUILD tensorflow/third_party/gpus/BUILD tensorflow/third_party/gpus/crosstool/BUILD tensorflow/third_party/gpus/crosstool/BUILD.tpl @@ -77,51 +73,55 @@ tensorflow/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_rocm.tp tensorflow/third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl tensorflow/third_party/gpus/cuda/BUILD tensorflow/third_party/gpus/cuda/LICENSE +tensorflow/third_party/gpus/cuda/cuda_config.h.tpl tensorflow/third_party/gpus/cuda/BUILD.tpl tensorflow/third_party/gpus/cuda/BUILD.windows.tpl tensorflow/third_party/gpus/cuda/build_defs.bzl.tpl -tensorflow/third_party/gpus/cuda/cuda_config.h.tpl -tensorflow/third_party/gpus/cuda_configure.bzl -tensorflow/third_party/gpus/find_cuda_config.py -tensorflow/third_party/gpus/rocm/BUILD.tpl tensorflow/third_party/gpus/rocm/BUILD -tensorflow/third_party/gpus/rocm/build_defs.bzl.tpl tensorflow/third_party/gpus/rocm/rocm_config.h.tpl +tensorflow/third_party/gpus/rocm/BUILD.tpl +tensorflow/third_party/gpus/rocm/build_defs.bzl.tpl +tensorflow/third_party/gpus/find_cuda_config.py +tensorflow/third_party/gpus/cuda_configure.bzl tensorflow/third_party/gpus/rocm_configure.bzl +tensorflow/third_party/googleapis.BUILD +tensorflow/third_party/functools32.BUILD +tensorflow/third_party/gif.BUILD tensorflow/third_party/grpc/BUILD +tensorflow/third_party/farmhash.BUILD tensorflow/third_party/icu/udata.patch tensorflow/third_party/jsoncpp.BUILD -tensorflow/third_party/kafka/BUILD tensorflow/third_party/kafka/config.patch -tensorflow/third_party/libxsmm.BUILD -tensorflow/third_party/linenoise.BUILD +tensorflow/third_party/kafka/BUILD tensorflow/third_party/llvm/BUILD tensorflow/third_party/llvm/expand_cmake_vars.py -tensorflow/third_party/llvm/llvm.autogenerated.BUILD tensorflow/third_party/llvm/llvm.bzl -tensorflow/third_party/lmdb.BUILD +tensorflow/third_party/llvm/llvm.autogenerated.BUILD +tensorflow/third_party/libxsmm.BUILD tensorflow/third_party/mkl/BUILD -tensorflow/third_party/mkl/LICENSE -tensorflow/third_party/mkl/MKL_LICENSE tensorflow/third_party/mkl/build_defs.bzl tensorflow/third_party/mkl/mkl.BUILD -tensorflow/third_party/mkl_dnn/LICENSE +tensorflow/third_party/mkl/MKL_LICENSE +tensorflow/third_party/mkl/LICENSE +tensorflow/third_party/lmdb.BUILD tensorflow/third_party/mkl_dnn/mkldnn.BUILD -tensorflow/third_party/mpi/.gitignore +tensorflow/third_party/mkl_dnn/LICENSE tensorflow/third_party/mpi/BUILD +tensorflow/third_party/mpi/.gitignore +tensorflow/third_party/linenoise.BUILD tensorflow/third_party/mpi_collectives/BUILD tensorflow/third_party/nanopb.BUILD -tensorflow/third_party/nccl/LICENSE tensorflow/third_party/nccl/BUILD tensorflow/third_party/nccl/archive.BUILD -tensorflow/third_party/nccl/archive.patch +tensorflow/third_party/nccl/system.BUILD.tpl +tensorflow/third_party/nccl/LICENSE tensorflow/third_party/nccl/build_defs.bzl.tpl tensorflow/third_party/nccl/nccl_configure.bzl -tensorflow/third_party/nccl/system.BUILD.tpl -tensorflow/third_party/ngraph/BUILD +tensorflow/third_party/nccl/archive.patch tensorflow/third_party/ngraph/LICENSE tensorflow/third_party/ngraph/NGRAPH_LICENSE tensorflow/third_party/ngraph/build_defs.bzl +tensorflow/third_party/ngraph/BUILD tensorflow/third_party/ngraph/ngraph.BUILD tensorflow/third_party/ngraph/nlohmann_json.BUILD tensorflow/third_party/ngraph/tbb.BUILD @@ -130,115 +130,108 @@ tensorflow/third_party/opt_einsum.BUILD tensorflow/third_party/pcre.BUILD tensorflow/third_party/png.BUILD tensorflow/third_party/png_fix_rpi.patch -tensorflow/third_party/pprof.BUILD tensorflow/third_party/protobuf/BUILD +tensorflow/third_party/pprof.BUILD tensorflow/third_party/py/BUILD tensorflow/third_party/py/BUILD.tpl tensorflow/third_party/py/numpy/BUILD tensorflow/third_party/py/python_configure.bzl tensorflow/third_party/pybind11.BUILD tensorflow/third_party/python_runtime/BUILD -tensorflow/third_party/repo.bzl -tensorflow/third_party/six.BUILD +tensorflow/third_party/sycl/crosstool/BUILD tensorflow/third_party/snappy.BUILD -tensorflow/third_party/sqlite.BUILD +tensorflow/third_party/repo.bzl tensorflow/third_party/swig.BUILD -tensorflow/third_party/sycl/crosstool/BUILD tensorflow/third_party/systemlibs/BUILD -tensorflow/third_party/systemlibs/BUILD.tpl tensorflow/third_party/systemlibs/absl_py.BUILD -tensorflow/third_party/systemlibs/absl_py.absl.flags.BUILD -tensorflow/third_party/systemlibs/absl_py.absl.testing.BUILD +tensorflow/third_party/systemlibs/BUILD.tpl tensorflow/third_party/systemlibs/astor.BUILD tensorflow/third_party/systemlibs/boringssl.BUILD -tensorflow/third_party/systemlibs/curl.BUILD +tensorflow/third_party/systemlibs/absl_py.absl.testing.BUILD +tensorflow/third_party/systemlibs/absl_py.absl.flags.BUILD +tensorflow/third_party/systemlibs/double_conversion.BUILD tensorflow/third_party/systemlibs/build_defs.bzl.tpl tensorflow/third_party/systemlibs/cython.BUILD -tensorflow/third_party/systemlibs/double_conversion.BUILD -tensorflow/third_party/systemlibs/gast.BUILD -tensorflow/third_party/systemlibs/gif.BUILD tensorflow/third_party/systemlibs/google_cloud_cpp.BUILD -tensorflow/third_party/systemlibs/googleapis.BUILD -tensorflow/third_party/systemlibs/grpc.BUILD +tensorflow/third_party/systemlibs/curl.BUILD +tensorflow/third_party/systemlibs/gif.BUILD tensorflow/third_party/systemlibs/google_cloud_cpp.google.cloud.bigtable.BUILD +tensorflow/third_party/systemlibs/googleapis.BUILD tensorflow/third_party/systemlibs/jsoncpp.BUILD -tensorflow/third_party/systemlibs/nsync.BUILD tensorflow/third_party/systemlibs/opt_einsum.BUILD -tensorflow/third_party/systemlibs/lmdb.BUILD -tensorflow/third_party/systemlibs/pcre.BUILD +tensorflow/third_party/systemlibs/nsync.BUILD tensorflow/third_party/systemlibs/png.BUILD tensorflow/third_party/systemlibs/protobuf.BUILD -tensorflow/third_party/systemlibs/protobuf.bzl +tensorflow/third_party/systemlibs/grpc.BUILD tensorflow/third_party/systemlibs/re2.BUILD -tensorflow/third_party/systemlibs/six.BUILD tensorflow/third_party/systemlibs/sqlite.BUILD -tensorflow/third_party/systemlibs/snappy.BUILD tensorflow/third_party/systemlibs/swig.BUILD +tensorflow/third_party/systemlibs/pcre.BUILD +tensorflow/third_party/systemlibs/protobuf.bzl +tensorflow/third_party/systemlibs/snappy.BUILD +tensorflow/third_party/systemlibs/gast.BUILD tensorflow/third_party/systemlibs/syslibs_configure.bzl tensorflow/third_party/systemlibs/termcolor.BUILD tensorflow/third_party/systemlibs/zlib.BUILD +tensorflow/third_party/systemlibs/six.BUILD +tensorflow/third_party/systemlibs/lmdb.BUILD +tensorflow/third_party/sqlite.BUILD tensorflow/third_party/tensorrt/BUILD tensorflow/third_party/tensorrt/BUILD.tpl +tensorflow/third_party/tensorrt/tensorrt/include/tensorrt_config.h.tpl tensorflow/third_party/tensorrt/LICENSE tensorflow/third_party/tensorrt/build_defs.bzl.tpl -tensorflow/third_party/tensorrt/tensorrt/include/tensorrt_config.h.tpl tensorflow/third_party/tensorrt/tensorrt_configure.bzl -tensorflow/third_party/termcolor.BUILD -tensorflow/third_party/tflite_mobilenet.BUILD -tensorflow/third_party/tflite_mobilenet_float.BUILD -tensorflow/third_party/tflite_mobilenet_quant.BUILD -tensorflow/third_party/tflite_ovic_testdata.BUILD -tensorflow/third_party/tflite_smartreply.BUILD -tensorflow/third_party/toolchains/BUILD +tensorflow/third_party/toolchains/clang6/README.md tensorflow/third_party/toolchains/clang6/BUILD tensorflow/third_party/toolchains/clang6/CROSSTOOL.tpl -tensorflow/third_party/toolchains/clang6/README.md -tensorflow/third_party/toolchains/clang6/clang.BUILD tensorflow/third_party/toolchains/clang6/repo.bzl +tensorflow/third_party/toolchains/clang6/clang.BUILD +tensorflow/third_party/toolchains/BUILD +tensorflow/third_party/toolchains/cpus/arm/arm_compiler_configure.bzl tensorflow/third_party/toolchains/cpus/arm/BUILD tensorflow/third_party/toolchains/cpus/arm/cc_config.bzl.tpl -tensorflow/third_party/toolchains/cpus/arm/arm_compiler_configure.bzl tensorflow/third_party/toolchains/cpus/py/BUILD tensorflow/third_party/toolchains/cpus/py3/BUILD tensorflow/third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/build_defs.bzl tensorflow/third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/BUILD tensorflow/third_party/toolchains/preconfig/centos6/cuda10.1-cudnn7/cuda/BUILD tensorflow/third_party/toolchains/preconfig/centos6/cuda10.1-cudnn7/cuda/build_defs.bzl -tensorflow/third_party/toolchains/preconfig/centos6/gcc7/cc_toolchain_config.bzl tensorflow/third_party/toolchains/preconfig/centos6/gcc7/BUILD +tensorflow/third_party/toolchains/preconfig/centos6/gcc7/cc_toolchain_config.bzl tensorflow/third_party/toolchains/preconfig/centos6/gcc7/dummy_toolchain.bzl -tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/BUILD tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/cc_toolchain_config.bzl +tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/BUILD tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.1/cc_toolchain_config.bzl tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.1/BUILD tensorflow/third_party/toolchains/preconfig/centos6/py/BUILD tensorflow/third_party/toolchains/preconfig/centos6/py3/BUILD -tensorflow/third_party/toolchains/preconfig/centos6/tensorrt5/BUILD tensorflow/third_party/toolchains/preconfig/centos6/tensorrt5/build_defs.bzl +tensorflow/third_party/toolchains/preconfig/centos6/tensorrt5/BUILD +tensorflow/third_party/toolchains/preconfig/generate/generate.bzl tensorflow/third_party/toolchains/preconfig/generate/archives.bzl tensorflow/third_party/toolchains/preconfig/generate/BUILD tensorflow/third_party/toolchains/preconfig/generate/containers.bzl -tensorflow/third_party/toolchains/preconfig/generate/generate.bzl tensorflow/third_party/toolchains/preconfig/generate/workspace.bzl -tensorflow/third_party/toolchains/preconfig/ubuntu14.04/cuda10.0-cudnn7/cuda/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu14.04/cuda10.0-cudnn7/cuda/build_defs.bzl +tensorflow/third_party/toolchains/preconfig/ubuntu14.04/cuda10.0-cudnn7/cuda/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu14.04/gcc-nvcc-cuda10.0/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu14.04/gcc-nvcc-cuda10.0/cc_toolchain_config.bzl tensorflow/third_party/toolchains/preconfig/ubuntu14.04/py3/BUILD -tensorflow/third_party/toolchains/preconfig/ubuntu14.04/tensorrt5/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu14.04/tensorrt5/build_defs.bzl +tensorflow/third_party/toolchains/preconfig/ubuntu14.04/tensorrt5/BUILD +tensorflow/third_party/toolchains/preconfig/ubuntu16.04/clang/dummy_toolchain.bzl tensorflow/third_party/toolchains/preconfig/ubuntu16.04/clang/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/clang/cc_toolchain_config.bzl -tensorflow/third_party/toolchains/preconfig/ubuntu16.04/clang/dummy_toolchain.bzl tensorflow/third_party/toolchains/preconfig/ubuntu16.04/cuda10.0-cudnn7/cuda/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/cuda10.0-cudnn7/cuda/build_defs.bzl tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc5-rocm/cc_toolchain_config.bzl tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc5-rocm/BUILD -tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010/cc_toolchain_config.bzl -tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010/dummy_toolchain.bzl -tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0/BUILD +tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010/BUILD +tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010/cc_toolchain_config.bzl tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0/cc_toolchain_config.bzl +tensorflow/third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/py/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/py3/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/py3_opt/BUILD @@ -247,36 +240,43 @@ tensorflow/third_party/toolchains/preconfig/ubuntu16.04/rocm/rocm/build_defs.bzl tensorflow/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5.1/BUILD tensorflow/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5.1/build_defs.bzl -tensorflow/third_party/toolchains/preconfig/win_1803/BUILD tensorflow/third_party/toolchains/preconfig/win_1803/bazel_025/BUILD +tensorflow/third_party/toolchains/preconfig/win_1803/BUILD tensorflow/third_party/toolchains/preconfig/win_1803/py36/BUILD -tensorflow/third_party/toolchains/remote/BUILD.tpl tensorflow/third_party/toolchains/remote/BUILD -tensorflow/third_party/toolchains/remote/configure.bzl +tensorflow/third_party/toolchains/remote/BUILD.tpl tensorflow/third_party/toolchains/remote/execution.bzl.tpl -tensorflow/third_party/wrapt.BUILD +tensorflow/third_party/toolchains/remote/configure.bzl +tensorflow/third_party/tflite_smartreply.BUILD +tensorflow/third_party/tflite_mobilenet.BUILD +tensorflow/third_party/tflite_mobilenet_quant.BUILD +tensorflow/third_party/tflite_mobilenet_float.BUILD +tensorflow/third_party/tflite_ovic_testdata.BUILD +tensorflow/third_party/termcolor.BUILD +tensorflow/third_party/six.BUILD tensorflow/third_party/zlib.BUILD +tensorflow/third_party/wrapt.BUILD tensorflow/tools/ci_build/remote/BUILD -tensorflow/tools/def_file_filter/BUILD tensorflow/tools/def_file_filter/BUILD.tpl -tensorflow/tools/def_file_filter/def_file_filter.py.tpl tensorflow/tools/def_file_filter/def_file_filter_configure.bzl -tensorflow/tools/lib_package/BUILD -tensorflow/tools/lib_package/LibTensorFlowTest.java -tensorflow/tools/lib_package/README.md +tensorflow/tools/def_file_filter/def_file_filter.py.tpl +tensorflow/tools/def_file_filter/BUILD tensorflow/tools/lib_package/concat_licenses.sh +tensorflow/tools/lib_package/README.md +tensorflow/tools/lib_package/LibTensorFlowTest.java +tensorflow/tools/lib_package/libtensorflow_java_test.sh +tensorflow/tools/lib_package/BUILD tensorflow/tools/lib_package/libtensorflow_test.c tensorflow/tools/lib_package/libtensorflow_test.sh -tensorflow/tools/lib_package/libtensorflow_java_test.sh tensorflow/tools/pip_package/README -tensorflow/tools/pip_package/BUILD +tensorflow/tools/pip_package/check_load_py_test.py tensorflow/tools/pip_package/MANIFEST.in tensorflow/tools/pip_package/build_pip_package.sh -tensorflow/tools/pip_package/check_load_py_test.py tensorflow/tools/pip_package/pip_smoke_test.py tensorflow/tools/pip_package/setup.py -tensorflow/tools/pip_package/simple_console_for_windows.py +tensorflow/tools/pip_package/BUILD tensorflow/tools/pip_package/simple_console.py +tensorflow/tools/pip_package/simple_console_for_windows.py tensorflow/virtual_root_template_v1.__init__.py tensorflow/virtual_root_template_v2.__init__.py llvm/llvm/projects/google_mlir/WORKSPACE \ No newline at end of file diff --git a/tensorflow/python/keras/engine/training_v2_utils.py b/tensorflow/python/keras/engine/training_v2_utils.py index 6dabecf4f02b4c..945440212278a0 100644 --- a/tensorflow/python/keras/engine/training_v2_utils.py +++ b/tensorflow/python/keras/engine/training_v2_utils.py @@ -58,7 +58,7 @@ def _get_or_make_execution_function(model, mode): def _make_execution_function(model, mode): """Creates a function to run one step of distributed model execution.""" - per_replica_function = _make_replica_execution_function(mode) + per_replica_function = _make_replica_execution_function(model, mode) def distributed_function(input_iterator): """A single step of the distributed execution across replicas.""" @@ -70,7 +70,7 @@ def distributed_function(input_iterator): # are PerReplicas too. strategy = distribution_strategy_context.get_strategy() outputs = strategy.experimental_run_v2( - per_replica_function, args=(model, x, y, sample_weights)) + per_replica_function, args=(x, y, sample_weights)) # Out of PerReplica outputs reduce or pick values to return. all_outputs = dist_utils.unwrap_output_dict( strategy, outputs, mode) @@ -150,14 +150,14 @@ def _get_input_from_iterator(iterator): return x, y, sample_weights -def _make_replica_execution_function(mode): +def _make_replica_execution_function(model, mode): """A single step of the distributed execution on a replica.""" if mode == ModeKeys.TRAIN: - func = train_on_batch + func = functools.partial(train_on_batch, model) elif mode == ModeKeys.TEST: - func = test_on_batch + func = functools.partial(test_on_batch, model) else: - def _predict_on_batch(model, x, y=None, sample_weights=None): + def _predict_on_batch(x, y=None, sample_weights=None): del y, sample_weights return predict_on_batch(model, x)