├── .bazelrc ├── .clang-format ├── .cmake-format.yaml ├── .gitattributes ├── .github └── workflows │ ├── python.yml │ └── tensornet-tools.yml ├── .gitignore ├── .gitlab-ci.yml ├── .pixi └── rc.sh ├── .pre-commit-config.yaml ├── .yamlfmt.yaml ├── CMakeLists.txt ├── CMakePresets.json ├── LICENSE ├── LICENSE-HEADER ├── README.md ├── WORKSPACE ├── cmake-build ├── cmake ├── AddModulePath.cmake ├── DebugUtils.cmake ├── FindTensorflow.cmake ├── IncludeWithWatch.cmake ├── PrintPythonModule.cmake └── find-tensorflow.py ├── config └── tn_build.yaml ├── configure.sh ├── core ├── BUILD ├── kernels │ ├── bn_table_ops.cc │ ├── bn_table_ops_dummy.cc │ ├── data │ │ ├── balance_dataset_ops.cc │ │ ├── balance_dataset_ops.h │ │ ├── balance_dataset_ops_dummy.cc │ │ └── balance_dataset_ops_dummy.h │ ├── dense_table_ops.cc │ ├── dense_table_ops_dummy.cc │ ├── resource_var_wrapper.h │ ├── sparse_table_ops.cc │ └── sparse_table_ops_dummy.cc ├── main │ └── py_wrapper.cc ├── ops │ ├── balance_dataset_ops.cc │ ├── bn_table_ops.cc │ ├── dense_table_ops.cc │ └── sparse_table_ops.cc ├── ps │ ├── optimizer │ │ ├── ada_grad_kernel.cc │ │ ├── ada_grad_kernel.h │ │ ├── adam_kernel.cc │ │ ├── adam_kernel.h │ │ ├── data_struct.cc │ │ ├── data_struct.h │ │ ├── ftrl_kernel.cc │ │ ├── ftrl_kernel.h │ │ ├── optimizer.cc │ │ ├── optimizer.h │ │ └── optimizer_kernel.h │ ├── ps_cluster.cc │ ├── ps_cluster.h │ ├── ps_local_server.cc │ ├── ps_local_server.h │ ├── ps_remote_server.cc │ ├── ps_remote_server.h │ ├── ps_server_interface.h │ ├── ps_service_impl.cc │ ├── ps_service_impl.h │ └── table │ │ ├── bn_table.cc │ │ ├── bn_table.h │ │ ├── dense_table.cc │ │ ├── dense_table.h │ │ ├── sparse_table.cc │ │ └── sparse_table.h ├── ps_interface │ ├── BUILD │ ├── ps_raw_interface.h │ └── ps_server.proto ├── public │ └── version.h └── utility │ ├── BUILD │ ├── allocator.h │ ├── file_io.cc │ ├── file_io.h │ ├── fix_redef.h │ ├── mpi_manager.cc │ ├── mpi_manager.h │ ├── net_util.cc │ ├── net_util.h │ ├── random.h │ └── semaphore.h ├── doc ├── async-arch.png ├── compile_deploy.md ├── inference.png ├── logo.png ├── tensornet-wide-deep.png ├── tf-wide-deep.png └── tutorial │ ├── .gitignore │ ├── 01-begin-with-wide-deep.ipynb │ ├── 02-run-in-cluster.ipynb │ ├── 03-split-to-sub-graph.ipynb │ ├── 04-deploy-tf-graph-online.ipynb │ ├── 05-export-sparse-feature-embedding.ipynb │ └── pic │ └── wide-deep.png ├── examples ├── .gitignore ├── common │ ├── config.py │ ├── feature_column.py │ ├── layers.py │ └── util.py ├── gen_example_data.py ├── integration-test.py ├── main.py ├── models │ ├── deep_cross.py │ ├── deepfm.py │ └── wide_deep.py └── online_serving │ ├── .gitignore │ ├── BUILD │ ├── graph.cc │ ├── main.cc │ └── test_env │ ├── .gitignore │ └── data │ ├── feature.data │ └── slot.data ├── manager ├── pixi.lock ├── pixiw ├── pyproject.toml ├── setup.py ├── tensornet ├── __init__.py ├── _opentelemetry.py ├── callbacks │ ├── __init__.py │ └── callbacks.py ├── core │ ├── .gitignore │ ├── __init__.py │ ├── gen_balance_dataset_ops.py │ ├── gen_bn_table_ops.py │ ├── gen_dense_table_ops.py │ └── gen_sparse_table_ops.py ├── data │ ├── __init__.py │ └── dataset.py ├── distribute │ ├── __init__.py │ └── ps_strategy.py ├── feature_column │ ├── __init__.py │ ├── category_column.py │ └── sequence_category_column.py ├── layers │ ├── __init__.py │ ├── embedding_features.py │ ├── normalization_layer.py │ ├── position_mapping.py │ └── sequence_embedding_features.py ├── metric │ ├── __init__.py │ └── metrics.py ├── model │ ├── Model.py │ ├── __init__.py │ └── compile_utils.py ├── optimizer │ ├── __init__.py │ └── optimizer.py └── version.py ├── tensornet_tools ├── LICENSE ├── README.md ├── bin │ └── tn_tools.sh ├── config │ └── tn_tool_env.yaml ├── pyproject.toml └── python │ ├── merge_extra_embedding.py │ ├── merge_sparse.py │ ├── resize_dense.py │ ├── resize_sparse.py │ └── utils.py ├── test ├── CMakeLists.txt ├── ps │ ├── optimizer │ │ ├── BUILD │ │ └── optimizer_kernel_test.cc │ └── table │ │ ├── BUILD │ │ └── sparse_table_test.cc └── utility │ ├── BUILD │ └── allocator_test.cc ├── tf_gen_op_wrapper.py └── thirdparty ├── openmpi ├── .gitignore └── BUILD ├── patches ├── 01-fix_dl_sym.patch ├── 02-fix-brpc-compile.patch ├── 03-fix-brpc-cxx-flags.patch ├── 04-boringssl.patch ├── 05-disable-install-brpc.patch └── BUILD └── tensorflow ├── .gitignore └── BUILD /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.bazelrc -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.clang-format -------------------------------------------------------------------------------- /.cmake-format.yaml: -------------------------------------------------------------------------------- 1 | --- 2 | line_width: 120 3 | -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/python.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.github/workflows/python.yml -------------------------------------------------------------------------------- /.github/workflows/tensornet-tools.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.github/workflows/tensornet-tools.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitlab-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.gitlab-ci.yml -------------------------------------------------------------------------------- /.pixi/rc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.pixi/rc.sh -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.yamlfmt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/.yamlfmt.yaml -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CMakePresets.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/CMakePresets.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE-HEADER: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/LICENSE-HEADER -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/README.md -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/WORKSPACE -------------------------------------------------------------------------------- /cmake-build: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake-build -------------------------------------------------------------------------------- /cmake/AddModulePath.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake/AddModulePath.cmake -------------------------------------------------------------------------------- /cmake/DebugUtils.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake/DebugUtils.cmake -------------------------------------------------------------------------------- /cmake/FindTensorflow.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake/FindTensorflow.cmake -------------------------------------------------------------------------------- /cmake/IncludeWithWatch.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake/IncludeWithWatch.cmake -------------------------------------------------------------------------------- /cmake/PrintPythonModule.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake/PrintPythonModule.cmake -------------------------------------------------------------------------------- /cmake/find-tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/cmake/find-tensorflow.py -------------------------------------------------------------------------------- /config/tn_build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/config/tn_build.yaml -------------------------------------------------------------------------------- /configure.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/configure.sh -------------------------------------------------------------------------------- /core/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/BUILD -------------------------------------------------------------------------------- /core/kernels/bn_table_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/bn_table_ops.cc -------------------------------------------------------------------------------- /core/kernels/bn_table_ops_dummy.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/bn_table_ops_dummy.cc -------------------------------------------------------------------------------- /core/kernels/data/balance_dataset_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/data/balance_dataset_ops.cc -------------------------------------------------------------------------------- /core/kernels/data/balance_dataset_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/data/balance_dataset_ops.h -------------------------------------------------------------------------------- /core/kernels/data/balance_dataset_ops_dummy.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/data/balance_dataset_ops_dummy.cc -------------------------------------------------------------------------------- /core/kernels/data/balance_dataset_ops_dummy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/data/balance_dataset_ops_dummy.h -------------------------------------------------------------------------------- /core/kernels/dense_table_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/dense_table_ops.cc -------------------------------------------------------------------------------- /core/kernels/dense_table_ops_dummy.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/dense_table_ops_dummy.cc -------------------------------------------------------------------------------- /core/kernels/resource_var_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/resource_var_wrapper.h -------------------------------------------------------------------------------- /core/kernels/sparse_table_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/sparse_table_ops.cc -------------------------------------------------------------------------------- /core/kernels/sparse_table_ops_dummy.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/kernels/sparse_table_ops_dummy.cc -------------------------------------------------------------------------------- /core/main/py_wrapper.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/main/py_wrapper.cc -------------------------------------------------------------------------------- /core/ops/balance_dataset_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ops/balance_dataset_ops.cc -------------------------------------------------------------------------------- /core/ops/bn_table_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ops/bn_table_ops.cc -------------------------------------------------------------------------------- /core/ops/dense_table_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ops/dense_table_ops.cc -------------------------------------------------------------------------------- /core/ops/sparse_table_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ops/sparse_table_ops.cc -------------------------------------------------------------------------------- /core/ps/optimizer/ada_grad_kernel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/ada_grad_kernel.cc -------------------------------------------------------------------------------- /core/ps/optimizer/ada_grad_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/ada_grad_kernel.h -------------------------------------------------------------------------------- /core/ps/optimizer/adam_kernel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/adam_kernel.cc -------------------------------------------------------------------------------- /core/ps/optimizer/adam_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/adam_kernel.h -------------------------------------------------------------------------------- /core/ps/optimizer/data_struct.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/data_struct.cc -------------------------------------------------------------------------------- /core/ps/optimizer/data_struct.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/data_struct.h -------------------------------------------------------------------------------- /core/ps/optimizer/ftrl_kernel.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/ftrl_kernel.cc -------------------------------------------------------------------------------- /core/ps/optimizer/ftrl_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/ftrl_kernel.h -------------------------------------------------------------------------------- /core/ps/optimizer/optimizer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/optimizer.cc -------------------------------------------------------------------------------- /core/ps/optimizer/optimizer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/optimizer.h -------------------------------------------------------------------------------- /core/ps/optimizer/optimizer_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/optimizer/optimizer_kernel.h -------------------------------------------------------------------------------- /core/ps/ps_cluster.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_cluster.cc -------------------------------------------------------------------------------- /core/ps/ps_cluster.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_cluster.h -------------------------------------------------------------------------------- /core/ps/ps_local_server.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_local_server.cc -------------------------------------------------------------------------------- /core/ps/ps_local_server.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_local_server.h -------------------------------------------------------------------------------- /core/ps/ps_remote_server.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_remote_server.cc -------------------------------------------------------------------------------- /core/ps/ps_remote_server.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_remote_server.h -------------------------------------------------------------------------------- /core/ps/ps_server_interface.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_server_interface.h -------------------------------------------------------------------------------- /core/ps/ps_service_impl.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_service_impl.cc -------------------------------------------------------------------------------- /core/ps/ps_service_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/ps_service_impl.h -------------------------------------------------------------------------------- /core/ps/table/bn_table.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/table/bn_table.cc -------------------------------------------------------------------------------- /core/ps/table/bn_table.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/table/bn_table.h -------------------------------------------------------------------------------- /core/ps/table/dense_table.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/table/dense_table.cc -------------------------------------------------------------------------------- /core/ps/table/dense_table.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/table/dense_table.h -------------------------------------------------------------------------------- /core/ps/table/sparse_table.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/table/sparse_table.cc -------------------------------------------------------------------------------- /core/ps/table/sparse_table.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps/table/sparse_table.h -------------------------------------------------------------------------------- /core/ps_interface/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps_interface/BUILD -------------------------------------------------------------------------------- /core/ps_interface/ps_raw_interface.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps_interface/ps_raw_interface.h -------------------------------------------------------------------------------- /core/ps_interface/ps_server.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/ps_interface/ps_server.proto -------------------------------------------------------------------------------- /core/public/version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/public/version.h -------------------------------------------------------------------------------- /core/utility/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/BUILD -------------------------------------------------------------------------------- /core/utility/allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/allocator.h -------------------------------------------------------------------------------- /core/utility/file_io.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/file_io.cc -------------------------------------------------------------------------------- /core/utility/file_io.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/file_io.h -------------------------------------------------------------------------------- /core/utility/fix_redef.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/fix_redef.h -------------------------------------------------------------------------------- /core/utility/mpi_manager.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/mpi_manager.cc -------------------------------------------------------------------------------- /core/utility/mpi_manager.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/mpi_manager.h -------------------------------------------------------------------------------- /core/utility/net_util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/net_util.cc -------------------------------------------------------------------------------- /core/utility/net_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/net_util.h -------------------------------------------------------------------------------- /core/utility/random.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/random.h -------------------------------------------------------------------------------- /core/utility/semaphore.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/core/utility/semaphore.h -------------------------------------------------------------------------------- /doc/async-arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/async-arch.png -------------------------------------------------------------------------------- /doc/compile_deploy.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/compile_deploy.md -------------------------------------------------------------------------------- /doc/inference.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/inference.png -------------------------------------------------------------------------------- /doc/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/logo.png -------------------------------------------------------------------------------- /doc/tensornet-wide-deep.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tensornet-wide-deep.png -------------------------------------------------------------------------------- /doc/tf-wide-deep.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tf-wide-deep.png -------------------------------------------------------------------------------- /doc/tutorial/.gitignore: -------------------------------------------------------------------------------- 1 | .ipynb_checkpoints/ 2 | -------------------------------------------------------------------------------- /doc/tutorial/01-begin-with-wide-deep.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tutorial/01-begin-with-wide-deep.ipynb -------------------------------------------------------------------------------- /doc/tutorial/02-run-in-cluster.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tutorial/02-run-in-cluster.ipynb -------------------------------------------------------------------------------- /doc/tutorial/03-split-to-sub-graph.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tutorial/03-split-to-sub-graph.ipynb -------------------------------------------------------------------------------- /doc/tutorial/04-deploy-tf-graph-online.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tutorial/04-deploy-tf-graph-online.ipynb -------------------------------------------------------------------------------- /doc/tutorial/05-export-sparse-feature-embedding.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tutorial/05-export-sparse-feature-embedding.ipynb -------------------------------------------------------------------------------- /doc/tutorial/pic/wide-deep.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/doc/tutorial/pic/wide-deep.png -------------------------------------------------------------------------------- /examples/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/.gitignore -------------------------------------------------------------------------------- /examples/common/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/common/config.py -------------------------------------------------------------------------------- /examples/common/feature_column.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/common/feature_column.py -------------------------------------------------------------------------------- /examples/common/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/common/layers.py -------------------------------------------------------------------------------- /examples/common/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/common/util.py -------------------------------------------------------------------------------- /examples/gen_example_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/gen_example_data.py -------------------------------------------------------------------------------- /examples/integration-test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/integration-test.py -------------------------------------------------------------------------------- /examples/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/main.py -------------------------------------------------------------------------------- /examples/models/deep_cross.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/models/deep_cross.py -------------------------------------------------------------------------------- /examples/models/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/models/deepfm.py -------------------------------------------------------------------------------- /examples/models/wide_deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/models/wide_deep.py -------------------------------------------------------------------------------- /examples/online_serving/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/online_serving/.gitignore -------------------------------------------------------------------------------- /examples/online_serving/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/online_serving/BUILD -------------------------------------------------------------------------------- /examples/online_serving/graph.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/online_serving/graph.cc -------------------------------------------------------------------------------- /examples/online_serving/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/online_serving/main.cc -------------------------------------------------------------------------------- /examples/online_serving/test_env/.gitignore: -------------------------------------------------------------------------------- 1 | libmodel.so 2 | tf_serving 3 | -------------------------------------------------------------------------------- /examples/online_serving/test_env/data/feature.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/examples/online_serving/test_env/data/feature.data -------------------------------------------------------------------------------- /examples/online_serving/test_env/data/slot.data: -------------------------------------------------------------------------------- 1 | 1,2,3,4 2 | -------------------------------------------------------------------------------- /manager: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/manager -------------------------------------------------------------------------------- /pixi.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/pixi.lock -------------------------------------------------------------------------------- /pixiw: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/pixiw -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/setup.py -------------------------------------------------------------------------------- /tensornet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/__init__.py -------------------------------------------------------------------------------- /tensornet/_opentelemetry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/_opentelemetry.py -------------------------------------------------------------------------------- /tensornet/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/callbacks/__init__.py -------------------------------------------------------------------------------- /tensornet/callbacks/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/callbacks/callbacks.py -------------------------------------------------------------------------------- /tensornet/core/.gitignore: -------------------------------------------------------------------------------- 1 | _pywrap_tn.so 2 | -------------------------------------------------------------------------------- /tensornet/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/core/__init__.py -------------------------------------------------------------------------------- /tensornet/core/gen_balance_dataset_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/core/gen_balance_dataset_ops.py -------------------------------------------------------------------------------- /tensornet/core/gen_bn_table_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/core/gen_bn_table_ops.py -------------------------------------------------------------------------------- /tensornet/core/gen_dense_table_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/core/gen_dense_table_ops.py -------------------------------------------------------------------------------- /tensornet/core/gen_sparse_table_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/core/gen_sparse_table_ops.py -------------------------------------------------------------------------------- /tensornet/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/data/__init__.py -------------------------------------------------------------------------------- /tensornet/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/data/dataset.py -------------------------------------------------------------------------------- /tensornet/distribute/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/distribute/__init__.py -------------------------------------------------------------------------------- /tensornet/distribute/ps_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/distribute/ps_strategy.py -------------------------------------------------------------------------------- /tensornet/feature_column/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/feature_column/__init__.py -------------------------------------------------------------------------------- /tensornet/feature_column/category_column.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/feature_column/category_column.py -------------------------------------------------------------------------------- /tensornet/feature_column/sequence_category_column.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/feature_column/sequence_category_column.py -------------------------------------------------------------------------------- /tensornet/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/layers/__init__.py -------------------------------------------------------------------------------- /tensornet/layers/embedding_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/layers/embedding_features.py -------------------------------------------------------------------------------- /tensornet/layers/normalization_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/layers/normalization_layer.py -------------------------------------------------------------------------------- /tensornet/layers/position_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/layers/position_mapping.py -------------------------------------------------------------------------------- /tensornet/layers/sequence_embedding_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/layers/sequence_embedding_features.py -------------------------------------------------------------------------------- /tensornet/metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/metric/__init__.py -------------------------------------------------------------------------------- /tensornet/metric/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/metric/metrics.py -------------------------------------------------------------------------------- /tensornet/model/Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/model/Model.py -------------------------------------------------------------------------------- /tensornet/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/model/__init__.py -------------------------------------------------------------------------------- /tensornet/model/compile_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/model/compile_utils.py -------------------------------------------------------------------------------- /tensornet/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/optimizer/__init__.py -------------------------------------------------------------------------------- /tensornet/optimizer/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/optimizer/optimizer.py -------------------------------------------------------------------------------- /tensornet/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet/version.py -------------------------------------------------------------------------------- /tensornet_tools/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/LICENSE -------------------------------------------------------------------------------- /tensornet_tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/README.md -------------------------------------------------------------------------------- /tensornet_tools/bin/tn_tools.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/bin/tn_tools.sh -------------------------------------------------------------------------------- /tensornet_tools/config/tn_tool_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/config/tn_tool_env.yaml -------------------------------------------------------------------------------- /tensornet_tools/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/pyproject.toml -------------------------------------------------------------------------------- /tensornet_tools/python/merge_extra_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/python/merge_extra_embedding.py -------------------------------------------------------------------------------- /tensornet_tools/python/merge_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/python/merge_sparse.py -------------------------------------------------------------------------------- /tensornet_tools/python/resize_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/python/resize_dense.py -------------------------------------------------------------------------------- /tensornet_tools/python/resize_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/python/resize_sparse.py -------------------------------------------------------------------------------- /tensornet_tools/python/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tensornet_tools/python/utils.py -------------------------------------------------------------------------------- /test/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/CMakeLists.txt -------------------------------------------------------------------------------- /test/ps/optimizer/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/ps/optimizer/BUILD -------------------------------------------------------------------------------- /test/ps/optimizer/optimizer_kernel_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/ps/optimizer/optimizer_kernel_test.cc -------------------------------------------------------------------------------- /test/ps/table/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/ps/table/BUILD -------------------------------------------------------------------------------- /test/ps/table/sparse_table_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/ps/table/sparse_table_test.cc -------------------------------------------------------------------------------- /test/utility/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/utility/BUILD -------------------------------------------------------------------------------- /test/utility/allocator_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/test/utility/allocator_test.cc -------------------------------------------------------------------------------- /tf_gen_op_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/tf_gen_op_wrapper.py -------------------------------------------------------------------------------- /thirdparty/openmpi/.gitignore: -------------------------------------------------------------------------------- 1 | include 2 | lib 3 | -------------------------------------------------------------------------------- /thirdparty/openmpi/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/openmpi/BUILD -------------------------------------------------------------------------------- /thirdparty/patches/01-fix_dl_sym.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/patches/01-fix_dl_sym.patch -------------------------------------------------------------------------------- /thirdparty/patches/02-fix-brpc-compile.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/patches/02-fix-brpc-compile.patch -------------------------------------------------------------------------------- /thirdparty/patches/03-fix-brpc-cxx-flags.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/patches/03-fix-brpc-cxx-flags.patch -------------------------------------------------------------------------------- /thirdparty/patches/04-boringssl.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/patches/04-boringssl.patch -------------------------------------------------------------------------------- /thirdparty/patches/05-disable-install-brpc.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/patches/05-disable-install-brpc.patch -------------------------------------------------------------------------------- /thirdparty/patches/BUILD: -------------------------------------------------------------------------------- 1 | exports_files(["*.patch"]) 2 | -------------------------------------------------------------------------------- /thirdparty/tensorflow/.gitignore: -------------------------------------------------------------------------------- 1 | include 2 | lib 3 | -------------------------------------------------------------------------------- /thirdparty/tensorflow/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qihoo360/tensornet/HEAD/thirdparty/tensorflow/BUILD --------------------------------------------------------------------------------