├── .bazelignore ├── .bazelrc ├── .bazelversion ├── .clang-format ├── .github ├── ISSUE_TEMPLATE │ ├── tensorflow_issue_template.yaml │ ├── tflite-converter-issue.md │ ├── tflite-in-play-services.md │ ├── tflite-op-request.md │ └── tflite-other.md ├── bot_config.yml └── workflows │ ├── arm-cd.yml │ ├── arm-ci-extended-cpp.yml │ ├── arm-ci-extended.yml │ ├── arm-ci.yml │ ├── cffconvert.yml │ ├── create_issue.js │ ├── issue-on-pr-rollback.yml │ ├── osv-scanner-scheduled.yml │ ├── pylint-presubmit.yml │ ├── release-branch-cherrypick.yml │ ├── scorecards-analysis.yml │ ├── sigbuild-docker-branch.yml │ ├── sigbuild-docker-presubmit.yml │ ├── sigbuild-docker.yml │ ├── stale-issues.yml │ ├── trusted-partners.yml │ ├── trusted_partners.js │ ├── update-nightly.yml │ └── update-rbe.yml ├── .gitignore ├── .pylintrc ├── .zenodo.json ├── AUTHORS ├── BUILD ├── CITATION.cff ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── ISSUES.md ├── ISSUE_TEMPLATE.md ├── LICENSE ├── README.md ├── RELEASE.md ├── SECURITY.md ├── WORKSPACE ├── arm_compiler.BUILD ├── ci ├── README.md ├── devinfra │ ├── README.md │ └── docker_windows │ │ └── Dockerfile └── official │ ├── README.md │ ├── any.sh │ ├── bisect.sh │ ├── code_check_changed_files.sh │ ├── code_check_full.sh │ ├── containers │ └── linux_arm64 │ │ ├── .gitignore │ │ ├── Dockerfile │ │ ├── apt.conf │ │ ├── build.sh │ │ ├── builder.devtoolset │ │ ├── build_devtoolset.sh │ │ ├── fixlinks_aarch64.sh │ │ ├── gcc9-fixups.patch │ │ ├── rpm-patch.sh │ │ └── stringop_trunc.patch │ │ ├── builder.packages.txt │ │ ├── builder.patchelf │ │ └── build_patchelf.sh │ │ ├── cuda.packages.txt │ │ ├── devel.bashrc │ │ ├── devel.packages.txt │ │ ├── devel.requirements.txt │ │ ├── devel.usertools │ │ ├── aarch64.bazelrc │ │ ├── aarch64_clang.bazelrc │ │ ├── code_check_changed_files.bats │ │ ├── code_check_full.bats │ │ ├── get_test_list.sh │ │ ├── rename_and_verify_wheels.sh │ │ ├── repack_libtensorflow.sh │ │ ├── setup_venv_test.sh │ │ ├── squash_testlogs.py │ │ ├── test.requirements.txt │ │ └── wheel_verification.bats │ │ ├── jax.requirements.txt │ │ ├── ld.so.conf │ │ ├── setup.packages.sh │ │ ├── setup.python.sh │ │ └── setup.sources.sh │ ├── debug_tfci.sh │ ├── envs │ ├── ci_default │ ├── ci_nightly_uploads │ ├── continuous_linux_arm64_cpu_py310 │ ├── continuous_linux_arm64_cpu_py311 │ ├── continuous_linux_arm64_cpu_py311_cross_compile │ ├── continuous_linux_arm64_cpu_py39 │ ├── continuous_linux_arm64_cpu_py39_cross_compile │ ├── continuous_linux_x86_cpu_py310 │ ├── continuous_linux_x86_cpu_py311 │ ├── continuous_linux_x86_cpu_py39 │ ├── continuous_linux_x86_cuda_py310 │ ├── continuous_linux_x86_cuda_py311 │ ├── continuous_linux_x86_cuda_py39 │ ├── continuous_macos_arm64_py310 │ ├── continuous_macos_arm64_py311 │ ├── continuous_macos_arm64_py39 │ ├── disable_all_uploads │ ├── nightly_libtensorflow_linux_x86_cpu │ ├── nightly_libtensorflow_linux_x86_cuda │ ├── nightly_libtensorflow_macos_arm64 │ ├── nightly_linux_arm64_cpu_py310 │ ├── nightly_linux_arm64_cpu_py311 │ ├── nightly_linux_arm64_cpu_py312 │ ├── nightly_linux_arm64_cpu_py39 │ ├── nightly_linux_x86_cpu_py310 │ ├── nightly_linux_x86_cpu_py311 │ ├── nightly_linux_x86_cpu_py312 │ ├── nightly_linux_x86_cpu_py39 │ ├── nightly_linux_x86_cuda_py310 │ ├── nightly_linux_x86_cuda_py311 │ ├── nightly_linux_x86_cuda_py312 │ ├── nightly_linux_x86_cuda_py39 │ ├── nightly_linux_x86_tpu_py310 │ ├── nightly_linux_x86_tpu_py311 │ ├── nightly_linux_x86_tpu_py312 │ ├── nightly_linux_x86_tpu_py39 │ ├── nightly_macos_arm64_py310 │ ├── nightly_macos_arm64_py311 │ ├── nightly_macos_arm64_py312 │ ├── nightly_macos_arm64_py39 │ └── sample │ ├── libtensorflow.sh │ ├── pycpp.sh │ ├── requirements_updater │ ├── BUILD.bazel │ ├── README.md │ ├── WORKSPACE │ ├── release_updater.sh │ ├── requirements.in │ ├── updater.sh │ └── updater_config_repository.bzl │ ├── utilities │ ├── cleanup_docker.sh │ ├── cleanup_summary.sh │ ├── code_check_changed_files.bats │ ├── code_check_full.bats │ ├── docker.sh │ ├── extract_resultstore_links.py │ ├── generate_index_html.sh │ ├── rename_and_verify_wheels.sh │ ├── repack_libtensorflow.sh │ ├── setup.sh │ └── setup_macos.sh │ ├── wheel.sh │ └── wheel_test │ ├── BUILD │ ├── README.md │ ├── WORKSPACE │ ├── test_import_api_packages.py │ └── update_requirements.sh ├── configure ├── configure.cmd ├── configure.py ├── models.BUILD ├── requirements_lock_3_10.txt ├── requirements_lock_3_11.txt ├── requirements_lock_3_12.txt ├── requirements_lock_3_9.txt ├── tensorflow ├── .clang-format ├── BUILD ├── __init__.py ├── api_template.__init__.py ├── api_template_v1.__init__.py ├── build_cleaner_spec.textproto ├── c │ ├── BUILD │ ├── README.md │ ├── c_api.cc │ ├── c_api.h │ ├── c_api_experimental.cc │ ├── c_api_experimental.h │ ├── c_api_experimental_test.cc │ ├── c_api_function.cc │ ├── c_api_function_test.cc │ ├── c_api_internal.h │ ├── c_api_macros.h │ ├── c_api_macros_internal.h │ ├── c_api_test.cc │ ├── c_op_requires.h │ ├── c_test.c │ ├── c_test_util.cc │ ├── c_test_util.h │ ├── checkpoint_reader.cc │ ├── checkpoint_reader.h │ ├── conversion_macros.h │ ├── eager │ │ ├── BUILD │ │ ├── abstract_context.h │ │ ├── abstract_function.h │ │ ├── abstract_op_attrs.h │ │ ├── abstract_operation.h │ │ ├── abstract_tensor_handle.cc │ │ ├── abstract_tensor_handle.h │ │ ├── c_api.cc │ │ ├── c_api.h │ │ ├── c_api_cluster_test.cc │ │ ├── c_api_debug.cc │ │ ├── c_api_debug_test.cc │ │ ├── c_api_distributed_test.cc │ │ ├── c_api_experimental.cc │ │ ├── c_api_experimental.h │ │ ├── c_api_experimental_reader.cc │ │ ├── c_api_experimental_reader.h │ │ ├── c_api_experimental_reader_test.cc │ │ ├── c_api_experimental_test.cc │ │ ├── c_api_internal.h │ │ ├── c_api_remote_function_test.cc │ │ ├── c_api_remote_test.cc │ │ ├── c_api_remote_test_util.cc │ │ ├── c_api_remote_test_util.h │ │ ├── c_api_test.cc │ │ ├── c_api_test_util.cc │ │ ├── c_api_test_util.h │ │ ├── c_api_unified_experimental.cc │ │ ├── c_api_unified_experimental.h │ │ ├── c_api_unified_experimental_eager.cc │ │ ├── c_api_unified_experimental_graph.cc │ │ ├── c_api_unified_experimental_internal.h │ │ ├── c_api_unified_experimental_test.cc │ │ ├── custom_device_test.cc │ │ ├── custom_device_testutil.cc │ │ ├── custom_device_testutil.h │ │ ├── dlpack.cc │ │ ├── dlpack.h │ │ ├── dlpack_test.cc │ │ ├── gradient_checker.cc │ │ ├── gradient_checker.h │ │ ├── gradient_checker_test.cc │ │ ├── gradients.cc │ │ ├── gradients.h │ │ ├── gradients_internal.h │ │ ├── gradients_test.cc │ │ ├── graph_function.cc │ │ ├── graph_function.h │ │ ├── immediate_execution_context.h │ │ ├── immediate_execution_distributed_manager.h │ │ ├── immediate_execution_operation.h │ │ ├── immediate_execution_tensor_handle.cc │ │ ├── immediate_execution_tensor_handle.h │ │ ├── parallel_device │ │ │ ├── BUILD │ │ │ ├── parallel_device.cc │ │ │ ├── parallel_device.h │ │ │ ├── parallel_device_lib.cc │ │ │ ├── parallel_device_lib.h │ │ │ ├── parallel_device_lib_test.cc │ │ │ ├── parallel_device_remote_test.cc │ │ │ ├── parallel_device_test.cc │ │ │ ├── parallel_device_testlib.cc │ │ │ └── parallel_device_testlib.h │ │ ├── tape.h │ │ ├── tfe_cancellation_manager_internal.h │ │ ├── tfe_context_internal.h │ │ ├── tfe_executor_internal.h │ │ ├── tfe_monitoring_internal.h │ │ ├── tfe_monitoring_reader_internal.h │ │ ├── tfe_op_attrs_internal.h │ │ ├── tfe_op_internal.h │ │ ├── tfe_tensor_debug_info_internal.h │ │ ├── tfe_tensorhandle_internal.h │ │ ├── tracing_utils.cc │ │ ├── tracing_utils.h │ │ ├── unified_api_test.cc │ │ ├── unified_api_testutil.cc │ │ └── unified_api_testutil.h │ ├── env.cc │ ├── env.h │ ├── env_test.cc │ ├── experimental │ │ ├── filesystem │ │ │ ├── BUILD │ │ │ ├── filesystem_interface.h │ │ │ ├── modular_filesystem.cc │ │ │ ├── modular_filesystem.h │ │ │ ├── modular_filesystem_registration.cc │ │ │ ├── modular_filesystem_registration.h │ │ │ ├── modular_filesystem_test.cc │ │ │ └── plugins │ │ │ │ ├── gcs │ │ │ │ ├── BUILD │ │ │ │ ├── cleanup.h │ │ │ │ ├── expiring_lru_cache.h │ │ │ │ ├── expiring_lru_cache_test.cc │ │ │ │ ├── gcs_filesystem.cc │ │ │ │ ├── gcs_filesystem.h │ │ │ │ ├── gcs_filesystem_test.cc │ │ │ │ ├── gcs_helper.cc │ │ │ │ ├── gcs_helper.h │ │ │ │ ├── ram_file_block_cache.cc │ │ │ │ ├── ram_file_block_cache.h │ │ │ │ └── ram_file_block_cache_test.cc │ │ │ │ ├── posix │ │ │ │ ├── BUILD │ │ │ │ ├── copy_file.h │ │ │ │ ├── copy_file_linux.cc │ │ │ │ ├── copy_file_portable.cc │ │ │ │ ├── posix_filesystem.cc │ │ │ │ ├── posix_filesystem.h │ │ │ │ ├── posix_filesystem_helper.cc │ │ │ │ ├── posix_filesystem_helper.h │ │ │ │ └── posix_filesystem_static.cc │ │ │ │ └── windows │ │ │ │ ├── BUILD │ │ │ │ └── windows_filesystem.cc │ │ ├── gradients │ │ │ ├── BUILD │ │ │ ├── array_grad.cc │ │ │ ├── array_grad.h │ │ │ ├── array_grad_test.cc │ │ │ ├── custom_gradient_test.cc │ │ │ ├── grad_test_helper.cc │ │ │ ├── grad_test_helper.h │ │ │ ├── math_grad.cc │ │ │ ├── math_grad.h │ │ │ ├── math_grad_test.cc │ │ │ ├── nn_grad.cc │ │ │ ├── nn_grad.h │ │ │ ├── nn_grad_test.cc │ │ │ ├── not_differentiable.cc │ │ │ ├── not_differentiable.h │ │ │ └── tape │ │ │ │ ├── BUILD │ │ │ │ ├── tape_context.cc │ │ │ │ ├── tape_context.h │ │ │ │ ├── tape_operation.cc │ │ │ │ └── tape_operation.h │ │ ├── grappler │ │ │ ├── BUILD │ │ │ ├── grappler.cc │ │ │ ├── grappler.h │ │ │ ├── grappler_internal.h │ │ │ └── grappler_test.cc │ │ ├── next_pluggable_device │ │ │ ├── BUILD │ │ │ ├── c_api.cc │ │ │ ├── c_api.h │ │ │ ├── tensor_pjrt_buffer_util.cc │ │ │ ├── tensor_pjrt_buffer_util.h │ │ │ └── tensor_pjrt_buffer_util_test.cc │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── array_ops.cc │ │ │ ├── array_ops.h │ │ │ ├── gen │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── common │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── case_format.cc │ │ │ │ │ ├── case_format.h │ │ │ │ │ ├── case_format_test.cc │ │ │ │ │ ├── controller.cc │ │ │ │ │ ├── controller.h │ │ │ │ │ ├── path_config.cc │ │ │ │ │ ├── path_config.h │ │ │ │ │ ├── source_code.cc │ │ │ │ │ ├── source_code.h │ │ │ │ │ ├── view_util.cc │ │ │ │ │ └── view_util.h │ │ │ │ ├── cpp │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── cpp_generator.cc │ │ │ │ │ ├── cpp_generator.h │ │ │ │ │ ├── cpp_generator_test.cc │ │ │ │ │ ├── golden │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── testing_ops.cc.golden │ │ │ │ │ │ └── testing_ops.h.golden │ │ │ │ │ ├── renderers │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── cpp_config.cc │ │ │ │ │ │ ├── cpp_config.h │ │ │ │ │ │ ├── cpp_file_renderer.cc │ │ │ │ │ │ ├── cpp_file_renderer.h │ │ │ │ │ │ ├── guard_renderer.cc │ │ │ │ │ │ ├── guard_renderer.h │ │ │ │ │ │ ├── include_renderer.cc │ │ │ │ │ │ ├── include_renderer.h │ │ │ │ │ │ ├── namespace_renderer.cc │ │ │ │ │ │ ├── namespace_renderer.h │ │ │ │ │ │ ├── op_comment_renderer.cc │ │ │ │ │ │ ├── op_comment_renderer.h │ │ │ │ │ │ ├── op_implementation_renderer.cc │ │ │ │ │ │ ├── op_implementation_renderer.h │ │ │ │ │ │ ├── op_renderer.cc │ │ │ │ │ │ ├── op_renderer.h │ │ │ │ │ │ ├── renderer.cc │ │ │ │ │ │ ├── renderer.h │ │ │ │ │ │ ├── renderer_context.h │ │ │ │ │ │ └── renderer_test.cc │ │ │ │ │ └── views │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── arg_type_view.cc │ │ │ │ │ │ ├── arg_type_view.h │ │ │ │ │ │ ├── arg_view.cc │ │ │ │ │ │ ├── arg_view.h │ │ │ │ │ │ ├── attr_view.cc │ │ │ │ │ │ ├── attr_view.h │ │ │ │ │ │ ├── op_argument_view.cc │ │ │ │ │ │ ├── op_argument_view.h │ │ │ │ │ │ ├── op_view.cc │ │ │ │ │ │ └── op_view.h │ │ │ │ ├── generate_cpp_main.cc │ │ │ │ └── model │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── arg_spec.cc │ │ │ │ │ ├── arg_spec.h │ │ │ │ │ ├── arg_type.cc │ │ │ │ │ ├── arg_type.h │ │ │ │ │ ├── attr_spec.cc │ │ │ │ │ ├── attr_spec.h │ │ │ │ │ ├── op_spec.cc │ │ │ │ │ └── op_spec.h │ │ │ ├── io_ops.cc │ │ │ ├── io_ops.h │ │ │ ├── math_ops.cc │ │ │ ├── math_ops.h │ │ │ ├── nn_ops.cc │ │ │ ├── nn_ops.h │ │ │ ├── resource_variable_ops.cc │ │ │ ├── resource_variable_ops.h │ │ │ └── update_cpp_ops.sh │ │ ├── pluggable_profiler │ │ │ ├── BUILD │ │ │ ├── pluggable_profiler.cc │ │ │ ├── pluggable_profiler.h │ │ │ └── pluggable_profiler_internal.h │ │ ├── saved_model │ │ │ ├── README.md │ │ │ ├── core │ │ │ │ ├── BUILD │ │ │ │ ├── concrete_function.h │ │ │ │ ├── constant_loading_test.cc │ │ │ │ ├── function_metadata.h │ │ │ │ ├── object_graph_traversal_test.cc │ │ │ │ ├── ops │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── restore_ops.cc │ │ │ │ │ ├── restore_ops.h │ │ │ │ │ ├── restore_ops_test.cc │ │ │ │ │ ├── variable_ops.cc │ │ │ │ │ ├── variable_ops.h │ │ │ │ │ └── variable_ops_test.cc │ │ │ │ ├── revived_types │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── asset.cc │ │ │ │ │ ├── asset.h │ │ │ │ │ ├── constant.cc │ │ │ │ │ ├── constant.h │ │ │ │ │ ├── flat_tensor_function.cc │ │ │ │ │ ├── flat_tensor_function.h │ │ │ │ │ ├── partially_revived_objects.cc │ │ │ │ │ ├── partially_revived_objects.h │ │ │ │ │ ├── restored_resource.cc │ │ │ │ │ ├── restored_resource.h │ │ │ │ │ ├── restored_resource_revival_state.h │ │ │ │ │ ├── revived_objects.h │ │ │ │ │ ├── tensorhandle_convertible.h │ │ │ │ │ ├── tf_concrete_function.cc │ │ │ │ │ ├── tf_concrete_function.h │ │ │ │ │ ├── tf_concrete_function_revival_state.h │ │ │ │ │ ├── tf_signature_def_function.cc │ │ │ │ │ ├── tf_signature_def_function.h │ │ │ │ │ ├── tf_signature_def_function_revival_state.h │ │ │ │ │ ├── variable.cc │ │ │ │ │ └── variable.h │ │ │ │ ├── saved_model_api.h │ │ │ │ ├── saved_model_utils.cc │ │ │ │ ├── saved_model_utils.h │ │ │ │ ├── saved_variable_loading_test.cc │ │ │ │ ├── signature_def_function.h │ │ │ │ ├── signature_def_function_metadata.cc │ │ │ │ ├── signature_def_function_metadata.h │ │ │ │ ├── signature_flattening_test.cc │ │ │ │ ├── tensor_spec.cc │ │ │ │ ├── tensor_spec.h │ │ │ │ ├── test_utils.cc │ │ │ │ ├── test_utils.h │ │ │ │ ├── tf_concrete_function_loading_test.cc │ │ │ │ ├── tf_concrete_function_test_protos.cc │ │ │ │ ├── tf_concrete_function_test_protos.h │ │ │ │ ├── tf_saved_model_api.cc │ │ │ │ └── tf_saved_model_api.h │ │ │ ├── internal │ │ │ │ ├── BUILD │ │ │ │ ├── concrete_function.cc │ │ │ │ ├── concrete_function_list.cc │ │ │ │ ├── concrete_function_list_type.h │ │ │ │ ├── concrete_function_type.h │ │ │ │ ├── function_metadata.cc │ │ │ │ ├── function_metadata_type.h │ │ │ │ ├── saved_model_api.cc │ │ │ │ ├── saved_model_api_test.cc │ │ │ │ ├── saved_model_api_type.h │ │ │ │ ├── signature_def_function.cc │ │ │ │ ├── signature_def_function_metadata.cc │ │ │ │ ├── signature_def_function_metadata_type.h │ │ │ │ ├── signature_def_function_type.h │ │ │ │ ├── signature_def_param.cc │ │ │ │ ├── signature_def_param_list.cc │ │ │ │ ├── signature_def_param_list_type.h │ │ │ │ ├── signature_def_param_type.h │ │ │ │ ├── tensor_spec.cc │ │ │ │ ├── tensor_spec_type.h │ │ │ │ └── testdata │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── SimpleWhileLoop │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ └── variables.index │ │ │ │ │ ├── UninitializedVariable │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ └── variables.index │ │ │ │ │ └── gen_saved_models.py │ │ │ └── public │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── c_saved_model_api.h │ │ │ │ ├── concrete_function.h │ │ │ │ ├── concrete_function_list.h │ │ │ │ ├── function_metadata.h │ │ │ │ ├── saved_model_api.h │ │ │ │ ├── signature_def_function.h │ │ │ │ ├── signature_def_function_metadata.h │ │ │ │ ├── signature_def_param.h │ │ │ │ ├── signature_def_param_list.h │ │ │ │ └── tensor_spec.h │ │ └── stream_executor │ │ │ ├── BUILD │ │ │ ├── stream_executor.cc │ │ │ ├── stream_executor.h │ │ │ ├── stream_executor_internal.h │ │ │ ├── stream_executor_test.cc │ │ │ ├── stream_executor_test_util.cc │ │ │ ├── stream_executor_test_util.h │ │ │ └── test │ │ │ ├── BUILD │ │ │ └── test_pluggable_device.cc │ ├── exported_symbols.lds │ ├── generate-pc.sh │ ├── kernels.cc │ ├── kernels.h │ ├── kernels │ │ ├── BUILD │ │ ├── bitcast_op.cc │ │ ├── bitcast_op_test.cc │ │ ├── histogram_summary_op.cc │ │ ├── merge_summary_op.cc │ │ ├── ops │ │ │ ├── bitcast.cc │ │ │ ├── histogram_summary.cc │ │ │ ├── merge_summary.cc │ │ │ └── summary.cc │ │ ├── summary_op.cc │ │ ├── summary_op_benchmark_test.cc │ │ ├── summary_op_test.cc │ │ ├── tensor_shape_utils.cc │ │ ├── tensor_shape_utils.h │ │ └── tensor_shape_utils_test.cc │ ├── kernels_experimental.cc │ ├── kernels_experimental.h │ ├── kernels_test.cc │ ├── logging.cc │ ├── logging.h │ ├── ops.cc │ ├── ops.h │ ├── ops_test.cc │ ├── python_api.cc │ ├── python_api.h │ ├── safe_ptr.cc │ ├── safe_ptr.h │ ├── tensor_interface.h │ ├── test_op.cc │ ├── test_op1.cc │ ├── testdata │ │ └── tf_record │ ├── tf_attrtype.h │ ├── tf_buffer.cc │ ├── tf_buffer.h │ ├── tf_buffer_internal.h │ ├── tf_datatype.cc │ ├── tf_datatype.h │ ├── tf_file_statistics.h │ ├── tf_shape.cc │ ├── tf_shape.h │ ├── tf_shape_internal.h │ ├── tf_status.cc │ ├── tf_status.h │ ├── tf_status_helper.cc │ ├── tf_status_helper.h │ ├── tf_status_helper_test.cc │ ├── tf_status_internal.h │ ├── tf_tensor.cc │ ├── tf_tensor.h │ ├── tf_tensor_helper.h │ ├── tf_tensor_internal.h │ ├── tf_tstring.cc │ ├── tf_tstring.h │ ├── version_script.lds │ └── while_loop_test.cc ├── cc │ ├── BUILD │ ├── client │ │ ├── client_session.cc │ │ ├── client_session.h │ │ └── client_session_test.cc │ ├── experimental │ │ ├── base │ │ │ ├── public │ │ │ │ ├── BUILD │ │ │ │ ├── runtime.h │ │ │ │ ├── runtime_builder.h │ │ │ │ ├── status.h │ │ │ │ ├── tensor.h │ │ │ │ └── tensorhandle.h │ │ │ └── tests │ │ │ │ ├── BUILD │ │ │ │ ├── tensor_test.cc │ │ │ │ ├── tensor_types_test_util.h │ │ │ │ └── tensorhandle_test.cc │ │ ├── libexport │ │ │ ├── BUILD │ │ │ ├── load.cc │ │ │ ├── load.h │ │ │ ├── load_test.cc │ │ │ ├── save.cc │ │ │ ├── save.h │ │ │ └── save_test.cc │ │ └── libtf │ │ │ ├── BUILD │ │ │ ├── function.cc │ │ │ ├── function.h │ │ │ ├── impl │ │ │ ├── BUILD │ │ │ ├── iostream.cc │ │ │ ├── iostream_test.cc │ │ │ ├── none.cc │ │ │ ├── none.h │ │ │ ├── none_test.cc │ │ │ ├── scalars.h │ │ │ ├── scalars_test.cc │ │ │ ├── string.cc │ │ │ ├── string.h │ │ │ ├── string_test.cc │ │ │ ├── tensor_spec.h │ │ │ └── tensor_spec_test.cc │ │ │ ├── mlir │ │ │ ├── BUILD │ │ │ ├── mlir_transform.cc │ │ │ └── mlir_transform.h │ │ │ ├── module.cc │ │ │ ├── module.h │ │ │ ├── object.cc │ │ │ ├── object.h │ │ │ ├── runtime │ │ │ ├── BUILD │ │ │ ├── core │ │ │ │ ├── BUILD │ │ │ │ ├── core.cc │ │ │ │ └── core.h │ │ │ ├── runtime.cc │ │ │ └── runtime.h │ │ │ ├── tests │ │ │ ├── function_test.cc │ │ │ ├── generate_testdata.py │ │ │ ├── mlir_transform_test.cc │ │ │ ├── module_test.cc │ │ │ ├── object_test.cc │ │ │ ├── perf_test.cc │ │ │ ├── runtime_test.cc │ │ │ ├── runtime_test.h │ │ │ ├── runtime_test_core.cc │ │ │ ├── tensor_test.cc │ │ │ ├── testdata │ │ │ │ ├── README │ │ │ │ ├── data-structure-model │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ └── variables.index │ │ │ │ └── simple-model │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── value_test.cc │ │ │ ├── variable_test.cc │ │ │ └── visit_test.cc │ │ │ ├── value.h │ │ │ └── value_iostream.h │ ├── framework │ │ ├── cc_op_gen.cc │ │ ├── cc_op_gen.h │ │ ├── cc_op_gen_main.cc │ │ ├── cc_op_gen_test.cc │ │ ├── cc_op_gen_util.cc │ │ ├── cc_op_gen_util.h │ │ ├── cc_ops_test.cc │ │ ├── fuzzing │ │ │ ├── BUILD │ │ │ ├── cc_op_fuzz_gen.cc │ │ │ ├── cc_op_fuzz_gen.h │ │ │ ├── cc_op_fuzz_gen_main.cc │ │ │ └── op_fuzzing.bzl │ │ ├── grad_op_registry.cc │ │ ├── grad_op_registry.h │ │ ├── gradient_checker.cc │ │ ├── gradient_checker.h │ │ ├── gradient_checker_test.cc │ │ ├── gradients.cc │ │ ├── gradients.h │ │ ├── gradients_test.cc │ │ ├── ops.cc │ │ ├── ops.h │ │ ├── scope.cc │ │ ├── scope.h │ │ ├── scope_internal.h │ │ ├── scope_test.cc │ │ ├── test_op.cc │ │ ├── testutil.cc │ │ ├── testutil.h │ │ ├── while_gradients.cc │ │ ├── while_gradients.h │ │ └── while_gradients_test.cc │ ├── gradients │ │ ├── README.md │ │ ├── array_grad.cc │ │ ├── array_grad_test.cc │ │ ├── data_flow_grad.cc │ │ ├── data_flow_grad_test.cc │ │ ├── functional_grad.cc │ │ ├── functional_grad_test.cc │ │ ├── grad_helper.cc │ │ ├── grad_helper.h │ │ ├── grad_testutil.cc │ │ ├── grad_testutil.h │ │ ├── image_grad.cc │ │ ├── image_grad_test.cc │ │ ├── linalg_grad.cc │ │ ├── linalg_grad_test.cc │ │ ├── manip_grad.cc │ │ ├── manip_grad_test.cc │ │ ├── math_grad.cc │ │ ├── math_grad_test.cc │ │ ├── nn_grad.cc │ │ ├── nn_grad_test.cc │ │ ├── resource_variable_grad.cc │ │ └── resource_variable_grad_test.cc │ ├── ops │ │ ├── const_op.cc │ │ ├── const_op.h │ │ ├── const_op_test.cc │ │ ├── standard_ops.h │ │ ├── while_loop.cc │ │ ├── while_loop.h │ │ └── while_loop_test.cc │ ├── saved_model │ │ ├── BUILD │ │ ├── README.md │ │ ├── bundle_v2.cc │ │ ├── bundle_v2.h │ │ ├── bundle_v2_test.cc │ │ ├── constants.h │ │ ├── experimental │ │ │ ├── public │ │ │ │ ├── BUILD │ │ │ │ ├── concrete_function.h │ │ │ │ ├── concrete_function_list.h │ │ │ │ ├── function_metadata.h │ │ │ │ ├── saved_model_api.h │ │ │ │ ├── signature_def_function.h │ │ │ │ └── signature_def_function_metadata.h │ │ │ └── tests │ │ │ │ ├── BUILD │ │ │ │ └── saved_model_api_test.cc │ │ ├── fingerprinting.cc │ │ ├── fingerprinting.h │ │ ├── fingerprinting_chunked_test.cc │ │ ├── fingerprinting_test.cc │ │ ├── fingerprinting_utils.cc │ │ ├── fingerprinting_utils.h │ │ ├── fingerprinting_utils_test.cc │ │ ├── image_format │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── internal_api.cc │ │ │ └── internal_api.h │ │ ├── loader.cc │ │ ├── loader.h │ │ ├── loader_util.cc │ │ ├── loader_util.h │ │ ├── metrics.cc │ │ ├── metrics.h │ │ ├── metrics_test.cc │ │ ├── python │ │ │ ├── BUILD │ │ │ └── loader.clif │ │ ├── reader.cc │ │ ├── reader.h │ │ ├── reader_test.cc │ │ ├── saved_model_bundle_lite_test.cc │ │ ├── saved_model_bundle_test.cc │ │ ├── saved_model_fuzz.cc │ │ ├── signature_constants.h │ │ ├── tag_constants.h │ │ ├── test_utils.h │ │ ├── testdata │ │ │ ├── AssetModule │ │ │ │ ├── assets │ │ │ │ │ └── test_asset.txt │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── CyclicModule │ │ │ │ ├── debug │ │ │ │ │ └── saved_model_debug_info.pb │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── OptimizerSlotVariableModule │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── SimpleV1Model │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── StaticHashTableModule │ │ │ │ ├── assets │ │ │ │ │ └── static_hashtable_asset.txt │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── VarsAndArithmeticObjectGraph │ │ │ │ ├── debug │ │ │ │ │ └── saved_model_debug_info.pb │ │ │ │ ├── fingerprint.pb │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── bert1 │ │ │ │ └── saved_model.pb │ │ │ ├── bert2 │ │ │ │ └── saved_model.pb │ │ │ ├── chunked_saved_model │ │ │ │ ├── chunked_model │ │ │ │ │ ├── saved_model.cpb │ │ │ │ │ └── saved_model.pbtxt │ │ │ │ └── non_chunked_model │ │ │ │ │ ├── fingerprint.pb │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── fuzz_generated │ │ │ │ ├── bad_node_attr │ │ │ │ │ ├── assets │ │ │ │ │ │ └── empty │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ └── variables.index │ │ │ │ ├── const_with_no_value │ │ │ │ │ ├── assets │ │ │ │ │ │ └── empty │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ └── variables.index │ │ │ │ └── negative_shape │ │ │ │ │ ├── assets │ │ │ │ │ └── empty │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── generate_chunked_models.py │ │ │ ├── generate_saved_models.py │ │ │ ├── half_plus_two │ │ │ │ └── 00000123 │ │ │ │ │ ├── assets │ │ │ │ │ └── foo.txt │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── half_plus_two_main_op │ │ │ │ └── 00000123 │ │ │ │ │ ├── assets │ │ │ │ │ └── foo.txt │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── half_plus_two_pbtxt │ │ │ │ └── 00000123 │ │ │ │ │ ├── assets │ │ │ │ │ └── foo.txt │ │ │ │ │ ├── saved_model.pbtxt │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── half_plus_two_v2 │ │ │ │ └── 00000123 │ │ │ │ │ ├── assets │ │ │ │ │ └── foo.txt │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ └── variables │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ └── variables.index │ │ │ ├── static_hashtable_asset.txt │ │ │ ├── test_asset.txt │ │ │ └── x_plus_y_v2_debuginfo │ │ │ │ ├── debug │ │ │ │ └── saved_model_debug_info.pb │ │ │ │ ├── saved_model.pb │ │ │ │ └── variables │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ └── variables.index │ │ ├── util.cc │ │ ├── util.h │ │ └── util_test.cc │ ├── tools │ │ ├── BUILD │ │ ├── freeze_saved_model.cc │ │ ├── freeze_saved_model.h │ │ └── freeze_saved_model_test.cc │ └── training │ │ ├── coordinator.cc │ │ ├── coordinator.h │ │ ├── coordinator_test.cc │ │ ├── queue_runner.cc │ │ ├── queue_runner.h │ │ └── queue_runner_test.cc ├── compat_template.__init__.py ├── compat_template_v1.__init__.py ├── compiler │ ├── aot │ │ ├── BUILD │ │ ├── aot_only_var_handle_op.cc │ │ ├── aot_only_var_handle_op.h │ │ ├── benchmark.cc │ │ ├── benchmark.h │ │ ├── benchmark_main.template │ │ ├── benchmark_test.cc │ │ ├── codegen.cc │ │ ├── codegen.h │ │ ├── codegen_test.cc │ │ ├── codegen_test_h.golden │ │ ├── codegen_test_o.golden │ │ ├── compile.cc │ │ ├── compile.h │ │ ├── embedded_protocol_buffers.cc │ │ ├── embedded_protocol_buffers.h │ │ ├── flags.cc │ │ ├── flags.h │ │ ├── quantize.h │ │ ├── test.cc │ │ ├── test_google.cc │ │ ├── test_graph_tfadd.config.pbtxt │ │ ├── test_graph_tfadd.pbtxt │ │ ├── test_graph_tfunknownop.config.pbtxt │ │ ├── test_graph_tfunknownop.pbtxt │ │ ├── test_graph_tfunknownop2.config.pbtxt │ │ ├── test_graph_tfunknownop3.config.pbtxt │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── bench_graph_tfmatmul.template.pbtxt │ │ │ ├── make_test_graphs.py │ │ │ ├── test_error_message.lit.pbtxt │ │ │ ├── test_error_message.lit.pbtxt.config.pbtxt │ │ │ ├── test_error_message.lit.pbtxt.debug.pbtxt │ │ │ ├── test_error_message.lit.pbtxt.fake_py.debug │ │ │ ├── test_graph_tfadd.config.pbtxt │ │ │ ├── test_graph_tfadd_with_ckpt.config.pbtxt │ │ │ ├── test_graph_tfassert_eq.config.pbtxt │ │ │ ├── test_graph_tfcond.config.pbtxt │ │ │ ├── test_graph_tffunction.config.pbtxt │ │ │ ├── test_graph_tfgather.config.pbtxt │ │ │ ├── test_graph_tfmatmul.config.pbtxt │ │ │ ├── test_graph_tfmatmulandadd.config.pbtxt │ │ │ ├── test_graph_tfsplits.config.pbtxt │ │ │ ├── test_graph_tftop_k.config.pbtxt │ │ │ ├── test_graph_tfvariable.config.pbtxt │ │ │ ├── test_graph_tfvariable_readonly.config.pbtxt │ │ │ ├── test_graph_tfvariable_sequential_updates.config.pbtxt │ │ │ └── tfcompile_test.cc │ │ ├── tfcompile.bzl │ │ └── tfcompile_main.cc │ ├── jit │ │ ├── BUILD │ │ ├── build_xla_ops_pass.cc │ │ ├── build_xla_ops_pass.h │ │ ├── build_xla_ops_pass_test.cc │ │ ├── clone_constants_for_better_clustering.cc │ │ ├── clone_constants_for_better_clustering.h │ │ ├── clone_constants_for_better_clustering_test.cc │ │ ├── cluster_scoping_pass.cc │ │ ├── cluster_scoping_pass.h │ │ ├── cluster_scoping_pass_test.cc │ │ ├── compilability_check_util.cc │ │ ├── compilability_check_util.h │ │ ├── compilability_check_util_test.cc │ │ ├── compilation_passes_test_main.cc │ │ ├── deadness_analysis.cc │ │ ├── deadness_analysis.h │ │ ├── deadness_analysis_internal.h │ │ ├── deadness_analysis_test.cc │ │ ├── defs.cc │ │ ├── defs.h │ │ ├── device_compilation_cache.h │ │ ├── device_compilation_cache_test.cc │ │ ├── device_compilation_cluster_signature.cc │ │ ├── device_compilation_cluster_signature.h │ │ ├── device_compilation_cluster_signature_test.cc │ │ ├── device_compilation_profiler.cc │ │ ├── device_compilation_profiler.h │ │ ├── device_compilation_profiler_test.cc │ │ ├── device_compiler.h │ │ ├── device_compiler_client.cc │ │ ├── device_compiler_client.h │ │ ├── device_compiler_client_test.cc │ │ ├── device_compiler_disable_test.cc │ │ ├── device_compiler_test.cc │ │ ├── device_context_test.cc │ │ ├── device_executable_persistor.h │ │ ├── device_executable_persistor_test.cc │ │ ├── device_util.cc │ │ ├── device_util.h │ │ ├── device_util_test.cc │ │ ├── encapsulate_subgraphs_pass.cc │ │ ├── encapsulate_subgraphs_pass.h │ │ ├── encapsulate_subgraphs_pass_test.cc │ │ ├── encapsulate_util.cc │ │ ├── encapsulate_util.h │ │ ├── encapsulate_util_test.cc │ │ ├── encapsulate_xla_computations_pass.cc │ │ ├── encapsulate_xla_computations_pass.h │ │ ├── encapsulate_xla_computations_pass_test.cc │ │ ├── extract_outside_compilation_pass.cc │ │ ├── extract_outside_compilation_pass.h │ │ ├── extract_outside_compilation_pass_test.cc │ │ ├── flags.cc │ │ ├── flags.h │ │ ├── force_xla_constants_on_host_pass.cc │ │ ├── force_xla_constants_on_host_pass.h │ │ ├── force_xla_constants_on_host_pass_test.cc │ │ ├── get_compiler_ir.cc │ │ ├── get_compiler_ir.h │ │ ├── increase_dynamism_for_auto_jit_pass.cc │ │ ├── increase_dynamism_for_auto_jit_pass.h │ │ ├── increase_dynamism_for_auto_jit_pass_test.cc │ │ ├── jit_compilation_pass_registration.cc │ │ ├── kernels │ │ │ ├── BUILD │ │ │ ├── xla_ops.cc │ │ │ └── xla_ops.h │ │ ├── mark_for_compilation_pass.cc │ │ ├── mark_for_compilation_pass.h │ │ ├── mark_for_compilation_pass_test.cc │ │ ├── mark_for_compilation_pass_test_helper.cc │ │ ├── mark_for_compilation_pass_test_helper.h │ │ ├── node_matchers.cc │ │ ├── node_matchers.h │ │ ├── node_matchers_test.cc │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── xla_ops.cc │ │ │ └── xla_ops_grad.py │ │ ├── partially_decluster_pass.cc │ │ ├── partially_decluster_pass.h │ │ ├── partially_decluster_pass_test.cc │ │ ├── pjrt_base_device.cc │ │ ├── pjrt_base_device.h │ │ ├── pjrt_compile_util.cc │ │ ├── pjrt_compile_util.h │ │ ├── pjrt_compile_util_test.cc │ │ ├── pjrt_device_compiler_client.cc │ │ ├── pjrt_device_compiler_client.h │ │ ├── pjrt_device_context.cc │ │ ├── pjrt_device_context.h │ │ ├── pjrt_tensor_buffer.h │ │ ├── pjrt_tensor_buffer_util.cc │ │ ├── pjrt_tensor_buffer_util.h │ │ ├── pjrt_tensor_buffer_util_test.cc │ │ ├── rearrange_function_argument_pass_test.cc │ │ ├── report_clustering_info_pass.cc │ │ ├── report_clustering_info_pass.h │ │ ├── resource_operation_safety_analysis.cc │ │ ├── resource_operation_safety_analysis.h │ │ ├── resource_operation_safety_analysis_test.cc │ │ ├── shape_inference.cc │ │ ├── shape_inference.h │ │ ├── shape_inference_helpers.cc │ │ ├── shape_inference_helpers.h │ │ ├── shape_inference_test.cc │ │ ├── test_util.cc │ │ ├── test_util.h │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── auto_clustering_test.cc │ │ │ ├── auto_clustering_test_helper.cc │ │ │ ├── auto_clustering_test_helper.h │ │ │ ├── device_compiler_serialize_options_test.cc │ │ │ ├── device_compiler_serialize_test.cc │ │ │ ├── device_compiler_test_helper.cc │ │ │ ├── device_compiler_test_helper.h │ │ │ ├── keras_imagenet_main.golden_summary │ │ │ ├── keras_imagenet_main.pbtxt │ │ │ ├── keras_imagenet_main_graph_mode.golden_summary │ │ │ ├── keras_imagenet_main_graph_mode.pbtxt │ │ │ ├── opens2s_gnmt_mixed_precision.golden_summary │ │ │ └── opens2s_gnmt_mixed_precision.pbtxt.gz │ │ ├── tf_graph_to_hlo_compiler.cc │ │ ├── tf_graph_to_hlo_compiler.h │ │ ├── tf_to_hlo_compiler.h │ │ ├── variable_info.cc │ │ ├── variable_info.h │ │ ├── variable_info_util.cc │ │ ├── variable_info_util.h │ │ ├── xla_activity.proto │ │ ├── xla_activity_listener.cc │ │ ├── xla_activity_listener.h │ │ ├── xla_activity_listener_test.cc │ │ ├── xla_activity_logging_listener.cc │ │ ├── xla_cluster_util.cc │ │ ├── xla_cluster_util.h │ │ ├── xla_cluster_util_test.cc │ │ ├── xla_compilation_cache.proto │ │ ├── xla_compile_on_demand_op.cc │ │ ├── xla_compile_on_demand_op.h │ │ ├── xla_compile_util.cc │ │ ├── xla_compile_util.h │ │ ├── xla_compile_util_test.cc │ │ ├── xla_compiler_options_util.cc │ │ ├── xla_compiler_options_util.h │ │ ├── xla_compiler_options_util_test.cc │ │ ├── xla_cpu_device.cc │ │ ├── xla_device.cc │ │ ├── xla_device.h │ │ ├── xla_device_compiler_client.cc │ │ ├── xla_device_compiler_client.h │ │ ├── xla_device_context.cc │ │ ├── xla_device_context.h │ │ ├── xla_device_ops.cc │ │ ├── xla_device_ops.h │ │ ├── xla_gpu_device.cc │ │ ├── xla_host_recv_device_context.cc │ │ ├── xla_host_recv_device_context.h │ │ ├── xla_host_send_device_context.cc │ │ ├── xla_host_send_device_context.h │ │ ├── xla_host_send_recv_device_context_test.cc │ │ ├── xla_kernel_creator.cc │ │ ├── xla_kernel_creator.h │ │ ├── xla_kernel_creator_test.cc │ │ ├── xla_launch_util.cc │ │ ├── xla_launch_util.h │ │ ├── xla_launch_util_gpu_test.cc │ │ ├── xla_launch_util_test.cc │ │ ├── xla_ops_on_regular_devices.cc │ │ ├── xla_platform_info.cc │ │ ├── xla_platform_info.h │ │ ├── xla_platform_info_test.cc │ │ ├── xla_tensor.cc │ │ ├── xla_tensor.h │ │ ├── xla_tpu_device.cc │ │ └── xla_tpu_device.h │ ├── mlir │ │ ├── BUILD │ │ ├── README.md │ │ ├── g3doc │ │ │ ├── README.md │ │ │ ├── _book.yaml │ │ │ ├── _includes │ │ │ │ └── tf_passes.md │ │ │ ├── _index.yaml │ │ │ ├── dialects.md │ │ │ ├── images │ │ │ │ └── mlir-infra.svg │ │ │ ├── overview.md │ │ │ └── tf_passes.md │ │ ├── glob_lit_test.bzl │ │ ├── init_mlir.cc │ │ ├── init_mlir.h │ │ ├── lite │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── common │ │ │ │ └── tfl_pass_config.h │ │ │ ├── converter_gen.cc │ │ │ ├── debug │ │ │ │ ├── BUILD │ │ │ │ ├── debug.cc │ │ │ │ ├── debug.h │ │ │ │ ├── debug_options.proto │ │ │ │ └── debug_test.cc │ │ │ ├── emit_error_reporter.cc │ │ │ ├── emit_error_reporter.h │ │ │ ├── experimental │ │ │ │ ├── common │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── outline_operations.cc │ │ │ │ │ └── outline_operations.h │ │ │ │ ├── remat │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── rematerializer.cc │ │ │ │ │ ├── rematerializer.h │ │ │ │ │ └── rematerializer_test.cc │ │ │ │ └── tac │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── common │ │ │ │ │ ├── cost.h │ │ │ │ │ ├── subgraph.h │ │ │ │ │ ├── targets.h │ │ │ │ │ ├── utils.cc │ │ │ │ │ └── utils.h │ │ │ │ │ ├── examples │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── example_hardware.cc │ │ │ │ │ └── example_hardware.h │ │ │ │ │ ├── execution_metadata_exporter.cc │ │ │ │ │ ├── execution_metadata_exporter.h │ │ │ │ │ ├── execution_metadata_exporter_test.cc │ │ │ │ │ ├── g3doc │ │ │ │ │ └── images │ │ │ │ │ │ ├── compute_cost.png │ │ │ │ │ │ ├── final_graph.png │ │ │ │ │ │ ├── pick_subgraphs.png │ │ │ │ │ │ ├── target_annotation.png │ │ │ │ │ │ └── target_optimization.png │ │ │ │ │ ├── hardwares │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── cpu_hardware.cc │ │ │ │ │ ├── gpu_hardware.cc │ │ │ │ │ ├── gpu_hardware.h │ │ │ │ │ ├── nnapi_hardware.cc │ │ │ │ │ ├── nnapi_hardware.h │ │ │ │ │ ├── simple_hardware.cc │ │ │ │ │ ├── simple_hardware.h │ │ │ │ │ ├── target_hardware.cc │ │ │ │ │ └── target_hardware.h │ │ │ │ │ ├── py_wrapper │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── _pywrap_tac_wrapper.pyi │ │ │ │ │ ├── tac_wrapper.cc │ │ │ │ │ ├── tac_wrapper.h │ │ │ │ │ └── tac_wrapper_pybind11.cc │ │ │ │ │ ├── runtime_metadata.fbs │ │ │ │ │ ├── tac.py │ │ │ │ │ ├── tac_filter.proto │ │ │ │ │ ├── tac_importer_exporter.h │ │ │ │ │ ├── tac_module.cc │ │ │ │ │ ├── tac_module.h │ │ │ │ │ ├── tac_translate.cc │ │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── compute-cost.mlir │ │ │ │ │ ├── device-transform-gpu.mlir │ │ │ │ │ ├── device-transform-nnapi.mlir │ │ │ │ │ ├── e2e │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── device-transform-nnapi.mlir │ │ │ │ │ │ └── simple-graph.mlir │ │ │ │ │ ├── fold-constants-to-subgraph.mlir │ │ │ │ │ ├── get-alternative-subgraph.mlir │ │ │ │ │ ├── get-op-cost.mlir │ │ │ │ │ ├── pick-subgraphs.mlir │ │ │ │ │ ├── raise-target-subgraphs.mlir │ │ │ │ │ ├── tac-filter.mlir │ │ │ │ │ └── target-annotation.mlir │ │ │ │ │ ├── tflite_import_export.cc │ │ │ │ │ ├── tflite_import_export.h │ │ │ │ │ ├── transforms │ │ │ │ │ ├── compute_cost.cc │ │ │ │ │ ├── cost_model.cc │ │ │ │ │ ├── cost_model.h │ │ │ │ │ ├── device_transform.cc │ │ │ │ │ ├── device_transform.h │ │ │ │ │ ├── device_transform_gpu.cc │ │ │ │ │ ├── device_transform_gpu.h │ │ │ │ │ ├── device_transform_nnapi.cc │ │ │ │ │ ├── device_transform_patterns.cc │ │ │ │ │ ├── device_transform_patterns.h │ │ │ │ │ ├── fold_constants_to_subgraph.cc │ │ │ │ │ ├── get_alternative_subgraph.cc │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── pick_subgraphs.cc │ │ │ │ │ ├── raise_target_subgraphs.cc │ │ │ │ │ ├── tac_filter.cc │ │ │ │ │ ├── tac_pass.h │ │ │ │ │ ├── target_annotation.cc │ │ │ │ │ └── transform_patterns.td │ │ │ │ │ └── utils │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── utils.cc │ │ │ │ │ └── utils.h │ │ │ ├── flatbuffer_export.cc │ │ │ ├── flatbuffer_export.h │ │ │ ├── flatbuffer_export_flags.h │ │ │ ├── flatbuffer_import.cc │ │ │ ├── flatbuffer_import.h │ │ │ ├── flatbuffer_operator.cc │ │ │ ├── flatbuffer_operator.h │ │ │ ├── flatbuffer_to_string.cc │ │ │ ├── flatbuffer_translate.cc │ │ │ ├── flatbuffer_translate.h │ │ │ ├── flatbuffer_translate_flags.h │ │ │ ├── ir │ │ │ │ ├── tfl_op_enums.td │ │ │ │ ├── tfl_op_interfaces.td │ │ │ │ ├── tfl_ops.cc │ │ │ │ ├── tfl_ops.h │ │ │ │ └── tfl_ops.td │ │ │ ├── json_to_flatbuffer.cc │ │ │ ├── metrics │ │ │ │ ├── BUILD │ │ │ │ ├── error_collector.cc │ │ │ │ ├── error_collector.h │ │ │ │ ├── error_collector_inst.cc │ │ │ │ ├── error_collector_inst.h │ │ │ │ ├── error_collector_inst_test.cc │ │ │ │ ├── testdata │ │ │ │ │ └── strided_slice.mlir │ │ │ │ ├── types_util.cc │ │ │ │ └── types_util.h │ │ │ ├── mlir_tflite_runner.cc │ │ │ ├── offset_buffer.h │ │ │ ├── offset_buffer_test.cc │ │ │ ├── python │ │ │ │ ├── BUILD │ │ │ │ ├── flatbuffer_to_mlir.cc │ │ │ │ ├── flatbuffer_to_mlir.h │ │ │ │ ├── graphdef_to_tfl_flatbuffer.cc │ │ │ │ ├── graphdef_to_tfl_flatbuffer.h │ │ │ │ ├── jax_to_tfl_flatbuffer.cc │ │ │ │ ├── jax_to_tfl_flatbuffer.h │ │ │ │ ├── saved_model_to_tfl_flatbuffer.cc │ │ │ │ ├── saved_model_to_tfl_flatbuffer.h │ │ │ │ ├── tf_tfl_flatbuffer_helpers.cc │ │ │ │ └── tf_tfl_flatbuffer_helpers.h │ │ │ ├── quantization │ │ │ │ ├── BUILD │ │ │ │ ├── device_target.cc │ │ │ │ ├── device_target.h │ │ │ │ ├── import_quant_stats_pass.cc │ │ │ │ ├── ir │ │ │ │ │ ├── .clang-format │ │ │ │ │ ├── .clang-tidy │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── ConvertConst.cc │ │ │ │ │ ├── ConvertSimQuant.cc │ │ │ │ │ ├── FakeQuantSupport.cc │ │ │ │ │ ├── FakeQuantSupport.h │ │ │ │ │ ├── Passes.h │ │ │ │ │ ├── Passes.td │ │ │ │ │ ├── QuantOps.cc │ │ │ │ │ ├── QuantOps.h │ │ │ │ │ ├── QuantOps.td │ │ │ │ │ ├── QuantOpsBase.td │ │ │ │ │ ├── QuantizeUtils.cc │ │ │ │ │ ├── QuantizeUtils.h │ │ │ │ │ ├── UniformSupport.cc │ │ │ │ │ └── UniformSupport.h │ │ │ │ ├── lite │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── quantize_model.cc │ │ │ │ │ ├── quantize_model.h │ │ │ │ │ ├── quantize_model_test.cc │ │ │ │ │ ├── quantize_weights.cc │ │ │ │ │ ├── quantize_weights.h │ │ │ │ │ ├── quantize_weights_test.cc │ │ │ │ │ ├── tfl_quantizer.cc │ │ │ │ │ ├── tfl_to_std.cc │ │ │ │ │ └── tfl_to_std.h │ │ │ │ ├── numerical_utils.cc │ │ │ │ ├── numerical_utils.h │ │ │ │ ├── numerical_utils_test.cc │ │ │ │ ├── quantization.td │ │ │ │ ├── quantization_config.cc │ │ │ │ ├── quantization_config.h │ │ │ │ ├── quantization_context.cc │ │ │ │ ├── quantization_context.h │ │ │ │ ├── quantization_driver.cc │ │ │ │ ├── quantization_info.proto │ │ │ │ ├── quantization_passes.h │ │ │ │ ├── quantization_traits.h │ │ │ │ ├── quantization_utils.cc │ │ │ │ ├── quantization_utils.h │ │ │ │ ├── tensorflow │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── fallback_to_flex_ops.cc │ │ │ │ │ ├── fallback_to_flex_patterns.td │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── fallback_to_flex_ops_default.mlir │ │ │ │ │ │ ├── fallback_to_flex_ops_legacy.mlir │ │ │ │ │ │ ├── tf_to_quant.mlir │ │ │ │ │ │ └── tf_to_quant_4bit.mlir │ │ │ │ │ └── tf_to_quant.cc │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ └── import_quant_stats.mlir │ │ │ │ └── tools │ │ │ │ │ ├── op_quant_spec_getters_gen.cc │ │ │ │ │ └── tflite_op_coverage_spec_getters_gen.cc │ │ │ ├── sparsity │ │ │ │ ├── BUILD │ │ │ │ ├── sparsify_model.cc │ │ │ │ ├── sparsify_model.h │ │ │ │ └── sparsify_model_test.cc │ │ │ ├── stablehlo │ │ │ │ ├── BUILD │ │ │ │ ├── odml_to_stablehlo.cc │ │ │ │ ├── serializer │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── flatbuffer_export.cc │ │ │ │ │ ├── flatbuffer_export.h │ │ │ │ │ ├── flatbuffer_operator.h │ │ │ │ │ ├── flatbuffer_translator.cc │ │ │ │ │ └── flatbuffer_translator.h │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── call_xla_module_to_stablehlo.mlir │ │ │ │ │ ├── compose-uniform-quantized-type.mlir │ │ │ │ │ ├── fold_broadcast.mlir │ │ │ │ │ ├── fuse_mhlo_convolution.mlir │ │ │ │ │ ├── legalize-inplaceupdate.mlir │ │ │ │ │ ├── legalize-skip-quantization-ops.mlir │ │ │ │ │ ├── legalize-stablehlo-tf-fb-tf.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-add.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-broadcast_in_dim.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-clamp.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-compare.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-concat.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-constant.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-conv.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-dot.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-gather.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-max.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-mul.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-pad.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-reshape.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-rsqrt.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-scatter.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl-sub.mlir │ │ │ │ │ ├── legalize-stablehlo-tfl.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-add.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-broadcast.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-clamp.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-concat.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-constant.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-conv.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-max.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-mul.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-pad.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-reshape.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-rsqrt.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo-sub.mlir │ │ │ │ │ ├── legalize-tfl-stablehlo.mlir │ │ │ │ │ ├── legalize_hlo.mlir │ │ │ │ │ ├── odml-to-stablehlo-allow-tf.mlir │ │ │ │ │ ├── odml-to-stablehlo-smuggle-resize.mlir │ │ │ │ │ ├── optimize.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-clamp.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-concat.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-conv.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-division.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-logistic.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-multiply.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo-resize-bilinear.mlir │ │ │ │ │ ├── tf-tfl-translate-serialize-stablehlo.mlir │ │ │ │ │ ├── tf-tfl-translate-tf-quantize.mlir │ │ │ │ │ ├── tfl_legalize_hlo.mlir │ │ │ │ │ ├── tfl_legalize_hlo_custom_call.mlir │ │ │ │ │ ├── unfold_splat_constant_pass.mlir │ │ │ │ │ ├── unfuse_mhlo_batch_norm.mlir │ │ │ │ │ └── uniform-quantized-stablehlo-to-tfl.mlir │ │ │ │ └── transforms │ │ │ │ │ ├── check_accepted_ops_pass.cc │ │ │ │ │ ├── check_accepted_ops_pass.h │ │ │ │ │ ├── compose_uniform_quantized_type_pass.cc │ │ │ │ │ ├── drop_savedmodel_semantics.cc │ │ │ │ │ ├── drop_savedmodel_semantics.h │ │ │ │ │ ├── fold_broadcast_pass.cc │ │ │ │ │ ├── fuse_convolution_pass.cc │ │ │ │ │ ├── legalize_hlo.cc │ │ │ │ │ ├── legalize_hlo_conversions │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── custom_call.cc │ │ │ │ │ ├── custom_call.h │ │ │ │ │ ├── dot_general.cc │ │ │ │ │ ├── dot_general.h │ │ │ │ │ ├── scatter.cc │ │ │ │ │ ├── scatter.h │ │ │ │ │ ├── util.cc │ │ │ │ │ └── util.h │ │ │ │ │ ├── legalize_hlo_patterns.td │ │ │ │ │ ├── legalize_tf_xla_call_module_to_stablehlo_pass.cc │ │ │ │ │ ├── legalize_tf_xla_call_module_to_stablehlo_pass.h │ │ │ │ │ ├── op_stat_pass.cc │ │ │ │ │ ├── op_stat_pass.h │ │ │ │ │ ├── optimize.cc │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── passes.td │ │ │ │ │ ├── rename_entrypoint_to_main.cc │ │ │ │ │ ├── rename_entrypoint_to_main.h │ │ │ │ │ ├── smuggle_disallowed_ops.cc │ │ │ │ │ ├── smuggle_disallowed_ops.h │ │ │ │ │ ├── stablehlo_tfl_pass.cc │ │ │ │ │ ├── stablehlo_tfl_pass.h │ │ │ │ │ ├── stablehlo_util.cc │ │ │ │ │ ├── stablehlo_util.h │ │ │ │ │ ├── tf_stablehlo_pass.cc │ │ │ │ │ ├── tf_stablehlo_pass.h │ │ │ │ │ ├── tfl_stablehlo_pass.cc │ │ │ │ │ ├── tfl_stablehlo_pass.h │ │ │ │ │ ├── tflite_legalize_hlo.cc │ │ │ │ │ ├── tflite_legalize_hlo_patterns.td │ │ │ │ │ ├── transforms.cc │ │ │ │ │ ├── transforms.h │ │ │ │ │ ├── unfold_splat_constant_pass.cc │ │ │ │ │ ├── unfuse_batch_norm_pass.cc │ │ │ │ │ └── uniform_quantized_stablehlo_to_tfl_pass.cc │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── analyze-variables.mlir │ │ │ │ ├── canonicalize.mlir │ │ │ │ ├── const-fold.mlir │ │ │ │ ├── debuginfo │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── concrete_function_error.py │ │ │ │ │ ├── saved_model_error.py │ │ │ │ │ ├── v1_1.0_224_frozen.wrong_attr.line.part.pbtxt │ │ │ │ │ ├── v1_1.0_224_frozen.wrong_attr.line.part.pbtxt.debug │ │ │ │ │ ├── v1_1.0_224_frozen.wrong_attr.stack.part.pbtxt │ │ │ │ │ └── v1_1.0_224_frozen.wrong_attr.stack.part.pbtxt.debug │ │ │ │ ├── decompose-hybrid-quantization.mlir │ │ │ │ ├── default_quant_params.mlir │ │ │ │ ├── dilated-conv.mlir │ │ │ │ ├── end2end │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add.pbtxt │ │ │ │ │ ├── back2back_fake_quant.pbtxt │ │ │ │ │ ├── control_flow_v1.pbtxt │ │ │ │ │ ├── conv_2d.pbtxt │ │ │ │ │ ├── conv_2d_nchw.pbtxt │ │ │ │ │ ├── custom_opdef.pbtxt │ │ │ │ │ ├── disallow_stateful_partitioned_call.pbtxt │ │ │ │ │ ├── fake_quant_per_channel.pbtxt │ │ │ │ │ ├── fake_quant_per_channel_4bit.pbtxt │ │ │ │ │ ├── fake_quant_without_identity.pbtxt │ │ │ │ │ ├── fake_quant_without_identity_4bit.pbtxt │ │ │ │ │ ├── graph-input-node.pbtxt │ │ │ │ │ ├── graph_with_placeholder_with_default.pbtxt │ │ │ │ │ ├── if_op.pbtxt │ │ │ │ │ ├── quant_stats.pbtxt │ │ │ │ │ ├── quant_stats.pbtxt.stats │ │ │ │ │ ├── unroll_batch_matmul.pbtxt │ │ │ │ │ └── unroll_batch_matmul_disabled.pbtxt │ │ │ │ ├── flatbuffer2mlir │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── basic_lstm.mlir │ │ │ │ │ ├── bucketize.mlir │ │ │ │ │ ├── constants.mlir │ │ │ │ │ ├── constants_offset.mlir │ │ │ │ │ ├── control_edges.mlir │ │ │ │ │ ├── custom_op.mlir │ │ │ │ │ ├── custom_op_offset.mlir │ │ │ │ │ ├── dynamic_shape.mlir │ │ │ │ │ ├── empty_input_output_names.json │ │ │ │ │ ├── external_constant.mlir │ │ │ │ │ ├── if_op.mlir │ │ │ │ │ ├── import_json.json │ │ │ │ │ ├── importer_test_min_max.cc │ │ │ │ │ ├── importer_test_min_max.cc.mlir │ │ │ │ │ ├── input_arrays.mlir │ │ │ │ │ ├── input_output_names_attr.mlir │ │ │ │ │ ├── legacy_reshape.json │ │ │ │ │ ├── lstm.json │ │ │ │ │ ├── lstm.mlir │ │ │ │ │ ├── many_attribute_op.mlir │ │ │ │ │ ├── math.mlir │ │ │ │ │ ├── matmul.mlir │ │ │ │ │ ├── mix_tflite_stablehlo.mlir │ │ │ │ │ ├── multi_output_op.json │ │ │ │ │ ├── optional.mlir │ │ │ │ │ ├── optional_input.json │ │ │ │ │ ├── output_arrays.mlir │ │ │ │ │ ├── pruning.mlir │ │ │ │ │ ├── pruning_function_input_as_output.mlir │ │ │ │ │ ├── quant_stats.mlir │ │ │ │ │ ├── quantization.mlir │ │ │ │ │ ├── reshape.mlir │ │ │ │ │ ├── signature.mlir │ │ │ │ │ ├── signature_with_multiple_entry_points.mlir │ │ │ │ │ ├── simple.mlir │ │ │ │ │ ├── stablehlo.mlir │ │ │ │ │ ├── stablehlo_const.mlir │ │ │ │ │ ├── stablehlo_custom_call.mlir │ │ │ │ │ ├── test_schema.fbs │ │ │ │ │ ├── tf_variant_type.mlir │ │ │ │ │ ├── unranked_function_output.mlir │ │ │ │ │ ├── unranked_tensor.mlir │ │ │ │ │ ├── variable.mlir │ │ │ │ │ └── while_op.mlir │ │ │ │ ├── fuse-tftext.mlir │ │ │ │ ├── get-arithmetic-count.mlir │ │ │ │ ├── guarantee_func_has_one_use.mlir │ │ │ │ ├── inlining.mlir │ │ │ │ ├── insert_call_once_op.mlir │ │ │ │ ├── legalize-tensorlist.mlir │ │ │ │ ├── legalize-tf-assert.mlir │ │ │ │ ├── legalize-tf-hashtables.mlir │ │ │ │ ├── legalize-tf-no-runtime-verification.mlir │ │ │ │ ├── legalize-tf-variables.mlir │ │ │ │ ├── legalize-tf-while.mlir │ │ │ │ ├── legalize-tf.mlir │ │ │ │ ├── legalize_jax_random.mlir │ │ │ │ ├── lift_tflite_flex_ops.mlir │ │ │ │ ├── lower-static-tensor-list-default-to-single-batch.mlir │ │ │ │ ├── lower-static-tensor-list-enable-dynamic-update-slice.mlir │ │ │ │ ├── lower-static-tensor-list.mlir │ │ │ │ ├── mlir2exec │ │ │ │ │ ├── BUILD │ │ │ │ │ └── tfl_while_op.mlir │ │ │ │ ├── mlir2flatbuffer │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── basic_lstm.mlir │ │ │ │ │ ├── bucketize.mlir │ │ │ │ │ ├── custom_op_with_tflite_op.mlir │ │ │ │ │ ├── custom_tensorlist_reserve.mlir │ │ │ │ │ ├── depthwise_conv2d.mlir │ │ │ │ │ ├── depthwise_conv2d_v2.mlir │ │ │ │ │ ├── disable_builtin.mlir │ │ │ │ │ ├── disable_custom.mlir │ │ │ │ │ ├── disable_flex.mlir │ │ │ │ │ ├── disable_flex_enable_builtin.mlir │ │ │ │ │ ├── dynamic_shape_constant.mlir │ │ │ │ │ ├── fake_quant.mlir │ │ │ │ │ ├── flex_exclusively.mlir │ │ │ │ │ ├── flex_op_with_complex128.mlir │ │ │ │ │ ├── flex_op_with_f64.mlir │ │ │ │ │ ├── flex_op_with_tflite_op.mlir │ │ │ │ │ ├── fully_connected.mlir │ │ │ │ │ ├── fully_connected_v2.mlir │ │ │ │ │ ├── hashtable_resource.mlir │ │ │ │ │ ├── if_op.mlir │ │ │ │ │ ├── logical.mlir │ │ │ │ │ ├── low_bit_packing.mlir │ │ │ │ │ ├── lstm.mlir │ │ │ │ │ ├── lstm_asym_attr.mlir │ │ │ │ │ ├── lstm_quantized.mlir │ │ │ │ │ ├── math.mlir │ │ │ │ │ ├── metadata.mlir │ │ │ │ │ ├── mul_v2.mlir │ │ │ │ │ ├── mul_v3.mlir │ │ │ │ │ ├── nn.mlir │ │ │ │ │ ├── numeric_verify.mlir │ │ │ │ │ ├── optional.mlir │ │ │ │ │ ├── quantization.mlir │ │ │ │ │ ├── reshape.mlir │ │ │ │ │ ├── signature_def.mlir │ │ │ │ │ ├── signature_def_output_override.mlir │ │ │ │ │ ├── signature_def_with_multiple_entry_points.mlir │ │ │ │ │ ├── signature_def_with_no_inputs.mlir │ │ │ │ │ ├── simple.mlir │ │ │ │ │ ├── simple_with_connected_control_nodes.mlir │ │ │ │ │ ├── simple_with_unconnected_control_nodes.mlir │ │ │ │ │ ├── svdf.mlir │ │ │ │ │ ├── svdf_v2.mlir │ │ │ │ │ ├── tf_entry_function.mlir │ │ │ │ │ ├── tfl_while_op.mlir │ │ │ │ │ ├── transpose_conv_optional.mlir │ │ │ │ │ ├── type_attr.mlir │ │ │ │ │ ├── u16_quant.mlir │ │ │ │ │ ├── unidirectional_sequence_lstm.mlir │ │ │ │ │ ├── unidirectional_sequence_rnn.mlir │ │ │ │ │ ├── unranked_tensor.mlir │ │ │ │ │ ├── unsorted_segment_prod.mlir │ │ │ │ │ ├── variable.mlir │ │ │ │ │ ├── variant_type_on_func.mlir │ │ │ │ │ ├── variant_type_on_op.mlir │ │ │ │ │ └── while_op.mlir │ │ │ │ ├── modify_io_nodes.mlir │ │ │ │ ├── ops.mlir │ │ │ │ ├── optimize-after-quantization.mlir │ │ │ │ ├── optimize.mlir │ │ │ │ ├── optimize_batch_matmul.mlir │ │ │ │ ├── optimize_functional_ops.mlir │ │ │ │ ├── optimize_no_verify.mlir │ │ │ │ ├── optimize_op_order.mlir │ │ │ │ ├── partitioned-topological-sort.mlir │ │ │ │ ├── pin-ops-with-side-effects.mlir │ │ │ │ ├── post-quantize-dynamic-range.mlir │ │ │ │ ├── post-quantize.mlir │ │ │ │ ├── prepare-composite-functions-tf.mlir │ │ │ │ ├── prepare-quantize-dynamic-range.mlir │ │ │ │ ├── prepare-quantize-post-training-16bits.mlir │ │ │ │ ├── prepare-quantize-post-training.mlir │ │ │ │ ├── prepare-quantize-signed.mlir │ │ │ │ ├── prepare-quantize.mlir │ │ │ │ ├── prepare-tf-fake-quant-4bit.mlir │ │ │ │ ├── prepare-tf-fake-quant.mlir │ │ │ │ ├── prepare-tf-with-allowing-bf16-and-f16-type-legalization.mlir │ │ │ │ ├── prepare-tf.mlir │ │ │ │ ├── quantize-dynamic-range.mlir │ │ │ │ ├── quantize-numeric-verify.mlir │ │ │ │ ├── quantize-variables.mlir │ │ │ │ ├── quantize.mlir │ │ │ │ ├── raise-custom-ops.mlir │ │ │ │ ├── reduce-type-precision.mlir │ │ │ │ ├── reduce_while_operands.mlir │ │ │ │ ├── shape-inference.mlir │ │ │ │ ├── split-merged-operands.mlir │ │ │ │ ├── tfl_while_op_licm.mlir │ │ │ │ ├── tfl_while_outline.mlir │ │ │ │ ├── trim-functions-tf.mlir │ │ │ │ └── unfold-large-splat-constant.mlir │ │ │ ├── tf_tfl_passes.cc │ │ │ ├── tf_tfl_passes.h │ │ │ ├── tf_tfl_translate.cc │ │ │ ├── tf_tfl_translate_cl.cc │ │ │ ├── tf_tfl_translate_cl.h │ │ │ ├── tf_to_tfl_flatbuffer.cc │ │ │ ├── tf_to_tfl_flatbuffer.h │ │ │ ├── transforms │ │ │ │ ├── analyze_variables.cc │ │ │ │ ├── decompose_hybrid_quantization.cc │ │ │ │ ├── default_quant_params.cc │ │ │ │ ├── dense_to_sparse.cc │ │ │ │ ├── dilated_conv.cc │ │ │ │ ├── dilated_conv.h │ │ │ │ ├── get_arithmetic_count.cc │ │ │ │ ├── insert_call_once_op.cc │ │ │ │ ├── legalize_hashtables.cc │ │ │ │ ├── legalize_jax_random.cc │ │ │ │ ├── legalize_patterns.td │ │ │ │ ├── legalize_tensorlist.cc │ │ │ │ ├── legalize_tensorlist.td │ │ │ │ ├── legalize_tf.cc │ │ │ │ ├── legalize_tf_while.cc │ │ │ │ ├── legalize_variables.cc │ │ │ │ ├── legalize_variables.td │ │ │ │ ├── lift_tflite_flex_ops.cc │ │ │ │ ├── lift_tflite_flex_ops.h │ │ │ │ ├── lower_static_tensor_list.cc │ │ │ │ ├── modify_io_nodes.cc │ │ │ │ ├── optimize.cc │ │ │ │ ├── optimize_batch_matmul.cc │ │ │ │ ├── optimize_batch_matmul.td │ │ │ │ ├── optimize_functional_ops.cc │ │ │ │ ├── optimize_op_order.cc │ │ │ │ ├── optimize_patterns.td │ │ │ │ ├── partitioned_topological_sort.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ ├── pin_ops_with_side_effects.cc │ │ │ │ ├── post_quantize.cc │ │ │ │ ├── post_quantize_patterns.td │ │ │ │ ├── prepare_composite_functions_tf.cc │ │ │ │ ├── prepare_patterns.td │ │ │ │ ├── prepare_quantize.cc │ │ │ │ ├── prepare_quantize_dynamic_range.cc │ │ │ │ ├── prepare_quantize_helper.cc │ │ │ │ ├── prepare_quantize_helper.h │ │ │ │ ├── prepare_tf.cc │ │ │ │ ├── quantize.cc │ │ │ │ ├── quantize_patterns.td │ │ │ │ ├── quantize_variables.cc │ │ │ │ ├── raise_custom_ops.cc │ │ │ │ ├── reduce_type_precision.cc │ │ │ │ ├── reduce_while_operands.cc │ │ │ │ ├── runtime_verify.cc │ │ │ │ ├── split_merged_operands.cc │ │ │ │ ├── tensorlist_patterns.td │ │ │ │ ├── trim_functions_tf.cc │ │ │ │ ├── unfold_large_splat_constant.cc │ │ │ │ └── while_loop_outline.cc │ │ │ └── utils │ │ │ │ ├── arithmetic_count_util.h │ │ │ │ ├── attribute_utils.cc │ │ │ │ ├── attribute_utils.h │ │ │ │ ├── constant_utils.cc │ │ │ │ ├── constant_utils.h │ │ │ │ ├── convert_type.cc │ │ │ │ ├── convert_type.h │ │ │ │ ├── fake_quant_utils.cc │ │ │ │ ├── fake_quant_utils.h │ │ │ │ ├── low_bit_utils.cc │ │ │ │ ├── low_bit_utils.h │ │ │ │ ├── lstm_utils.cc │ │ │ │ ├── lstm_utils.h │ │ │ │ ├── lstm_utils_test.cc │ │ │ │ ├── nms_utils.cc │ │ │ │ ├── nms_utils.h │ │ │ │ ├── perception_ops_utils.cc │ │ │ │ ├── perception_ops_utils.h │ │ │ │ ├── perception_ops_utils_test.cc │ │ │ │ ├── size_utils.cc │ │ │ │ ├── size_utils.h │ │ │ │ ├── size_utils_test.cc │ │ │ │ ├── stateful_ops_utils.cc │ │ │ │ ├── stateful_ops_utils.h │ │ │ │ ├── tftext_utils.cc │ │ │ │ ├── tftext_utils.h │ │ │ │ ├── tftext_utils_test.cc │ │ │ │ ├── utils.h │ │ │ │ ├── utils.td │ │ │ │ ├── validators.cc │ │ │ │ ├── validators.h │ │ │ │ ├── variables_utils.cc │ │ │ │ └── variables_utils.h │ │ ├── mlir_graph_optimization_pass.cc │ │ ├── mlir_graph_optimization_pass.h │ │ ├── mlir_graph_optimization_pass_registration.cc │ │ ├── mlir_graph_optimization_pass_test.cc │ │ ├── op_or_arg_name_mapper.cc │ │ ├── op_or_arg_name_mapper.h │ │ ├── python │ │ │ ├── BUILD │ │ │ ├── mlir.cc │ │ │ ├── mlir.h │ │ │ └── mlir_wrapper │ │ │ │ ├── BUILD │ │ │ │ ├── attrs.cc │ │ │ │ ├── basic_classes.cc │ │ │ │ ├── builders.cc │ │ │ │ ├── filecheck_wrapper.cc │ │ │ │ ├── filecheck_wrapper.pyi │ │ │ │ ├── mlir_wrapper.cc │ │ │ │ ├── mlir_wrapper.h │ │ │ │ ├── mlir_wrapper.pyi │ │ │ │ ├── ops.cc │ │ │ │ └── types.cc │ │ ├── quantization │ │ │ ├── stablehlo │ │ │ │ ├── BUILD │ │ │ │ ├── cc │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── component.h │ │ │ │ │ ├── debugger.cc │ │ │ │ │ ├── debugger.h │ │ │ │ │ ├── graph_def.h │ │ │ │ │ ├── graph_def_test.cc │ │ │ │ │ ├── io.cc │ │ │ │ │ ├── io.h │ │ │ │ │ └── io_test.cc │ │ │ │ ├── internal_visibility_allowlist.bzl │ │ │ │ ├── ops │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── stablehlo_op_quant_spec.cc │ │ │ │ │ └── stablehlo_op_quant_spec.h │ │ │ │ ├── passes │ │ │ │ │ ├── bridge │ │ │ │ │ │ ├── convert_mhlo_quant_to_int.cc │ │ │ │ │ │ ├── convert_tf_quant_ops_to_mhlo.cc │ │ │ │ │ │ ├── convert_tf_quant_to_mhlo_int_test.cc │ │ │ │ │ │ ├── convert_tf_quant_types.cc │ │ │ │ │ │ ├── convert_tf_quant_types_test.cc │ │ │ │ │ │ ├── legalize_tf_quant_test.cc │ │ │ │ │ │ ├── passes.cc │ │ │ │ │ │ ├── passes.h │ │ │ │ │ │ ├── passes.td │ │ │ │ │ │ └── verify_quant_legalization.cc │ │ │ │ │ ├── lift_quantizable_spots_as_functions.cc │ │ │ │ │ ├── lift_quantizable_spots_as_functions_fusion.td │ │ │ │ │ ├── lift_quantizable_spots_as_functions_simple.td │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── passes.td │ │ │ │ │ ├── post_quantize.cc │ │ │ │ │ ├── prepare_quantize.cc │ │ │ │ │ ├── quantization_pattern.h │ │ │ │ │ ├── quantize.cc │ │ │ │ │ ├── quantize_composite_functions.cc │ │ │ │ │ ├── quantize_weight.cc │ │ │ │ │ ├── replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc │ │ │ │ │ ├── restore_function_name.cc │ │ │ │ │ ├── unfuse_mhlo_batch_norm.cc │ │ │ │ │ ├── unwrap_xla_call_module_op.cc │ │ │ │ │ └── utils.td │ │ │ │ ├── python │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── integration_test │ │ │ │ │ │ ├── quantize_model_test.py │ │ │ │ │ │ └── quantize_model_test_base.py │ │ │ │ │ ├── pywrap_quantization.cc │ │ │ │ │ ├── pywrap_quantization.pyi │ │ │ │ │ └── quantization.py │ │ │ │ ├── quantization_config.proto │ │ │ │ ├── quantization_options.proto │ │ │ │ ├── quantize_passes.cc │ │ │ │ ├── quantize_passes.h │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── bridge │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── convert-mhlo-quant-to-int-no-chlo.mlir │ │ │ │ │ │ ├── convert-mhlo-quant-to-int.mlir │ │ │ │ │ │ ├── convert-tf-quant-types.mlir │ │ │ │ │ │ ├── convert_tf_quant_ops_to_mhlo.mlir │ │ │ │ │ │ └── verify-quant-legalization.mlir │ │ │ │ │ ├── fill_quantization_options_test.cc │ │ │ │ │ ├── lift_quantizable_spots_as_functions.mlir │ │ │ │ │ ├── post_quantize.mlir │ │ │ │ │ ├── prepare_quantize.mlir │ │ │ │ │ ├── prepare_quantize_int4.mlir │ │ │ │ │ ├── prepare_quantize_per_channel.mlir │ │ │ │ │ ├── quantize.mlir │ │ │ │ │ ├── quantize_composite_functions.mlir │ │ │ │ │ ├── quantize_same_scale.mlir │ │ │ │ │ ├── replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.mlir │ │ │ │ │ ├── restore_function_name.mlir │ │ │ │ │ ├── stablehlo_op_quant_spec_test.cc │ │ │ │ │ ├── unfuse_mhlo_batch_norm.mlir │ │ │ │ │ └── unwrap_xla_call_module_op.mlir │ │ │ │ ├── tools │ │ │ │ │ └── stablehlo_quant_opt.cc │ │ │ │ ├── uniform_quantized_types.cc │ │ │ │ ├── uniform_quantized_types.h │ │ │ │ ├── uniform_quantized_types_test.cc │ │ │ │ └── utils │ │ │ │ │ ├── fill_quantization_options.cc │ │ │ │ │ ├── fill_quantization_options.h │ │ │ │ │ ├── math_utils.cc │ │ │ │ │ ├── math_utils.h │ │ │ │ │ ├── math_utils_test.cc │ │ │ │ │ ├── stablehlo_type_utils.h │ │ │ │ │ ├── stablehlo_type_utils_test.cc │ │ │ │ │ ├── tf_type_utils.cc │ │ │ │ │ ├── tf_type_utils.h │ │ │ │ │ └── tf_type_utils_test.cc │ │ │ └── tensorflow │ │ │ │ ├── BUILD │ │ │ │ ├── calibrator │ │ │ │ ├── BUILD │ │ │ │ ├── calibration_algorithm.py │ │ │ │ ├── calibration_algorithm_test.py │ │ │ │ ├── calibration_statistics.proto │ │ │ │ ├── calibration_statistics_collector_average_min_max.cc │ │ │ │ ├── calibration_statistics_collector_average_min_max.h │ │ │ │ ├── calibration_statistics_collector_base.h │ │ │ │ ├── calibration_statistics_collector_histogram.cc │ │ │ │ ├── calibration_statistics_collector_histogram.h │ │ │ │ ├── calibration_statistics_collector_min_max.cc │ │ │ │ ├── calibration_statistics_collector_min_max.h │ │ │ │ ├── calibration_statistics_collector_test.cc │ │ │ │ ├── calibrator_singleton.cc │ │ │ │ ├── calibrator_singleton.h │ │ │ │ ├── calibrator_singleton_test.cc │ │ │ │ ├── custom_aggregator_op.cc │ │ │ │ ├── custom_aggregator_op.py │ │ │ │ ├── integration_test │ │ │ │ │ └── custom_aggregator_op_test.py │ │ │ │ ├── pywrap_calibration.cc │ │ │ │ └── pywrap_calibration.pyi │ │ │ │ ├── cc │ │ │ │ ├── BUILD │ │ │ │ ├── const_op_size.cc │ │ │ │ ├── const_op_size.h │ │ │ │ ├── const_op_size_test.cc │ │ │ │ ├── constant_fold.cc │ │ │ │ ├── constant_fold.h │ │ │ │ ├── constant_fold_test.cc │ │ │ │ ├── convert_asset_args.cc │ │ │ │ ├── convert_asset_args.h │ │ │ │ ├── convert_asset_args_test.cc │ │ │ │ ├── quantization_unit_loc.cc │ │ │ │ ├── quantization_unit_loc.h │ │ │ │ ├── run_passes.cc │ │ │ │ ├── run_passes.h │ │ │ │ ├── save_variables.cc │ │ │ │ ├── save_variables.h │ │ │ │ ├── save_variables_test.cc │ │ │ │ ├── status_macro.h │ │ │ │ └── status_macro_test.cc │ │ │ │ ├── debugging │ │ │ │ ├── BUILD │ │ │ │ ├── dump_tensor_op.cc │ │ │ │ ├── mlir_dump.cc │ │ │ │ ├── mlir_dump.h │ │ │ │ └── mlir_dump_test.cc │ │ │ │ ├── exported_model.proto │ │ │ │ ├── gen_quantized_function_library.py │ │ │ │ ├── internal_visibility_allowlist.bzl │ │ │ │ ├── ops │ │ │ │ ├── BUILD │ │ │ │ ├── tf_op_quant_spec.cc │ │ │ │ ├── tf_op_quant_spec.h │ │ │ │ ├── tf_op_quant_spec_test.cc │ │ │ │ ├── tf_quantize_op.cc │ │ │ │ ├── tf_quantize_op.h │ │ │ │ ├── tf_quantize_op_test.cc │ │ │ │ ├── uniform_op_quant_spec.cc │ │ │ │ └── uniform_op_quant_spec.h │ │ │ │ ├── passes │ │ │ │ ├── add_dump_tensor_op.cc │ │ │ │ ├── add_quantization_unit_loc.cc │ │ │ │ ├── cast_bf16_ops_to_f32.cc │ │ │ │ ├── cast_bf16_ops_to_f32.td │ │ │ │ ├── constants.h │ │ │ │ ├── convert_custom_aggregation_op_to_quant_stats.cc │ │ │ │ ├── convert_fake_quant_to_qdq.cc │ │ │ │ ├── convert_tf_xla_op_to_tf_op.cc │ │ │ │ ├── convert_tf_xla_op_to_tf_op.td │ │ │ │ ├── convert_tpu_model_to_cpu.cc │ │ │ │ ├── convert_tpu_model_to_cpu.td │ │ │ │ ├── duplicate_shape_determining_constants.cc │ │ │ │ ├── insert_custom_aggregation_ops.cc │ │ │ │ ├── insert_main_function.cc │ │ │ │ ├── insert_quantized_functions.cc │ │ │ │ ├── insert_restore_op.cc │ │ │ │ ├── insert_save_op.cc │ │ │ │ ├── issue_ids_of_custom_aggregation_ops.cc │ │ │ │ ├── lift_hashtable_ops_as_args.cc │ │ │ │ ├── lift_quantizable_spots_as_functions.cc │ │ │ │ ├── lift_quantizable_spots_as_functions.td │ │ │ │ ├── lift_quantizable_spots_as_functions_drq.cc │ │ │ │ ├── lift_quantizable_spots_as_functions_drq.td │ │ │ │ ├── manipulate_model_attr.cc │ │ │ │ ├── manipulate_model_attr.h │ │ │ │ ├── mark_functions_noinline.cc │ │ │ │ ├── merge_duplicate_resource_ops.cc │ │ │ │ ├── merge_initializer_function_ops_to_main.cc │ │ │ │ ├── merge_save_function_ops_to_main.cc │ │ │ │ ├── optimize.cc │ │ │ │ ├── optimize.td │ │ │ │ ├── passes.h │ │ │ │ ├── post_quantize.cc │ │ │ │ ├── post_quantize.td │ │ │ │ ├── prepare_lifting.cc │ │ │ │ ├── prepare_lifting.td │ │ │ │ ├── prepare_quantize.cc │ │ │ │ ├── prepare_quantize.td │ │ │ │ ├── prepare_quantize_drq.cc │ │ │ │ ├── preprocess_op.cc │ │ │ │ ├── preprocess_op.td │ │ │ │ ├── propagate_quantize_type.cc │ │ │ │ ├── quantize.cc │ │ │ │ ├── quantize_composite_functions.cc │ │ │ │ ├── quantize_composite_functions.td │ │ │ │ ├── quantize_weights.cc │ │ │ │ ├── quantized_function_library.mlir │ │ │ │ ├── quantized_function_library_tf_drq.mlir │ │ │ │ ├── quantized_function_library_uniform_quantized.mlir │ │ │ │ ├── quantized_function_library_uniform_quantized_drq.mlir │ │ │ │ ├── quantized_function_library_xla_weight_only.mlir │ │ │ │ ├── remove_identity_op_pattern.cc │ │ │ │ ├── remove_identity_op_pattern.h │ │ │ │ ├── remove_var_init_by_const.cc │ │ │ │ ├── replace_cast_hacks_with_tf_xla_ops.cc │ │ │ │ ├── replace_cast_hacks_with_tf_xla_ops.td │ │ │ │ ├── tf_quant_ops.cc │ │ │ │ ├── tf_quant_ops.h │ │ │ │ ├── tf_quant_ops.td │ │ │ │ ├── tf_quant_opt.cc │ │ │ │ ├── unfreeze_constants.cc │ │ │ │ ├── utils.cc │ │ │ │ ├── utils.h │ │ │ │ └── utils.td │ │ │ │ ├── python │ │ │ │ ├── BUILD │ │ │ │ ├── integration_test │ │ │ │ │ ├── concurrency_test.py │ │ │ │ │ ├── quantize_model_test.py │ │ │ │ │ └── quantize_model_test_base.py │ │ │ │ ├── py_function_lib.h │ │ │ │ ├── py_function_lib.py │ │ │ │ ├── py_function_lib_test.py │ │ │ │ ├── pywrap_function_lib.cc │ │ │ │ ├── pywrap_function_lib.pyi │ │ │ │ ├── pywrap_quantize_model.cc │ │ │ │ ├── pywrap_quantize_model.pyi │ │ │ │ ├── pywrap_quantize_model_test.py │ │ │ │ ├── quantize_model.cc │ │ │ │ ├── quantize_model.h │ │ │ │ ├── quantize_model.py │ │ │ │ ├── representative_dataset.py │ │ │ │ ├── representative_dataset_test.py │ │ │ │ ├── save_model.py │ │ │ │ └── type_casters.h │ │ │ │ ├── quantization_options.proto │ │ │ │ ├── quantize_passes.cc │ │ │ │ ├── quantize_passes.h │ │ │ │ ├── quantize_preprocess.cc │ │ │ │ ├── quantize_preprocess.h │ │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── add_dump_tensor_op.mlir │ │ │ │ ├── add_quantization_unit_loc.mlir │ │ │ │ ├── cast_bf16_ops_to_f32.mlir │ │ │ │ ├── convert_custom_aggregation_op_to_quant_stats.mlir │ │ │ │ ├── convert_fake_quant_to_qdq.mlir │ │ │ │ ├── convert_tf_xla_op_to_tf_op.mlir │ │ │ │ ├── convert_tpu_model_to_cpu.mlir │ │ │ │ ├── duplicate_shape_determining_constants.mlir │ │ │ │ ├── fake_quant_e2e_flow.mlir │ │ │ │ ├── fake_quant_e2e_xla.mlir │ │ │ │ ├── insert_custom_aggregation_ops.mlir │ │ │ │ ├── insert_main_function.mlir │ │ │ │ ├── insert_quantized_functions.mlir │ │ │ │ ├── insert_quantized_functions_drq.mlir │ │ │ │ ├── insert_quantized_functions_weight_only.mlir │ │ │ │ ├── insert_restore_op.mlir │ │ │ │ ├── insert_save_op.mlir │ │ │ │ ├── issue_ids_of_custom_aggregation_ops.mlir │ │ │ │ ├── lift_hashtable_ops_as_args.mlir │ │ │ │ ├── lift_quantizable_spots_as_functions.mlir │ │ │ │ ├── lift_quantizable_spots_as_functions_drq.mlir │ │ │ │ ├── lift_quantizable_spots_as_functions_drq_min_elements.mlir │ │ │ │ ├── lift_quantizable_spots_as_functions_xla.mlir │ │ │ │ ├── lift_quantizable_spots_as_functions_xla_selective_quantization.mlir │ │ │ │ ├── mark_functions_noinline.mlir │ │ │ │ ├── merge_duplicate_resource_ops.mlir │ │ │ │ ├── merge_initializer_function_ops_to_main.mlir │ │ │ │ ├── merge_save_function_ops_to_main.mlir │ │ │ │ ├── optimize.mlir │ │ │ │ ├── prepare_lifting.mlir │ │ │ │ ├── prepare_quantize.mlir │ │ │ │ ├── prepare_quantize_drq.mlir │ │ │ │ ├── prepare_quantize_drq_per_channel.mlir │ │ │ │ ├── prepare_quantize_ptq.mlir │ │ │ │ ├── prepare_quantize_ptq_per_channel.mlir │ │ │ │ ├── preprocess_op.mlir │ │ │ │ ├── preprocess_op_weight_only.mlir │ │ │ │ ├── propagate_quantize_type.mlir │ │ │ │ ├── quantize.mlir │ │ │ │ ├── quantize_composit_functions_debugging.mlir │ │ │ │ ├── quantize_composite_functions.mlir │ │ │ │ ├── quantize_composite_functions_drq.mlir │ │ │ │ ├── quantize_composite_functions_weight_only.mlir │ │ │ │ ├── quantize_composite_functions_xla.mlir │ │ │ │ ├── quantize_drq.mlir │ │ │ │ ├── quantize_weights.mlir │ │ │ │ ├── quantize_xla.mlir │ │ │ │ ├── remove_var_init_by_const.mlir │ │ │ │ ├── replace_cast_hacks_with_tf_xla_ops.mlir │ │ │ │ ├── replace_cast_hacks_with_tf_xla_ops_large_constants.mlir │ │ │ │ └── unfreeze_constants.mlir │ │ │ │ └── utils │ │ │ │ ├── BUILD │ │ │ │ ├── fake_quant_utils.cc │ │ │ │ ├── fake_quant_utils.h │ │ │ │ ├── lift_as_function_call_utils.cc │ │ │ │ ├── lift_as_function_call_utils.h │ │ │ │ ├── lift_as_function_call_utils.td │ │ │ │ ├── tf_quantize_op_utils.cc │ │ │ │ ├── tf_quantize_op_utils.h │ │ │ │ ├── tf_to_uniform_attribute_utils.cc │ │ │ │ ├── tf_to_uniform_attribute_utils.h │ │ │ │ ├── tf_to_uniform_attribute_utils_test.cc │ │ │ │ ├── tf_to_xla_attribute_utils.cc │ │ │ │ ├── tf_to_xla_attribute_utils.h │ │ │ │ └── tf_to_xla_attribute_utils_test.cc │ │ ├── register_common_dialects.cc │ │ ├── register_common_dialects.h │ │ ├── register_common_dialects_test.cc │ │ ├── runlit.cfg.py │ │ ├── runlit.site.cfg.py │ │ ├── stablehlo │ │ │ ├── BUILD │ │ │ ├── stablehlo.cc │ │ │ ├── stablehlo.py │ │ │ └── stablehlo_test.py │ │ ├── tensorflow │ │ │ ├── BUILD │ │ │ ├── analysis │ │ │ │ ├── per_function_aggregate_analysis.h │ │ │ │ ├── resource_alias_analysis.cc │ │ │ │ ├── resource_alias_analysis.h │ │ │ │ ├── resource_dataflow.cc │ │ │ │ ├── resource_dataflow.h │ │ │ │ ├── resource_value_typed_analyzer.cc │ │ │ │ ├── resource_value_typed_analyzer.h │ │ │ │ ├── side_effect_analysis.cc │ │ │ │ └── side_effect_analysis.h │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ ├── c_api_unified_experimental_mlir.cc │ │ │ │ └── c_api_unified_experimental_mlir_registration.cc │ │ │ ├── dialect_registration.h │ │ │ ├── g3doc │ │ │ │ ├── enable_mlir_bridge.md │ │ │ │ ├── images │ │ │ │ │ └── space_to_depth_transform.png │ │ │ │ ├── space_to_depth.md │ │ │ │ └── tf_dialects.md │ │ │ ├── ir │ │ │ │ ├── tf_arith_ops_folder.cc │ │ │ │ ├── tf_arith_ops_folder.h │ │ │ │ ├── tf_attributes.h │ │ │ │ ├── tf_device.cc │ │ │ │ ├── tf_device.h │ │ │ │ ├── tf_device_ops.td │ │ │ │ ├── tf_dialect.h │ │ │ │ ├── tf_executor.cc │ │ │ │ ├── tf_executor.h │ │ │ │ ├── tf_executor_ops.td │ │ │ │ ├── tf_generated_ops.td │ │ │ │ ├── tf_op_base.td │ │ │ │ ├── tf_op_interfaces.cc │ │ │ │ ├── tf_op_interfaces.h │ │ │ │ ├── tf_op_interfaces.td │ │ │ │ ├── tf_ops.cc │ │ │ │ ├── tf_ops.h │ │ │ │ ├── tf_ops.td │ │ │ │ ├── tf_ops_a_m.cc │ │ │ │ ├── tf_ops_a_m.h │ │ │ │ ├── tf_ops_canonicalization_helper.h │ │ │ │ ├── tf_ops_device_helper.cc │ │ │ │ ├── tf_ops_device_helper.h │ │ │ │ ├── tf_ops_layout_helper.cc │ │ │ │ ├── tf_ops_layout_helper.h │ │ │ │ ├── tf_ops_n_z.cc │ │ │ │ ├── tf_ops_n_z.h │ │ │ │ ├── tf_ops_tensor_helper.cc │ │ │ │ ├── tf_ops_tensor_helper.h │ │ │ │ ├── tf_remaining_ops.cc │ │ │ │ ├── tf_remaining_ops.h │ │ │ │ ├── tf_saved_model.cc │ │ │ │ ├── tf_saved_model.h │ │ │ │ ├── tf_saved_model_ops.td │ │ │ │ ├── tf_saved_model_test.cc │ │ │ │ ├── tf_side_effects.h │ │ │ │ ├── tf_structs.cc │ │ │ │ ├── tf_structs.h │ │ │ │ ├── tf_traits.h │ │ │ │ ├── tf_types.def │ │ │ │ ├── tf_types.h │ │ │ │ ├── tf_verifiers.cc │ │ │ │ ├── tf_verifiers.h │ │ │ │ ├── tfrt_ops.cc │ │ │ │ ├── tfrt_ops.h │ │ │ │ ├── tfrt_ops.td │ │ │ │ ├── tpu_embedding_ops_registry.cc │ │ │ │ └── tpu_embedding_ops_registry.h │ │ │ ├── ops │ │ │ │ ├── mlir_local_var_op.cc │ │ │ │ └── mlir_passthrough_op.cc │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── add_functions_for_exported_names.mlir │ │ │ │ ├── annotate-parameter-replication.mlir │ │ │ │ ├── batchmatmul_to_einsum.mlir │ │ │ │ ├── breakup-islands.mlir │ │ │ │ ├── cannonicalize_ops_outside_compilation.mlir │ │ │ │ ├── canonicalize.mlir │ │ │ │ ├── canonicalize_compile_and_replicate_attributes.mlir │ │ │ │ ├── check_control_dependencies.mlir │ │ │ │ ├── cluster_formation.mlir │ │ │ │ ├── cluster_ops_by_policy.mlir │ │ │ │ ├── cluster_outlining.mlir │ │ │ │ ├── cluster_tf_ops_pass.mlir │ │ │ │ ├── colocate_tpu_copy_with_dynamic_shape.mlir │ │ │ │ ├── compile_mlir_util │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add.mlir │ │ │ │ │ ├── argument-sharding-invalid.mlir │ │ │ │ │ ├── argument-sharding.mlir │ │ │ │ │ ├── constant-folding-hook.mlir │ │ │ │ │ ├── constant-folding.mlir │ │ │ │ │ ├── convert_mhlo_quant_to_int.mlir │ │ │ │ │ ├── graph-resource.mlir │ │ │ │ │ ├── graph-resource.pbtxt │ │ │ │ │ ├── graph.pbtxt │ │ │ │ │ ├── mlir-module-serialized-str-attr.mlir │ │ │ │ │ ├── replicate-tensor-list-init-ops.mlir │ │ │ │ │ ├── result-sharding.mlir │ │ │ │ │ ├── serialized-mlir-module-str-attr-invalid.mlir │ │ │ │ │ ├── serialized-mlir-module-str-attr.mlir │ │ │ │ │ ├── shape-inference-after-legalization.mlir │ │ │ │ │ ├── shape-inference.mlir │ │ │ │ │ └── stablehlo_add.mlir │ │ │ │ ├── constant-fold.mlir │ │ │ │ ├── constant_op_device_assignment.mlir │ │ │ │ ├── convert-tf-control-flow-to-scf.mlir │ │ │ │ ├── convert_control_to_data_outputs.mlir │ │ │ │ ├── convert_launch_func_to_tf_call.mlir │ │ │ │ ├── convert_session_initializer_to_function.mlir │ │ │ │ ├── convert_to_legacy_compile_and_replicate_attributes.mlir │ │ │ │ ├── decompose_reduce_dataset.mlir │ │ │ │ ├── decompose_resource_ops.mlir │ │ │ │ ├── device_assignment.mlir │ │ │ │ ├── device_assignment_by_func_attr.mlir │ │ │ │ ├── device_attribute_to_launch.mlir │ │ │ │ ├── device_canonicalize.mlir │ │ │ │ ├── device_copy.mlir │ │ │ │ ├── drop_while_shape_invariant.mlir │ │ │ │ ├── einsum.mlir │ │ │ │ ├── embedding_pipelining.mlir │ │ │ │ ├── embedding_program_key.mlir │ │ │ │ ├── embedding_sequencing.mlir │ │ │ │ ├── empty-main.mlir │ │ │ │ ├── end-to-end-tpu-reshard-variables.mlir │ │ │ │ ├── executor_canonicalize.mlir │ │ │ │ ├── executor_island_coarsening.mlir │ │ │ │ ├── executor_island_materialize_const.mlir │ │ │ │ ├── executor_tpuv1_island_coarsening │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── executor_tpuv1_island_coarsening.mlir │ │ │ │ │ └── while_op.mlir │ │ │ │ ├── executor_tpuv1_island_inlining │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── executor_tpuv1_inline_tpu_island.mlir │ │ │ │ │ └── while_op.mlir │ │ │ │ ├── executor_tpuv1_outline_island │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── case_op.mlir │ │ │ │ │ ├── executor_tpuv1_outline_tpu_island.mlir │ │ │ │ │ └── while_op.mlir │ │ │ │ ├── extract_head_tail_outside_compilation.mlir │ │ │ │ ├── extract_outside_compilation.mlir │ │ │ │ ├── extract_tpu_copy_with_dynamic_shape_op.mlir │ │ │ │ ├── fold-broadcast.mlir │ │ │ │ ├── freeze_variables.mlir │ │ │ │ ├── func-attr-invalid.mlir │ │ │ │ ├── func-attr.mlir │ │ │ │ ├── functional-control-flow-to-cfg.mlir │ │ │ │ ├── functional-control-flow-to-regions.mlir │ │ │ │ ├── functionalize-if-fail.mlir │ │ │ │ ├── functionalize-if.mlir │ │ │ │ ├── fused_kernel_matcher.mlir │ │ │ │ ├── gpu_fusion.mlir │ │ │ │ ├── graph_pruning.mlir │ │ │ │ ├── graph_pruning_preserve_ops.mlir │ │ │ │ ├── graphdef2mlir │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add.pbtxt │ │ │ │ │ ├── arg-as-fetch.pbtxt │ │ │ │ │ ├── arg-control-dep.pbtxt │ │ │ │ │ ├── arg-data-type-with-subtype.pbtxt │ │ │ │ │ ├── arg-data-type.pbtxt │ │ │ │ │ ├── arg-multi-data-type-with-subtype.pbtxt │ │ │ │ │ ├── arg-retval-attrs.pbtxt │ │ │ │ │ ├── batch_use_same_function │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ └── saved_model.pbtxt │ │ │ │ │ ├── case_op.pbtxt │ │ │ │ │ ├── const-values.pbtxt │ │ │ │ │ ├── device-arg-retval-attr.pbtxt │ │ │ │ │ ├── empty-input-shapes.pbtxt │ │ │ │ │ ├── empty-value-attr.pbtxt │ │ │ │ │ ├── error-message-with-source-info.pbtxt │ │ │ │ │ ├── error-message-with-source-info.pbtxt.debug │ │ │ │ │ ├── error-message-with-source-info.pbtxt.fake_py.debug │ │ │ │ │ ├── feed-as-fetch.pbtxt │ │ │ │ │ ├── feed-control-dep.pbtxt │ │ │ │ │ ├── force_shared_name_for_resource_ops.pbtxt │ │ │ │ │ ├── function-func-attr.pbtxt │ │ │ │ │ ├── functional-if-ops.pbtxt │ │ │ │ │ ├── functional-while-ops.pbtxt │ │ │ │ │ ├── graph-as-function-control-ret.pbtxt │ │ │ │ │ ├── graph-as-function-retval-of-arg.pbtxt │ │ │ │ │ ├── graph-as-function.pbtxt │ │ │ │ │ ├── graph-custom-operation.pbtxt │ │ │ │ │ ├── graph-default-attr.pbtxt │ │ │ │ │ ├── graph-device-retval.pbtxt │ │ │ │ │ ├── graph-empty-tensor-content.pbtxt │ │ │ │ │ ├── graph-func-attr.pbtxt │ │ │ │ │ ├── graph-function-call.pbtxt │ │ │ │ │ ├── graph-function-control-ret-diff-island.pbtxt │ │ │ │ │ ├── graph-function-control-ret-same-island.pbtxt │ │ │ │ │ ├── graph-function-defs.pbtxt │ │ │ │ │ ├── graph-function-input-shapes.pbtxt │ │ │ │ │ ├── graph-function-name-bug.pbtxt │ │ │ │ │ ├── graph-function-resource-args.pbtxt │ │ │ │ │ ├── graph-gradient-def.pbtxt │ │ │ │ │ ├── graph-input-func-arg-name-collision.pbtxt │ │ │ │ │ ├── graph-library.pbtxt │ │ │ │ │ ├── graph-malformed.pbtxt │ │ │ │ │ ├── graph-scalar-input.pbtxt │ │ │ │ │ ├── graph-uint8-return.pbtxt │ │ │ │ │ ├── graph-undefined-output.pbtxt │ │ │ │ │ ├── graph-version-info.pbtxt │ │ │ │ │ ├── graph-while-loop.pbtxt │ │ │ │ │ ├── invalid-output-index.pbtxt │ │ │ │ │ ├── legacy-fed-input-without-inputs.pbtxt │ │ │ │ │ ├── merge_node_with_function.pbtxt │ │ │ │ │ ├── mlir_passthrough_op.pbtxt │ │ │ │ │ ├── multi-output-feeds.pbtxt │ │ │ │ │ ├── multiple-use-next-iteration.pbtxt │ │ │ │ │ ├── node-locations.pbtxt │ │ │ │ │ ├── output-shapes-attr.pbtxt │ │ │ │ │ ├── output-shapes.pbtxt │ │ │ │ │ ├── parse_example.pbtxt │ │ │ │ │ ├── parse_example_v2.pbtxt │ │ │ │ │ ├── partial-device-name.pbtxt │ │ │ │ │ ├── prune_unused_nodes.pbtxt │ │ │ │ │ ├── quint8-const.pbtxt │ │ │ │ │ ├── shape-attrs.pbtxt │ │ │ │ │ ├── stateful-attribute.pbtxt │ │ │ │ │ ├── string-attr.pbtxt │ │ │ │ │ ├── switch_n.pbtxt │ │ │ │ │ ├── target.pbtxt │ │ │ │ │ ├── tensor-list.pbtxt │ │ │ │ │ ├── tf-data-pipeline.pbtxt │ │ │ │ │ └── unregistered_kernel.pbtxt │ │ │ │ ├── group_by_dialect.mlir │ │ │ │ ├── guarantee-all-funcs-one-use.mlir │ │ │ │ ├── hoist_loop_invariant.mlir │ │ │ │ ├── hoist_replicate_invariant_resource_writes.mlir │ │ │ │ ├── host_launch_to_outside_compiled.mlir │ │ │ │ ├── init_text_file_to_import.mlir │ │ │ │ ├── init_text_file_to_import_invalid.mlir │ │ │ │ ├── init_text_file_to_import_saved_model.mlir │ │ │ │ ├── inlining.mlir │ │ │ │ ├── isolate-placer.mlir │ │ │ │ ├── launch_outlining.mlir │ │ │ │ ├── launch_to_device_attribute.mlir │ │ │ │ ├── launch_to_device_attribute_legacy.mlir │ │ │ │ ├── layout_optimization_layout_assignment_gpu_cc_60.mlir │ │ │ │ ├── layout_optimization_layout_assignment_gpu_cc_70.mlir │ │ │ │ ├── layout_optimization_layout_assignment_to_nchw.mlir │ │ │ │ ├── layout_optimization_layout_assignment_to_nhwc.mlir │ │ │ │ ├── layout_optimization_move_transposes_begin.mlir │ │ │ │ ├── layout_optimization_move_transposes_end.mlir │ │ │ │ ├── layout_optimization_to_nchw.mlir │ │ │ │ ├── layout_optimization_to_nhwc.mlir │ │ │ │ ├── legalize_tfg.mlir │ │ │ │ ├── legalize_tfg_arg_control_dep.mlir │ │ │ │ ├── legalize_tfg_with_control_flow.mlir │ │ │ │ ├── localize_var_handles.mlir │ │ │ │ ├── lower_globals_to_ml_program.mlir │ │ │ │ ├── lower_globals_to_ml_program_invalid.mlir │ │ │ │ ├── lower_quantized.mlir │ │ │ │ ├── lower_tf.mlir │ │ │ │ ├── lower_variable_ops_to_ml_program.mlir │ │ │ │ ├── mark_input_output_aliases.mlir │ │ │ │ ├── mark_ops_for_outside_compilation.mlir │ │ │ │ ├── materialize_passthrough_op.mlir │ │ │ │ ├── merge_control_flow.mlir │ │ │ │ ├── mlir2graph │ │ │ │ │ ├── BUILD │ │ │ │ │ └── convert_tensor.mlir │ │ │ │ ├── mlir2graphdef │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── aliasing_arg_attr.mlir │ │ │ │ │ ├── case.mlir │ │ │ │ │ ├── convert_tensor.mlir │ │ │ │ │ ├── derived_shape_attr.mlir │ │ │ │ │ ├── derived_size_attr.mlir │ │ │ │ │ ├── device-arg-retval-attr.mlir │ │ │ │ │ ├── export_main_to_flib.mlir │ │ │ │ │ ├── fetch_feed_names.mlir │ │ │ │ │ ├── func_attr.mlir │ │ │ │ │ ├── func_list_attr.mlir │ │ │ │ │ ├── function-control-ret.mlir │ │ │ │ │ ├── function-order.mlir │ │ │ │ │ ├── function-resource-args-handle-info.mlir │ │ │ │ │ ├── function-resource-args.mlir │ │ │ │ │ ├── functional-if-ops.mlir │ │ │ │ │ ├── functional-while-ops.mlir │ │ │ │ │ ├── graph-as-function.mlir │ │ │ │ │ ├── infer_derived_attribute.mlir │ │ │ │ │ ├── invalid_input.mlir │ │ │ │ │ ├── legalized_name.mlir │ │ │ │ │ ├── missing-main.mlir │ │ │ │ │ ├── noop.mlir │ │ │ │ │ ├── optional_symbol_ref.mlir │ │ │ │ │ ├── output-shapes-attr.mlir │ │ │ │ │ ├── parse_example.mlir │ │ │ │ │ ├── parse_example_v2.mlir │ │ │ │ │ ├── preserve-entry-func-names.mlir │ │ │ │ │ ├── ref-type-attr.mlir │ │ │ │ │ ├── ref-while-loop.mlir │ │ │ │ │ ├── shape_list_attr.mlir │ │ │ │ │ ├── simple.mlir │ │ │ │ │ ├── simple_tf_dialect_op.mlir │ │ │ │ │ ├── stringescape.mlir │ │ │ │ │ ├── switchn.mlir │ │ │ │ │ ├── tf-gradient-attr.mlir │ │ │ │ │ ├── tf-legacy-call.mlir │ │ │ │ │ ├── tf_add.mlir │ │ │ │ │ ├── tf_identity_n.mlir │ │ │ │ │ ├── tf_tpu_embedding_ops.mlir │ │ │ │ │ ├── type_attr.mlir │ │ │ │ │ ├── type_list_attr.mlir │ │ │ │ │ ├── unique_name.mlir │ │ │ │ │ ├── unique_output_name.mlir │ │ │ │ │ └── while-loop.mlir │ │ │ │ ├── mlprogram.mlir │ │ │ │ ├── move_tpu_compile_to_front.mlir │ │ │ │ ├── name_anonymous_iterators.mlir │ │ │ │ ├── optimize-arg-operand-constraint.mlir │ │ │ │ ├── optimize.mlir │ │ │ │ ├── order_by_dialect.mlir │ │ │ │ ├── outside_compiled_to_host_launch.mlir │ │ │ │ ├── parallel_execute_to_islands.mlir │ │ │ │ ├── parallel_execute_to_islands_legacy.mlir │ │ │ │ ├── prepare_tpu_computation_for_tf_export.mlir │ │ │ │ ├── print.mlir │ │ │ │ ├── promote_resources_to_args.mlir │ │ │ │ ├── promote_resources_to_args_functions.mlir │ │ │ │ ├── promote_var_handles_to_args.mlir │ │ │ │ ├── readonly_references_to_resources.mlir │ │ │ │ ├── reducer │ │ │ │ │ └── unsupported-op-test.sh │ │ │ │ ├── region-control-flow-to-functional.mlir │ │ │ │ ├── remove_unused_arguments.mlir │ │ │ │ ├── remove_unused_while_results.mlir │ │ │ │ ├── replica_id_to_device_ordinal.mlir │ │ │ │ ├── replicate_invariant_op_hoisting.mlir │ │ │ │ ├── replicate_tensor_list_init_ops.mlir │ │ │ │ ├── replicate_to_island.mlir │ │ │ │ ├── replicate_to_island_legacy.mlir │ │ │ │ ├── resource-alias-analysis-test.mlir │ │ │ │ ├── resource-device-inference.mlir │ │ │ │ ├── resource_analyzer.mlir │ │ │ │ ├── resource_inlining.mlir │ │ │ │ ├── resource_op_lifting.mlir │ │ │ │ ├── rewrite_tpu_embedding_ops.mlir │ │ │ │ ├── roundtrip-tf-executor.mlir │ │ │ │ ├── shape_inference.mlir │ │ │ │ ├── side-effect-analysis-test.mlir │ │ │ │ ├── sink_constant.mlir │ │ │ │ ├── split_into_island_per_op.mlir │ │ │ │ ├── stack_ops_decomposition.mlir │ │ │ │ ├── strip_noinline.mlir │ │ │ │ ├── strip_saved_module_metadata.mlir │ │ │ │ ├── strip_tf_attributes.mlir │ │ │ │ ├── tensor_array_ops_decomposition.mlir │ │ │ │ ├── tensor_list_ops_decomposition.mlir │ │ │ │ ├── tf-executor-to-functional.mlir │ │ │ │ ├── tf-functional-to-executor.mlir │ │ │ │ ├── tf-ops.mlir │ │ │ │ ├── tf-reduce-identity.mlir │ │ │ │ ├── tf_data_fuse_map_and_batch.mlir │ │ │ │ ├── tf_data_fuse_pmap_and_batch.mlir │ │ │ │ ├── tf_device_index_selector.mlir │ │ │ │ ├── tf_device_ops.mlir │ │ │ │ ├── tf_device_ops_invalid.mlir │ │ │ │ ├── tf_executor_ops.mlir │ │ │ │ ├── tf_executor_ops_invalid.mlir │ │ │ │ ├── tf_executor_ops_location_roundtrip.mlir │ │ │ │ ├── tf_executor_ops_printer.mlir │ │ │ │ ├── tf_executor_ops_side_effect.mlir │ │ │ │ ├── tf_optimize.mlir │ │ │ │ ├── tf_saved_model │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── basic.py │ │ │ │ │ ├── basic_v1.py │ │ │ │ │ ├── basic_v1_no_variable_lifting.py │ │ │ │ │ ├── call_to_exported.py │ │ │ │ │ ├── common.py │ │ │ │ │ ├── common_v1.py │ │ │ │ │ ├── control_flow_duplicate_v1.py │ │ │ │ │ ├── control_flow_upgrade_legacy_v1.py │ │ │ │ │ ├── cyclic_object_graph.py │ │ │ │ │ ├── dag_object_graph.py │ │ │ │ │ ├── debug_info.py │ │ │ │ │ ├── defun_export.py │ │ │ │ │ ├── duplicate_method_names_v1.py │ │ │ │ │ ├── exported_python_args.py │ │ │ │ │ ├── hash_table_asset_v1.py │ │ │ │ │ ├── hash_table_v1.py │ │ │ │ │ ├── import_restore_v1.py │ │ │ │ │ ├── include_variables_in_init_v1.py │ │ │ │ │ ├── keras.py │ │ │ │ │ ├── multi_arguments_results_v1.py │ │ │ │ │ ├── multi_variables_v1.py │ │ │ │ │ ├── no_input_shape_v1.py │ │ │ │ │ ├── partially_shaped_variables.py │ │ │ │ │ ├── remove_init_variable_v1.py │ │ │ │ │ ├── shapes_for_arguments.py │ │ │ │ │ ├── shared_variable_v1.py │ │ │ │ │ ├── structured_input.py │ │ │ │ │ └── structured_output.py │ │ │ │ ├── tf_saved_model_asset_sinking.mlir │ │ │ │ ├── tf_saved_model_deduplicate_bound_input_bindings.mlir │ │ │ │ ├── tf_saved_model_freeze_assets.mlir │ │ │ │ ├── tf_saved_model_freeze_global_tensors.mlir │ │ │ │ ├── tf_saved_model_freeze_global_tensors_mutable_tensors.mlir │ │ │ │ ├── tf_saved_model_initialize_variables_in_session_init.mlir │ │ │ │ ├── tf_saved_model_initialize_variables_in_session_init_fail.mlir │ │ │ │ ├── tf_saved_model_lift_variables.mlir │ │ │ │ ├── tf_saved_model_lift_variables_invalid_session.mlir │ │ │ │ ├── tf_saved_model_mark_initialized_variables.mlir │ │ │ │ ├── tf_saved_model_ops.mlir │ │ │ │ ├── tf_saved_model_ops_invalid.mlir │ │ │ │ ├── tf_saved_model_optimize_global_tensors.mlir │ │ │ │ ├── tf_saved_model_optimize_global_tensors_interprocedural.mlir │ │ │ │ ├── tf_saved_model_remove_vars_in_session_initializer.mlir │ │ │ │ ├── tf_side_effect.mlir │ │ │ │ ├── tf_to_hlo_pipeline │ │ │ │ │ ├── BUILD │ │ │ │ │ └── sccp-post-shape-inference.mlir │ │ │ │ ├── tf_trait_folds.mlir │ │ │ │ ├── tfrt_ops.mlir │ │ │ │ ├── tpu-annotate-dynamic-shape-inputs.mlir │ │ │ │ ├── tpu-cluster-cleanup-attributes.mlir │ │ │ │ ├── tpu-dynamic-layout-pass.mlir │ │ │ │ ├── tpu-merge-variables-with-execute.mlir │ │ │ │ ├── tpu-multiple-while-body-func.mlir │ │ │ │ ├── tpu-resource-read-for-write.mlir │ │ │ │ ├── tpu-variable-runtime-reformatting.mlir │ │ │ │ ├── tpu_bridge_v1 │ │ │ │ │ ├── BUILD │ │ │ │ │ └── end_to_end.mlir │ │ │ │ ├── tpu_cluster_formation.mlir │ │ │ │ ├── tpu_colocate_composite_resource_ops.mlir │ │ │ │ ├── tpu_colocate_splits.mlir │ │ │ │ ├── tpu_device_propagation.mlir │ │ │ │ ├── tpu_host_computation_expansion.mlir │ │ │ │ ├── tpu_identity_pruning.mlir │ │ │ │ ├── tpu_parallel_execute_sink_resource_write.mlir │ │ │ │ ├── tpu_partitioned_op_conversion.mlir │ │ │ │ ├── tpu_reorder_replicate_and_partitioned_inputs.mlir │ │ │ │ ├── tpu_resource_partitioning.mlir │ │ │ │ ├── tpu_rewrite.mlir │ │ │ │ ├── tpu_sharding_identification.mlir │ │ │ │ ├── tpu_space_to_depth_pass.mlir │ │ │ │ ├── tpu_tail_with_tobool_op.mlir │ │ │ │ ├── tpu_update_embedding_enqueue_op_inputs.mlir │ │ │ │ ├── tpu_validate_inputs.mlir │ │ │ │ ├── transpose-op.mlir │ │ │ │ ├── unroll-batch-matmul.mlir │ │ │ │ ├── update_control_dependencies.mlir │ │ │ │ ├── verify_for_export.mlir │ │ │ │ ├── warn_when_using_deprecated_dumps.mlir │ │ │ │ ├── while_licm.mlir │ │ │ │ ├── xla_call_module_deserialization.mlir │ │ │ │ ├── xla_call_module_round_trip.mlir │ │ │ │ ├── xla_call_module_serialization.mlir │ │ │ │ ├── xla_cluster_formation.mlir │ │ │ │ ├── xla_inline_device_ops.mlir │ │ │ │ ├── xla_rewrite.mlir │ │ │ │ ├── xla_rewrite_v2.mlir │ │ │ │ ├── xla_sharding_util_test.cc │ │ │ │ └── xla_validate_iputs.mlir │ │ │ ├── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── add_functions_for_exported_names.cc │ │ │ │ ├── annotate_parameter_replication.cc │ │ │ │ ├── batchmatmul_to_einsum.cc │ │ │ │ ├── breakup-islands.cc │ │ │ │ ├── bridge.cc │ │ │ │ ├── bridge.h │ │ │ │ ├── canonicalize.td │ │ │ │ ├── canonicalize_compile_and_replicate_attributes.cc │ │ │ │ ├── check_control_dependencies.cc │ │ │ │ ├── cluster_formation.cc │ │ │ │ ├── cluster_ops_by_policy.cc │ │ │ │ ├── cluster_ops_by_policy.h │ │ │ │ ├── cluster_outlining.cc │ │ │ │ ├── cluster_tf_ops_pass.cc │ │ │ │ ├── collection_ops_util.cc │ │ │ │ ├── collection_ops_util.h │ │ │ │ ├── colocate_tpu_copy_with_dynamic_shape.cc │ │ │ │ ├── constant_fold.cc │ │ │ │ ├── constant_fold.h │ │ │ │ ├── constant_fold_utils.cc │ │ │ │ ├── constant_fold_utils.h │ │ │ │ ├── constant_op_device_assignment.cc │ │ │ │ ├── convert_control_to_data_outputs.cc │ │ │ │ ├── convert_launch_func_to_tf_call.cc │ │ │ │ ├── convert_session_initializer_to_function.cc │ │ │ │ ├── convert_tf_control_flow_to_scf.cc │ │ │ │ ├── convert_to_legacy_compile_and_replicate_attributes.cc │ │ │ │ ├── decode_attributes_hook.cc │ │ │ │ ├── decompose_reduce_dataset.cc │ │ │ │ ├── decompose_resource_ops.cc │ │ │ │ ├── decompose_resource_ops.h │ │ │ │ ├── decompose_resource_ops.td │ │ │ │ ├── decompose_resource_ops_pass.cc │ │ │ │ ├── deduplicate_bound_input_bindings.cc │ │ │ │ ├── device_attribute_to_launch.cc │ │ │ │ ├── device_index_selector.cc │ │ │ │ ├── drop_while_shape_invariant.cc │ │ │ │ ├── einsum.cc │ │ │ │ ├── einsum.h │ │ │ │ ├── embedding_pipelining.cc │ │ │ │ ├── embedding_program_key.cc │ │ │ │ ├── embedding_sequencing.cc │ │ │ │ ├── executor_island_coarsening.cc │ │ │ │ ├── executor_tpuv1_inline_tpu_island.cc │ │ │ │ ├── executor_tpuv1_island_coarsening.cc │ │ │ │ ├── executor_tpuv1_outline_tpu_island.cc │ │ │ │ ├── extract_tpu_copy_with_dynamic_shape_op.cc │ │ │ │ ├── fold_broadcast.cc │ │ │ │ ├── freeze_global_tensors.cc │ │ │ │ ├── freeze_saved_model_assets.cc │ │ │ │ ├── functional_control_flow_to_cfg.cc │ │ │ │ ├── functional_control_flow_to_regions.cc │ │ │ │ ├── fused_kernel_matcher.cc │ │ │ │ ├── gpu_fusion.cc │ │ │ │ ├── graph_optimization_pass.cc │ │ │ │ ├── graph_optimization_pass.h │ │ │ │ ├── graph_optimization_pass_registration.cc │ │ │ │ ├── graph_pruning.cc │ │ │ │ ├── group_by_dialect.cc │ │ │ │ ├── group_by_dialect.h │ │ │ │ ├── guarantee_all_funcs_one_use.cc │ │ │ │ ├── hoist_loop_invariant.cc │ │ │ │ ├── hoist_replicate_invariant_resource_writes.cc │ │ │ │ ├── host_launch_to_outside_compiled.cc │ │ │ │ ├── host_runtime │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── lower_cluster_to_runtime_ops.cc │ │ │ │ │ ├── lower_cluster_to_runtime_ops.h │ │ │ │ │ ├── lower_cluster_to_runtime_ops_test.cc │ │ │ │ │ └── testdata │ │ │ │ │ │ ├── basic_cluster.mlir │ │ │ │ │ │ ├── empty_func.mlir │ │ │ │ │ │ └── malformed_cluster.mlir │ │ │ │ ├── init_text_file_to_import.cc │ │ │ │ ├── init_text_file_to_import_test_pass.cc │ │ │ │ ├── initialize_variables_in_session_init.cc │ │ │ │ ├── initialize_variables_in_session_init.h │ │ │ │ ├── initialize_variables_in_session_init_test_pass.cc │ │ │ │ ├── launch_to_device_attribute.cc │ │ │ │ ├── layout_optimization.cc │ │ │ │ ├── lift_variables.cc │ │ │ │ ├── lift_variables.h │ │ │ │ ├── lift_variables_test_pass.cc │ │ │ │ ├── lift_variables_test_pass.h │ │ │ │ ├── localize_var_handles.cc │ │ │ │ ├── lower_globals_to_ml_program.cc │ │ │ │ ├── lower_globals_to_ml_program.h │ │ │ │ ├── lower_quantized.cc │ │ │ │ ├── lower_tf.cc │ │ │ │ ├── lower_tf.h │ │ │ │ ├── lower_tf.td │ │ │ │ ├── lower_tf_test_pass.cc │ │ │ │ ├── lower_variable_ops_to_ml_program.cc │ │ │ │ ├── mark_initialized_variables.cc │ │ │ │ ├── mark_initialized_variables.h │ │ │ │ ├── mark_initialized_variables_test_pass.cc │ │ │ │ ├── mark_input_output_aliases.cc │ │ │ │ ├── materialize_mlir_passthrough_op.cc │ │ │ │ ├── merge_control_flow.cc │ │ │ │ ├── mlprogram.cc │ │ │ │ ├── mlprogram.h │ │ │ │ ├── move_tpu_compile_to_front.cc │ │ │ │ ├── name_anonymous_iterators.cc │ │ │ │ ├── optimize.cc │ │ │ │ ├── optimize.td │ │ │ │ ├── optimize_global_tensors.cc │ │ │ │ ├── order_by_dialect.cc │ │ │ │ ├── order_by_dialect.h │ │ │ │ ├── outside_compiled_to_host_launch.cc │ │ │ │ ├── parallel_execute_to_islands.cc │ │ │ │ ├── passes.h │ │ │ │ ├── prepare_tpu_computation_for_tf_export.cc │ │ │ │ ├── print.cc │ │ │ │ ├── promote_resources_to_args.cc │ │ │ │ ├── readonly_references_to_resources.cc │ │ │ │ ├── reducer │ │ │ │ │ └── tf_mlir_reduce_patterns.td │ │ │ │ ├── region_control_flow_to_functional.cc │ │ │ │ ├── remove_unused_arguments.cc │ │ │ │ ├── remove_unused_while_results.cc │ │ │ │ ├── remove_vars_in_session_initializer.cc │ │ │ │ ├── replica_id_to_device_ordinal.cc │ │ │ │ ├── replicate_invariant_op_hoisting.cc │ │ │ │ ├── replicate_tensor_list_init_ops_pass.cc │ │ │ │ ├── replicate_to_island.cc │ │ │ │ ├── resource_analyzer_test_pass.cc │ │ │ │ ├── resource_device_inference.cc │ │ │ │ ├── resource_op_lifting.cc │ │ │ │ ├── resource_op_lifting_cleanup.cc │ │ │ │ ├── resource_op_lifting_cleanup.h │ │ │ │ ├── rewrite_tpu_embedding_ops.cc │ │ │ │ ├── rewrite_util.cc │ │ │ │ ├── rewrite_util.h │ │ │ │ ├── rewrite_util.td │ │ │ │ ├── set_tpu_infeed_layout.cc │ │ │ │ ├── set_tpu_infeed_layout.h │ │ │ │ ├── shape_inference.cc │ │ │ │ ├── shape_inference.h │ │ │ │ ├── shape_inference_pass.cc │ │ │ │ ├── sink_constant.cc │ │ │ │ ├── stack_ops_decomposition.cc │ │ │ │ ├── strip_noinline_attribute.cc │ │ │ │ ├── strip_saved_module_metadata.cc │ │ │ │ ├── strip_tf_attributes.cc │ │ │ │ ├── tensor_array_ops_decomposition.cc │ │ │ │ ├── tensor_device_copy_conversion.cc │ │ │ │ ├── tensor_list_ops_decomposition.cc │ │ │ │ ├── test_cluster_ops_by_policy.cc │ │ │ │ ├── test_passes.h │ │ │ │ ├── test_resource_alias_analysis.cc │ │ │ │ ├── test_side_effect_analysis.cc │ │ │ │ ├── tf_data_optimization.cc │ │ │ │ ├── tf_data_optimization.h │ │ │ │ ├── tf_data_optimization.td │ │ │ │ ├── tf_data_optimization_pass.cc │ │ │ │ ├── tf_device_assignment.cc │ │ │ │ ├── tf_device_passes.td │ │ │ │ ├── tf_executor_to_functional.cc │ │ │ │ ├── tf_functional_to_executor.cc │ │ │ │ ├── tf_graph_optimization_pass.cc │ │ │ │ ├── tf_graph_optimization_pass.h │ │ │ │ ├── tf_passes.td │ │ │ │ ├── tf_saved_model_asset_sinking_pass.cc │ │ │ │ ├── tf_saved_model_asset_sinking_pass.h │ │ │ │ ├── tf_saved_model_freeze_variables.cc │ │ │ │ ├── tf_saved_model_freeze_variables.h │ │ │ │ ├── tf_saved_model_freeze_variables_test_pass.cc │ │ │ │ ├── tf_saved_model_passes.h │ │ │ │ ├── tf_savedmodel_passes.td │ │ │ │ ├── tf_test_passes.td │ │ │ │ ├── tfg-to-tfe.cc │ │ │ │ ├── tpu_annotate_dynamic_shape_inputs.cc │ │ │ │ ├── tpu_cluster_cleanup_attributes.cc │ │ │ │ ├── tpu_colocate_composite_resource_ops.cc │ │ │ │ ├── tpu_colocate_splits.cc │ │ │ │ ├── tpu_device_propagation.cc │ │ │ │ ├── tpu_dynamic_layout_pass.cc │ │ │ │ ├── tpu_host_computation_expansion.cc │ │ │ │ ├── tpu_identity_pruning.cc │ │ │ │ ├── tpu_merge_variables_with_execute.cc │ │ │ │ ├── tpu_parallel_execute_sink_resource_write.cc │ │ │ │ ├── tpu_partitioned_op_conversion.cc │ │ │ │ ├── tpu_reorder_replicate_and_partitioned_inputs.cc │ │ │ │ ├── tpu_resource_partitioning.cc │ │ │ │ ├── tpu_resource_read_for_write.cc │ │ │ │ ├── tpu_rewrite_pass.cc │ │ │ │ ├── tpu_sharding_identification_pass.cc │ │ │ │ ├── tpu_space_to_depth_pass.cc │ │ │ │ ├── tpu_update_embedding_enqueue_op_inputs.cc │ │ │ │ ├── tpu_validate_inputs.cc │ │ │ │ ├── tpu_variable_runtime_reformatting.cc │ │ │ │ ├── unroll_batch_matmul.cc │ │ │ │ ├── unroll_batch_matmul.h │ │ │ │ ├── update_control_dependencies.cc │ │ │ │ ├── verify_no_outside_compilation_markers_pass.cc │ │ │ │ ├── verify_no_outside_compilation_markers_pass_test.cc │ │ │ │ ├── verify_suitable_for_graph_export_pass.cc │ │ │ │ ├── xla_call_module_deserialization.cc │ │ │ │ ├── xla_call_module_serialization.cc │ │ │ │ ├── xla_inline_device_ops.cc │ │ │ │ ├── xla_rewrite.cc │ │ │ │ ├── xla_rewrite_v2.cc │ │ │ │ └── xla_validate_inputs.cc │ │ │ ├── translate │ │ │ │ ├── export_graphdef.cc │ │ │ │ ├── export_graphdef.h │ │ │ │ ├── export_tf_dialect_op.cc │ │ │ │ ├── export_tf_dialect_op.h │ │ │ │ ├── import_model.cc │ │ │ │ ├── import_model.h │ │ │ │ ├── mlir_import_options.h │ │ │ │ ├── mlir_roundtrip_flags.cc │ │ │ │ ├── mlir_roundtrip_flags.h │ │ │ │ ├── mlir_roundtrip_pass.cc │ │ │ │ ├── mlir_roundtrip_pass.h │ │ │ │ ├── mlir_roundtrip_pass_registration.cc │ │ │ │ ├── split_into_island_per_op_pass.cc │ │ │ │ ├── split_into_island_per_op_pass.h │ │ │ │ ├── tf_mlir_translate.cc │ │ │ │ ├── tf_mlir_translate.h │ │ │ │ ├── tf_mlir_translate_cl.cc │ │ │ │ ├── tf_mlir_translate_cl.h │ │ │ │ ├── tf_mlir_translate_registration.cc │ │ │ │ ├── tf_mlir_translate_registration_test.cc │ │ │ │ ├── translate_tf_dialect_op.cc │ │ │ │ ├── upgrade_graph.cc │ │ │ │ └── upgrade_graph.h │ │ │ └── utils │ │ │ │ ├── attribute_utils.cc │ │ │ │ ├── attribute_utils.h │ │ │ │ ├── bridge_logger.cc │ │ │ │ ├── bridge_logger.h │ │ │ │ ├── bridge_logger_test.cc │ │ │ │ ├── call_graph_util.cc │ │ │ │ ├── call_graph_util.h │ │ │ │ ├── call_graph_util_test.cc │ │ │ │ ├── cluster_util.cc │ │ │ │ ├── cluster_util.h │ │ │ │ ├── cluster_util_test.cc │ │ │ │ ├── convert_attr.cc │ │ │ │ ├── convert_attr.h │ │ │ │ ├── convert_tensor.cc │ │ │ │ ├── convert_tensor.h │ │ │ │ ├── convert_tensor_test.cc │ │ │ │ ├── convert_type.cc │ │ │ │ ├── convert_type.h │ │ │ │ ├── convert_type_test.cc │ │ │ │ ├── data_dumper_logger_config.cc │ │ │ │ ├── data_dumper_logger_config.h │ │ │ │ ├── data_dumper_logger_config_test.cc │ │ │ │ ├── device_util.cc │ │ │ │ ├── device_util.h │ │ │ │ ├── device_util_test.cc │ │ │ │ ├── dump_graph.cc │ │ │ │ ├── dump_graph.h │ │ │ │ ├── dump_graph_test.cc │ │ │ │ ├── dump_mlir_util.cc │ │ │ │ ├── dump_mlir_util.h │ │ │ │ ├── dump_mlir_util_test.cc │ │ │ │ ├── dynamic_shape_utils.cc │ │ │ │ ├── dynamic_shape_utils.h │ │ │ │ ├── error_util.cc │ │ │ │ ├── error_util.h │ │ │ │ ├── error_util_test.cc │ │ │ │ ├── eval_util.cc │ │ │ │ ├── eval_util.h │ │ │ │ ├── export_utils.cc │ │ │ │ ├── export_utils.h │ │ │ │ ├── fake_session.cc │ │ │ │ ├── fake_session.h │ │ │ │ ├── import_utils.cc │ │ │ │ ├── import_utils.h │ │ │ │ ├── location_utils.cc │ │ │ │ ├── location_utils.h │ │ │ │ ├── mangling_util.cc │ │ │ │ ├── mangling_util.h │ │ │ │ ├── mlprogram_util.cc │ │ │ │ ├── mlprogram_util.h │ │ │ │ ├── parallel_execute_util.cc │ │ │ │ ├── parallel_execute_util.h │ │ │ │ ├── parse_text_proto.cc │ │ │ │ ├── parse_text_proto.h │ │ │ │ ├── serialize_mlir_module_utils.cc │ │ │ │ ├── serialize_mlir_module_utils.h │ │ │ │ ├── session_utils.cc │ │ │ │ ├── session_utils.h │ │ │ │ ├── shape_inference_utils.cc │ │ │ │ ├── shape_inference_utils.h │ │ │ │ ├── side_effect_analysis_util.cc │ │ │ │ ├── side_effect_analysis_util.h │ │ │ │ ├── stablehlo_custom_call.cc │ │ │ │ ├── stablehlo_custom_call.h │ │ │ │ ├── string_util.cc │ │ │ │ ├── string_util.h │ │ │ │ ├── tf_xla_mlir_translate.cc │ │ │ │ ├── topological_sort.cc │ │ │ │ ├── topological_sort.h │ │ │ │ ├── tpu_cluster_util.cc │ │ │ │ ├── tpu_cluster_util.h │ │ │ │ ├── tpu_rewrite_device_util.cc │ │ │ │ ├── tpu_rewrite_device_util.h │ │ │ │ ├── tpu_rewrite_device_util_test.cc │ │ │ │ ├── translate_utils.cc │ │ │ │ ├── translate_utils.h │ │ │ │ ├── verification_utils.cc │ │ │ │ ├── verification_utils.h │ │ │ │ ├── verify_suitable_for_graph_export.cc │ │ │ │ ├── verify_suitable_for_graph_export.h │ │ │ │ ├── visitor.cc │ │ │ │ ├── visitor.h │ │ │ │ ├── xla_call_module_attrs.h │ │ │ │ ├── xla_rewrite_util.cc │ │ │ │ ├── xla_rewrite_util.h │ │ │ │ ├── xla_rewrite_util_test.cc │ │ │ │ ├── xla_sharding_util.cc │ │ │ │ └── xla_sharding_util.h │ │ ├── tf2xla │ │ │ ├── BUILD │ │ │ ├── api │ │ │ │ ├── v1 │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── cluster_tf.cc │ │ │ │ │ ├── cluster_tf.h │ │ │ │ │ ├── cluster_tf_test.cc │ │ │ │ │ ├── compile_mlir_util.cc │ │ │ │ │ ├── compile_mlir_util.h │ │ │ │ │ ├── compile_mlir_util_test.cc │ │ │ │ │ ├── compile_tf_graph.cc │ │ │ │ │ ├── compile_tf_graph.h │ │ │ │ │ ├── compile_tf_graph_test.cc │ │ │ │ │ ├── testdata │ │ │ │ │ │ ├── empty_func.mlir │ │ │ │ │ │ ├── invalid_executor.mlir │ │ │ │ │ │ └── multiple_submodules.mlir │ │ │ │ │ ├── tf_dialect_to_executor.cc │ │ │ │ │ ├── tf_dialect_to_executor.h │ │ │ │ │ └── tf_dialect_to_executor_test.cc │ │ │ │ └── v2 │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── cluster_tf.cc │ │ │ │ │ ├── cluster_tf.h │ │ │ │ │ ├── cluster_tf_test.cc │ │ │ │ │ ├── device_type.proto │ │ │ │ │ ├── legalize_tf.cc │ │ │ │ │ ├── legalize_tf.h │ │ │ │ │ ├── legalize_tf_test.cc │ │ │ │ │ ├── testdata │ │ │ │ │ ├── empty_func.mlir │ │ │ │ │ ├── invalid_executor.mlir │ │ │ │ │ └── outside_compilation.mlir │ │ │ │ │ ├── tf_dialect_to_executor.cc │ │ │ │ │ ├── tf_dialect_to_executor.h │ │ │ │ │ └── tf_dialect_to_executor_test.cc │ │ │ ├── internal │ │ │ │ ├── BUILD │ │ │ │ ├── clustering_bridge_passes.cc │ │ │ │ ├── clustering_bridge_passes.h │ │ │ │ ├── clustering_bridge_passes_test.cc │ │ │ │ ├── compilation_timer.h │ │ │ │ ├── compilation_timer_test.cc │ │ │ │ ├── hlo_post_processing │ │ │ │ │ └── BUILD │ │ │ │ ├── inference │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── inference_metrics_pass.cc │ │ │ │ │ ├── inference_metrics_pass_test.cc │ │ │ │ │ ├── inference_passes.h │ │ │ │ │ └── inference_passes.td │ │ │ │ ├── legalize_tf_mlir.cc │ │ │ │ ├── legalize_tf_mlir.h │ │ │ │ ├── legalize_tf_mlir_test.cc │ │ │ │ ├── legalize_tf_to_hlo.cc │ │ │ │ ├── legalize_tf_to_hlo.h │ │ │ │ ├── legalize_tf_to_hlo_test.cc │ │ │ │ ├── logging_hooks.cc │ │ │ │ ├── logging_hooks.h │ │ │ │ ├── logging_hooks_test.cc │ │ │ │ ├── mlir_pass_instrumentation.cc │ │ │ │ ├── mlir_pass_instrumentation.h │ │ │ │ ├── mlir_pass_instrumentation_test.cc │ │ │ │ ├── passes │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── clustering_passes.h │ │ │ │ │ ├── clustering_passes.td │ │ │ │ │ ├── dialect_to_executor_passes.h │ │ │ │ │ ├── dialect_to_executor_passes.td │ │ │ │ │ ├── extract_head_tail_outside_compilation.cc │ │ │ │ │ ├── extract_outside_compilation.cc │ │ │ │ │ ├── mark_ops_for_outside_compilation.cc │ │ │ │ │ ├── tpu_cluster_formation.cc │ │ │ │ │ ├── verify_clustering_pass.cc │ │ │ │ │ ├── verify_clustering_pass_test.cc │ │ │ │ │ ├── verify_clustering_pass_test.mlir │ │ │ │ │ ├── verify_input_dialect_to_executor_pass.cc │ │ │ │ │ ├── verify_input_dialect_to_executor_pass_test.mlir │ │ │ │ │ └── xla_cluster_formation.cc │ │ │ │ ├── test_matchers.h │ │ │ │ ├── test_matchers_test.cc │ │ │ │ ├── testdata │ │ │ │ │ └── dead_const.mlir │ │ │ │ └── utils │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── dialect_detection_utils.cc │ │ │ │ │ ├── dialect_detection_utils.h │ │ │ │ │ └── dialect_detection_utils_test.cc │ │ │ ├── mlir_bridge_rollout_policy.cc │ │ │ ├── mlir_bridge_rollout_policy.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── adjust-layout.mlir │ │ │ │ ├── hlo_xla_runtime_pipeline.mlir │ │ │ │ ├── hlo_xla_sparsification.mlir │ │ │ │ ├── legalize-tf-BatchMatMulV2.mlir │ │ │ │ ├── legalize-tf-binary-elementwise.mlir │ │ │ │ ├── legalize-tf-collective.mlir │ │ │ │ ├── legalize-tf-communication.mlir │ │ │ │ ├── legalize-tf-include-tf2xla-fallback.mlir │ │ │ │ ├── legalize-tf-prefer-tf2xla.mlir │ │ │ │ ├── legalize-tf-quant.mlir │ │ │ │ ├── legalize-tf-with-tf2xla-hlo-importer.mlir │ │ │ │ ├── legalize-tf.mlir │ │ │ │ ├── tfxla_device_specific_transformations_cpu.mlir │ │ │ │ ├── tfxla_device_specific_transformations_gpu.mlir │ │ │ │ ├── verify-tfxla-legalization-no-chlo.mlir │ │ │ │ └── verify-tfxla-legalization.mlir │ │ │ └── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── infeed_ops_xla_adjust_layout.cc │ │ │ │ ├── legalization_op_config.cc │ │ │ │ ├── legalization_op_config.h │ │ │ │ ├── legalization_op_config_test.cc │ │ │ │ ├── legalize_tf.cc │ │ │ │ ├── legalize_tf_collective.cc │ │ │ │ ├── legalize_tf_communication.cc │ │ │ │ ├── legalize_tf_patterns.td │ │ │ │ ├── legalize_tf_with_tf2xla.cc │ │ │ │ ├── passes.h │ │ │ │ ├── test_utils.cc │ │ │ │ ├── test_utils.h │ │ │ │ ├── tf2xla_rewriter.cc │ │ │ │ ├── tf2xla_rewriter.h │ │ │ │ ├── tf2xla_rewriter_test.cc │ │ │ │ ├── tf_xla_passes.td │ │ │ │ ├── tfxla_device_specific_transforms.cc │ │ │ │ ├── utils.cc │ │ │ │ ├── utils.h │ │ │ │ ├── verify_tfxla_legalization.cc │ │ │ │ ├── verify_tfxla_legalization_test.cc │ │ │ │ ├── xla_legalize_targets.cc │ │ │ │ ├── xla_legalize_targets.h │ │ │ │ ├── xla_legalize_targets_test.cc │ │ │ │ ├── xla_legalize_tf.cc │ │ │ │ ├── xla_legalize_tf_passes.td │ │ │ │ └── xla_legalize_tf_test.cc │ │ ├── tf_mlir_opt_main.cc │ │ ├── tf_mlir_reduce_main.cc │ │ ├── tf_mlir_translate_main.cc │ │ ├── tfr │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── build_defs.bzl │ │ │ ├── define_op_template.py │ │ │ ├── examples │ │ │ │ ├── customization │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── ops_defs.py │ │ │ │ │ └── test_ops_test.py │ │ │ │ ├── mnist │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── mnist_ops_test.py │ │ │ │ │ ├── mnist_train.py │ │ │ │ │ ├── mnist_train_test.py │ │ │ │ │ └── ops_defs.py │ │ │ │ └── pad │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── ops_defs.py │ │ │ │ │ └── pad_ops_test.py │ │ │ ├── integration │ │ │ │ ├── graph_decompose_pass.cc │ │ │ │ ├── graph_decompose_pass.h │ │ │ │ ├── graph_decompose_test.py │ │ │ │ ├── node_expansion_pass.cc │ │ │ │ ├── node_expansion_pass.h │ │ │ │ ├── node_expansion_test.py │ │ │ │ ├── tfr_decompose_ctx.cc │ │ │ │ ├── tfr_decompose_ctx.h │ │ │ │ └── tfr_decompose_ctx_test.cc │ │ │ ├── ir │ │ │ │ ├── tfr_ops.cc │ │ │ │ ├── tfr_ops.h │ │ │ │ ├── tfr_ops.td │ │ │ │ └── tfr_types.h │ │ │ ├── passes │ │ │ │ ├── canonicalize.cc │ │ │ │ ├── decompose.cc │ │ │ │ ├── decompose_patterns.td │ │ │ │ ├── passes.h │ │ │ │ ├── raise_to_tf.cc │ │ │ │ ├── rewrite_quantized_io.cc │ │ │ │ └── tfr_opt.cc │ │ │ ├── python │ │ │ │ ├── composite.py │ │ │ │ ├── op_reg_gen.py │ │ │ │ ├── op_reg_gen_test.py │ │ │ │ ├── test_utils.py │ │ │ │ ├── tfr_gen.py │ │ │ │ ├── tfr_gen_test.py │ │ │ │ └── tfr_wrapper.cc │ │ │ ├── resources │ │ │ │ ├── BUILD │ │ │ │ ├── composite_ops.cc │ │ │ │ ├── decomposition_lib.mlir │ │ │ │ └── test_ops.cc │ │ │ ├── tests │ │ │ │ ├── canonicalize.mlir │ │ │ │ ├── control_flow.mlir │ │ │ │ ├── decompose.mlir │ │ │ │ ├── end2end.mlir │ │ │ │ ├── ops.mlir │ │ │ │ ├── raise_to_tf.mlir │ │ │ │ └── rewrite_quantized_io.mlir │ │ │ ├── tfr_wrapper.pyi │ │ │ └── utils │ │ │ │ ├── utils.cc │ │ │ │ └── utils.h │ │ ├── tfrt │ │ │ ├── BUILD │ │ │ ├── analysis │ │ │ │ ├── analysis.proto │ │ │ │ ├── cost_analysis.cc │ │ │ │ ├── cost_analysis.h │ │ │ │ ├── tensor_array_side_effect_analysis.cc │ │ │ │ ├── tensor_array_side_effect_analysis.h │ │ │ │ ├── test_cost_analysis_pass.cc │ │ │ │ └── test_tensor_array_side_effect_analysis.cc │ │ │ ├── backend_compiler.h │ │ │ ├── constants.h │ │ │ ├── function │ │ │ │ ├── function.cc │ │ │ │ └── function.h │ │ │ ├── ir │ │ │ │ ├── BUILD │ │ │ │ ├── gpu_ops.cc │ │ │ │ ├── gpu_ops.h │ │ │ │ ├── gpu_ops.td │ │ │ │ ├── mlrt │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── mlrt_dialect.cc │ │ │ │ │ ├── mlrt_dialect.h │ │ │ │ │ ├── mlrt_dialect.td │ │ │ │ │ ├── mlrt_ops.cc │ │ │ │ │ ├── mlrt_ops.h │ │ │ │ │ ├── mlrt_ops.td │ │ │ │ │ ├── tf_mlrt_dialect.td │ │ │ │ │ ├── tf_mlrt_ops.cc │ │ │ │ │ ├── tf_mlrt_ops.h │ │ │ │ │ ├── tf_mlrt_ops.td │ │ │ │ │ ├── tf_mlrt_tpu_ops.cc │ │ │ │ │ ├── tf_mlrt_tpu_ops.h │ │ │ │ │ ├── tf_mlrt_tpu_ops.td │ │ │ │ │ └── tf_ops.td │ │ │ │ ├── tfrt_fallback.cc │ │ │ │ ├── tfrt_fallback.h │ │ │ │ ├── tfrt_fallback.td │ │ │ │ ├── tfrt_fallback_async.cc │ │ │ │ ├── tfrt_fallback_async.h │ │ │ │ ├── tfrt_fallback_async.td │ │ │ │ ├── tfrt_fallback_common.cc │ │ │ │ ├── tfrt_fallback_common.h │ │ │ │ ├── tfrt_fallback_sync.cc │ │ │ │ ├── tfrt_fallback_sync.h │ │ │ │ ├── tfrt_fallback_sync.td │ │ │ │ ├── tfrt_fallback_util.cc │ │ │ │ └── tfrt_fallback_util.h │ │ │ ├── lhlo-tfrt-opt.cc │ │ │ ├── runtime_fallback │ │ │ │ ├── runtime_fallback_combine.cc │ │ │ │ ├── runtime_fallback_executor.cc │ │ │ │ ├── runtime_fallback_executor.h │ │ │ │ ├── runtime_fallback_ops.cc │ │ │ │ ├── runtime_fallback_ops.h │ │ │ │ └── runtime_fallback_ops.td │ │ │ ├── saved_model │ │ │ │ ├── saved_model.cc │ │ │ │ └── saved_model.h │ │ │ ├── test_opkernels.cc │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── analysis │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── cost_analysis.mlir │ │ │ │ │ ├── tensor_array_side_effect_analysis.mlir │ │ │ │ │ ├── testdata │ │ │ │ │ │ └── test.mlir │ │ │ │ │ └── update_op_cost_in_tfrt_mlir_test.cc │ │ │ │ ├── batch_function_fallback_resource_variable_as_captured_tensor.mlir │ │ │ │ ├── batch_function_lowering.mlir │ │ │ │ ├── convert_ref_variables.mlir │ │ │ │ ├── cross_device_transfer.mlir │ │ │ │ ├── deduplicate_if_results.mlir │ │ │ │ ├── fuse_tpu_compile_and_execute_ops.mlir │ │ │ │ ├── hoist_invariant_ops.mlir │ │ │ │ ├── hoist_invariant_ops_mlrt.mlir │ │ │ │ ├── ifrt │ │ │ │ │ ├── BUILD │ │ │ │ │ └── rewrite_cluster_to_ifrt_call.mlir │ │ │ │ ├── ir │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── fallback_opt.mlir │ │ │ │ │ ├── testdata │ │ │ │ │ │ └── test.mlir │ │ │ │ │ └── tfrt_fallback_util_test.cc │ │ │ │ ├── lhlo_to_jitrt │ │ │ │ │ └── BUILD │ │ │ │ ├── mlrt │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── assign_op_key.mlir │ │ │ │ │ ├── async_while.mlir │ │ │ │ │ ├── fuse_mlrt_ops.mlir │ │ │ │ │ ├── inline.mlir │ │ │ │ │ ├── parallelization.mlir │ │ │ │ │ ├── tf_to_mlrt.mlir │ │ │ │ │ ├── tpu_conversions.mlir │ │ │ │ │ └── while_to_map_fn.mlir │ │ │ │ ├── optimize.mlir │ │ │ │ ├── remove_device_attribute.mlir │ │ │ │ ├── runtime_lowering_gpu.mlir │ │ │ │ ├── runtime_lowering_tpu.mlir │ │ │ │ ├── saved_model │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── saved_model_test.cc │ │ │ │ │ └── testdata │ │ │ │ │ │ ├── test.mlir │ │ │ │ │ │ ├── xla_launch.mlir │ │ │ │ │ │ └── xla_launch_xla_reduce_window.mlir │ │ │ │ ├── sink_in_invariant_ops.mlir │ │ │ │ ├── tf_to_corert │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── attributes.mlir │ │ │ │ │ ├── basic.mlir │ │ │ │ │ ├── batch_function_deduplicate.mlir │ │ │ │ │ ├── batch_function_deduplicate_failed.mlir │ │ │ │ │ ├── const_tensor.mlir │ │ │ │ │ ├── control_flow.mlir │ │ │ │ │ ├── decompose_resource_op.mlir │ │ │ │ │ ├── derived_attrs.mlir │ │ │ │ │ ├── device_conversion.mlir │ │ │ │ │ ├── errors.mlir │ │ │ │ │ ├── fallback.mlir │ │ │ │ │ ├── fallback_canonicalization.mlir │ │ │ │ │ ├── fallback_inline.mlir │ │ │ │ │ ├── func_attributes.mlir │ │ │ │ │ ├── func_attributes_multiple_callers.mlir │ │ │ │ │ ├── func_use_fallback_tensor.mlir │ │ │ │ │ ├── insert_fallback_tensor_copy.mlir │ │ │ │ │ ├── merge_tf_if_ops.mlir │ │ │ │ │ ├── optimize_tf_control_flow_side_effect.mlir │ │ │ │ │ ├── remove_tf_if_const_args.mlir │ │ │ │ │ ├── reorder_assert.mlir │ │ │ │ │ ├── side_effects.mlir │ │ │ │ │ ├── tf_to_corert_pipeline.mlir │ │ │ │ │ ├── tf_to_corert_pipeline_refvar.mlir │ │ │ │ │ └── whileop.mlir │ │ │ │ ├── tfrt_fallback │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── async_op_thread.mlir │ │ │ │ │ ├── batching_fallback.mlir │ │ │ │ │ ├── compile.benchmark.large.mlir │ │ │ │ │ ├── compile.benchmark.small.mlir │ │ │ │ │ ├── convert_tensorhandle_to_fallback_tensor.mlir │ │ │ │ │ ├── fallback.mlir │ │ │ │ │ ├── fallback_tensor_conversion_host.mlir │ │ │ │ │ ├── kernel_fallback_op_handler.mlir │ │ │ │ │ ├── mnist.mlir │ │ │ │ │ ├── runtime_fallback_op_handler.mlir │ │ │ │ │ ├── tf_delegate.mlir │ │ │ │ │ ├── tf_ops.mlir │ │ │ │ │ ├── tf_ops_error.mlir │ │ │ │ │ └── tfrt_forwarding.mlir │ │ │ │ ├── xla_launch_fallback.mlir │ │ │ │ ├── xla_launch_lowering.mlir │ │ │ │ └── xla_rewrite.mlir │ │ │ ├── tf-tfrt-opt.cc │ │ │ ├── tfrt_fallback_registration.cc │ │ │ ├── tfrt_fallback_registration.h │ │ │ ├── tfrt_fallback_translate_registration.cc │ │ │ ├── tools │ │ │ │ └── tfrt_translate │ │ │ │ │ └── static_registration.cc │ │ │ ├── transforms │ │ │ │ ├── attr_lowering_utils.cc │ │ │ │ ├── attr_lowering_utils.h │ │ │ │ ├── corert_converter.cc │ │ │ │ ├── corert_converter.h │ │ │ │ ├── cross_device_transfer.cc │ │ │ │ ├── deduplicate_batch_function.cc │ │ │ │ ├── deduplicate_if_result_pass.cc │ │ │ │ ├── fallback_converter.cc │ │ │ │ ├── fallback_converter.h │ │ │ │ ├── fuse_tpu_compile_and_execute_ops.cc │ │ │ │ ├── gpu_passes.cc │ │ │ │ ├── gpu_passes.h │ │ │ │ ├── ifrt │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── ifrt_backend_compiler.cc │ │ │ │ │ ├── ifrt_backend_compiler.h │ │ │ │ │ ├── ifrt_backend_compiler_test.cc │ │ │ │ │ ├── rewrite_cluster_to_ifrt_call.cc │ │ │ │ │ ├── rewrite_cluster_to_ifrt_call.h │ │ │ │ │ ├── testdata │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── ifrt_cluster.mlir │ │ │ │ │ │ ├── tf2hlo_empty.mlir │ │ │ │ │ │ └── tf2hlo_tuple.mlir │ │ │ │ │ ├── tf2hlo.cc │ │ │ │ │ ├── tf2hlo.h │ │ │ │ │ ├── tf2hlo_test.cc │ │ │ │ │ ├── tf_ifrt_passes.cc │ │ │ │ │ └── tf_ifrt_passes.h │ │ │ │ ├── insert_tensor_copy.cc │ │ │ │ ├── lower_saved_model.cc │ │ │ │ ├── merge_tf_if_ops.cc │ │ │ │ ├── mlrt │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── assign_op_key.cc │ │ │ │ │ ├── assign_op_key.h │ │ │ │ │ ├── async_while.cc │ │ │ │ │ ├── async_while.h │ │ │ │ │ ├── execute_op_registry.h │ │ │ │ │ ├── fuse_mlrt_ops.cc │ │ │ │ │ ├── fuse_mlrt_ops.h │ │ │ │ │ ├── import_model.cc │ │ │ │ │ ├── import_model.h │ │ │ │ │ ├── parallelization.cc │ │ │ │ │ ├── parallelization.h │ │ │ │ │ ├── passes.cc │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── tf_to_mlrt.cc │ │ │ │ │ ├── tf_to_mlrt.h │ │ │ │ │ ├── tpu_conversion_patterns.cc │ │ │ │ │ ├── tpu_conversion_patterns.h │ │ │ │ │ ├── util.cc │ │ │ │ │ ├── util.h │ │ │ │ │ ├── while_to_map_fn.cc │ │ │ │ │ └── while_to_map_fn.h │ │ │ │ ├── optimize.cc │ │ │ │ ├── optimize_tf_control_flow_side_effect.cc │ │ │ │ ├── passes.cc │ │ │ │ ├── passes.h │ │ │ │ ├── remove_device_attribute.cc │ │ │ │ ├── remove_tf_if_const_args.cc │ │ │ │ ├── reorder_assert.cc │ │ │ │ ├── set_shape_invariant_in_while_ops.cc │ │ │ │ ├── set_shape_invariant_in_while_ops.h │ │ │ │ ├── sink_in_invariant_ops.cc │ │ │ │ ├── tf_to_tfrt.cc │ │ │ │ ├── tfrt_pipeline_options.h │ │ │ │ ├── tpu_passes.h │ │ │ │ ├── update_op_cost_in_tfrt_mlir.cc │ │ │ │ ├── update_op_cost_in_tfrt_mlir.h │ │ │ │ ├── utils.cc │ │ │ │ ├── utils.h │ │ │ │ └── xla_rewrite_pass.cc │ │ │ ├── translate │ │ │ │ ├── import_model.cc │ │ │ │ ├── import_model.h │ │ │ │ ├── mlrt │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── mlir_to_bytecode.cc │ │ │ │ │ ├── mlir_to_bytecode.h │ │ │ │ │ ├── mlir_to_bytecode_test.cc │ │ │ │ │ ├── test_utils.cc │ │ │ │ │ ├── test_utils.h │ │ │ │ │ └── testdata │ │ │ │ │ │ ├── basic.mlir │ │ │ │ │ │ ├── basic_attributes.mlir │ │ │ │ │ │ ├── custom_attributes.mlir │ │ │ │ │ │ └── unsupported_attributes.mlir │ │ │ │ ├── tfrt_compile_options.cc │ │ │ │ └── tfrt_compile_options.h │ │ │ └── utils │ │ │ │ ├── export.cc │ │ │ │ ├── export.h │ │ │ │ ├── host_context.cc │ │ │ │ └── host_context.h │ │ ├── tools │ │ │ └── kernel_gen │ │ │ │ ├── BUILD │ │ │ │ ├── compile_cache_item.proto │ │ │ │ ├── hlo_to_kernel.cc │ │ │ │ ├── ir │ │ │ │ ├── BUILD │ │ │ │ ├── tf_framework_ops.cc │ │ │ │ ├── tf_framework_ops.h │ │ │ │ ├── tf_framework_ops.td │ │ │ │ └── tf_status.td │ │ │ │ ├── kernel_creator.cc │ │ │ │ ├── kernel_creator.h │ │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── buffer_deallocation.mlir │ │ │ │ ├── buffer_reuse.mlir │ │ │ │ ├── bufferize.mlir │ │ │ │ ├── copy_cleanup.mlir │ │ │ │ ├── embed_tf_framework.mlir │ │ │ │ ├── func_to_jit_invocations.mlir │ │ │ │ ├── hlo_to_kernel │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add_v2.mlir │ │ │ │ │ ├── add_v2_unsigned.mlir │ │ │ │ │ └── tanh.mlir │ │ │ │ ├── invalid.mlir │ │ │ │ ├── isinf.mlir │ │ │ │ ├── ops.mlir │ │ │ │ ├── parallel_loops_to_sequential.mlir │ │ │ │ ├── rewrite_tf_framework_assert.mlir │ │ │ │ ├── tanh.mlir │ │ │ │ ├── tf-legalize-to-lmhlo.mlir │ │ │ │ ├── tf_abi_knowledge.mlir │ │ │ │ ├── tf_framework_legalize_to_llvm.mlir │ │ │ │ └── tf_kernel_gpu_launch_to_llvm.mlir │ │ │ │ ├── tf_framework_c_interface.cc │ │ │ │ ├── tf_framework_c_interface.h │ │ │ │ ├── tf_gpu_runtime_wrappers.cc │ │ │ │ ├── tf_gpu_runtime_wrappers.h │ │ │ │ ├── tf_jit_cache.cc │ │ │ │ ├── tf_jit_cache.h │ │ │ │ ├── tools │ │ │ │ └── kernel-gen-opt │ │ │ │ │ └── kernel-gen-opt.cc │ │ │ │ └── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── buffer_reuse_pass.cc │ │ │ │ ├── bufferize.cc │ │ │ │ ├── bufferize_pass.cc │ │ │ │ ├── copy_cleanup_pass.cc │ │ │ │ ├── embed_tf_framework.cc │ │ │ │ ├── embed_tf_framework_pass.cc │ │ │ │ ├── func_to_jit_invocations.cc │ │ │ │ ├── fuse_inner_parallel_loops_pass.cc │ │ │ │ ├── gpu_kernel_to_blob_pass.cc │ │ │ │ ├── parallel_loops_to_sequential.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ ├── rewrite_tf_framework_assert.cc │ │ │ │ ├── rewriters.h │ │ │ │ ├── same_shape_propagation.cc │ │ │ │ ├── shape_to_descriptors_pass.cc │ │ │ │ ├── tensorflow_abi_knowledge_propagation.cc │ │ │ │ ├── tf_framework_legalize_to_llvm.cc │ │ │ │ ├── tf_kernel_to_llvm_pass.cc │ │ │ │ ├── utils.cc │ │ │ │ └── utils.h │ │ ├── tosa │ │ │ ├── BUILD │ │ │ ├── g3doc │ │ │ │ └── legalization.md │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── convert-tfl-uint8.mlir │ │ │ │ ├── convert_metadata.mlir │ │ │ │ ├── fuse-bias-tf.mlir │ │ │ │ ├── lower-complex-types.mlir │ │ │ │ ├── lower_global_tensors.mlir │ │ │ │ ├── multi_add.mlir │ │ │ │ ├── retain_call_once_funcs.mlir │ │ │ │ ├── strip-quant-types.mlir │ │ │ │ ├── strip_metadata.mlir │ │ │ │ ├── tf-tfl-to-tosa-pipeline.mlir │ │ │ │ ├── tf-to-tosa-pipeline.mlir │ │ │ │ ├── tfl-to-tosa-dequantize_softmax.mlir │ │ │ │ ├── tfl-to-tosa-pipeline-filtered.mlir │ │ │ │ ├── tfl-to-tosa-pipeline.mlir │ │ │ │ ├── tfl-to-tosa-stateful.mlir │ │ │ │ └── verify_fully_converted.mlir │ │ │ ├── tf_passes.cc │ │ │ ├── tf_passes.h │ │ │ ├── tf_tfl_passes.cc │ │ │ ├── tf_tfl_passes.h │ │ │ ├── tfl_passes.cc │ │ │ ├── tfl_passes.h │ │ │ └── transforms │ │ │ │ ├── convert_metadata.cc │ │ │ │ ├── convert_tfl_uint8.cc │ │ │ │ ├── dequantize_tfl_softmax.cc │ │ │ │ ├── fuse_bias_tf.cc │ │ │ │ ├── legalize_common.cc │ │ │ │ ├── legalize_common.h │ │ │ │ ├── legalize_tf.cc │ │ │ │ ├── legalize_tf_tfl.cc │ │ │ │ ├── legalize_tfl.cc │ │ │ │ ├── legalize_tfl_stateful.cc │ │ │ │ ├── legalize_utils.cc │ │ │ │ ├── legalize_utils.h │ │ │ │ ├── lower_complex_types.cc │ │ │ │ ├── lower_global_tensors.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ ├── retain_call_once_funcs.cc │ │ │ │ ├── strip_metadata.cc │ │ │ │ ├── strip_quant_types.cc │ │ │ │ ├── tf_legalize_patterns.td │ │ │ │ ├── tfl_legalize_patterns.td │ │ │ │ └── verify_fully_converted.cc │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── array_container_utils.h │ │ │ ├── name_utils.cc │ │ │ ├── name_utils.h │ │ │ └── string_container_utils.h │ ├── plugin │ │ ├── BUILD │ │ └── README.md │ ├── tests │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── adadelta_test.py │ │ ├── adagrad_da_test.py │ │ ├── adagrad_test.py │ │ ├── adam_test.py │ │ ├── add_n_test.py │ │ ├── approx_topk_test.py │ │ ├── argminmax_test.py │ │ ├── async_comp_test.py │ │ ├── binary_ops_test.py │ │ ├── bincount_op_test.py │ │ ├── bucketize_op_test.py │ │ ├── build_defs.bzl │ │ ├── case_test.py │ │ ├── cast_ops_test.py │ │ ├── categorical_op_test.py │ │ ├── cholesky_op_test.py │ │ ├── clustering_test.py │ │ ├── complex_div_test.py │ │ ├── concat_ops_test.py │ │ ├── cond_test.py │ │ ├── const_arg_test.py │ │ ├── const_test.py │ │ ├── conv2d_test.py │ │ ├── conv3d_test.py │ │ ├── conv_node_name_test.py │ │ ├── data_format_ops_test.py │ │ ├── dense_layer_test.py │ │ ├── depthwise_conv_op_test.py │ │ ├── dynamic_slice_ops_test.py │ │ ├── dynamic_stitch_test.py │ │ ├── eager_test.py │ │ ├── einsum_op_test.py │ │ ├── ensure_shape_op_test.py │ │ ├── extract_image_patches_op_test.py │ │ ├── fake_quant_ops_test.py │ │ ├── fft_test.py │ │ ├── fifo_queue_test.py │ │ ├── ftrl_ops_test.py │ │ ├── ftrl_test.py │ │ ├── function_test.py │ │ ├── fused_batchnorm_test.py │ │ ├── gather_nd_op_test.py │ │ ├── gather_test.py │ │ ├── giant_const_op_test.py │ │ ├── image_ops_jit_compile_test.py │ │ ├── image_ops_test.py │ │ ├── jit_test.py │ │ ├── listdiff_op_test.py │ │ ├── lrn_ops_test.py │ │ ├── lstm.py │ │ ├── lstm_layer_inference.config.pbtxt │ │ ├── lstm_layer_inference.pbtxt │ │ ├── lstm_test.py │ │ ├── manip_ops_test.py │ │ ├── matrix_band_part_test.py │ │ ├── matrix_diag_ops_test.py │ │ ├── matrix_inverse_op_test.py │ │ ├── matrix_solve_op_test.py │ │ ├── matrix_triangular_solve_op_test.py │ │ ├── mean_op_test.py │ │ ├── momentum_test.py │ │ ├── nary_ops_test.py │ │ ├── nullary_ops_test.py │ │ ├── placeholder_test.py │ │ ├── plugin.bzl │ │ ├── pooling_ops_3d_test.py │ │ ├── pooling_ops_test.py │ │ ├── proximal_adagrad_test.py │ │ ├── proximal_gradient_descent_test.py │ │ ├── qr_op_test.py │ │ ├── quantized_ops_test.py │ │ ├── random_ops_test.py │ │ ├── randomized_tests.cc │ │ ├── reduce_ops_test.py │ │ ├── reduce_window_test.py │ │ ├── repeat_op_test.py │ │ ├── reshape_op_test.py │ │ ├── reverse_ops_test.py │ │ ├── reverse_sequence_op_args_test.py │ │ ├── reverse_sequence_op_test.py │ │ ├── rmsprop_test.py │ │ ├── runtime_shape_check_test.py │ │ ├── scan_ops_test.py │ │ ├── scatter_nd_op_test.py │ │ ├── searchsorted_op_test.py │ │ ├── segment_reduction_ops_test.py │ │ ├── self_adjoint_eig_op_test.py │ │ ├── sharding_util_ops_test.py │ │ ├── slice_ops_test.py │ │ ├── sort_ops_test.py │ │ ├── spacetobatch_op_test.py │ │ ├── sparse_to_dense_op_test.py │ │ ├── special_math_test.py │ │ ├── stack_ops_test.py │ │ ├── stateful_random_ops_test.py │ │ ├── stateless_random_ops_test.py │ │ ├── stochastic_cast_op_test.py │ │ ├── svd_op_test.py │ │ ├── tensor_array_ops_test.py │ │ ├── tensor_float_32_test.py │ │ ├── tensor_list_ops_test.py │ │ ├── ternary_ops_test.py │ │ ├── test_utils.py │ │ ├── tridiagonal_matmul_ops_test.py │ │ ├── tridiagonal_solve_ops_test.py │ │ ├── unary_ops_composition_test.cc │ │ ├── unary_ops_test.py │ │ ├── unique_ops_test.py │ │ ├── unstack_test.py │ │ ├── variable_ops_test.py │ │ ├── where_op_test.py │ │ ├── while_test.py │ │ ├── xla_call_module_test.py │ │ ├── xla_custom_call_ops_test.py │ │ ├── xla_device_gpu_test.py │ │ ├── xla_device_test.py │ │ ├── xla_dump_to_sponge_test.py │ │ ├── xla_dump_to_test.py │ │ ├── xla_ops_test.py │ │ ├── xla_test.py │ │ └── xla_test_test.py │ ├── tf2tensorrt │ │ ├── BUILD │ │ ├── _pywrap_py_utils.pyi │ │ ├── common │ │ │ ├── datavec.h │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── convert │ │ │ ├── algorithm_selector.cc │ │ │ ├── algorithm_selector.h │ │ │ ├── algorithm_selector_test.cc │ │ │ ├── convert_graph.cc │ │ │ ├── convert_graph.h │ │ │ ├── convert_graph_test.cc │ │ │ ├── convert_nodes.cc │ │ │ ├── convert_nodes.h │ │ │ ├── convert_nodes_test.cc │ │ │ ├── logger_registry.cc │ │ │ ├── logger_registry.h │ │ │ ├── logger_registry_test.cc │ │ │ ├── op_converter.h │ │ │ ├── op_converter_registry.cc │ │ │ ├── op_converter_registry.h │ │ │ ├── op_converter_registry_test.cc │ │ │ ├── op_converter_test.cc │ │ │ ├── ops │ │ │ │ ├── binary_ops.cc │ │ │ │ ├── data_format_vec_permute.cc │ │ │ │ ├── einsum.cc │ │ │ │ ├── fill_ops.cc │ │ │ │ ├── layer_utils.h │ │ │ │ ├── like_ops.cc │ │ │ │ ├── log_softmax.cc │ │ │ │ ├── quantization_ops.cc │ │ │ │ ├── quantization_ops.h │ │ │ │ ├── quantization_ops_test.cc │ │ │ │ ├── selectv2.cc │ │ │ │ ├── slice_ops.cc │ │ │ │ ├── slice_ops.h │ │ │ │ ├── softmax.cc │ │ │ │ ├── tile.cc │ │ │ │ ├── unary_ops.cc │ │ │ │ └── variable_ops.cc │ │ │ ├── timing_cache.cc │ │ │ ├── timing_cache.h │ │ │ ├── trt_layout_optimization_pass.cc │ │ │ ├── trt_layout_optimization_pass.h │ │ │ ├── trt_optimization_pass.cc │ │ │ ├── trt_optimization_pass.h │ │ │ ├── trt_parameters.cc │ │ │ ├── trt_parameters.h │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ ├── weights.cc │ │ │ └── weights.h │ │ ├── kernels │ │ │ ├── get_calibration_data_op.cc │ │ │ ├── trt_engine_op.cc │ │ │ ├── trt_engine_op_test.cc │ │ │ ├── trt_engine_resource_ops.cc │ │ │ └── trt_engine_resource_ops_test.cc │ │ ├── ops │ │ │ ├── get_calibration_data_op.cc │ │ │ ├── trt_engine_op.cc │ │ │ └── trt_engine_resource_ops.cc │ │ ├── plugin │ │ │ ├── trt_plugin.cc │ │ │ └── trt_plugin.h │ │ ├── segment │ │ │ ├── segment.cc │ │ │ ├── segment.h │ │ │ ├── segment_test.cc │ │ │ ├── union_find.cc │ │ │ └── union_find.h │ │ ├── stub │ │ │ ├── NvInferPlugin_7_0.inc │ │ │ ├── NvInfer_7_0.inc │ │ │ ├── NvInfer_8_0.inc │ │ │ ├── nvinfer_plugin_stub.cc │ │ │ └── nvinfer_stub.cc │ │ ├── tensorrt_test.cc │ │ ├── trt_convert_api.cc │ │ ├── trt_convert_api.h │ │ ├── trt_convert_api_test.cc │ │ └── utils │ │ │ ├── py_utils.cc │ │ │ ├── py_utils.h │ │ │ ├── py_utils_wrapper.cc │ │ │ ├── trt_allocator.cc │ │ │ ├── trt_allocator.h │ │ │ ├── trt_allocator_test.cc │ │ │ ├── trt_engine_instance.proto │ │ │ ├── trt_engine_utils.cc │ │ │ ├── trt_engine_utils.h │ │ │ ├── trt_execution_context.h │ │ │ ├── trt_experimental_features.cc │ │ │ ├── trt_experimental_features.h │ │ │ ├── trt_int8_calibrator.cc │ │ │ ├── trt_int8_calibrator.h │ │ │ ├── trt_logger.cc │ │ │ ├── trt_logger.h │ │ │ ├── trt_lru_cache.cc │ │ │ ├── trt_lru_cache.h │ │ │ ├── trt_lru_cache_test.cc │ │ │ ├── trt_shape_optimization_profiles.cc │ │ │ ├── trt_shape_optimization_profiles.h │ │ │ ├── trt_shape_optimization_profiles_test.cc │ │ │ ├── trt_tensor_proxy.h │ │ │ ├── trt_testutils.cc │ │ │ ├── trt_testutils.h │ │ │ └── trt_testutils_test.cc │ └── tf2xla │ │ ├── BUILD │ │ ├── cc │ │ └── BUILD │ │ ├── const_analysis.cc │ │ ├── const_analysis.h │ │ ├── const_analysis_test.cc │ │ ├── cpu_function_runtime_test.cc │ │ ├── frontend_attributes_util.cc │ │ ├── frontend_attributes_util.h │ │ ├── functionalize_cond.cc │ │ ├── functionalize_cond.h │ │ ├── functionalize_cond_test.cc │ │ ├── functionalize_control_flow.cc │ │ ├── functionalize_control_flow.h │ │ ├── functionalize_control_flow_pass_registration.cc │ │ ├── functionalize_control_flow_test.cc │ │ ├── functionalize_control_flow_util.cc │ │ ├── functionalize_control_flow_util.h │ │ ├── functionalize_while.cc │ │ ├── functionalize_while.h │ │ ├── fused_batchnorm_reserve_space_test.cc │ │ ├── g3doc │ │ ├── cpu_supported_ops.md │ │ └── gpu_supported_ops.md │ │ ├── graph_compiler.cc │ │ ├── graph_compiler.h │ │ ├── graph_compiler_test.cc │ │ ├── graph_compiler_util.cc │ │ ├── graph_compiler_util.h │ │ ├── host_compute_metadata.proto │ │ ├── kernels │ │ ├── BUILD │ │ ├── aggregate_ops.cc │ │ ├── all_reduce_op.cc │ │ ├── approx_topk_op.cc │ │ ├── arg_op.cc │ │ ├── assert_op.cc │ │ ├── batch_matmul_op.cc │ │ ├── batch_norm_op.cc │ │ ├── batchtospace_op.cc │ │ ├── bcast_ops.cc │ │ ├── beta_op.cc │ │ ├── bias_ops.cc │ │ ├── binary_ops.cc │ │ ├── bincount_op.cc │ │ ├── broadcast_to_op.cc │ │ ├── bucketize_op.cc │ │ ├── callback.proto │ │ ├── case_op.cc │ │ ├── case_op.h │ │ ├── cast_op.cc │ │ ├── categorical_op.cc │ │ ├── check_numerics_op.cc │ │ ├── cholesky_op.cc │ │ ├── clip_by_value_op.cc │ │ ├── concat_op.cc │ │ ├── const_op.cc │ │ ├── conv_op_helpers.cc │ │ ├── conv_op_helpers.h │ │ ├── conv_ops.cc │ │ ├── cross_op.cc │ │ ├── cwise_ops.cc │ │ ├── cwise_ops.h │ │ ├── data_format_ops.cc │ │ ├── depthtospace_op.cc │ │ ├── dequantize_op.cc │ │ ├── device_index_op.cc │ │ ├── diag_op.cc │ │ ├── dynamic_partition_op.cc │ │ ├── dynamic_slice_ops.cc │ │ ├── dynamic_stitch_op.cc │ │ ├── einsum_op.cc │ │ ├── elu_op.cc │ │ ├── elu_op.h │ │ ├── empty_op.cc │ │ ├── ensure_shape_op.cc │ │ ├── extract_image_patches_op.cc │ │ ├── fake_param_op.cc │ │ ├── fake_quantize_ops.cc │ │ ├── fft_ops.cc │ │ ├── fill_op.cc │ │ ├── function_ops.cc │ │ ├── fused_conv_ops.cc │ │ ├── gather_op.cc │ │ ├── gather_op_helpers.h │ │ ├── gather_scatter_ops.cc │ │ ├── identity_op.cc │ │ ├── if_op.cc │ │ ├── if_op.h │ │ ├── if_while_utils.cc │ │ ├── if_while_utils.h │ │ ├── image_ops.cc │ │ ├── image_resize_ops.cc │ │ ├── image_resize_ops.h │ │ ├── in_topk_op.cc │ │ ├── index_ops.cc │ │ ├── index_ops.h │ │ ├── l2loss_op.cc │ │ ├── light_outside_compilation.cc │ │ ├── light_outside_compilation.h │ │ ├── listdiff_op.cc │ │ ├── lower_upper_bound_ops.cc │ │ ├── lrn_ops.cc │ │ ├── matmul_op.cc │ │ ├── matrix_band_part_op.cc │ │ ├── matrix_diag_ops.cc │ │ ├── matrix_inverse_op.cc │ │ ├── matrix_solve_op.cc │ │ ├── matrix_triangular_solve_op.cc │ │ ├── mirror_pad_op.cc │ │ ├── next_after_op.cc │ │ ├── no_op.cc │ │ ├── one_hot_op.cc │ │ ├── pack_op.cc │ │ ├── pad_op.cc │ │ ├── pooling_ops.cc │ │ ├── qr_op.cc │ │ ├── quantize_and_dequantize_op.cc │ │ ├── random_ops.cc │ │ ├── random_ops_util.cc │ │ ├── random_ops_util.h │ │ ├── reduce_window_op.cc │ │ ├── reduction_ops.cc │ │ ├── reduction_ops.h │ │ ├── reduction_ops_common.cc │ │ ├── relu_op.cc │ │ ├── relu_op.h │ │ ├── replica_id_op.cc │ │ ├── resampler_addon_ops.cc │ │ ├── resampler_ops.cc │ │ ├── resampler_ops.h │ │ ├── reshape_op.cc │ │ ├── retval_op.cc │ │ ├── reverse_op.cc │ │ ├── reverse_sequence_op.cc │ │ ├── rng_converter_utils.cc │ │ ├── rng_converter_utils.h │ │ ├── rng_converter_utils_test.cc │ │ ├── roll_op.cc │ │ ├── scan_ops.cc │ │ ├── scatter_nd_op.cc │ │ ├── segment_reduction_ops.cc │ │ ├── select_op.cc │ │ ├── sendrecv_ops.cc │ │ ├── sequence_ops.cc │ │ ├── shape_op.cc │ │ ├── shape_util.cc │ │ ├── shape_util.h │ │ ├── sharding_op.cc │ │ ├── sharding_util_ops.cc │ │ ├── slice_op.cc │ │ ├── softmax_op.cc │ │ ├── sort_ops.cc │ │ ├── spacetobatch_op.cc │ │ ├── spacetodepth_op.cc │ │ ├── sparse_to_dense_op.cc │ │ ├── split_op.cc │ │ ├── spmd_manual_sharding_ops.cc │ │ ├── stack_ops.cc │ │ ├── stateful_random_ops.cc │ │ ├── stateless_random_ops.cc │ │ ├── stateless_random_ops_v2.cc │ │ ├── stochastic_cast_op.cc │ │ ├── strided_slice_op.cc │ │ ├── tensor_array_ops.cc │ │ ├── tensor_list_ops.cc │ │ ├── tensor_list_utils.cc │ │ ├── tensor_list_utils.h │ │ ├── tile_ops.cc │ │ ├── to_bool_op.cc │ │ ├── topk_op.cc │ │ ├── training_ops.cc │ │ ├── transpose_op.cc │ │ ├── tridiagonal_ops.cc │ │ ├── unary_ops.cc │ │ ├── unary_ops_composition.cc │ │ ├── uniform_quantized_ops.cc │ │ ├── unique_op.cc │ │ ├── unpack_op.cc │ │ ├── variable_ops.cc │ │ ├── where_op.cc │ │ ├── while_op.cc │ │ ├── while_op.h │ │ ├── xla_broadcast_helper_op.cc │ │ ├── xla_call_module_loader.cc │ │ ├── xla_call_module_loader.h │ │ ├── xla_call_module_op.cc │ │ ├── xla_conv_op.cc │ │ ├── xla_custom_call_op.cc │ │ ├── xla_custom_call_v2_op.cc │ │ ├── xla_dequantize_op.cc │ │ ├── xla_dot_op.cc │ │ ├── xla_optimization_barrier_op.cc │ │ ├── xla_pad_op.cc │ │ ├── xla_reduce_op.cc │ │ ├── xla_reduce_precision_op.cc │ │ ├── xla_select_and_scatter_op.cc │ │ ├── xla_self_adjoint_eig_op.cc │ │ └── xla_svd_op.cc │ │ ├── layout_util.cc │ │ ├── layout_util.h │ │ ├── lib │ │ ├── BUILD │ │ ├── broadcast.cc │ │ ├── broadcast.h │ │ ├── data_format.cc │ │ ├── data_format.h │ │ ├── random.cc │ │ ├── random.h │ │ ├── scatter.cc │ │ ├── scatter.h │ │ ├── util.cc │ │ └── util.h │ │ ├── light_outside_compilation_kernels_for_test.cc │ │ ├── light_outside_compilation_test.py │ │ ├── literal_util.cc │ │ ├── literal_util.h │ │ ├── literal_util_test.cc │ │ ├── mlir_bridge_pass.cc │ │ ├── mlir_bridge_pass.h │ │ ├── mlir_bridge_pass_registration.cc │ │ ├── mlir_tf2xla.cc │ │ ├── mlir_xla_op_kernel.cc │ │ ├── mlir_xla_op_kernel.h │ │ ├── ops │ │ ├── BUILD │ │ └── xla_ops.cc │ │ ├── python │ │ ├── BUILD │ │ ├── xla.py │ │ └── xla_op_registry.clif │ │ ├── rearrange_function_argument.cc │ │ ├── rearrange_function_argument.h │ │ ├── resource_operation_table.cc │ │ ├── resource_operation_table.h │ │ ├── resource_operation_table_test.cc │ │ ├── resource_util.cc │ │ ├── resource_util.h │ │ ├── resource_util_test.cc │ │ ├── shape_util.cc │ │ ├── shape_util.h │ │ ├── sharding_util.cc │ │ ├── sharding_util.h │ │ ├── sharding_util_test.cc │ │ ├── side_effect_util.cc │ │ ├── side_effect_util.h │ │ ├── test_util.cc │ │ ├── test_util.h │ │ ├── tf2xla.cc │ │ ├── tf2xla.h │ │ ├── tf2xla.proto │ │ ├── tf2xla_defs.h │ │ ├── tf2xla_opset.cc │ │ ├── tf2xla_opset.h │ │ ├── tf2xla_opset_test.cc │ │ ├── tf2xla_supported_ops.cc │ │ ├── tf2xla_supported_ops.h │ │ ├── tf2xla_supported_ops_main.cc │ │ ├── tf2xla_test.cc │ │ ├── tf2xla_util.cc │ │ ├── tf2xla_util.h │ │ ├── tf2xla_util_test.cc │ │ ├── type_util.cc │ │ ├── type_util.h │ │ ├── type_util_test.cc │ │ ├── xla_argument.cc │ │ ├── xla_argument.h │ │ ├── xla_compilation_device.cc │ │ ├── xla_compilation_device.h │ │ ├── xla_compiled_cpu_function.cc │ │ ├── xla_compiled_cpu_function.h │ │ ├── xla_compiler.cc │ │ ├── xla_compiler.h │ │ ├── xla_compiler_test.cc │ │ ├── xla_context.cc │ │ ├── xla_context.h │ │ ├── xla_cpu_backend.cc │ │ ├── xla_expression.cc │ │ ├── xla_expression.h │ │ ├── xla_expression_test.cc │ │ ├── xla_gpu_backend.cc │ │ ├── xla_helpers.cc │ │ ├── xla_helpers.h │ │ ├── xla_jit_compiled_cpu_function.cc │ │ ├── xla_jit_compiled_cpu_function.h │ │ ├── xla_jit_compiled_cpu_function_test.cc │ │ ├── xla_op_kernel.cc │ │ ├── xla_op_kernel.h │ │ ├── xla_op_registry.cc │ │ ├── xla_op_registry.h │ │ ├── xla_op_registry_test.cc │ │ ├── xla_resource.cc │ │ ├── xla_resource.h │ │ └── xla_tpu_backend.cc ├── core │ ├── BUILD │ ├── activity_watcher │ │ ├── BUILD │ │ ├── activity.cc │ │ ├── activity.h │ │ ├── activity_utils.cc │ │ └── activity_utils.h │ ├── api_def │ │ ├── BUILD │ │ ├── README.md │ │ ├── api_test.cc │ │ ├── base_api │ │ │ ├── BUILD │ │ │ ├── api_def_Abort.pbtxt │ │ │ ├── api_def_Abs.pbtxt │ │ │ ├── api_def_AccumulateNV2.pbtxt │ │ │ ├── api_def_AccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_AccumulatorNumAccumulated.pbtxt │ │ │ ├── api_def_AccumulatorSetGlobalStep.pbtxt │ │ │ ├── api_def_AccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_Acos.pbtxt │ │ │ ├── api_def_Acosh.pbtxt │ │ │ ├── api_def_Add.pbtxt │ │ │ ├── api_def_AddManySparseToTensorsMap.pbtxt │ │ │ ├── api_def_AddN.pbtxt │ │ │ ├── api_def_AddSparseToTensorsMap.pbtxt │ │ │ ├── api_def_AddV2.pbtxt │ │ │ ├── api_def_AdjustContrast.pbtxt │ │ │ ├── api_def_AdjustContrastv2.pbtxt │ │ │ ├── api_def_AdjustHue.pbtxt │ │ │ ├── api_def_AdjustSaturation.pbtxt │ │ │ ├── api_def_All.pbtxt │ │ │ ├── api_def_AllCandidateSampler.pbtxt │ │ │ ├── api_def_AllToAll.pbtxt │ │ │ ├── api_def_Angle.pbtxt │ │ │ ├── api_def_AnonymousHashTable.pbtxt │ │ │ ├── api_def_AnonymousIterator.pbtxt │ │ │ ├── api_def_AnonymousIteratorV2.pbtxt │ │ │ ├── api_def_AnonymousIteratorV3.pbtxt │ │ │ ├── api_def_AnonymousMemoryCache.pbtxt │ │ │ ├── api_def_AnonymousMultiDeviceIterator.pbtxt │ │ │ ├── api_def_AnonymousMultiDeviceIteratorV3.pbtxt │ │ │ ├── api_def_AnonymousMutableDenseHashTable.pbtxt │ │ │ ├── api_def_AnonymousMutableHashTable.pbtxt │ │ │ ├── api_def_AnonymousMutableHashTableOfTensors.pbtxt │ │ │ ├── api_def_AnonymousRandomSeedGenerator.pbtxt │ │ │ ├── api_def_AnonymousSeedGenerator.pbtxt │ │ │ ├── api_def_Any.pbtxt │ │ │ ├── api_def_ApplyAdaMax.pbtxt │ │ │ ├── api_def_ApplyAdadelta.pbtxt │ │ │ ├── api_def_ApplyAdagrad.pbtxt │ │ │ ├── api_def_ApplyAdagradDA.pbtxt │ │ │ ├── api_def_ApplyAdagradV2.pbtxt │ │ │ ├── api_def_ApplyAdam.pbtxt │ │ │ ├── api_def_ApplyAddSign.pbtxt │ │ │ ├── api_def_ApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ApplyFtrl.pbtxt │ │ │ ├── api_def_ApplyFtrlV2.pbtxt │ │ │ ├── api_def_ApplyGradientDescent.pbtxt │ │ │ ├── api_def_ApplyMomentum.pbtxt │ │ │ ├── api_def_ApplyPowerSign.pbtxt │ │ │ ├── api_def_ApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ApplyRMSProp.pbtxt │ │ │ ├── api_def_ApproxTopK.pbtxt │ │ │ ├── api_def_ApproximateEqual.pbtxt │ │ │ ├── api_def_ArgMax.pbtxt │ │ │ ├── api_def_ArgMin.pbtxt │ │ │ ├── api_def_AsString.pbtxt │ │ │ ├── api_def_Asin.pbtxt │ │ │ ├── api_def_Asinh.pbtxt │ │ │ ├── api_def_Assert.pbtxt │ │ │ ├── api_def_AssertCardinalityDataset.pbtxt │ │ │ ├── api_def_AssertNextDataset.pbtxt │ │ │ ├── api_def_AssertPrevDataset.pbtxt │ │ │ ├── api_def_Assign.pbtxt │ │ │ ├── api_def_AssignAdd.pbtxt │ │ │ ├── api_def_AssignAddVariableOp.pbtxt │ │ │ ├── api_def_AssignSub.pbtxt │ │ │ ├── api_def_AssignSubVariableOp.pbtxt │ │ │ ├── api_def_AssignVariableOp.pbtxt │ │ │ ├── api_def_AssignVariableXlaConcatND.pbtxt │ │ │ ├── api_def_Atan.pbtxt │ │ │ ├── api_def_Atan2.pbtxt │ │ │ ├── api_def_Atanh.pbtxt │ │ │ ├── api_def_AudioSpectrogram.pbtxt │ │ │ ├── api_def_AudioSummary.pbtxt │ │ │ ├── api_def_AudioSummaryV2.pbtxt │ │ │ ├── api_def_AutoShardDataset.pbtxt │ │ │ ├── api_def_AvgPool.pbtxt │ │ │ ├── api_def_AvgPool3D.pbtxt │ │ │ ├── api_def_AvgPool3DGrad.pbtxt │ │ │ ├── api_def_AvgPoolGrad.pbtxt │ │ │ ├── api_def_BandedTriangularSolve.pbtxt │ │ │ ├── api_def_Barrier.pbtxt │ │ │ ├── api_def_BarrierClose.pbtxt │ │ │ ├── api_def_BarrierIncompleteSize.pbtxt │ │ │ ├── api_def_BarrierInsertMany.pbtxt │ │ │ ├── api_def_BarrierReadySize.pbtxt │ │ │ ├── api_def_BarrierTakeMany.pbtxt │ │ │ ├── api_def_Batch.pbtxt │ │ │ ├── api_def_BatchCholesky.pbtxt │ │ │ ├── api_def_BatchCholeskyGrad.pbtxt │ │ │ ├── api_def_BatchDataset.pbtxt │ │ │ ├── api_def_BatchDatasetV2.pbtxt │ │ │ ├── api_def_BatchFFT.pbtxt │ │ │ ├── api_def_BatchFFT2D.pbtxt │ │ │ ├── api_def_BatchFFT3D.pbtxt │ │ │ ├── api_def_BatchFunction.pbtxt │ │ │ ├── api_def_BatchIFFT.pbtxt │ │ │ ├── api_def_BatchIFFT2D.pbtxt │ │ │ ├── api_def_BatchIFFT3D.pbtxt │ │ │ ├── api_def_BatchMatMul.pbtxt │ │ │ ├── api_def_BatchMatMulV2.pbtxt │ │ │ ├── api_def_BatchMatMulV3.pbtxt │ │ │ ├── api_def_BatchMatrixBandPart.pbtxt │ │ │ ├── api_def_BatchMatrixDeterminant.pbtxt │ │ │ ├── api_def_BatchMatrixDiag.pbtxt │ │ │ ├── api_def_BatchMatrixDiagPart.pbtxt │ │ │ ├── api_def_BatchMatrixInverse.pbtxt │ │ │ ├── api_def_BatchMatrixSetDiag.pbtxt │ │ │ ├── api_def_BatchMatrixSolve.pbtxt │ │ │ ├── api_def_BatchMatrixSolveLs.pbtxt │ │ │ ├── api_def_BatchMatrixTriangularSolve.pbtxt │ │ │ ├── api_def_BatchNormWithGlobalNormalization.pbtxt │ │ │ ├── api_def_BatchNormWithGlobalNormalizationGrad.pbtxt │ │ │ ├── api_def_BatchSelfAdjointEig.pbtxt │ │ │ ├── api_def_BatchSelfAdjointEigV2.pbtxt │ │ │ ├── api_def_BatchSvd.pbtxt │ │ │ ├── api_def_BatchToSpace.pbtxt │ │ │ ├── api_def_BatchToSpaceND.pbtxt │ │ │ ├── api_def_BesselI0.pbtxt │ │ │ ├── api_def_BesselI0e.pbtxt │ │ │ ├── api_def_BesselI1.pbtxt │ │ │ ├── api_def_BesselI1e.pbtxt │ │ │ ├── api_def_BesselJ0.pbtxt │ │ │ ├── api_def_BesselJ1.pbtxt │ │ │ ├── api_def_BesselK0.pbtxt │ │ │ ├── api_def_BesselK0e.pbtxt │ │ │ ├── api_def_BesselK1.pbtxt │ │ │ ├── api_def_BesselK1e.pbtxt │ │ │ ├── api_def_BesselY0.pbtxt │ │ │ ├── api_def_BesselY1.pbtxt │ │ │ ├── api_def_Betainc.pbtxt │ │ │ ├── api_def_BiasAdd.pbtxt │ │ │ ├── api_def_BiasAddGrad.pbtxt │ │ │ ├── api_def_BiasAddV1.pbtxt │ │ │ ├── api_def_Bincount.pbtxt │ │ │ ├── api_def_Bitcast.pbtxt │ │ │ ├── api_def_BitwiseAnd.pbtxt │ │ │ ├── api_def_BitwiseOr.pbtxt │ │ │ ├── api_def_BitwiseXor.pbtxt │ │ │ ├── api_def_BlockLSTM.pbtxt │ │ │ ├── api_def_BlockLSTMGrad.pbtxt │ │ │ ├── api_def_BlockLSTMGradV2.pbtxt │ │ │ ├── api_def_BlockLSTMV2.pbtxt │ │ │ ├── api_def_BoostedTreesAggregateStats.pbtxt │ │ │ ├── api_def_BoostedTreesBucketize.pbtxt │ │ │ ├── api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt │ │ │ ├── api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt │ │ │ ├── api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt │ │ │ ├── api_def_BoostedTreesCenterBias.pbtxt │ │ │ ├── api_def_BoostedTreesCreateEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesCreateQuantileStreamResource.pbtxt │ │ │ ├── api_def_BoostedTreesDeserializeEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt │ │ │ ├── api_def_BoostedTreesExampleDebugOutputs.pbtxt │ │ │ ├── api_def_BoostedTreesFlushQuantileSummaries.pbtxt │ │ │ ├── api_def_BoostedTreesGetEnsembleStates.pbtxt │ │ │ ├── api_def_BoostedTreesMakeQuantileSummaries.pbtxt │ │ │ ├── api_def_BoostedTreesMakeStatsSummary.pbtxt │ │ │ ├── api_def_BoostedTreesPredict.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceDeserialize.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceHandleOp.pbtxt │ │ │ ├── api_def_BoostedTreesSerializeEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesSparseAggregateStats.pbtxt │ │ │ ├── api_def_BoostedTreesSparseCalculateBestFeatureSplit.pbtxt │ │ │ ├── api_def_BoostedTreesTrainingPredict.pbtxt │ │ │ ├── api_def_BoostedTreesUpdateEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesUpdateEnsembleV2.pbtxt │ │ │ ├── api_def_BroadcastArgs.pbtxt │ │ │ ├── api_def_BroadcastGradientArgs.pbtxt │ │ │ ├── api_def_BroadcastTo.pbtxt │ │ │ ├── api_def_Bucketize.pbtxt │ │ │ ├── api_def_BytesProducedStatsDataset.pbtxt │ │ │ ├── api_def_CSRSparseMatrixComponents.pbtxt │ │ │ ├── api_def_CSRSparseMatrixToDense.pbtxt │ │ │ ├── api_def_CSRSparseMatrixToSparseTensor.pbtxt │ │ │ ├── api_def_CSVDataset.pbtxt │ │ │ ├── api_def_CSVDatasetV2.pbtxt │ │ │ ├── api_def_CTCBeamSearchDecoder.pbtxt │ │ │ ├── api_def_CTCGreedyDecoder.pbtxt │ │ │ ├── api_def_CTCLoss.pbtxt │ │ │ ├── api_def_CTCLossV2.pbtxt │ │ │ ├── api_def_CacheDataset.pbtxt │ │ │ ├── api_def_CacheDatasetV2.pbtxt │ │ │ ├── api_def_Case.pbtxt │ │ │ ├── api_def_Cast.pbtxt │ │ │ ├── api_def_Ceil.pbtxt │ │ │ ├── api_def_CheckNumerics.pbtxt │ │ │ ├── api_def_CheckNumericsV2.pbtxt │ │ │ ├── api_def_Cholesky.pbtxt │ │ │ ├── api_def_CholeskyGrad.pbtxt │ │ │ ├── api_def_ChooseFastestBranchDataset.pbtxt │ │ │ ├── api_def_ChooseFastestDataset.pbtxt │ │ │ ├── api_def_ClipByValue.pbtxt │ │ │ ├── api_def_CloseSummaryWriter.pbtxt │ │ │ ├── api_def_CollateTPUEmbeddingMemory.pbtxt │ │ │ ├── api_def_CollectiveAllToAllV2.pbtxt │ │ │ ├── api_def_CollectiveAllToAllV3.pbtxt │ │ │ ├── api_def_CollectiveAssignGroupV2.pbtxt │ │ │ ├── api_def_CollectiveBcastRecv.pbtxt │ │ │ ├── api_def_CollectiveBcastRecvV2.pbtxt │ │ │ ├── api_def_CollectiveBcastSend.pbtxt │ │ │ ├── api_def_CollectiveBcastSendV2.pbtxt │ │ │ ├── api_def_CollectiveGather.pbtxt │ │ │ ├── api_def_CollectiveGatherV2.pbtxt │ │ │ ├── api_def_CollectiveInitializeCommunicator.pbtxt │ │ │ ├── api_def_CollectivePermute.pbtxt │ │ │ ├── api_def_CollectiveReduce.pbtxt │ │ │ ├── api_def_CollectiveReduceScatterV2.pbtxt │ │ │ ├── api_def_CollectiveReduceV2.pbtxt │ │ │ ├── api_def_CollectiveReduceV3.pbtxt │ │ │ ├── api_def_CombinedNonMaxSuppression.pbtxt │ │ │ ├── api_def_Complex.pbtxt │ │ │ ├── api_def_ComplexAbs.pbtxt │ │ │ ├── api_def_CompositeTensorVariantFromComponents.pbtxt │ │ │ ├── api_def_CompositeTensorVariantToComponents.pbtxt │ │ │ ├── api_def_CompressElement.pbtxt │ │ │ ├── api_def_ComputeAccidentalHits.pbtxt │ │ │ ├── api_def_ComputeBatchSize.pbtxt │ │ │ ├── api_def_ComputeDedupDataSize.pbtxt │ │ │ ├── api_def_ComputeDedupDataTupleMask.pbtxt │ │ │ ├── api_def_Concat.pbtxt │ │ │ ├── api_def_ConcatOffset.pbtxt │ │ │ ├── api_def_ConcatV2.pbtxt │ │ │ ├── api_def_ConcatenateDataset.pbtxt │ │ │ ├── api_def_ConditionalAccumulator.pbtxt │ │ │ ├── api_def_ConfigureAndInitializeGlobalTPU.pbtxt │ │ │ ├── api_def_ConfigureDistributedTPU.pbtxt │ │ │ ├── api_def_ConfigureTPUEmbedding.pbtxt │ │ │ ├── api_def_ConfigureTPUEmbeddingHost.pbtxt │ │ │ ├── api_def_ConfigureTPUEmbeddingMemory.pbtxt │ │ │ ├── api_def_Conj.pbtxt │ │ │ ├── api_def_ConjugateTranspose.pbtxt │ │ │ ├── api_def_ConnectTPUEmbeddingHosts.pbtxt │ │ │ ├── api_def_Const.pbtxt │ │ │ ├── api_def_ConsumeMutexLock.pbtxt │ │ │ ├── api_def_ControlTrigger.pbtxt │ │ │ ├── api_def_Conv.pbtxt │ │ │ ├── api_def_Conv2D.pbtxt │ │ │ ├── api_def_Conv2DBackpropFilter.pbtxt │ │ │ ├── api_def_Conv2DBackpropFilterV2.pbtxt │ │ │ ├── api_def_Conv2DBackpropInput.pbtxt │ │ │ ├── api_def_Conv2DBackpropInputV2.pbtxt │ │ │ ├── api_def_Conv3D.pbtxt │ │ │ ├── api_def_Conv3DBackpropFilter.pbtxt │ │ │ ├── api_def_Conv3DBackpropFilterV2.pbtxt │ │ │ ├── api_def_Conv3DBackpropInput.pbtxt │ │ │ ├── api_def_Conv3DBackpropInputV2.pbtxt │ │ │ ├── api_def_ConvertToCooTensor.pbtxt │ │ │ ├── api_def_Copy.pbtxt │ │ │ ├── api_def_CopyHost.pbtxt │ │ │ ├── api_def_CopyToMesh.pbtxt │ │ │ ├── api_def_CopyToMeshGrad.pbtxt │ │ │ ├── api_def_Cos.pbtxt │ │ │ ├── api_def_Cosh.pbtxt │ │ │ ├── api_def_CountUpTo.pbtxt │ │ │ ├── api_def_CreateSummaryDbWriter.pbtxt │ │ │ ├── api_def_CreateSummaryFileWriter.pbtxt │ │ │ ├── api_def_CropAndResize.pbtxt │ │ │ ├── api_def_CropAndResizeGradBoxes.pbtxt │ │ │ ├── api_def_CropAndResizeGradImage.pbtxt │ │ │ ├── api_def_Cross.pbtxt │ │ │ ├── api_def_CrossReplicaSum.pbtxt │ │ │ ├── api_def_CudnnRNN.pbtxt │ │ │ ├── api_def_CudnnRNNBackprop.pbtxt │ │ │ ├── api_def_CudnnRNNBackpropV2.pbtxt │ │ │ ├── api_def_CudnnRNNBackpropV3.pbtxt │ │ │ ├── api_def_CudnnRNNCanonicalToParams.pbtxt │ │ │ ├── api_def_CudnnRNNCanonicalToParamsV2.pbtxt │ │ │ ├── api_def_CudnnRNNParamsSize.pbtxt │ │ │ ├── api_def_CudnnRNNParamsToCanonical.pbtxt │ │ │ ├── api_def_CudnnRNNParamsToCanonicalV2.pbtxt │ │ │ ├── api_def_CudnnRNNV2.pbtxt │ │ │ ├── api_def_CudnnRNNV3.pbtxt │ │ │ ├── api_def_Cumprod.pbtxt │ │ │ ├── api_def_Cumsum.pbtxt │ │ │ ├── api_def_CumulativeLogsumexp.pbtxt │ │ │ ├── api_def_DTensorRestoreV2.pbtxt │ │ │ ├── api_def_DTensorSetGlobalTPUArray.pbtxt │ │ │ ├── api_def_DataFormatDimMap.pbtxt │ │ │ ├── api_def_DataFormatVecPermute.pbtxt │ │ │ ├── api_def_DataServiceDataset.pbtxt │ │ │ ├── api_def_DataServiceDatasetV2.pbtxt │ │ │ ├── api_def_DataServiceDatasetV3.pbtxt │ │ │ ├── api_def_DataServiceDatasetV4.pbtxt │ │ │ ├── api_def_DatasetCardinality.pbtxt │ │ │ ├── api_def_DatasetFromGraph.pbtxt │ │ │ ├── api_def_DatasetToGraph.pbtxt │ │ │ ├── api_def_DatasetToGraphV2.pbtxt │ │ │ ├── api_def_DatasetToSingleElement.pbtxt │ │ │ ├── api_def_DatasetToTFRecord.pbtxt │ │ │ ├── api_def_Dawsn.pbtxt │ │ │ ├── api_def_DebugGradientIdentity.pbtxt │ │ │ ├── api_def_DebugGradientRefIdentity.pbtxt │ │ │ ├── api_def_DebugIdentity.pbtxt │ │ │ ├── api_def_DebugIdentityV2.pbtxt │ │ │ ├── api_def_DebugIdentityV3.pbtxt │ │ │ ├── api_def_DebugNanCount.pbtxt │ │ │ ├── api_def_DebugNumericSummary.pbtxt │ │ │ ├── api_def_DebugNumericSummaryV2.pbtxt │ │ │ ├── api_def_DecodeAndCropJpeg.pbtxt │ │ │ ├── api_def_DecodeBase64.pbtxt │ │ │ ├── api_def_DecodeBmp.pbtxt │ │ │ ├── api_def_DecodeCSV.pbtxt │ │ │ ├── api_def_DecodeCompressed.pbtxt │ │ │ ├── api_def_DecodeGif.pbtxt │ │ │ ├── api_def_DecodeImage.pbtxt │ │ │ ├── api_def_DecodeJSONExample.pbtxt │ │ │ ├── api_def_DecodeJpeg.pbtxt │ │ │ ├── api_def_DecodePaddedRaw.pbtxt │ │ │ ├── api_def_DecodePng.pbtxt │ │ │ ├── api_def_DecodeProtoV2.pbtxt │ │ │ ├── api_def_DecodeRaw.pbtxt │ │ │ ├── api_def_DecodeWav.pbtxt │ │ │ ├── api_def_DeepCopy.pbtxt │ │ │ ├── api_def_DeleteIterator.pbtxt │ │ │ ├── api_def_DeleteMemoryCache.pbtxt │ │ │ ├── api_def_DeleteMultiDeviceIterator.pbtxt │ │ │ ├── api_def_DeleteRandomSeedGenerator.pbtxt │ │ │ ├── api_def_DeleteSeedGenerator.pbtxt │ │ │ ├── api_def_DeleteSessionTensor.pbtxt │ │ │ ├── api_def_DenseBincount.pbtxt │ │ │ ├── api_def_DenseCountSparseOutput.pbtxt │ │ │ ├── api_def_DenseToCSRSparseMatrix.pbtxt │ │ │ ├── api_def_DenseToDenseSetOperation.pbtxt │ │ │ ├── api_def_DenseToSparseBatchDataset.pbtxt │ │ │ ├── api_def_DenseToSparseSetOperation.pbtxt │ │ │ ├── api_def_DepthToSpace.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNative.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNativeBackpropInput.pbtxt │ │ │ ├── api_def_Dequantize.pbtxt │ │ │ ├── api_def_DeserializeIterator.pbtxt │ │ │ ├── api_def_DeserializeManySparse.pbtxt │ │ │ ├── api_def_DeserializeSparse.pbtxt │ │ │ ├── api_def_DestroyResourceOp.pbtxt │ │ │ ├── api_def_DestroyTemporaryVariable.pbtxt │ │ │ ├── api_def_DeviceIndex.pbtxt │ │ │ ├── api_def_Diag.pbtxt │ │ │ ├── api_def_DiagPart.pbtxt │ │ │ ├── api_def_Digamma.pbtxt │ │ │ ├── api_def_Dilation2D.pbtxt │ │ │ ├── api_def_Dilation2DBackpropFilter.pbtxt │ │ │ ├── api_def_Dilation2DBackpropInput.pbtxt │ │ │ ├── api_def_DirectedInterleaveDataset.pbtxt │ │ │ ├── api_def_DisableCopyOnRead.pbtxt │ │ │ ├── api_def_DistributedSave.pbtxt │ │ │ ├── api_def_Div.pbtxt │ │ │ ├── api_def_DivNoNan.pbtxt │ │ │ ├── api_def_DrawBoundingBoxes.pbtxt │ │ │ ├── api_def_DrawBoundingBoxesV2.pbtxt │ │ │ ├── api_def_DummyIterationCounter.pbtxt │ │ │ ├── api_def_DummyMemoryCache.pbtxt │ │ │ ├── api_def_DummySeedGenerator.pbtxt │ │ │ ├── api_def_DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt │ │ │ ├── api_def_DynamicEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt │ │ │ ├── api_def_DynamicPartition.pbtxt │ │ │ ├── api_def_DynamicStitch.pbtxt │ │ │ ├── api_def_EagerPyFunc.pbtxt │ │ │ ├── api_def_EditDistance.pbtxt │ │ │ ├── api_def_Eig.pbtxt │ │ │ ├── api_def_Einsum.pbtxt │ │ │ ├── api_def_Elu.pbtxt │ │ │ ├── api_def_EluGrad.pbtxt │ │ │ ├── api_def_Empty.pbtxt │ │ │ ├── api_def_EmptyTensorList.pbtxt │ │ │ ├── api_def_EmptyTensorMap.pbtxt │ │ │ ├── api_def_EncodeBase64.pbtxt │ │ │ ├── api_def_EncodeJpeg.pbtxt │ │ │ ├── api_def_EncodeJpegVariableQuality.pbtxt │ │ │ ├── api_def_EncodePng.pbtxt │ │ │ ├── api_def_EncodeProto.pbtxt │ │ │ ├── api_def_EncodeWav.pbtxt │ │ │ ├── api_def_EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt │ │ │ ├── api_def_EnqueueTPUEmbeddingBatch.pbtxt │ │ │ ├── api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt │ │ │ ├── api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt │ │ │ ├── api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt │ │ │ ├── api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt │ │ │ ├── api_def_EnsureShape.pbtxt │ │ │ ├── api_def_Enter.pbtxt │ │ │ ├── api_def_Equal.pbtxt │ │ │ ├── api_def_Erf.pbtxt │ │ │ ├── api_def_Erfc.pbtxt │ │ │ ├── api_def_Erfinv.pbtxt │ │ │ ├── api_def_EuclideanNorm.pbtxt │ │ │ ├── api_def_ExecuteTPUEmbeddingPartitioner.pbtxt │ │ │ ├── api_def_Exit.pbtxt │ │ │ ├── api_def_Exp.pbtxt │ │ │ ├── api_def_ExpandDims.pbtxt │ │ │ ├── api_def_ExperimentalAssertNextDataset.pbtxt │ │ │ ├── api_def_ExperimentalAutoShardDataset.pbtxt │ │ │ ├── api_def_ExperimentalBytesProducedStatsDataset.pbtxt │ │ │ ├── api_def_ExperimentalCSVDataset.pbtxt │ │ │ ├── api_def_ExperimentalChooseFastestDataset.pbtxt │ │ │ ├── api_def_ExperimentalDatasetCardinality.pbtxt │ │ │ ├── api_def_ExperimentalDatasetToTFRecord.pbtxt │ │ │ ├── api_def_ExperimentalDenseToSparseBatchDataset.pbtxt │ │ │ ├── api_def_ExperimentalDirectedInterleaveDataset.pbtxt │ │ │ ├── api_def_ExperimentalGroupByReducerDataset.pbtxt │ │ │ ├── api_def_ExperimentalGroupByWindowDataset.pbtxt │ │ │ ├── api_def_ExperimentalIgnoreErrorsDataset.pbtxt │ │ │ ├── api_def_ExperimentalIteratorGetDevice.pbtxt │ │ │ ├── api_def_ExperimentalLMDBDataset.pbtxt │ │ │ ├── api_def_ExperimentalLatencyStatsDataset.pbtxt │ │ │ ├── api_def_ExperimentalMapAndBatchDataset.pbtxt │ │ │ ├── api_def_ExperimentalMapDataset.pbtxt │ │ │ ├── api_def_ExperimentalMatchingFilesDataset.pbtxt │ │ │ ├── api_def_ExperimentalMaxIntraOpParallelismDataset.pbtxt │ │ │ ├── api_def_ExperimentalNonSerializableDataset.pbtxt │ │ │ ├── api_def_ExperimentalParallelInterleaveDataset.pbtxt │ │ │ ├── api_def_ExperimentalParseExampleDataset.pbtxt │ │ │ ├── api_def_ExperimentalPrivateThreadPoolDataset.pbtxt │ │ │ ├── api_def_ExperimentalRandomDataset.pbtxt │ │ │ ├── api_def_ExperimentalRebatchDataset.pbtxt │ │ │ ├── api_def_ExperimentalScanDataset.pbtxt │ │ │ ├── api_def_ExperimentalSetStatsAggregatorDataset.pbtxt │ │ │ ├── api_def_ExperimentalSleepDataset.pbtxt │ │ │ ├── api_def_ExperimentalSlidingWindowDataset.pbtxt │ │ │ ├── api_def_ExperimentalSqlDataset.pbtxt │ │ │ ├── api_def_ExperimentalStatsAggregatorHandle.pbtxt │ │ │ ├── api_def_ExperimentalStatsAggregatorSummary.pbtxt │ │ │ ├── api_def_ExperimentalTakeWhileDataset.pbtxt │ │ │ ├── api_def_ExperimentalThreadPoolDataset.pbtxt │ │ │ ├── api_def_ExperimentalThreadPoolHandle.pbtxt │ │ │ ├── api_def_ExperimentalUnbatchDataset.pbtxt │ │ │ ├── api_def_ExperimentalUniqueDataset.pbtxt │ │ │ ├── api_def_Expint.pbtxt │ │ │ ├── api_def_Expm1.pbtxt │ │ │ ├── api_def_ExtractGlimpse.pbtxt │ │ │ ├── api_def_ExtractGlimpseV2.pbtxt │ │ │ ├── api_def_ExtractImagePatches.pbtxt │ │ │ ├── api_def_ExtractJpegShape.pbtxt │ │ │ ├── api_def_ExtractVolumePatches.pbtxt │ │ │ ├── api_def_FFT.pbtxt │ │ │ ├── api_def_FFT2D.pbtxt │ │ │ ├── api_def_FFT3D.pbtxt │ │ │ ├── api_def_FFTND.pbtxt │ │ │ ├── api_def_FIFOQueue.pbtxt │ │ │ ├── api_def_FIFOQueueV2.pbtxt │ │ │ ├── api_def_Fact.pbtxt │ │ │ ├── api_def_FakeParam.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxArgs.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxArgsGradient.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVars.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsGradient.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt │ │ │ ├── api_def_FakeQueue.pbtxt │ │ │ ├── api_def_FileSystemSetConfiguration.pbtxt │ │ │ ├── api_def_Fill.pbtxt │ │ │ ├── api_def_FilterByLastComponentDataset.pbtxt │ │ │ ├── api_def_FilterDataset.pbtxt │ │ │ ├── api_def_FinalizeDataset.pbtxt │ │ │ ├── api_def_FinalizeTPUEmbedding.pbtxt │ │ │ ├── api_def_Fingerprint.pbtxt │ │ │ ├── api_def_FixedLengthRecordDataset.pbtxt │ │ │ ├── api_def_FixedLengthRecordDatasetV2.pbtxt │ │ │ ├── api_def_FixedLengthRecordReader.pbtxt │ │ │ ├── api_def_FixedLengthRecordReaderV2.pbtxt │ │ │ ├── api_def_FixedUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_FlatMapDataset.pbtxt │ │ │ ├── api_def_Floor.pbtxt │ │ │ ├── api_def_FloorDiv.pbtxt │ │ │ ├── api_def_FloorMod.pbtxt │ │ │ ├── api_def_FlushSummaryWriter.pbtxt │ │ │ ├── api_def_For.pbtxt │ │ │ ├── api_def_FractionalAvgPool.pbtxt │ │ │ ├── api_def_FractionalAvgPoolGrad.pbtxt │ │ │ ├── api_def_FractionalMaxPool.pbtxt │ │ │ ├── api_def_FractionalMaxPoolGrad.pbtxt │ │ │ ├── api_def_FresnelCos.pbtxt │ │ │ ├── api_def_FresnelSin.pbtxt │ │ │ ├── api_def_FusedBatchNorm.pbtxt │ │ │ ├── api_def_FusedBatchNormGrad.pbtxt │ │ │ ├── api_def_FusedBatchNormGradV2.pbtxt │ │ │ ├── api_def_FusedBatchNormGradV3.pbtxt │ │ │ ├── api_def_FusedBatchNormV2.pbtxt │ │ │ ├── api_def_FusedBatchNormV3.pbtxt │ │ │ ├── api_def_FusedPadConv2D.pbtxt │ │ │ ├── api_def_FusedResizeAndPadConv2D.pbtxt │ │ │ ├── api_def_GRUBlockCell.pbtxt │ │ │ ├── api_def_GRUBlockCellGrad.pbtxt │ │ │ ├── api_def_Gather.pbtxt │ │ │ ├── api_def_GatherNd.pbtxt │ │ │ ├── api_def_GatherV2.pbtxt │ │ │ ├── api_def_GenerateBoundingBoxProposals.pbtxt │ │ │ ├── api_def_GenerateVocabRemapping.pbtxt │ │ │ ├── api_def_GeneratorDataset.pbtxt │ │ │ ├── api_def_GetElementAtIndex.pbtxt │ │ │ ├── api_def_GetMinibatchSplitsWithPhysicalReplica.pbtxt │ │ │ ├── api_def_GetMinibatchesInCsrWithPhysicalReplica.pbtxt │ │ │ ├── api_def_GetOptions.pbtxt │ │ │ ├── api_def_GetSessionHandle.pbtxt │ │ │ ├── api_def_GetSessionHandleV2.pbtxt │ │ │ ├── api_def_GetSessionTensor.pbtxt │ │ │ ├── api_def_GlobalIterId.pbtxt │ │ │ ├── api_def_Greater.pbtxt │ │ │ ├── api_def_GreaterEqual.pbtxt │ │ │ ├── api_def_GroupByReducerDataset.pbtxt │ │ │ ├── api_def_GroupByWindowDataset.pbtxt │ │ │ ├── api_def_GuaranteeConst.pbtxt │ │ │ ├── api_def_HSVToRGB.pbtxt │ │ │ ├── api_def_HashTable.pbtxt │ │ │ ├── api_def_HashTableV2.pbtxt │ │ │ ├── api_def_HistogramFixedWidth.pbtxt │ │ │ ├── api_def_HistogramSummary.pbtxt │ │ │ ├── api_def_HostConst.pbtxt │ │ │ ├── api_def_IFFT.pbtxt │ │ │ ├── api_def_IFFT2D.pbtxt │ │ │ ├── api_def_IFFT3D.pbtxt │ │ │ ├── api_def_IFFTND.pbtxt │ │ │ ├── api_def_IRFFT.pbtxt │ │ │ ├── api_def_IRFFT2D.pbtxt │ │ │ ├── api_def_IRFFT3D.pbtxt │ │ │ ├── api_def_IRFFTND.pbtxt │ │ │ ├── api_def_Identity.pbtxt │ │ │ ├── api_def_IdentityN.pbtxt │ │ │ ├── api_def_IdentityReader.pbtxt │ │ │ ├── api_def_IdentityReaderV2.pbtxt │ │ │ ├── api_def_If.pbtxt │ │ │ ├── api_def_Igamma.pbtxt │ │ │ ├── api_def_IgammaGradA.pbtxt │ │ │ ├── api_def_Igammac.pbtxt │ │ │ ├── api_def_IgnoreErrorsDataset.pbtxt │ │ │ ├── api_def_Imag.pbtxt │ │ │ ├── api_def_ImageProjectiveTransformV2.pbtxt │ │ │ ├── api_def_ImageProjectiveTransformV3.pbtxt │ │ │ ├── api_def_ImageSummary.pbtxt │ │ │ ├── api_def_ImmutableConst.pbtxt │ │ │ ├── api_def_ImportEvent.pbtxt │ │ │ ├── api_def_InTopK.pbtxt │ │ │ ├── api_def_InTopKV2.pbtxt │ │ │ ├── api_def_InfeedDequeue.pbtxt │ │ │ ├── api_def_InfeedDequeueTuple.pbtxt │ │ │ ├── api_def_InfeedEnqueue.pbtxt │ │ │ ├── api_def_InfeedEnqueuePrelinearizedBuffer.pbtxt │ │ │ ├── api_def_InfeedEnqueueTuple.pbtxt │ │ │ ├── api_def_InitializeTable.pbtxt │ │ │ ├── api_def_InitializeTableFromDataset.pbtxt │ │ │ ├── api_def_InitializeTableFromTextFile.pbtxt │ │ │ ├── api_def_InitializeTableFromTextFileV2.pbtxt │ │ │ ├── api_def_InitializeTableV2.pbtxt │ │ │ ├── api_def_InplaceAdd.pbtxt │ │ │ ├── api_def_InplaceSub.pbtxt │ │ │ ├── api_def_InplaceUpdate.pbtxt │ │ │ ├── api_def_InterleaveDataset.pbtxt │ │ │ ├── api_def_Inv.pbtxt │ │ │ ├── api_def_InvGrad.pbtxt │ │ │ ├── api_def_Invert.pbtxt │ │ │ ├── api_def_InvertPermutation.pbtxt │ │ │ ├── api_def_IsBoostedTreesEnsembleInitialized.pbtxt │ │ │ ├── api_def_IsBoostedTreesQuantileStreamResourceInitialized.pbtxt │ │ │ ├── api_def_IsFinite.pbtxt │ │ │ ├── api_def_IsInf.pbtxt │ │ │ ├── api_def_IsNan.pbtxt │ │ │ ├── api_def_IsTPUEmbeddingInitialized.pbtxt │ │ │ ├── api_def_IsVariableInitialized.pbtxt │ │ │ ├── api_def_IsotonicRegression.pbtxt │ │ │ ├── api_def_Iterator.pbtxt │ │ │ ├── api_def_IteratorFromStringHandle.pbtxt │ │ │ ├── api_def_IteratorFromStringHandleV2.pbtxt │ │ │ ├── api_def_IteratorGetDevice.pbtxt │ │ │ ├── api_def_IteratorGetNext.pbtxt │ │ │ ├── api_def_IteratorGetNextAsOptional.pbtxt │ │ │ ├── api_def_IteratorGetNextSync.pbtxt │ │ │ ├── api_def_IteratorToStringHandle.pbtxt │ │ │ ├── api_def_IteratorV2.pbtxt │ │ │ ├── api_def_KMC2ChainInitialization.pbtxt │ │ │ ├── api_def_KmeansPlusPlusInitialization.pbtxt │ │ │ ├── api_def_KthOrderStatistic.pbtxt │ │ │ ├── api_def_L2Loss.pbtxt │ │ │ ├── api_def_LMDBDataset.pbtxt │ │ │ ├── api_def_LMDBReader.pbtxt │ │ │ ├── api_def_LRN.pbtxt │ │ │ ├── api_def_LRNGrad.pbtxt │ │ │ ├── api_def_LSTMBlockCell.pbtxt │ │ │ ├── api_def_LSTMBlockCellGrad.pbtxt │ │ │ ├── api_def_LatencyStatsDataset.pbtxt │ │ │ ├── api_def_LeakyRelu.pbtxt │ │ │ ├── api_def_LeakyReluGrad.pbtxt │ │ │ ├── api_def_LearnedUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_LeftShift.pbtxt │ │ │ ├── api_def_LegacyParallelInterleaveDatasetV2.pbtxt │ │ │ ├── api_def_Less.pbtxt │ │ │ ├── api_def_LessEqual.pbtxt │ │ │ ├── api_def_Lgamma.pbtxt │ │ │ ├── api_def_LinSpace.pbtxt │ │ │ ├── api_def_ListDataset.pbtxt │ │ │ ├── api_def_ListDiff.pbtxt │ │ │ ├── api_def_LoadAllTPUEmbeddingParameters.pbtxt │ │ │ ├── api_def_LoadAndRemapMatrix.pbtxt │ │ │ ├── api_def_LoadDataset.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingADAMParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingAdadeltaParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingAdagradMomentumParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingAdagradParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingFTRLParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingMomentumParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingProximalAdagradParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingProximalYogiParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingRMSPropParameters.pbtxt │ │ │ ├── api_def_LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt │ │ │ ├── api_def_Log.pbtxt │ │ │ ├── api_def_Log1p.pbtxt │ │ │ ├── api_def_LogMatrixDeterminant.pbtxt │ │ │ ├── api_def_LogSoftmax.pbtxt │ │ │ ├── api_def_LogUniformCandidateSampler.pbtxt │ │ │ ├── api_def_LogicalAnd.pbtxt │ │ │ ├── api_def_LogicalNot.pbtxt │ │ │ ├── api_def_LogicalOr.pbtxt │ │ │ ├── api_def_LookupTableExport.pbtxt │ │ │ ├── api_def_LookupTableExportV2.pbtxt │ │ │ ├── api_def_LookupTableFind.pbtxt │ │ │ ├── api_def_LookupTableFindV2.pbtxt │ │ │ ├── api_def_LookupTableImport.pbtxt │ │ │ ├── api_def_LookupTableImportV2.pbtxt │ │ │ ├── api_def_LookupTableInsert.pbtxt │ │ │ ├── api_def_LookupTableInsertV2.pbtxt │ │ │ ├── api_def_LookupTableRemoveV2.pbtxt │ │ │ ├── api_def_LookupTableSize.pbtxt │ │ │ ├── api_def_LookupTableSizeV2.pbtxt │ │ │ ├── api_def_LoopCond.pbtxt │ │ │ ├── api_def_LowerBound.pbtxt │ │ │ ├── api_def_Lu.pbtxt │ │ │ ├── api_def_MakeIterator.pbtxt │ │ │ ├── api_def_MakeUnique.pbtxt │ │ │ ├── api_def_MapAndBatchDataset.pbtxt │ │ │ ├── api_def_MapClear.pbtxt │ │ │ ├── api_def_MapDataset.pbtxt │ │ │ ├── api_def_MapDefun.pbtxt │ │ │ ├── api_def_MapIncompleteSize.pbtxt │ │ │ ├── api_def_MapPeek.pbtxt │ │ │ ├── api_def_MapSize.pbtxt │ │ │ ├── api_def_MapStage.pbtxt │ │ │ ├── api_def_MapUnstage.pbtxt │ │ │ ├── api_def_MapUnstageNoKey.pbtxt │ │ │ ├── api_def_MatMul.pbtxt │ │ │ ├── api_def_MatchingFiles.pbtxt │ │ │ ├── api_def_MatchingFilesDataset.pbtxt │ │ │ ├── api_def_MatrixBandPart.pbtxt │ │ │ ├── api_def_MatrixDeterminant.pbtxt │ │ │ ├── api_def_MatrixDiag.pbtxt │ │ │ ├── api_def_MatrixDiagPart.pbtxt │ │ │ ├── api_def_MatrixDiagPartV2.pbtxt │ │ │ ├── api_def_MatrixDiagPartV3.pbtxt │ │ │ ├── api_def_MatrixDiagV2.pbtxt │ │ │ ├── api_def_MatrixDiagV3.pbtxt │ │ │ ├── api_def_MatrixExponential.pbtxt │ │ │ ├── api_def_MatrixInverse.pbtxt │ │ │ ├── api_def_MatrixLogarithm.pbtxt │ │ │ ├── api_def_MatrixSetDiag.pbtxt │ │ │ ├── api_def_MatrixSetDiagV2.pbtxt │ │ │ ├── api_def_MatrixSetDiagV3.pbtxt │ │ │ ├── api_def_MatrixSolve.pbtxt │ │ │ ├── api_def_MatrixSolveLs.pbtxt │ │ │ ├── api_def_MatrixSquareRoot.pbtxt │ │ │ ├── api_def_MatrixTriangularSolve.pbtxt │ │ │ ├── api_def_Max.pbtxt │ │ │ ├── api_def_MaxIntraOpParallelismDataset.pbtxt │ │ │ ├── api_def_MaxPool.pbtxt │ │ │ ├── api_def_MaxPool3D.pbtxt │ │ │ ├── api_def_MaxPool3DGrad.pbtxt │ │ │ ├── api_def_MaxPool3DGradGrad.pbtxt │ │ │ ├── api_def_MaxPoolGrad.pbtxt │ │ │ ├── api_def_MaxPoolGradGrad.pbtxt │ │ │ ├── api_def_MaxPoolGradGradV2.pbtxt │ │ │ ├── api_def_MaxPoolGradGradWithArgmax.pbtxt │ │ │ ├── api_def_MaxPoolGradV2.pbtxt │ │ │ ├── api_def_MaxPoolGradWithArgmax.pbtxt │ │ │ ├── api_def_MaxPoolV2.pbtxt │ │ │ ├── api_def_MaxPoolWithArgmax.pbtxt │ │ │ ├── api_def_Maximum.pbtxt │ │ │ ├── api_def_Mean.pbtxt │ │ │ ├── api_def_Merge.pbtxt │ │ │ ├── api_def_MergeDedupData.pbtxt │ │ │ ├── api_def_MergeSummary.pbtxt │ │ │ ├── api_def_MergeV2Checkpoints.pbtxt │ │ │ ├── api_def_Mfcc.pbtxt │ │ │ ├── api_def_Min.pbtxt │ │ │ ├── api_def_Minimum.pbtxt │ │ │ ├── api_def_MirrorPad.pbtxt │ │ │ ├── api_def_MirrorPadGrad.pbtxt │ │ │ ├── api_def_MlirPassthroughOp.pbtxt │ │ │ ├── api_def_Mod.pbtxt │ │ │ ├── api_def_ModelDataset.pbtxt │ │ │ ├── api_def_Mul.pbtxt │ │ │ ├── api_def_MulNoNan.pbtxt │ │ │ ├── api_def_MultiDeviceIterator.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorFromStringHandle.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorGetNextFromShard.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorInit.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorToStringHandle.pbtxt │ │ │ ├── api_def_Multinomial.pbtxt │ │ │ ├── api_def_MutableDenseHashTable.pbtxt │ │ │ ├── api_def_MutableDenseHashTableV2.pbtxt │ │ │ ├── api_def_MutableHashTable.pbtxt │ │ │ ├── api_def_MutableHashTableOfTensors.pbtxt │ │ │ ├── api_def_MutableHashTableOfTensorsV2.pbtxt │ │ │ ├── api_def_MutableHashTableV2.pbtxt │ │ │ ├── api_def_MutexLock.pbtxt │ │ │ ├── api_def_MutexV2.pbtxt │ │ │ ├── api_def_NcclAllReduce.pbtxt │ │ │ ├── api_def_NcclBroadcast.pbtxt │ │ │ ├── api_def_NcclReduce.pbtxt │ │ │ ├── api_def_Ndtri.pbtxt │ │ │ ├── api_def_NearestNeighbors.pbtxt │ │ │ ├── api_def_Neg.pbtxt │ │ │ ├── api_def_NegTrain.pbtxt │ │ │ ├── api_def_NextAfter.pbtxt │ │ │ ├── api_def_NextIteration.pbtxt │ │ │ ├── api_def_NoOp.pbtxt │ │ │ ├── api_def_NonDeterministicInts.pbtxt │ │ │ ├── api_def_NonMaxSuppression.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV2.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV3.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV4.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV5.pbtxt │ │ │ ├── api_def_NonMaxSuppressionWithOverlaps.pbtxt │ │ │ ├── api_def_NonSerializableDataset.pbtxt │ │ │ ├── api_def_NotEqual.pbtxt │ │ │ ├── api_def_NthElement.pbtxt │ │ │ ├── api_def_OneHot.pbtxt │ │ │ ├── api_def_OneShotIterator.pbtxt │ │ │ ├── api_def_OnesLike.pbtxt │ │ │ ├── api_def_OptimizeDataset.pbtxt │ │ │ ├── api_def_OptimizeDatasetV2.pbtxt │ │ │ ├── api_def_OptionalFromValue.pbtxt │ │ │ ├── api_def_OptionalGetValue.pbtxt │ │ │ ├── api_def_OptionalHasValue.pbtxt │ │ │ ├── api_def_OptionalNone.pbtxt │ │ │ ├── api_def_OptionsDataset.pbtxt │ │ │ ├── api_def_OrderedMapClear.pbtxt │ │ │ ├── api_def_OrderedMapIncompleteSize.pbtxt │ │ │ ├── api_def_OrderedMapPeek.pbtxt │ │ │ ├── api_def_OrderedMapSize.pbtxt │ │ │ ├── api_def_OrderedMapStage.pbtxt │ │ │ ├── api_def_OrderedMapUnstage.pbtxt │ │ │ ├── api_def_OrderedMapUnstageNoKey.pbtxt │ │ │ ├── api_def_OutfeedDequeue.pbtxt │ │ │ ├── api_def_OutfeedDequeueTuple.pbtxt │ │ │ ├── api_def_OutfeedDequeueTupleV2.pbtxt │ │ │ ├── api_def_OutfeedDequeueV2.pbtxt │ │ │ ├── api_def_OutfeedEnqueue.pbtxt │ │ │ ├── api_def_OutfeedEnqueueTuple.pbtxt │ │ │ ├── api_def_Pack.pbtxt │ │ │ ├── api_def_Pad.pbtxt │ │ │ ├── api_def_PadV2.pbtxt │ │ │ ├── api_def_PaddedBatchDataset.pbtxt │ │ │ ├── api_def_PaddedBatchDatasetV2.pbtxt │ │ │ ├── api_def_PaddingFIFOQueue.pbtxt │ │ │ ├── api_def_PaddingFIFOQueueV2.pbtxt │ │ │ ├── api_def_ParallelBatchDataset.pbtxt │ │ │ ├── api_def_ParallelConcat.pbtxt │ │ │ ├── api_def_ParallelDynamicStitch.pbtxt │ │ │ ├── api_def_ParallelFilterDataset.pbtxt │ │ │ ├── api_def_ParallelInterleaveDataset.pbtxt │ │ │ ├── api_def_ParallelInterleaveDatasetV2.pbtxt │ │ │ ├── api_def_ParallelInterleaveDatasetV3.pbtxt │ │ │ ├── api_def_ParallelInterleaveDatasetV4.pbtxt │ │ │ ├── api_def_ParallelMapDataset.pbtxt │ │ │ ├── api_def_ParallelMapDatasetV2.pbtxt │ │ │ ├── api_def_ParameterizedTruncatedNormal.pbtxt │ │ │ ├── api_def_ParseExample.pbtxt │ │ │ ├── api_def_ParseExampleDataset.pbtxt │ │ │ ├── api_def_ParseExampleDatasetV2.pbtxt │ │ │ ├── api_def_ParseExampleV2.pbtxt │ │ │ ├── api_def_ParseSequenceExample.pbtxt │ │ │ ├── api_def_ParseSequenceExampleV2.pbtxt │ │ │ ├── api_def_ParseSingleExample.pbtxt │ │ │ ├── api_def_ParseSingleSequenceExample.pbtxt │ │ │ ├── api_def_ParseTensor.pbtxt │ │ │ ├── api_def_PartitionedCall.pbtxt │ │ │ ├── api_def_Placeholder.pbtxt │ │ │ ├── api_def_PlaceholderV2.pbtxt │ │ │ ├── api_def_PlaceholderWithDefault.pbtxt │ │ │ ├── api_def_Polygamma.pbtxt │ │ │ ├── api_def_PopulationCount.pbtxt │ │ │ ├── api_def_Pow.pbtxt │ │ │ ├── api_def_PrefetchDataset.pbtxt │ │ │ ├── api_def_Prelinearize.pbtxt │ │ │ ├── api_def_PrelinearizeTuple.pbtxt │ │ │ ├── api_def_PreventGradient.pbtxt │ │ │ ├── api_def_Print.pbtxt │ │ │ ├── api_def_PrintV2.pbtxt │ │ │ ├── api_def_PriorityQueue.pbtxt │ │ │ ├── api_def_PriorityQueueV2.pbtxt │ │ │ ├── api_def_PrivateThreadPoolDataset.pbtxt │ │ │ ├── api_def_Prod.pbtxt │ │ │ ├── api_def_PyFunc.pbtxt │ │ │ ├── api_def_PyFuncStateless.pbtxt │ │ │ ├── api_def_Qr.pbtxt │ │ │ ├── api_def_QuantizeAndDequantize.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV2.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV3.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV4.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV4Grad.pbtxt │ │ │ ├── api_def_QuantizeDownAndShrinkRange.pbtxt │ │ │ ├── api_def_QuantizeV2.pbtxt │ │ │ ├── api_def_QuantizedAdd.pbtxt │ │ │ ├── api_def_QuantizedAvgPool.pbtxt │ │ │ ├── api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt │ │ │ ├── api_def_QuantizedBiasAdd.pbtxt │ │ │ ├── api_def_QuantizedConcat.pbtxt │ │ │ ├── api_def_QuantizedConv2D.pbtxt │ │ │ ├── api_def_QuantizedConv2DAndRelu.pbtxt │ │ │ ├── api_def_QuantizedConv2DAndReluAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedConv2DAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedConv2DPerChannel.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBias.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBiasAndRelu.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBiasAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBiasSumAndRelu.pbtxt │ │ │ ├── api_def_QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedDepthwiseConv2D.pbtxt │ │ │ ├── api_def_QuantizedDepthwiseConv2DWithBias.pbtxt │ │ │ ├── api_def_QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt │ │ │ ├── api_def_QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedInstanceNorm.pbtxt │ │ │ ├── api_def_QuantizedMatMul.pbtxt │ │ │ ├── api_def_QuantizedMatMulWithBias.pbtxt │ │ │ ├── api_def_QuantizedMatMulWithBiasAndDequantize.pbtxt │ │ │ ├── api_def_QuantizedMatMulWithBiasAndRelu.pbtxt │ │ │ ├── api_def_QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedMatMulWithBiasAndRequantize.pbtxt │ │ │ ├── api_def_QuantizedMaxPool.pbtxt │ │ │ ├── api_def_QuantizedMul.pbtxt │ │ │ ├── api_def_QuantizedRelu.pbtxt │ │ │ ├── api_def_QuantizedRelu6.pbtxt │ │ │ ├── api_def_QuantizedReluX.pbtxt │ │ │ ├── api_def_QuantizedReshape.pbtxt │ │ │ ├── api_def_QuantizedResizeBilinear.pbtxt │ │ │ ├── api_def_QueueClose.pbtxt │ │ │ ├── api_def_QueueCloseV2.pbtxt │ │ │ ├── api_def_QueueDequeue.pbtxt │ │ │ ├── api_def_QueueDequeueMany.pbtxt │ │ │ ├── api_def_QueueDequeueManyV2.pbtxt │ │ │ ├── api_def_QueueDequeueUpTo.pbtxt │ │ │ ├── api_def_QueueDequeueUpToV2.pbtxt │ │ │ ├── api_def_QueueDequeueV2.pbtxt │ │ │ ├── api_def_QueueEnqueue.pbtxt │ │ │ ├── api_def_QueueEnqueueMany.pbtxt │ │ │ ├── api_def_QueueEnqueueManyV2.pbtxt │ │ │ ├── api_def_QueueEnqueueV2.pbtxt │ │ │ ├── api_def_QueueIsClosed.pbtxt │ │ │ ├── api_def_QueueIsClosedV2.pbtxt │ │ │ ├── api_def_QueueSize.pbtxt │ │ │ ├── api_def_QueueSizeV2.pbtxt │ │ │ ├── api_def_RFFT.pbtxt │ │ │ ├── api_def_RFFT2D.pbtxt │ │ │ ├── api_def_RFFT3D.pbtxt │ │ │ ├── api_def_RFFTND.pbtxt │ │ │ ├── api_def_RGBToHSV.pbtxt │ │ │ ├── api_def_RaggedBincount.pbtxt │ │ │ ├── api_def_RaggedCountSparseOutput.pbtxt │ │ │ ├── api_def_RaggedCross.pbtxt │ │ │ ├── api_def_RaggedFillEmptyRows.pbtxt │ │ │ ├── api_def_RaggedFillEmptyRowsGrad.pbtxt │ │ │ ├── api_def_RaggedGather.pbtxt │ │ │ ├── api_def_RaggedRange.pbtxt │ │ │ ├── api_def_RaggedTensorFromVariant.pbtxt │ │ │ ├── api_def_RaggedTensorToSparse.pbtxt │ │ │ ├── api_def_RaggedTensorToTensor.pbtxt │ │ │ ├── api_def_RaggedTensorToVariant.pbtxt │ │ │ ├── api_def_RaggedTensorToVariantGradient.pbtxt │ │ │ ├── api_def_RandomCrop.pbtxt │ │ │ ├── api_def_RandomDataset.pbtxt │ │ │ ├── api_def_RandomDatasetV2.pbtxt │ │ │ ├── api_def_RandomGamma.pbtxt │ │ │ ├── api_def_RandomGammaGrad.pbtxt │ │ │ ├── api_def_RandomIndexShuffle.pbtxt │ │ │ ├── api_def_RandomPoisson.pbtxt │ │ │ ├── api_def_RandomPoissonV2.pbtxt │ │ │ ├── api_def_RandomShuffle.pbtxt │ │ │ ├── api_def_RandomShuffleQueue.pbtxt │ │ │ ├── api_def_RandomShuffleQueueV2.pbtxt │ │ │ ├── api_def_RandomStandardNormal.pbtxt │ │ │ ├── api_def_RandomUniform.pbtxt │ │ │ ├── api_def_RandomUniformInt.pbtxt │ │ │ ├── api_def_Range.pbtxt │ │ │ ├── api_def_RangeDataset.pbtxt │ │ │ ├── api_def_Rank.pbtxt │ │ │ ├── api_def_ReadFile.pbtxt │ │ │ ├── api_def_ReadVariableOp.pbtxt │ │ │ ├── api_def_ReadVariableXlaSplitND.pbtxt │ │ │ ├── api_def_ReaderNumRecordsProduced.pbtxt │ │ │ ├── api_def_ReaderNumRecordsProducedV2.pbtxt │ │ │ ├── api_def_ReaderNumWorkUnitsCompleted.pbtxt │ │ │ ├── api_def_ReaderNumWorkUnitsCompletedV2.pbtxt │ │ │ ├── api_def_ReaderRead.pbtxt │ │ │ ├── api_def_ReaderReadUpTo.pbtxt │ │ │ ├── api_def_ReaderReadUpToV2.pbtxt │ │ │ ├── api_def_ReaderReadV2.pbtxt │ │ │ ├── api_def_ReaderReset.pbtxt │ │ │ ├── api_def_ReaderResetV2.pbtxt │ │ │ ├── api_def_ReaderRestoreState.pbtxt │ │ │ ├── api_def_ReaderRestoreStateV2.pbtxt │ │ │ ├── api_def_ReaderSerializeState.pbtxt │ │ │ ├── api_def_ReaderSerializeStateV2.pbtxt │ │ │ ├── api_def_Real.pbtxt │ │ │ ├── api_def_RealDiv.pbtxt │ │ │ ├── api_def_RebatchDataset.pbtxt │ │ │ ├── api_def_RebatchDatasetV2.pbtxt │ │ │ ├── api_def_Reciprocal.pbtxt │ │ │ ├── api_def_ReciprocalGrad.pbtxt │ │ │ ├── api_def_RecordInput.pbtxt │ │ │ ├── api_def_Recv.pbtxt │ │ │ ├── api_def_RecvTPUEmbeddingActivations.pbtxt │ │ │ ├── api_def_ReduceDataset.pbtxt │ │ │ ├── api_def_ReduceJoin.pbtxt │ │ │ ├── api_def_RefEnter.pbtxt │ │ │ ├── api_def_RefExit.pbtxt │ │ │ ├── api_def_RefIdentity.pbtxt │ │ │ ├── api_def_RefMerge.pbtxt │ │ │ ├── api_def_RefNextIteration.pbtxt │ │ │ ├── api_def_RefSelect.pbtxt │ │ │ ├── api_def_RefSwitch.pbtxt │ │ │ ├── api_def_RegexFullMatch.pbtxt │ │ │ ├── api_def_RegexReplace.pbtxt │ │ │ ├── api_def_RegisterDataset.pbtxt │ │ │ ├── api_def_RegisterDatasetV2.pbtxt │ │ │ ├── api_def_Relayout.pbtxt │ │ │ ├── api_def_RelayoutLike.pbtxt │ │ │ ├── api_def_Relu.pbtxt │ │ │ ├── api_def_Relu6.pbtxt │ │ │ ├── api_def_Relu6Grad.pbtxt │ │ │ ├── api_def_ReluGrad.pbtxt │ │ │ ├── api_def_RemoteCall.pbtxt │ │ │ ├── api_def_RepeatDataset.pbtxt │ │ │ ├── api_def_RequantizationRange.pbtxt │ │ │ ├── api_def_RequantizationRangePerChannel.pbtxt │ │ │ ├── api_def_Requantize.pbtxt │ │ │ ├── api_def_RequantizePerChannel.pbtxt │ │ │ ├── api_def_Reshape.pbtxt │ │ │ ├── api_def_ResizeArea.pbtxt │ │ │ ├── api_def_ResizeBicubic.pbtxt │ │ │ ├── api_def_ResizeBicubicGrad.pbtxt │ │ │ ├── api_def_ResizeBilinear.pbtxt │ │ │ ├── api_def_ResizeBilinearGrad.pbtxt │ │ │ ├── api_def_ResizeNearestNeighbor.pbtxt │ │ │ ├── api_def_ResizeNearestNeighborGrad.pbtxt │ │ │ ├── api_def_ResourceAccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_ResourceAccumulatorNumAccumulated.pbtxt │ │ │ ├── api_def_ResourceAccumulatorSetGlobalStep.pbtxt │ │ │ ├── api_def_ResourceAccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_ResourceApplyAdaMax.pbtxt │ │ │ ├── api_def_ResourceApplyAdadelta.pbtxt │ │ │ ├── api_def_ResourceApplyAdagrad.pbtxt │ │ │ ├── api_def_ResourceApplyAdagradDA.pbtxt │ │ │ ├── api_def_ResourceApplyAdagradV2.pbtxt │ │ │ ├── api_def_ResourceApplyAdam.pbtxt │ │ │ ├── api_def_ResourceApplyAdamWithAmsgrad.pbtxt │ │ │ ├── api_def_ResourceApplyAddSign.pbtxt │ │ │ ├── api_def_ResourceApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ResourceApplyFtrl.pbtxt │ │ │ ├── api_def_ResourceApplyFtrlV2.pbtxt │ │ │ ├── api_def_ResourceApplyGradientDescent.pbtxt │ │ │ ├── api_def_ResourceApplyKerasMomentum.pbtxt │ │ │ ├── api_def_ResourceApplyMomentum.pbtxt │ │ │ ├── api_def_ResourceApplyPowerSign.pbtxt │ │ │ ├── api_def_ResourceApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ResourceApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ResourceApplyRMSProp.pbtxt │ │ │ ├── api_def_ResourceConditionalAccumulator.pbtxt │ │ │ ├── api_def_ResourceCountUpTo.pbtxt │ │ │ ├── api_def_ResourceGather.pbtxt │ │ │ ├── api_def_ResourceGatherNd.pbtxt │ │ │ ├── api_def_ResourceScatterAdd.pbtxt │ │ │ ├── api_def_ResourceScatterDiv.pbtxt │ │ │ ├── api_def_ResourceScatterMax.pbtxt │ │ │ ├── api_def_ResourceScatterMin.pbtxt │ │ │ ├── api_def_ResourceScatterMul.pbtxt │ │ │ ├── api_def_ResourceScatterNdAdd.pbtxt │ │ │ ├── api_def_ResourceScatterNdMax.pbtxt │ │ │ ├── api_def_ResourceScatterNdMin.pbtxt │ │ │ ├── api_def_ResourceScatterNdSub.pbtxt │ │ │ ├── api_def_ResourceScatterNdUpdate.pbtxt │ │ │ ├── api_def_ResourceScatterSub.pbtxt │ │ │ ├── api_def_ResourceScatterUpdate.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdadelta.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagrad.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagradDA.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagradV2.pbtxt │ │ │ ├── api_def_ResourceSparseApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ResourceSparseApplyFtrl.pbtxt │ │ │ ├── api_def_ResourceSparseApplyFtrlV2.pbtxt │ │ │ ├── api_def_ResourceSparseApplyKerasMomentum.pbtxt │ │ │ ├── api_def_ResourceSparseApplyMomentum.pbtxt │ │ │ ├── api_def_ResourceSparseApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ResourceSparseApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ResourceSparseApplyRMSProp.pbtxt │ │ │ ├── api_def_ResourceStridedSliceAssign.pbtxt │ │ │ ├── api_def_Restore.pbtxt │ │ │ ├── api_def_RestoreSlice.pbtxt │ │ │ ├── api_def_RestoreV2.pbtxt │ │ │ ├── api_def_RetrieveAllTPUEmbeddingParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingADAMParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingAdadeltaParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingAdagradMomentumParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingAdagradParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingFTRLParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingMomentumParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingProximalYogiParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingRMSPropParameters.pbtxt │ │ │ ├── api_def_RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt │ │ │ ├── api_def_Reverse.pbtxt │ │ │ ├── api_def_ReverseSequence.pbtxt │ │ │ ├── api_def_ReverseV2.pbtxt │ │ │ ├── api_def_RewriteDataset.pbtxt │ │ │ ├── api_def_RightShift.pbtxt │ │ │ ├── api_def_Rint.pbtxt │ │ │ ├── api_def_RiscAbs.pbtxt │ │ │ ├── api_def_RiscAdd.pbtxt │ │ │ ├── api_def_RiscBinaryArithmetic.pbtxt │ │ │ ├── api_def_RiscBinaryComparison.pbtxt │ │ │ ├── api_def_RiscBitcast.pbtxt │ │ │ ├── api_def_RiscBroadcast.pbtxt │ │ │ ├── api_def_RiscCast.pbtxt │ │ │ ├── api_def_RiscCeil.pbtxt │ │ │ ├── api_def_RiscCholesky.pbtxt │ │ │ ├── api_def_RiscConcat.pbtxt │ │ │ ├── api_def_RiscCondition.pbtxt │ │ │ ├── api_def_RiscConv.pbtxt │ │ │ ├── api_def_RiscCos.pbtxt │ │ │ ├── api_def_RiscDiv.pbtxt │ │ │ ├── api_def_RiscDot.pbtxt │ │ │ ├── api_def_RiscExp.pbtxt │ │ │ ├── api_def_RiscFft.pbtxt │ │ │ ├── api_def_RiscFloor.pbtxt │ │ │ ├── api_def_RiscGather.pbtxt │ │ │ ├── api_def_RiscImag.pbtxt │ │ │ ├── api_def_RiscIsFinite.pbtxt │ │ │ ├── api_def_RiscLog.pbtxt │ │ │ ├── api_def_RiscLogicalAnd.pbtxt │ │ │ ├── api_def_RiscLogicalNot.pbtxt │ │ │ ├── api_def_RiscLogicalOr.pbtxt │ │ │ ├── api_def_RiscMax.pbtxt │ │ │ ├── api_def_RiscMin.pbtxt │ │ │ ├── api_def_RiscMul.pbtxt │ │ │ ├── api_def_RiscNeg.pbtxt │ │ │ ├── api_def_RiscPad.pbtxt │ │ │ ├── api_def_RiscPool.pbtxt │ │ │ ├── api_def_RiscPow.pbtxt │ │ │ ├── api_def_RiscRandomUniform.pbtxt │ │ │ ├── api_def_RiscReal.pbtxt │ │ │ ├── api_def_RiscReduce.pbtxt │ │ │ ├── api_def_RiscRem.pbtxt │ │ │ ├── api_def_RiscReshape.pbtxt │ │ │ ├── api_def_RiscReverse.pbtxt │ │ │ ├── api_def_RiscScatter.pbtxt │ │ │ ├── api_def_RiscShape.pbtxt │ │ │ ├── api_def_RiscSign.pbtxt │ │ │ ├── api_def_RiscSlice.pbtxt │ │ │ ├── api_def_RiscSort.pbtxt │ │ │ ├── api_def_RiscSqueeze.pbtxt │ │ │ ├── api_def_RiscSub.pbtxt │ │ │ ├── api_def_RiscTranspose.pbtxt │ │ │ ├── api_def_RiscTriangularSolve.pbtxt │ │ │ ├── api_def_RiscUnary.pbtxt │ │ │ ├── api_def_RiscWhile.pbtxt │ │ │ ├── api_def_RngReadAndSkip.pbtxt │ │ │ ├── api_def_RngSkip.pbtxt │ │ │ ├── api_def_Roll.pbtxt │ │ │ ├── api_def_Round.pbtxt │ │ │ ├── api_def_Rsqrt.pbtxt │ │ │ ├── api_def_RsqrtGrad.pbtxt │ │ │ ├── api_def_SampleDistortedBoundingBox.pbtxt │ │ │ ├── api_def_SampleDistortedBoundingBoxV2.pbtxt │ │ │ ├── api_def_SamplingDataset.pbtxt │ │ │ ├── api_def_Save.pbtxt │ │ │ ├── api_def_SaveDataset.pbtxt │ │ │ ├── api_def_SaveDatasetV2.pbtxt │ │ │ ├── api_def_SaveSlices.pbtxt │ │ │ ├── api_def_SaveV2.pbtxt │ │ │ ├── api_def_ScalarSummary.pbtxt │ │ │ ├── api_def_ScaleAndTranslate.pbtxt │ │ │ ├── api_def_ScaleAndTranslateGrad.pbtxt │ │ │ ├── api_def_ScanDataset.pbtxt │ │ │ ├── api_def_ScatterAdd.pbtxt │ │ │ ├── api_def_ScatterDiv.pbtxt │ │ │ ├── api_def_ScatterMax.pbtxt │ │ │ ├── api_def_ScatterMin.pbtxt │ │ │ ├── api_def_ScatterMul.pbtxt │ │ │ ├── api_def_ScatterNd.pbtxt │ │ │ ├── api_def_ScatterNdAdd.pbtxt │ │ │ ├── api_def_ScatterNdMax.pbtxt │ │ │ ├── api_def_ScatterNdMin.pbtxt │ │ │ ├── api_def_ScatterNdNonAliasingAdd.pbtxt │ │ │ ├── api_def_ScatterNdSub.pbtxt │ │ │ ├── api_def_ScatterNdUpdate.pbtxt │ │ │ ├── api_def_ScatterSub.pbtxt │ │ │ ├── api_def_ScatterUpdate.pbtxt │ │ │ ├── api_def_SdcaFprint.pbtxt │ │ │ ├── api_def_SdcaOptimizer.pbtxt │ │ │ ├── api_def_SdcaOptimizerV2.pbtxt │ │ │ ├── api_def_SdcaShrinkL1.pbtxt │ │ │ ├── api_def_SegmentMax.pbtxt │ │ │ ├── api_def_SegmentMaxV2.pbtxt │ │ │ ├── api_def_SegmentMean.pbtxt │ │ │ ├── api_def_SegmentMin.pbtxt │ │ │ ├── api_def_SegmentMinV2.pbtxt │ │ │ ├── api_def_SegmentProd.pbtxt │ │ │ ├── api_def_SegmentProdV2.pbtxt │ │ │ ├── api_def_SegmentSum.pbtxt │ │ │ ├── api_def_SegmentSumV2.pbtxt │ │ │ ├── api_def_Select.pbtxt │ │ │ ├── api_def_SelectV2.pbtxt │ │ │ ├── api_def_SelfAdjointEig.pbtxt │ │ │ ├── api_def_SelfAdjointEigV2.pbtxt │ │ │ ├── api_def_Selu.pbtxt │ │ │ ├── api_def_SeluGrad.pbtxt │ │ │ ├── api_def_Send.pbtxt │ │ │ ├── api_def_SendTPUEmbeddingGradients.pbtxt │ │ │ ├── api_def_SerializeIterator.pbtxt │ │ │ ├── api_def_SerializeManySparse.pbtxt │ │ │ ├── api_def_SerializeSparse.pbtxt │ │ │ ├── api_def_SerializeTensor.pbtxt │ │ │ ├── api_def_SetSize.pbtxt │ │ │ ├── api_def_SetStatsAggregatorDataset.pbtxt │ │ │ ├── api_def_Shape.pbtxt │ │ │ ├── api_def_ShapeN.pbtxt │ │ │ ├── api_def_ShardDataset.pbtxt │ │ │ ├── api_def_ShardedFilename.pbtxt │ │ │ ├── api_def_ShardedFilespec.pbtxt │ │ │ ├── api_def_ShuffleAndRepeatDataset.pbtxt │ │ │ ├── api_def_ShuffleAndRepeatDatasetV2.pbtxt │ │ │ ├── api_def_ShuffleDataset.pbtxt │ │ │ ├── api_def_ShuffleDatasetV2.pbtxt │ │ │ ├── api_def_ShuffleDatasetV3.pbtxt │ │ │ ├── api_def_ShutdownDistributedTPU.pbtxt │ │ │ ├── api_def_ShutdownTPUSystem.pbtxt │ │ │ ├── api_def_Sigmoid.pbtxt │ │ │ ├── api_def_SigmoidGrad.pbtxt │ │ │ ├── api_def_Sign.pbtxt │ │ │ ├── api_def_Sin.pbtxt │ │ │ ├── api_def_Sinh.pbtxt │ │ │ ├── api_def_Size.pbtxt │ │ │ ├── api_def_SkipDataset.pbtxt │ │ │ ├── api_def_Skipgram.pbtxt │ │ │ ├── api_def_SleepDataset.pbtxt │ │ │ ├── api_def_Slice.pbtxt │ │ │ ├── api_def_SlidingWindowDataset.pbtxt │ │ │ ├── api_def_Snapshot.pbtxt │ │ │ ├── api_def_SnapshotChunkDataset.pbtxt │ │ │ ├── api_def_SnapshotDataset.pbtxt │ │ │ ├── api_def_SnapshotDatasetReader.pbtxt │ │ │ ├── api_def_SnapshotDatasetV2.pbtxt │ │ │ ├── api_def_SnapshotNestedDatasetReader.pbtxt │ │ │ ├── api_def_SobolSample.pbtxt │ │ │ ├── api_def_Softmax.pbtxt │ │ │ ├── api_def_SoftmaxCrossEntropyWithLogits.pbtxt │ │ │ ├── api_def_Softplus.pbtxt │ │ │ ├── api_def_SoftplusGrad.pbtxt │ │ │ ├── api_def_Softsign.pbtxt │ │ │ ├── api_def_SoftsignGrad.pbtxt │ │ │ ├── api_def_SpaceToBatch.pbtxt │ │ │ ├── api_def_SpaceToBatchND.pbtxt │ │ │ ├── api_def_SpaceToDepth.pbtxt │ │ │ ├── api_def_SparseAccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_SparseAccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_SparseAdd.pbtxt │ │ │ ├── api_def_SparseAddGrad.pbtxt │ │ │ ├── api_def_SparseApplyAdadelta.pbtxt │ │ │ ├── api_def_SparseApplyAdagrad.pbtxt │ │ │ ├── api_def_SparseApplyAdagradDA.pbtxt │ │ │ ├── api_def_SparseApplyAdagradV2.pbtxt │ │ │ ├── api_def_SparseApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_SparseApplyFtrl.pbtxt │ │ │ ├── api_def_SparseApplyFtrlV2.pbtxt │ │ │ ├── api_def_SparseApplyMomentum.pbtxt │ │ │ ├── api_def_SparseApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_SparseApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_SparseApplyRMSProp.pbtxt │ │ │ ├── api_def_SparseBincount.pbtxt │ │ │ ├── api_def_SparseConcat.pbtxt │ │ │ ├── api_def_SparseConditionalAccumulator.pbtxt │ │ │ ├── api_def_SparseCountSparseOutput.pbtxt │ │ │ ├── api_def_SparseCross.pbtxt │ │ │ ├── api_def_SparseCrossHashed.pbtxt │ │ │ ├── api_def_SparseCrossV2.pbtxt │ │ │ ├── api_def_SparseDenseCwiseAdd.pbtxt │ │ │ ├── api_def_SparseDenseCwiseDiv.pbtxt │ │ │ ├── api_def_SparseDenseCwiseMul.pbtxt │ │ │ ├── api_def_SparseFillEmptyRows.pbtxt │ │ │ ├── api_def_SparseFillEmptyRowsGrad.pbtxt │ │ │ ├── api_def_SparseMatMul.pbtxt │ │ │ ├── api_def_SparseMatrixAdd.pbtxt │ │ │ ├── api_def_SparseMatrixMatMul.pbtxt │ │ │ ├── api_def_SparseMatrixMul.pbtxt │ │ │ ├── api_def_SparseMatrixNNZ.pbtxt │ │ │ ├── api_def_SparseMatrixOrderingAMD.pbtxt │ │ │ ├── api_def_SparseMatrixSoftmax.pbtxt │ │ │ ├── api_def_SparseMatrixSoftmaxGrad.pbtxt │ │ │ ├── api_def_SparseMatrixSparseCholesky.pbtxt │ │ │ ├── api_def_SparseMatrixSparseMatMul.pbtxt │ │ │ ├── api_def_SparseMatrixTranspose.pbtxt │ │ │ ├── api_def_SparseMatrixZeros.pbtxt │ │ │ ├── api_def_SparseReduceMax.pbtxt │ │ │ ├── api_def_SparseReduceMaxSparse.pbtxt │ │ │ ├── api_def_SparseReduceSum.pbtxt │ │ │ ├── api_def_SparseReduceSumSparse.pbtxt │ │ │ ├── api_def_SparseReorder.pbtxt │ │ │ ├── api_def_SparseReshape.pbtxt │ │ │ ├── api_def_SparseSegmentMean.pbtxt │ │ │ ├── api_def_SparseSegmentMeanGrad.pbtxt │ │ │ ├── api_def_SparseSegmentMeanGradV2.pbtxt │ │ │ ├── api_def_SparseSegmentMeanWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtN.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNGrad.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNGradV2.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSegmentSum.pbtxt │ │ │ ├── api_def_SparseSegmentSumGrad.pbtxt │ │ │ ├── api_def_SparseSegmentSumGradV2.pbtxt │ │ │ ├── api_def_SparseSegmentSumWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSlice.pbtxt │ │ │ ├── api_def_SparseSliceGrad.pbtxt │ │ │ ├── api_def_SparseSoftmax.pbtxt │ │ │ ├── api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt │ │ │ ├── api_def_SparseSparseMaximum.pbtxt │ │ │ ├── api_def_SparseSparseMinimum.pbtxt │ │ │ ├── api_def_SparseSplit.pbtxt │ │ │ ├── api_def_SparseTensorDenseAdd.pbtxt │ │ │ ├── api_def_SparseTensorDenseMatMul.pbtxt │ │ │ ├── api_def_SparseTensorSliceDataset.pbtxt │ │ │ ├── api_def_SparseTensorToCSRSparseMatrix.pbtxt │ │ │ ├── api_def_SparseToDense.pbtxt │ │ │ ├── api_def_SparseToSparseSetOperation.pbtxt │ │ │ ├── api_def_Spence.pbtxt │ │ │ ├── api_def_Split.pbtxt │ │ │ ├── api_def_SplitDedupData.pbtxt │ │ │ ├── api_def_SplitV.pbtxt │ │ │ ├── api_def_SqlDataset.pbtxt │ │ │ ├── api_def_Sqrt.pbtxt │ │ │ ├── api_def_SqrtGrad.pbtxt │ │ │ ├── api_def_Square.pbtxt │ │ │ ├── api_def_SquaredDifference.pbtxt │ │ │ ├── api_def_Squeeze.pbtxt │ │ │ ├── api_def_Stack.pbtxt │ │ │ ├── api_def_StackClose.pbtxt │ │ │ ├── api_def_StackCloseV2.pbtxt │ │ │ ├── api_def_StackPop.pbtxt │ │ │ ├── api_def_StackPopV2.pbtxt │ │ │ ├── api_def_StackPush.pbtxt │ │ │ ├── api_def_StackPushV2.pbtxt │ │ │ ├── api_def_StackV2.pbtxt │ │ │ ├── api_def_Stage.pbtxt │ │ │ ├── api_def_StageClear.pbtxt │ │ │ ├── api_def_StagePeek.pbtxt │ │ │ ├── api_def_StageSize.pbtxt │ │ │ ├── api_def_StatefulPartitionedCall.pbtxt │ │ │ ├── api_def_StatefulRandomBinomial.pbtxt │ │ │ ├── api_def_StatefulStandardNormal.pbtxt │ │ │ ├── api_def_StatefulStandardNormalV2.pbtxt │ │ │ ├── api_def_StatefulTruncatedNormal.pbtxt │ │ │ ├── api_def_StatefulUniform.pbtxt │ │ │ ├── api_def_StatefulUniformFullInt.pbtxt │ │ │ ├── api_def_StatefulUniformInt.pbtxt │ │ │ ├── api_def_StatelessCase.pbtxt │ │ │ ├── api_def_StatelessIf.pbtxt │ │ │ ├── api_def_StatelessMultinomial.pbtxt │ │ │ ├── api_def_StatelessParameterizedTruncatedNormal.pbtxt │ │ │ ├── api_def_StatelessRandomBinomial.pbtxt │ │ │ ├── api_def_StatelessRandomGammaV2.pbtxt │ │ │ ├── api_def_StatelessRandomGammaV3.pbtxt │ │ │ ├── api_def_StatelessRandomGetAlg.pbtxt │ │ │ ├── api_def_StatelessRandomGetKeyCounter.pbtxt │ │ │ ├── api_def_StatelessRandomGetKeyCounterAlg.pbtxt │ │ │ ├── api_def_StatelessRandomNormal.pbtxt │ │ │ ├── api_def_StatelessRandomNormalV2.pbtxt │ │ │ ├── api_def_StatelessRandomPoisson.pbtxt │ │ │ ├── api_def_StatelessRandomUniform.pbtxt │ │ │ ├── api_def_StatelessRandomUniformFullInt.pbtxt │ │ │ ├── api_def_StatelessRandomUniformFullIntV2.pbtxt │ │ │ ├── api_def_StatelessRandomUniformInt.pbtxt │ │ │ ├── api_def_StatelessRandomUniformIntV2.pbtxt │ │ │ ├── api_def_StatelessRandomUniformV2.pbtxt │ │ │ ├── api_def_StatelessSampleDistortedBoundingBox.pbtxt │ │ │ ├── api_def_StatelessShuffle.pbtxt │ │ │ ├── api_def_StatelessTruncatedNormal.pbtxt │ │ │ ├── api_def_StatelessTruncatedNormalV2.pbtxt │ │ │ ├── api_def_StatelessWhile.pbtxt │ │ │ ├── api_def_StaticRegexFullMatch.pbtxt │ │ │ ├── api_def_StaticRegexReplace.pbtxt │ │ │ ├── api_def_StatsAggregatorHandle.pbtxt │ │ │ ├── api_def_StatsAggregatorHandleV2.pbtxt │ │ │ ├── api_def_StatsAggregatorSetSummaryWriter.pbtxt │ │ │ ├── api_def_StatsAggregatorSummary.pbtxt │ │ │ ├── api_def_StochasticCastToInt.pbtxt │ │ │ ├── api_def_StopGradient.pbtxt │ │ │ ├── api_def_StoreMinibatchStatisticsInFdo.pbtxt │ │ │ ├── api_def_StridedSlice.pbtxt │ │ │ ├── api_def_StridedSliceAssign.pbtxt │ │ │ ├── api_def_StridedSliceGrad.pbtxt │ │ │ ├── api_def_StringFormat.pbtxt │ │ │ ├── api_def_StringJoin.pbtxt │ │ │ ├── api_def_StringLength.pbtxt │ │ │ ├── api_def_StringLower.pbtxt │ │ │ ├── api_def_StringNGrams.pbtxt │ │ │ ├── api_def_StringSplit.pbtxt │ │ │ ├── api_def_StringSplitV2.pbtxt │ │ │ ├── api_def_StringStrip.pbtxt │ │ │ ├── api_def_StringToHashBucket.pbtxt │ │ │ ├── api_def_StringToHashBucketFast.pbtxt │ │ │ ├── api_def_StringToHashBucketStrong.pbtxt │ │ │ ├── api_def_StringToNumber.pbtxt │ │ │ ├── api_def_StringUpper.pbtxt │ │ │ ├── api_def_Sub.pbtxt │ │ │ ├── api_def_Substr.pbtxt │ │ │ ├── api_def_Sum.pbtxt │ │ │ ├── api_def_SummaryWriter.pbtxt │ │ │ ├── api_def_Svd.pbtxt │ │ │ ├── api_def_Switch.pbtxt │ │ │ ├── api_def_SymbolicGradient.pbtxt │ │ │ ├── api_def_SyncDevice.pbtxt │ │ │ ├── api_def_TFRecordDataset.pbtxt │ │ │ ├── api_def_TFRecordDatasetV2.pbtxt │ │ │ ├── api_def_TFRecordReader.pbtxt │ │ │ ├── api_def_TFRecordReaderV2.pbtxt │ │ │ ├── api_def_TPUAnnotateTensorsWithDynamicShape.pbtxt │ │ │ ├── api_def_TPUCompilationResult.pbtxt │ │ │ ├── api_def_TPUCompile.pbtxt │ │ │ ├── api_def_TPUCompileSucceededAssert.pbtxt │ │ │ ├── api_def_TPUCopyWithDynamicShape.pbtxt │ │ │ ├── api_def_TPUEmbeddingActivations.pbtxt │ │ │ ├── api_def_TPUExecute.pbtxt │ │ │ ├── api_def_TPUExecuteAndUpdateVariables.pbtxt │ │ │ ├── api_def_TPUOrdinalSelector.pbtxt │ │ │ ├── api_def_TPUPartitionedCall.pbtxt │ │ │ ├── api_def_TPUPartitionedInput.pbtxt │ │ │ ├── api_def_TPUPartitionedInputV2.pbtxt │ │ │ ├── api_def_TPUPartitionedOutput.pbtxt │ │ │ ├── api_def_TPUPartitionedOutputV2.pbtxt │ │ │ ├── api_def_TPUReplicateMetadata.pbtxt │ │ │ ├── api_def_TPUReplicatedInput.pbtxt │ │ │ ├── api_def_TPUReplicatedOutput.pbtxt │ │ │ ├── api_def_TPUReshardVariables.pbtxt │ │ │ ├── api_def_TPURoundRobin.pbtxt │ │ │ ├── api_def_TakeDataset.pbtxt │ │ │ ├── api_def_TakeManySparseFromTensorsMap.pbtxt │ │ │ ├── api_def_TakeWhileDataset.pbtxt │ │ │ ├── api_def_Tan.pbtxt │ │ │ ├── api_def_Tanh.pbtxt │ │ │ ├── api_def_TanhGrad.pbtxt │ │ │ ├── api_def_TemporaryVariable.pbtxt │ │ │ ├── api_def_TensorArray.pbtxt │ │ │ ├── api_def_TensorArrayClose.pbtxt │ │ │ ├── api_def_TensorArrayCloseV2.pbtxt │ │ │ ├── api_def_TensorArrayCloseV3.pbtxt │ │ │ ├── api_def_TensorArrayConcat.pbtxt │ │ │ ├── api_def_TensorArrayConcatV2.pbtxt │ │ │ ├── api_def_TensorArrayConcatV3.pbtxt │ │ │ ├── api_def_TensorArrayGather.pbtxt │ │ │ ├── api_def_TensorArrayGatherV2.pbtxt │ │ │ ├── api_def_TensorArrayGatherV3.pbtxt │ │ │ ├── api_def_TensorArrayGrad.pbtxt │ │ │ ├── api_def_TensorArrayGradV2.pbtxt │ │ │ ├── api_def_TensorArrayGradV3.pbtxt │ │ │ ├── api_def_TensorArrayGradWithShape.pbtxt │ │ │ ├── api_def_TensorArrayPack.pbtxt │ │ │ ├── api_def_TensorArrayRead.pbtxt │ │ │ ├── api_def_TensorArrayReadV2.pbtxt │ │ │ ├── api_def_TensorArrayReadV3.pbtxt │ │ │ ├── api_def_TensorArrayScatter.pbtxt │ │ │ ├── api_def_TensorArrayScatterV2.pbtxt │ │ │ ├── api_def_TensorArrayScatterV3.pbtxt │ │ │ ├── api_def_TensorArraySize.pbtxt │ │ │ ├── api_def_TensorArraySizeV2.pbtxt │ │ │ ├── api_def_TensorArraySizeV3.pbtxt │ │ │ ├── api_def_TensorArraySplit.pbtxt │ │ │ ├── api_def_TensorArraySplitV2.pbtxt │ │ │ ├── api_def_TensorArraySplitV3.pbtxt │ │ │ ├── api_def_TensorArrayUnpack.pbtxt │ │ │ ├── api_def_TensorArrayV2.pbtxt │ │ │ ├── api_def_TensorArrayV3.pbtxt │ │ │ ├── api_def_TensorArrayWrite.pbtxt │ │ │ ├── api_def_TensorArrayWriteV2.pbtxt │ │ │ ├── api_def_TensorArrayWriteV3.pbtxt │ │ │ ├── api_def_TensorDataset.pbtxt │ │ │ ├── api_def_TensorGetItem.pbtxt │ │ │ ├── api_def_TensorListConcat.pbtxt │ │ │ ├── api_def_TensorListConcatLists.pbtxt │ │ │ ├── api_def_TensorListConcatV2.pbtxt │ │ │ ├── api_def_TensorListElementShape.pbtxt │ │ │ ├── api_def_TensorListFromTensor.pbtxt │ │ │ ├── api_def_TensorListGather.pbtxt │ │ │ ├── api_def_TensorListGetItem.pbtxt │ │ │ ├── api_def_TensorListLength.pbtxt │ │ │ ├── api_def_TensorListPopBack.pbtxt │ │ │ ├── api_def_TensorListPushBack.pbtxt │ │ │ ├── api_def_TensorListPushBackBatch.pbtxt │ │ │ ├── api_def_TensorListReserve.pbtxt │ │ │ ├── api_def_TensorListResize.pbtxt │ │ │ ├── api_def_TensorListScatter.pbtxt │ │ │ ├── api_def_TensorListScatterIntoExistingList.pbtxt │ │ │ ├── api_def_TensorListScatterV2.pbtxt │ │ │ ├── api_def_TensorListSetItem.pbtxt │ │ │ ├── api_def_TensorListSplit.pbtxt │ │ │ ├── api_def_TensorListStack.pbtxt │ │ │ ├── api_def_TensorMapErase.pbtxt │ │ │ ├── api_def_TensorMapHasKey.pbtxt │ │ │ ├── api_def_TensorMapInsert.pbtxt │ │ │ ├── api_def_TensorMapLookup.pbtxt │ │ │ ├── api_def_TensorMapSize.pbtxt │ │ │ ├── api_def_TensorMapStackKeys.pbtxt │ │ │ ├── api_def_TensorScatterAdd.pbtxt │ │ │ ├── api_def_TensorScatterMax.pbtxt │ │ │ ├── api_def_TensorScatterMin.pbtxt │ │ │ ├── api_def_TensorScatterSub.pbtxt │ │ │ ├── api_def_TensorScatterUpdate.pbtxt │ │ │ ├── api_def_TensorSetItem.pbtxt │ │ │ ├── api_def_TensorSliceDataset.pbtxt │ │ │ ├── api_def_TensorStridedSliceUpdate.pbtxt │ │ │ ├── api_def_TensorSummary.pbtxt │ │ │ ├── api_def_TensorSummaryV2.pbtxt │ │ │ ├── api_def_TextLineDataset.pbtxt │ │ │ ├── api_def_TextLineReader.pbtxt │ │ │ ├── api_def_TextLineReaderV2.pbtxt │ │ │ ├── api_def_ThreadPoolDataset.pbtxt │ │ │ ├── api_def_ThreadPoolHandle.pbtxt │ │ │ ├── api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_Tile.pbtxt │ │ │ ├── api_def_TileGrad.pbtxt │ │ │ ├── api_def_Timestamp.pbtxt │ │ │ ├── api_def_ToBool.pbtxt │ │ │ ├── api_def_TopK.pbtxt │ │ │ ├── api_def_TopKUnique.pbtxt │ │ │ ├── api_def_TopKV2.pbtxt │ │ │ ├── api_def_TopKWithUnique.pbtxt │ │ │ ├── api_def_TpuHandleToProtoKey.pbtxt │ │ │ ├── api_def_Transpose.pbtxt │ │ │ ├── api_def_TridiagonalMatMul.pbtxt │ │ │ ├── api_def_TridiagonalSolve.pbtxt │ │ │ ├── api_def_TruncateDiv.pbtxt │ │ │ ├── api_def_TruncateMod.pbtxt │ │ │ ├── api_def_TruncatedNormal.pbtxt │ │ │ ├── api_def_Unbatch.pbtxt │ │ │ ├── api_def_UnbatchDataset.pbtxt │ │ │ ├── api_def_UnbatchGrad.pbtxt │ │ │ ├── api_def_UncompressElement.pbtxt │ │ │ ├── api_def_UnicodeDecode.pbtxt │ │ │ ├── api_def_UnicodeDecodeWithOffsets.pbtxt │ │ │ ├── api_def_UnicodeEncode.pbtxt │ │ │ ├── api_def_UnicodeScript.pbtxt │ │ │ ├── api_def_UnicodeTranscode.pbtxt │ │ │ ├── api_def_UniformCandidateSampler.pbtxt │ │ │ ├── api_def_UniformDequantize.pbtxt │ │ │ ├── api_def_UniformQuantize.pbtxt │ │ │ ├── api_def_UniformQuantizedAdd.pbtxt │ │ │ ├── api_def_UniformQuantizedClipByValue.pbtxt │ │ │ ├── api_def_UniformQuantizedConvolution.pbtxt │ │ │ ├── api_def_UniformQuantizedConvolutionHybrid.pbtxt │ │ │ ├── api_def_UniformQuantizedDot.pbtxt │ │ │ ├── api_def_UniformQuantizedDotHybrid.pbtxt │ │ │ ├── api_def_UniformRequantize.pbtxt │ │ │ ├── api_def_Unique.pbtxt │ │ │ ├── api_def_UniqueDataset.pbtxt │ │ │ ├── api_def_UniqueV2.pbtxt │ │ │ ├── api_def_UniqueWithCounts.pbtxt │ │ │ ├── api_def_UniqueWithCountsV2.pbtxt │ │ │ ├── api_def_Unpack.pbtxt │ │ │ ├── api_def_UnravelIndex.pbtxt │ │ │ ├── api_def_UnsortedSegmentJoin.pbtxt │ │ │ ├── api_def_UnsortedSegmentMax.pbtxt │ │ │ ├── api_def_UnsortedSegmentMin.pbtxt │ │ │ ├── api_def_UnsortedSegmentProd.pbtxt │ │ │ ├── api_def_UnsortedSegmentSum.pbtxt │ │ │ ├── api_def_Unstage.pbtxt │ │ │ ├── api_def_UnwrapDatasetVariant.pbtxt │ │ │ ├── api_def_UpperBound.pbtxt │ │ │ ├── api_def_VarHandleOp.pbtxt │ │ │ ├── api_def_VarIsInitializedOp.pbtxt │ │ │ ├── api_def_Variable.pbtxt │ │ │ ├── api_def_VariableShape.pbtxt │ │ │ ├── api_def_VariableV2.pbtxt │ │ │ ├── api_def_Where.pbtxt │ │ │ ├── api_def_While.pbtxt │ │ │ ├── api_def_WholeFileReader.pbtxt │ │ │ ├── api_def_WholeFileReaderV2.pbtxt │ │ │ ├── api_def_WindowDataset.pbtxt │ │ │ ├── api_def_WindowOp.pbtxt │ │ │ ├── api_def_WorkerHeartbeat.pbtxt │ │ │ ├── api_def_WrapDatasetVariant.pbtxt │ │ │ ├── api_def_WriteAudioSummary.pbtxt │ │ │ ├── api_def_WriteFile.pbtxt │ │ │ ├── api_def_WriteGraphSummary.pbtxt │ │ │ ├── api_def_WriteHistogramSummary.pbtxt │ │ │ ├── api_def_WriteImageSummary.pbtxt │ │ │ ├── api_def_WriteRawProtoSummary.pbtxt │ │ │ ├── api_def_WriteScalarSummary.pbtxt │ │ │ ├── api_def_WriteSummary.pbtxt │ │ │ ├── api_def_Xdivy.pbtxt │ │ │ ├── api_def_XlaConcatND.pbtxt │ │ │ ├── api_def_XlaHostCompute.pbtxt │ │ │ ├── api_def_XlaRecvFromHost.pbtxt │ │ │ ├── api_def_XlaRecvTPUEmbeddingActivations.pbtxt │ │ │ ├── api_def_XlaRecvTPUEmbeddingDeduplicationData.pbtxt │ │ │ ├── api_def_XlaSendTPUEmbeddingGradients.pbtxt │ │ │ ├── api_def_XlaSendToHost.pbtxt │ │ │ ├── api_def_XlaSparseCoreAdagrad.pbtxt │ │ │ ├── api_def_XlaSparseCoreAdagradMomentum.pbtxt │ │ │ ├── api_def_XlaSparseCoreAdam.pbtxt │ │ │ ├── api_def_XlaSparseCoreFtrl.pbtxt │ │ │ ├── api_def_XlaSparseCoreSgd.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmul.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithAdagradAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithAdamAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithFtrlAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithSgdAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulWithCsrInput.pbtxt │ │ │ ├── api_def_XlaSplitND.pbtxt │ │ │ ├── api_def_Xlog1py.pbtxt │ │ │ ├── api_def_Xlogy.pbtxt │ │ │ ├── api_def_ZerosLike.pbtxt │ │ │ ├── api_def_Zeta.pbtxt │ │ │ └── api_def_ZipDataset.pbtxt │ │ ├── excluded_ops.cc │ │ ├── excluded_ops.h │ │ ├── java_api │ │ │ ├── BUILD │ │ │ ├── api_def_Abort.pbtxt │ │ │ ├── api_def_Abs.pbtxt │ │ │ ├── api_def_AccumulateNV2.pbtxt │ │ │ ├── api_def_AccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_AccumulatorNumAccumulated.pbtxt │ │ │ ├── api_def_AccumulatorSetGlobalStep.pbtxt │ │ │ ├── api_def_AccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_Acos.pbtxt │ │ │ ├── api_def_Acosh.pbtxt │ │ │ ├── api_def_Add.pbtxt │ │ │ ├── api_def_AddManySparseToTensorsMap.pbtxt │ │ │ ├── api_def_AddN.pbtxt │ │ │ ├── api_def_AddSparseToTensorsMap.pbtxt │ │ │ ├── api_def_AddV2.pbtxt │ │ │ ├── api_def_AdjustContrast.pbtxt │ │ │ ├── api_def_AdjustContrastv2.pbtxt │ │ │ ├── api_def_AdjustHue.pbtxt │ │ │ ├── api_def_AdjustSaturation.pbtxt │ │ │ ├── api_def_All.pbtxt │ │ │ ├── api_def_AllCandidateSampler.pbtxt │ │ │ ├── api_def_Angle.pbtxt │ │ │ ├── api_def_AnonymousIterator.pbtxt │ │ │ ├── api_def_Any.pbtxt │ │ │ ├── api_def_ApplyAdaMax.pbtxt │ │ │ ├── api_def_ApplyAdadelta.pbtxt │ │ │ ├── api_def_ApplyAdagrad.pbtxt │ │ │ ├── api_def_ApplyAdagradDA.pbtxt │ │ │ ├── api_def_ApplyAdam.pbtxt │ │ │ ├── api_def_ApplyAddSign.pbtxt │ │ │ ├── api_def_ApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ApplyFtrl.pbtxt │ │ │ ├── api_def_ApplyFtrlV2.pbtxt │ │ │ ├── api_def_ApplyGradientDescent.pbtxt │ │ │ ├── api_def_ApplyMomentum.pbtxt │ │ │ ├── api_def_ApplyPowerSign.pbtxt │ │ │ ├── api_def_ApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ApplyRMSProp.pbtxt │ │ │ ├── api_def_ApproximateEqual.pbtxt │ │ │ ├── api_def_ArgMax.pbtxt │ │ │ ├── api_def_ArgMin.pbtxt │ │ │ ├── api_def_AsString.pbtxt │ │ │ ├── api_def_Asin.pbtxt │ │ │ ├── api_def_Asinh.pbtxt │ │ │ ├── api_def_Assert.pbtxt │ │ │ ├── api_def_Assign.pbtxt │ │ │ ├── api_def_AssignAdd.pbtxt │ │ │ ├── api_def_AssignAddVariableOp.pbtxt │ │ │ ├── api_def_AssignSub.pbtxt │ │ │ ├── api_def_AssignSubVariableOp.pbtxt │ │ │ ├── api_def_AssignVariableOp.pbtxt │ │ │ ├── api_def_Atan.pbtxt │ │ │ ├── api_def_Atan2.pbtxt │ │ │ ├── api_def_Atanh.pbtxt │ │ │ ├── api_def_AudioSpectrogram.pbtxt │ │ │ ├── api_def_AudioSummary.pbtxt │ │ │ ├── api_def_AudioSummaryV2.pbtxt │ │ │ ├── api_def_AvgPool.pbtxt │ │ │ ├── api_def_AvgPool3D.pbtxt │ │ │ ├── api_def_AvgPool3DGrad.pbtxt │ │ │ ├── api_def_AvgPoolGrad.pbtxt │ │ │ ├── api_def_Barrier.pbtxt │ │ │ ├── api_def_BarrierClose.pbtxt │ │ │ ├── api_def_BarrierIncompleteSize.pbtxt │ │ │ ├── api_def_BarrierInsertMany.pbtxt │ │ │ ├── api_def_BarrierReadySize.pbtxt │ │ │ ├── api_def_BarrierTakeMany.pbtxt │ │ │ ├── api_def_Batch.pbtxt │ │ │ ├── api_def_BatchCholesky.pbtxt │ │ │ ├── api_def_BatchCholeskyGrad.pbtxt │ │ │ ├── api_def_BatchDataset.pbtxt │ │ │ ├── api_def_BatchDatasetV2.pbtxt │ │ │ ├── api_def_BatchFFT.pbtxt │ │ │ ├── api_def_BatchFFT2D.pbtxt │ │ │ ├── api_def_BatchFFT3D.pbtxt │ │ │ ├── api_def_BatchFunction.pbtxt │ │ │ ├── api_def_BatchIFFT.pbtxt │ │ │ ├── api_def_BatchIFFT2D.pbtxt │ │ │ ├── api_def_BatchIFFT3D.pbtxt │ │ │ ├── api_def_BatchMatMul.pbtxt │ │ │ ├── api_def_BatchMatrixBandPart.pbtxt │ │ │ ├── api_def_BatchMatrixDeterminant.pbtxt │ │ │ ├── api_def_BatchMatrixDiag.pbtxt │ │ │ ├── api_def_BatchMatrixDiagPart.pbtxt │ │ │ ├── api_def_BatchMatrixInverse.pbtxt │ │ │ ├── api_def_BatchMatrixSetDiag.pbtxt │ │ │ ├── api_def_BatchMatrixSolve.pbtxt │ │ │ ├── api_def_BatchMatrixSolveLs.pbtxt │ │ │ ├── api_def_BatchMatrixTriangularSolve.pbtxt │ │ │ ├── api_def_BatchNormWithGlobalNormalization.pbtxt │ │ │ ├── api_def_BatchNormWithGlobalNormalizationGrad.pbtxt │ │ │ ├── api_def_BatchSelfAdjointEig.pbtxt │ │ │ ├── api_def_BatchSelfAdjointEigV2.pbtxt │ │ │ ├── api_def_BatchSvd.pbtxt │ │ │ ├── api_def_BatchToSpace.pbtxt │ │ │ ├── api_def_BatchToSpaceND.pbtxt │ │ │ ├── api_def_BesselI0e.pbtxt │ │ │ ├── api_def_BesselI1e.pbtxt │ │ │ ├── api_def_Betainc.pbtxt │ │ │ ├── api_def_BiasAdd.pbtxt │ │ │ ├── api_def_BiasAddGrad.pbtxt │ │ │ ├── api_def_BiasAddV1.pbtxt │ │ │ ├── api_def_BigQueryReader.pbtxt │ │ │ ├── api_def_Bincount.pbtxt │ │ │ ├── api_def_Bitcast.pbtxt │ │ │ ├── api_def_BitwiseAnd.pbtxt │ │ │ ├── api_def_BitwiseOr.pbtxt │ │ │ ├── api_def_BitwiseXor.pbtxt │ │ │ ├── api_def_BoostedTreesBucketize.pbtxt │ │ │ ├── api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt │ │ │ ├── api_def_BoostedTreesCenterBias.pbtxt │ │ │ ├── api_def_BoostedTreesCreateEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesCreateQuantileStreamResource.pbtxt │ │ │ ├── api_def_BoostedTreesDeserializeEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt │ │ │ ├── api_def_BoostedTreesExampleDebugOutputs.pbtxt │ │ │ ├── api_def_BoostedTreesGetEnsembleStates.pbtxt │ │ │ ├── api_def_BoostedTreesMakeQuantileSummaries.pbtxt │ │ │ ├── api_def_BoostedTreesMakeStatsSummary.pbtxt │ │ │ ├── api_def_BoostedTreesPredict.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceAddSummaries.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceDeserialize.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt │ │ │ ├── api_def_BoostedTreesQuantileStreamResourceHandleOp.pbtxt │ │ │ ├── api_def_BoostedTreesSerializeEnsemble.pbtxt │ │ │ ├── api_def_BoostedTreesTrainingPredict.pbtxt │ │ │ ├── api_def_BoostedTreesUpdateEnsemble.pbtxt │ │ │ ├── api_def_BroadcastArgs.pbtxt │ │ │ ├── api_def_BroadcastGradientArgs.pbtxt │ │ │ ├── api_def_BroadcastTo.pbtxt │ │ │ ├── api_def_Bucketize.pbtxt │ │ │ ├── api_def_BytesProducedStatsDataset.pbtxt │ │ │ ├── api_def_CTCBeamSearchDecoder.pbtxt │ │ │ ├── api_def_CTCGreedyDecoder.pbtxt │ │ │ ├── api_def_CTCLoss.pbtxt │ │ │ ├── api_def_CacheDataset.pbtxt │ │ │ ├── api_def_Cast.pbtxt │ │ │ ├── api_def_Ceil.pbtxt │ │ │ ├── api_def_CheckNumerics.pbtxt │ │ │ ├── api_def_CheckNumericsV2.pbtxt │ │ │ ├── api_def_Cholesky.pbtxt │ │ │ ├── api_def_CholeskyGrad.pbtxt │ │ │ ├── api_def_ClipByValue.pbtxt │ │ │ ├── api_def_CloseSummaryWriter.pbtxt │ │ │ ├── api_def_CollectiveBcastRecv.pbtxt │ │ │ ├── api_def_CollectiveBcastSend.pbtxt │ │ │ ├── api_def_CollectiveReduce.pbtxt │ │ │ ├── api_def_Complex.pbtxt │ │ │ ├── api_def_ComplexAbs.pbtxt │ │ │ ├── api_def_ComputeAccidentalHits.pbtxt │ │ │ ├── api_def_Concat.pbtxt │ │ │ ├── api_def_ConcatOffset.pbtxt │ │ │ ├── api_def_ConcatV2.pbtxt │ │ │ ├── api_def_ConcatenateDataset.pbtxt │ │ │ ├── api_def_ConditionalAccumulator.pbtxt │ │ │ ├── api_def_Conj.pbtxt │ │ │ ├── api_def_ConjugateTranspose.pbtxt │ │ │ ├── api_def_Const.pbtxt │ │ │ ├── api_def_ConsumeMutexLock.pbtxt │ │ │ ├── api_def_ControlTrigger.pbtxt │ │ │ ├── api_def_Conv2D.pbtxt │ │ │ ├── api_def_Conv2DBackpropFilter.pbtxt │ │ │ ├── api_def_Conv2DBackpropInput.pbtxt │ │ │ ├── api_def_Conv3D.pbtxt │ │ │ ├── api_def_Conv3DBackpropFilter.pbtxt │ │ │ ├── api_def_Conv3DBackpropFilterV2.pbtxt │ │ │ ├── api_def_Conv3DBackpropInput.pbtxt │ │ │ ├── api_def_Conv3DBackpropInputV2.pbtxt │ │ │ ├── api_def_Copy.pbtxt │ │ │ ├── api_def_CopyHost.pbtxt │ │ │ ├── api_def_Cos.pbtxt │ │ │ ├── api_def_Cosh.pbtxt │ │ │ ├── api_def_CountUpTo.pbtxt │ │ │ ├── api_def_CreateSummaryDbWriter.pbtxt │ │ │ ├── api_def_CreateSummaryFileWriter.pbtxt │ │ │ ├── api_def_CropAndResize.pbtxt │ │ │ ├── api_def_CropAndResizeGradBoxes.pbtxt │ │ │ ├── api_def_CropAndResizeGradImage.pbtxt │ │ │ ├── api_def_Cross.pbtxt │ │ │ ├── api_def_CudnnRNN.pbtxt │ │ │ ├── api_def_CudnnRNNBackprop.pbtxt │ │ │ ├── api_def_CudnnRNNBackpropV2.pbtxt │ │ │ ├── api_def_CudnnRNNCanonicalToParams.pbtxt │ │ │ ├── api_def_CudnnRNNParamsSize.pbtxt │ │ │ ├── api_def_CudnnRNNParamsToCanonical.pbtxt │ │ │ ├── api_def_CudnnRNNV2.pbtxt │ │ │ ├── api_def_Cumprod.pbtxt │ │ │ ├── api_def_Cumsum.pbtxt │ │ │ ├── api_def_DataFormatDimMap.pbtxt │ │ │ ├── api_def_DataFormatVecPermute.pbtxt │ │ │ ├── api_def_DatasetToGraph.pbtxt │ │ │ ├── api_def_DatasetToSingleElement.pbtxt │ │ │ ├── api_def_DatasetToTFRecord.pbtxt │ │ │ ├── api_def_DebugGradientIdentity.pbtxt │ │ │ ├── api_def_DebugGradientRefIdentity.pbtxt │ │ │ ├── api_def_DebugIdentity.pbtxt │ │ │ ├── api_def_DebugIdentityV2.pbtxt │ │ │ ├── api_def_DebugNanCount.pbtxt │ │ │ ├── api_def_DebugNumericSummary.pbtxt │ │ │ ├── api_def_DebugNumericSummaryV2.pbtxt │ │ │ ├── api_def_DecodeAndCropJpeg.pbtxt │ │ │ ├── api_def_DecodeBase64.pbtxt │ │ │ ├── api_def_DecodeBmp.pbtxt │ │ │ ├── api_def_DecodeCSV.pbtxt │ │ │ ├── api_def_DecodeCompressed.pbtxt │ │ │ ├── api_def_DecodeGif.pbtxt │ │ │ ├── api_def_DecodeJSONExample.pbtxt │ │ │ ├── api_def_DecodeJpeg.pbtxt │ │ │ ├── api_def_DecodePng.pbtxt │ │ │ ├── api_def_DecodeProtoV2.pbtxt │ │ │ ├── api_def_DecodeRaw.pbtxt │ │ │ ├── api_def_DecodeWav.pbtxt │ │ │ ├── api_def_DeepCopy.pbtxt │ │ │ ├── api_def_DeleteSessionTensor.pbtxt │ │ │ ├── api_def_DenseToDenseSetOperation.pbtxt │ │ │ ├── api_def_DenseToSparseBatchDataset.pbtxt │ │ │ ├── api_def_DenseToSparseSetOperation.pbtxt │ │ │ ├── api_def_DepthToSpace.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNative.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNativeBackpropInput.pbtxt │ │ │ ├── api_def_Dequantize.pbtxt │ │ │ ├── api_def_DeserializeIterator.pbtxt │ │ │ ├── api_def_DeserializeManySparse.pbtxt │ │ │ ├── api_def_DeserializeSparse.pbtxt │ │ │ ├── api_def_DestroyResourceOp.pbtxt │ │ │ ├── api_def_DestroyTemporaryVariable.pbtxt │ │ │ ├── api_def_Diag.pbtxt │ │ │ ├── api_def_DiagPart.pbtxt │ │ │ ├── api_def_Digamma.pbtxt │ │ │ ├── api_def_Dilation2D.pbtxt │ │ │ ├── api_def_Dilation2DBackpropFilter.pbtxt │ │ │ ├── api_def_Dilation2DBackpropInput.pbtxt │ │ │ ├── api_def_Div.pbtxt │ │ │ ├── api_def_DivNoNan.pbtxt │ │ │ ├── api_def_DrawBoundingBoxes.pbtxt │ │ │ ├── api_def_DynamicPartition.pbtxt │ │ │ ├── api_def_DynamicStitch.pbtxt │ │ │ ├── api_def_EagerPyFunc.pbtxt │ │ │ ├── api_def_EditDistance.pbtxt │ │ │ ├── api_def_Elu.pbtxt │ │ │ ├── api_def_EluGrad.pbtxt │ │ │ ├── api_def_Empty.pbtxt │ │ │ ├── api_def_EmptyTensorList.pbtxt │ │ │ ├── api_def_EncodeBase64.pbtxt │ │ │ ├── api_def_EncodeJpeg.pbtxt │ │ │ ├── api_def_EncodeJpegVariableQuality.pbtxt │ │ │ ├── api_def_EncodePng.pbtxt │ │ │ ├── api_def_EncodeProto.pbtxt │ │ │ ├── api_def_EncodeWav.pbtxt │ │ │ ├── api_def_EnqueueInQueueDataset.pbtxt │ │ │ ├── api_def_EnsureShape.pbtxt │ │ │ ├── api_def_Enter.pbtxt │ │ │ ├── api_def_Equal.pbtxt │ │ │ ├── api_def_Erf.pbtxt │ │ │ ├── api_def_Erfc.pbtxt │ │ │ ├── api_def_Exit.pbtxt │ │ │ ├── api_def_Exp.pbtxt │ │ │ ├── api_def_ExpandDims.pbtxt │ │ │ ├── api_def_ExperimentalAssertNextDataset.pbtxt │ │ │ ├── api_def_ExperimentalCSVDataset.pbtxt │ │ │ ├── api_def_ExperimentalDirectedInterleaveDataset.pbtxt │ │ │ ├── api_def_ExperimentalFunctionBufferingResource.pbtxt │ │ │ ├── api_def_ExperimentalFunctionBufferingResourceGetNext.pbtxt │ │ │ ├── api_def_ExperimentalFunctionBufferingResourceReset.pbtxt │ │ │ ├── api_def_ExperimentalIgnoreErrorsDataset.pbtxt │ │ │ ├── api_def_ExperimentalIteratorGetDevice.pbtxt │ │ │ ├── api_def_ExperimentalLMDBDataset.pbtxt │ │ │ ├── api_def_ExperimentalMapDataset.pbtxt │ │ │ ├── api_def_ExperimentalNonSerializableDataset.pbtxt │ │ │ ├── api_def_ExperimentalSleepDataset.pbtxt │ │ │ ├── api_def_ExperimentalThreadPoolDataset.pbtxt │ │ │ ├── api_def_ExperimentalThreadPoolHandle.pbtxt │ │ │ ├── api_def_ExperimentalUniqueDataset.pbtxt │ │ │ ├── api_def_Expm1.pbtxt │ │ │ ├── api_def_ExtractGlimpse.pbtxt │ │ │ ├── api_def_ExtractImagePatches.pbtxt │ │ │ ├── api_def_ExtractJpegShape.pbtxt │ │ │ ├── api_def_ExtractVolumePatches.pbtxt │ │ │ ├── api_def_FFT.pbtxt │ │ │ ├── api_def_FFT2D.pbtxt │ │ │ ├── api_def_FFT3D.pbtxt │ │ │ ├── api_def_FIFOQueue.pbtxt │ │ │ ├── api_def_FIFOQueueV2.pbtxt │ │ │ ├── api_def_Fact.pbtxt │ │ │ ├── api_def_FakeParam.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxArgs.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxArgsGradient.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVars.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsGradient.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt │ │ │ ├── api_def_FakeQueue.pbtxt │ │ │ ├── api_def_Fill.pbtxt │ │ │ ├── api_def_FilterByLastComponentDataset.pbtxt │ │ │ ├── api_def_FilterDataset.pbtxt │ │ │ ├── api_def_FixedLengthRecordDataset.pbtxt │ │ │ ├── api_def_FixedLengthRecordDatasetV2.pbtxt │ │ │ ├── api_def_FixedLengthRecordReader.pbtxt │ │ │ ├── api_def_FixedLengthRecordReaderV2.pbtxt │ │ │ ├── api_def_FixedUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_FlatMapDataset.pbtxt │ │ │ ├── api_def_Floor.pbtxt │ │ │ ├── api_def_FloorDiv.pbtxt │ │ │ ├── api_def_FloorMod.pbtxt │ │ │ ├── api_def_FlushSummaryWriter.pbtxt │ │ │ ├── api_def_For.pbtxt │ │ │ ├── api_def_FractionalAvgPool.pbtxt │ │ │ ├── api_def_FractionalAvgPoolGrad.pbtxt │ │ │ ├── api_def_FractionalMaxPool.pbtxt │ │ │ ├── api_def_FractionalMaxPoolGrad.pbtxt │ │ │ ├── api_def_FusedBatchNorm.pbtxt │ │ │ ├── api_def_FusedBatchNormGrad.pbtxt │ │ │ ├── api_def_FusedBatchNormGradV2.pbtxt │ │ │ ├── api_def_FusedBatchNormV2.pbtxt │ │ │ ├── api_def_FusedPadConv2D.pbtxt │ │ │ ├── api_def_FusedResizeAndPadConv2D.pbtxt │ │ │ ├── api_def_Gather.pbtxt │ │ │ ├── api_def_GatherNd.pbtxt │ │ │ ├── api_def_GatherV2.pbtxt │ │ │ ├── api_def_GcsConfigureBlockCache.pbtxt │ │ │ ├── api_def_GcsConfigureCredentials.pbtxt │ │ │ ├── api_def_GenerateBigQueryReaderPartitions.pbtxt │ │ │ ├── api_def_GenerateVocabRemapping.pbtxt │ │ │ ├── api_def_GeneratorDataset.pbtxt │ │ │ ├── api_def_GetSessionHandle.pbtxt │ │ │ ├── api_def_GetSessionHandleV2.pbtxt │ │ │ ├── api_def_GetSessionTensor.pbtxt │ │ │ ├── api_def_Greater.pbtxt │ │ │ ├── api_def_GreaterEqual.pbtxt │ │ │ ├── api_def_GroupByReducerDataset.pbtxt │ │ │ ├── api_def_GroupByWindowDataset.pbtxt │ │ │ ├── api_def_GuaranteeConst.pbtxt │ │ │ ├── api_def_HSVToRGB.pbtxt │ │ │ ├── api_def_HashTable.pbtxt │ │ │ ├── api_def_HashTableV2.pbtxt │ │ │ ├── api_def_HistogramFixedWidth.pbtxt │ │ │ ├── api_def_HistogramSummary.pbtxt │ │ │ ├── api_def_HostConst.pbtxt │ │ │ ├── api_def_IFFT.pbtxt │ │ │ ├── api_def_IFFT2D.pbtxt │ │ │ ├── api_def_IFFT3D.pbtxt │ │ │ ├── api_def_IRFFT.pbtxt │ │ │ ├── api_def_IRFFT2D.pbtxt │ │ │ ├── api_def_IRFFT3D.pbtxt │ │ │ ├── api_def_Identity.pbtxt │ │ │ ├── api_def_IdentityN.pbtxt │ │ │ ├── api_def_IdentityReader.pbtxt │ │ │ ├── api_def_IdentityReaderV2.pbtxt │ │ │ ├── api_def_If.pbtxt │ │ │ ├── api_def_Igamma.pbtxt │ │ │ ├── api_def_IgammaGradA.pbtxt │ │ │ ├── api_def_Igammac.pbtxt │ │ │ ├── api_def_Imag.pbtxt │ │ │ ├── api_def_ImageSummary.pbtxt │ │ │ ├── api_def_ImmutableConst.pbtxt │ │ │ ├── api_def_ImportEvent.pbtxt │ │ │ ├── api_def_InTopK.pbtxt │ │ │ ├── api_def_InTopKV2.pbtxt │ │ │ ├── api_def_InitializeTable.pbtxt │ │ │ ├── api_def_InitializeTableFromTextFile.pbtxt │ │ │ ├── api_def_InitializeTableFromTextFileV2.pbtxt │ │ │ ├── api_def_InitializeTableV2.pbtxt │ │ │ ├── api_def_InplaceAdd.pbtxt │ │ │ ├── api_def_InplaceSub.pbtxt │ │ │ ├── api_def_InplaceUpdate.pbtxt │ │ │ ├── api_def_InterleaveDataset.pbtxt │ │ │ ├── api_def_Inv.pbtxt │ │ │ ├── api_def_InvGrad.pbtxt │ │ │ ├── api_def_Invert.pbtxt │ │ │ ├── api_def_InvertPermutation.pbtxt │ │ │ ├── api_def_IsBoostedTreesEnsembleInitialized.pbtxt │ │ │ ├── api_def_IsBoostedTreesQuantileStreamResourceInitialized.pbtxt │ │ │ ├── api_def_IsFinite.pbtxt │ │ │ ├── api_def_IsInf.pbtxt │ │ │ ├── api_def_IsNan.pbtxt │ │ │ ├── api_def_IsVariableInitialized.pbtxt │ │ │ ├── api_def_Iterator.pbtxt │ │ │ ├── api_def_IteratorFromStringHandle.pbtxt │ │ │ ├── api_def_IteratorFromStringHandleV2.pbtxt │ │ │ ├── api_def_IteratorGetNext.pbtxt │ │ │ ├── api_def_IteratorGetNextAsOptional.pbtxt │ │ │ ├── api_def_IteratorGetNextSync.pbtxt │ │ │ ├── api_def_IteratorToStringHandle.pbtxt │ │ │ ├── api_def_IteratorV2.pbtxt │ │ │ ├── api_def_L2Loss.pbtxt │ │ │ ├── api_def_LMDBReader.pbtxt │ │ │ ├── api_def_LRN.pbtxt │ │ │ ├── api_def_LRNGrad.pbtxt │ │ │ ├── api_def_LatencyStatsDataset.pbtxt │ │ │ ├── api_def_LeakyRelu.pbtxt │ │ │ ├── api_def_LeakyReluGrad.pbtxt │ │ │ ├── api_def_LearnedUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_LeftShift.pbtxt │ │ │ ├── api_def_Less.pbtxt │ │ │ ├── api_def_LessEqual.pbtxt │ │ │ ├── api_def_Lgamma.pbtxt │ │ │ ├── api_def_LinSpace.pbtxt │ │ │ ├── api_def_ListDiff.pbtxt │ │ │ ├── api_def_LoadAndRemapMatrix.pbtxt │ │ │ ├── api_def_Log.pbtxt │ │ │ ├── api_def_Log1p.pbtxt │ │ │ ├── api_def_LogMatrixDeterminant.pbtxt │ │ │ ├── api_def_LogSoftmax.pbtxt │ │ │ ├── api_def_LogUniformCandidateSampler.pbtxt │ │ │ ├── api_def_LogicalAnd.pbtxt │ │ │ ├── api_def_LogicalNot.pbtxt │ │ │ ├── api_def_LogicalOr.pbtxt │ │ │ ├── api_def_LookupTableExport.pbtxt │ │ │ ├── api_def_LookupTableExportV2.pbtxt │ │ │ ├── api_def_LookupTableFind.pbtxt │ │ │ ├── api_def_LookupTableFindV2.pbtxt │ │ │ ├── api_def_LookupTableImport.pbtxt │ │ │ ├── api_def_LookupTableImportV2.pbtxt │ │ │ ├── api_def_LookupTableInsert.pbtxt │ │ │ ├── api_def_LookupTableInsertV2.pbtxt │ │ │ ├── api_def_LookupTableRemoveV2.pbtxt │ │ │ ├── api_def_LookupTableSize.pbtxt │ │ │ ├── api_def_LookupTableSizeV2.pbtxt │ │ │ ├── api_def_LoopCond.pbtxt │ │ │ ├── api_def_LowerBound.pbtxt │ │ │ ├── api_def_MakeIterator.pbtxt │ │ │ ├── api_def_MapAndBatchDataset.pbtxt │ │ │ ├── api_def_MapAndBatchDatasetV2.pbtxt │ │ │ ├── api_def_MapClear.pbtxt │ │ │ ├── api_def_MapDataset.pbtxt │ │ │ ├── api_def_MapDefun.pbtxt │ │ │ ├── api_def_MapIncompleteSize.pbtxt │ │ │ ├── api_def_MapPeek.pbtxt │ │ │ ├── api_def_MapSize.pbtxt │ │ │ ├── api_def_MapStage.pbtxt │ │ │ ├── api_def_MapUnstage.pbtxt │ │ │ ├── api_def_MapUnstageNoKey.pbtxt │ │ │ ├── api_def_MatMul.pbtxt │ │ │ ├── api_def_MatchingFiles.pbtxt │ │ │ ├── api_def_MatchingFilesDataset.pbtxt │ │ │ ├── api_def_MatrixBandPart.pbtxt │ │ │ ├── api_def_MatrixDeterminant.pbtxt │ │ │ ├── api_def_MatrixDiag.pbtxt │ │ │ ├── api_def_MatrixDiagPart.pbtxt │ │ │ ├── api_def_MatrixExponential.pbtxt │ │ │ ├── api_def_MatrixInverse.pbtxt │ │ │ ├── api_def_MatrixLogarithm.pbtxt │ │ │ ├── api_def_MatrixSetDiag.pbtxt │ │ │ ├── api_def_MatrixSolve.pbtxt │ │ │ ├── api_def_MatrixSolveLs.pbtxt │ │ │ ├── api_def_MatrixSquareRoot.pbtxt │ │ │ ├── api_def_MatrixTriangularSolve.pbtxt │ │ │ ├── api_def_Max.pbtxt │ │ │ ├── api_def_MaxPool.pbtxt │ │ │ ├── api_def_MaxPool3D.pbtxt │ │ │ ├── api_def_MaxPool3DGrad.pbtxt │ │ │ ├── api_def_MaxPool3DGradGrad.pbtxt │ │ │ ├── api_def_MaxPoolGrad.pbtxt │ │ │ ├── api_def_MaxPoolGradGrad.pbtxt │ │ │ ├── api_def_MaxPoolGradGradV2.pbtxt │ │ │ ├── api_def_MaxPoolGradGradWithArgmax.pbtxt │ │ │ ├── api_def_MaxPoolGradV2.pbtxt │ │ │ ├── api_def_MaxPoolGradWithArgmax.pbtxt │ │ │ ├── api_def_MaxPoolV2.pbtxt │ │ │ ├── api_def_MaxPoolWithArgmax.pbtxt │ │ │ ├── api_def_Maximum.pbtxt │ │ │ ├── api_def_Mean.pbtxt │ │ │ ├── api_def_Merge.pbtxt │ │ │ ├── api_def_MergeSummary.pbtxt │ │ │ ├── api_def_MergeV2Checkpoints.pbtxt │ │ │ ├── api_def_Mfcc.pbtxt │ │ │ ├── api_def_Min.pbtxt │ │ │ ├── api_def_Minimum.pbtxt │ │ │ ├── api_def_MirrorPad.pbtxt │ │ │ ├── api_def_MirrorPadGrad.pbtxt │ │ │ ├── api_def_Mod.pbtxt │ │ │ ├── api_def_ModelDataset.pbtxt │ │ │ ├── api_def_Mul.pbtxt │ │ │ ├── api_def_MultiDeviceIterator.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorFromStringHandle.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorGetNextFromShard.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorInit.pbtxt │ │ │ ├── api_def_MultiDeviceIteratorToStringHandle.pbtxt │ │ │ ├── api_def_Multinomial.pbtxt │ │ │ ├── api_def_MutableDenseHashTable.pbtxt │ │ │ ├── api_def_MutableDenseHashTableV2.pbtxt │ │ │ ├── api_def_MutableHashTable.pbtxt │ │ │ ├── api_def_MutableHashTableOfTensors.pbtxt │ │ │ ├── api_def_MutableHashTableOfTensorsV2.pbtxt │ │ │ ├── api_def_MutableHashTableV2.pbtxt │ │ │ ├── api_def_MutexLock.pbtxt │ │ │ ├── api_def_MutexV2.pbtxt │ │ │ ├── api_def_NcclAllReduce.pbtxt │ │ │ ├── api_def_NcclBroadcast.pbtxt │ │ │ ├── api_def_NcclReduce.pbtxt │ │ │ ├── api_def_Neg.pbtxt │ │ │ ├── api_def_NegTrain.pbtxt │ │ │ ├── api_def_NextIteration.pbtxt │ │ │ ├── api_def_NoOp.pbtxt │ │ │ ├── api_def_NonMaxSuppression.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV2.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV3.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV4.pbtxt │ │ │ ├── api_def_NonMaxSuppressionWithOverlaps.pbtxt │ │ │ ├── api_def_NotEqual.pbtxt │ │ │ ├── api_def_NthElement.pbtxt │ │ │ ├── api_def_OneHot.pbtxt │ │ │ ├── api_def_OneShotIterator.pbtxt │ │ │ ├── api_def_OnesLike.pbtxt │ │ │ ├── api_def_OptimizeDataset.pbtxt │ │ │ ├── api_def_OptionalFromValue.pbtxt │ │ │ ├── api_def_OptionalGetValue.pbtxt │ │ │ ├── api_def_OptionalHasValue.pbtxt │ │ │ ├── api_def_OptionalNone.pbtxt │ │ │ ├── api_def_OrderedMapClear.pbtxt │ │ │ ├── api_def_OrderedMapIncompleteSize.pbtxt │ │ │ ├── api_def_OrderedMapPeek.pbtxt │ │ │ ├── api_def_OrderedMapSize.pbtxt │ │ │ ├── api_def_OrderedMapStage.pbtxt │ │ │ ├── api_def_OrderedMapUnstage.pbtxt │ │ │ ├── api_def_OrderedMapUnstageNoKey.pbtxt │ │ │ ├── api_def_Pack.pbtxt │ │ │ ├── api_def_Pad.pbtxt │ │ │ ├── api_def_PadV2.pbtxt │ │ │ ├── api_def_PaddedBatchDataset.pbtxt │ │ │ ├── api_def_PaddedBatchDatasetV2.pbtxt │ │ │ ├── api_def_PaddingFIFOQueue.pbtxt │ │ │ ├── api_def_PaddingFIFOQueueV2.pbtxt │ │ │ ├── api_def_ParallelConcat.pbtxt │ │ │ ├── api_def_ParallelDynamicStitch.pbtxt │ │ │ ├── api_def_ParallelInterleaveDataset.pbtxt │ │ │ ├── api_def_ParallelInterleaveDatasetV2.pbtxt │ │ │ ├── api_def_ParallelMapDataset.pbtxt │ │ │ ├── api_def_ParameterizedTruncatedNormal.pbtxt │ │ │ ├── api_def_ParseExample.pbtxt │ │ │ ├── api_def_ParseExampleDataset.pbtxt │ │ │ ├── api_def_ParseSequenceExample.pbtxt │ │ │ ├── api_def_ParseSingleExample.pbtxt │ │ │ ├── api_def_ParseSingleSequenceExample.pbtxt │ │ │ ├── api_def_ParseTensor.pbtxt │ │ │ ├── api_def_PartitionedCall.pbtxt │ │ │ ├── api_def_Placeholder.pbtxt │ │ │ ├── api_def_PlaceholderV2.pbtxt │ │ │ ├── api_def_PlaceholderWithDefault.pbtxt │ │ │ ├── api_def_Polygamma.pbtxt │ │ │ ├── api_def_PopulationCount.pbtxt │ │ │ ├── api_def_Pow.pbtxt │ │ │ ├── api_def_PrefetchDataset.pbtxt │ │ │ ├── api_def_PrependFromQueueAndPaddedBatchDataset.pbtxt │ │ │ ├── api_def_PreventGradient.pbtxt │ │ │ ├── api_def_Print.pbtxt │ │ │ ├── api_def_PrintV2.pbtxt │ │ │ ├── api_def_PriorityQueue.pbtxt │ │ │ ├── api_def_PriorityQueueV2.pbtxt │ │ │ ├── api_def_Prod.pbtxt │ │ │ ├── api_def_PyFunc.pbtxt │ │ │ ├── api_def_PyFuncStateless.pbtxt │ │ │ ├── api_def_Qr.pbtxt │ │ │ ├── api_def_QuantizeAndDequantize.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV2.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV3.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV4.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV4Grad.pbtxt │ │ │ ├── api_def_QuantizeDownAndShrinkRange.pbtxt │ │ │ ├── api_def_QuantizeV2.pbtxt │ │ │ ├── api_def_QuantizedAdd.pbtxt │ │ │ ├── api_def_QuantizedAvgPool.pbtxt │ │ │ ├── api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt │ │ │ ├── api_def_QuantizedBiasAdd.pbtxt │ │ │ ├── api_def_QuantizedConcat.pbtxt │ │ │ ├── api_def_QuantizedConv2D.pbtxt │ │ │ ├── api_def_QuantizedInstanceNorm.pbtxt │ │ │ ├── api_def_QuantizedMatMul.pbtxt │ │ │ ├── api_def_QuantizedMaxPool.pbtxt │ │ │ ├── api_def_QuantizedMul.pbtxt │ │ │ ├── api_def_QuantizedRelu.pbtxt │ │ │ ├── api_def_QuantizedRelu6.pbtxt │ │ │ ├── api_def_QuantizedReluX.pbtxt │ │ │ ├── api_def_QuantizedReshape.pbtxt │ │ │ ├── api_def_QuantizedResizeBilinear.pbtxt │ │ │ ├── api_def_QueueClose.pbtxt │ │ │ ├── api_def_QueueCloseV2.pbtxt │ │ │ ├── api_def_QueueDequeue.pbtxt │ │ │ ├── api_def_QueueDequeueMany.pbtxt │ │ │ ├── api_def_QueueDequeueManyV2.pbtxt │ │ │ ├── api_def_QueueDequeueUpTo.pbtxt │ │ │ ├── api_def_QueueDequeueUpToV2.pbtxt │ │ │ ├── api_def_QueueDequeueV2.pbtxt │ │ │ ├── api_def_QueueEnqueue.pbtxt │ │ │ ├── api_def_QueueEnqueueMany.pbtxt │ │ │ ├── api_def_QueueEnqueueManyV2.pbtxt │ │ │ ├── api_def_QueueEnqueueV2.pbtxt │ │ │ ├── api_def_QueueIsClosed.pbtxt │ │ │ ├── api_def_QueueIsClosedV2.pbtxt │ │ │ ├── api_def_QueueSize.pbtxt │ │ │ ├── api_def_QueueSizeV2.pbtxt │ │ │ ├── api_def_RFFT.pbtxt │ │ │ ├── api_def_RFFT2D.pbtxt │ │ │ ├── api_def_RFFT3D.pbtxt │ │ │ ├── api_def_RGBToHSV.pbtxt │ │ │ ├── api_def_RaggedGather.pbtxt │ │ │ ├── api_def_RaggedRange.pbtxt │ │ │ ├── api_def_RaggedTensorToSparse.pbtxt │ │ │ ├── api_def_RandomCrop.pbtxt │ │ │ ├── api_def_RandomDataset.pbtxt │ │ │ ├── api_def_RandomGamma.pbtxt │ │ │ ├── api_def_RandomGammaGrad.pbtxt │ │ │ ├── api_def_RandomPoisson.pbtxt │ │ │ ├── api_def_RandomPoissonV2.pbtxt │ │ │ ├── api_def_RandomShuffle.pbtxt │ │ │ ├── api_def_RandomShuffleQueue.pbtxt │ │ │ ├── api_def_RandomShuffleQueueV2.pbtxt │ │ │ ├── api_def_RandomStandardNormal.pbtxt │ │ │ ├── api_def_RandomUniform.pbtxt │ │ │ ├── api_def_RandomUniformInt.pbtxt │ │ │ ├── api_def_Range.pbtxt │ │ │ ├── api_def_RangeDataset.pbtxt │ │ │ ├── api_def_Rank.pbtxt │ │ │ ├── api_def_ReadFile.pbtxt │ │ │ ├── api_def_ReadVariableOp.pbtxt │ │ │ ├── api_def_ReaderNumRecordsProduced.pbtxt │ │ │ ├── api_def_ReaderNumRecordsProducedV2.pbtxt │ │ │ ├── api_def_ReaderNumWorkUnitsCompleted.pbtxt │ │ │ ├── api_def_ReaderNumWorkUnitsCompletedV2.pbtxt │ │ │ ├── api_def_ReaderRead.pbtxt │ │ │ ├── api_def_ReaderReadUpTo.pbtxt │ │ │ ├── api_def_ReaderReadUpToV2.pbtxt │ │ │ ├── api_def_ReaderReadV2.pbtxt │ │ │ ├── api_def_ReaderReset.pbtxt │ │ │ ├── api_def_ReaderResetV2.pbtxt │ │ │ ├── api_def_ReaderRestoreState.pbtxt │ │ │ ├── api_def_ReaderRestoreStateV2.pbtxt │ │ │ ├── api_def_ReaderSerializeState.pbtxt │ │ │ ├── api_def_ReaderSerializeStateV2.pbtxt │ │ │ ├── api_def_Real.pbtxt │ │ │ ├── api_def_RealDiv.pbtxt │ │ │ ├── api_def_Reciprocal.pbtxt │ │ │ ├── api_def_ReciprocalGrad.pbtxt │ │ │ ├── api_def_RecordInput.pbtxt │ │ │ ├── api_def_ReduceDataset.pbtxt │ │ │ ├── api_def_ReduceJoin.pbtxt │ │ │ ├── api_def_RefEnter.pbtxt │ │ │ ├── api_def_RefExit.pbtxt │ │ │ ├── api_def_RefIdentity.pbtxt │ │ │ ├── api_def_RefMerge.pbtxt │ │ │ ├── api_def_RefNextIteration.pbtxt │ │ │ ├── api_def_RefSelect.pbtxt │ │ │ ├── api_def_RefSwitch.pbtxt │ │ │ ├── api_def_RegexFullMatch.pbtxt │ │ │ ├── api_def_RegexReplace.pbtxt │ │ │ ├── api_def_Relu.pbtxt │ │ │ ├── api_def_Relu6.pbtxt │ │ │ ├── api_def_Relu6Grad.pbtxt │ │ │ ├── api_def_ReluGrad.pbtxt │ │ │ ├── api_def_RemoteCall.pbtxt │ │ │ ├── api_def_RemoteFusedGraphExecute.pbtxt │ │ │ ├── api_def_RepeatDataset.pbtxt │ │ │ ├── api_def_RequantizationRange.pbtxt │ │ │ ├── api_def_Requantize.pbtxt │ │ │ ├── api_def_Reshape.pbtxt │ │ │ ├── api_def_ResizeArea.pbtxt │ │ │ ├── api_def_ResizeBicubic.pbtxt │ │ │ ├── api_def_ResizeBicubicGrad.pbtxt │ │ │ ├── api_def_ResizeBilinear.pbtxt │ │ │ ├── api_def_ResizeBilinearGrad.pbtxt │ │ │ ├── api_def_ResizeNearestNeighbor.pbtxt │ │ │ ├── api_def_ResizeNearestNeighborGrad.pbtxt │ │ │ ├── api_def_ResourceApplyAdaMax.pbtxt │ │ │ ├── api_def_ResourceApplyAdadelta.pbtxt │ │ │ ├── api_def_ResourceApplyAdagrad.pbtxt │ │ │ ├── api_def_ResourceApplyAdagradDA.pbtxt │ │ │ ├── api_def_ResourceApplyAdam.pbtxt │ │ │ ├── api_def_ResourceApplyAddSign.pbtxt │ │ │ ├── api_def_ResourceApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ResourceApplyFtrl.pbtxt │ │ │ ├── api_def_ResourceApplyFtrlV2.pbtxt │ │ │ ├── api_def_ResourceApplyGradientDescent.pbtxt │ │ │ ├── api_def_ResourceApplyMomentum.pbtxt │ │ │ ├── api_def_ResourceApplyPowerSign.pbtxt │ │ │ ├── api_def_ResourceApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ResourceApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ResourceApplyRMSProp.pbtxt │ │ │ ├── api_def_ResourceCountUpTo.pbtxt │ │ │ ├── api_def_ResourceGather.pbtxt │ │ │ ├── api_def_ResourceScatterAdd.pbtxt │ │ │ ├── api_def_ResourceScatterDiv.pbtxt │ │ │ ├── api_def_ResourceScatterMax.pbtxt │ │ │ ├── api_def_ResourceScatterMin.pbtxt │ │ │ ├── api_def_ResourceScatterMul.pbtxt │ │ │ ├── api_def_ResourceScatterNdAdd.pbtxt │ │ │ ├── api_def_ResourceScatterNdUpdate.pbtxt │ │ │ ├── api_def_ResourceScatterSub.pbtxt │ │ │ ├── api_def_ResourceScatterUpdate.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdadelta.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagrad.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagradDA.pbtxt │ │ │ ├── api_def_ResourceSparseApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ResourceSparseApplyFtrl.pbtxt │ │ │ ├── api_def_ResourceSparseApplyFtrlV2.pbtxt │ │ │ ├── api_def_ResourceSparseApplyMomentum.pbtxt │ │ │ ├── api_def_ResourceSparseApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ResourceSparseApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ResourceSparseApplyRMSProp.pbtxt │ │ │ ├── api_def_ResourceStridedSliceAssign.pbtxt │ │ │ ├── api_def_Restore.pbtxt │ │ │ ├── api_def_RestoreSlice.pbtxt │ │ │ ├── api_def_RestoreV2.pbtxt │ │ │ ├── api_def_Reverse.pbtxt │ │ │ ├── api_def_ReverseSequence.pbtxt │ │ │ ├── api_def_ReverseV2.pbtxt │ │ │ ├── api_def_RightShift.pbtxt │ │ │ ├── api_def_Rint.pbtxt │ │ │ ├── api_def_Roll.pbtxt │ │ │ ├── api_def_Round.pbtxt │ │ │ ├── api_def_Rsqrt.pbtxt │ │ │ ├── api_def_RsqrtGrad.pbtxt │ │ │ ├── api_def_SampleDistortedBoundingBox.pbtxt │ │ │ ├── api_def_SampleDistortedBoundingBoxV2.pbtxt │ │ │ ├── api_def_Save.pbtxt │ │ │ ├── api_def_SaveSlices.pbtxt │ │ │ ├── api_def_SaveV2.pbtxt │ │ │ ├── api_def_ScalarSummary.pbtxt │ │ │ ├── api_def_ScanDataset.pbtxt │ │ │ ├── api_def_ScatterAdd.pbtxt │ │ │ ├── api_def_ScatterDiv.pbtxt │ │ │ ├── api_def_ScatterMax.pbtxt │ │ │ ├── api_def_ScatterMin.pbtxt │ │ │ ├── api_def_ScatterMul.pbtxt │ │ │ ├── api_def_ScatterNd.pbtxt │ │ │ ├── api_def_ScatterNdAdd.pbtxt │ │ │ ├── api_def_ScatterNdNonAliasingAdd.pbtxt │ │ │ ├── api_def_ScatterNdSub.pbtxt │ │ │ ├── api_def_ScatterNdUpdate.pbtxt │ │ │ ├── api_def_ScatterSub.pbtxt │ │ │ ├── api_def_ScatterUpdate.pbtxt │ │ │ ├── api_def_SdcaFprint.pbtxt │ │ │ ├── api_def_SdcaOptimizer.pbtxt │ │ │ ├── api_def_SdcaOptimizerV2.pbtxt │ │ │ ├── api_def_SdcaShrinkL1.pbtxt │ │ │ ├── api_def_SegmentMax.pbtxt │ │ │ ├── api_def_SegmentMean.pbtxt │ │ │ ├── api_def_SegmentMin.pbtxt │ │ │ ├── api_def_SegmentProd.pbtxt │ │ │ ├── api_def_SegmentSum.pbtxt │ │ │ ├── api_def_Select.pbtxt │ │ │ ├── api_def_SelfAdjointEig.pbtxt │ │ │ ├── api_def_SelfAdjointEigV2.pbtxt │ │ │ ├── api_def_Selu.pbtxt │ │ │ ├── api_def_SeluGrad.pbtxt │ │ │ ├── api_def_SerializeIterator.pbtxt │ │ │ ├── api_def_SerializeManySparse.pbtxt │ │ │ ├── api_def_SerializeSparse.pbtxt │ │ │ ├── api_def_SerializeTensor.pbtxt │ │ │ ├── api_def_SetSize.pbtxt │ │ │ ├── api_def_SetStatsAggregatorDataset.pbtxt │ │ │ ├── api_def_Shape.pbtxt │ │ │ ├── api_def_ShapeN.pbtxt │ │ │ ├── api_def_ShardedFilename.pbtxt │ │ │ ├── api_def_ShardedFilespec.pbtxt │ │ │ ├── api_def_ShuffleAndRepeatDataset.pbtxt │ │ │ ├── api_def_ShuffleDataset.pbtxt │ │ │ ├── api_def_Sigmoid.pbtxt │ │ │ ├── api_def_SigmoidGrad.pbtxt │ │ │ ├── api_def_Sign.pbtxt │ │ │ ├── api_def_Sin.pbtxt │ │ │ ├── api_def_Sinh.pbtxt │ │ │ ├── api_def_Size.pbtxt │ │ │ ├── api_def_SkipDataset.pbtxt │ │ │ ├── api_def_Skipgram.pbtxt │ │ │ ├── api_def_Slice.pbtxt │ │ │ ├── api_def_SlideDataset.pbtxt │ │ │ ├── api_def_Snapshot.pbtxt │ │ │ ├── api_def_Softmax.pbtxt │ │ │ ├── api_def_SoftmaxCrossEntropyWithLogits.pbtxt │ │ │ ├── api_def_Softplus.pbtxt │ │ │ ├── api_def_SoftplusGrad.pbtxt │ │ │ ├── api_def_Softsign.pbtxt │ │ │ ├── api_def_SoftsignGrad.pbtxt │ │ │ ├── api_def_SpaceToBatch.pbtxt │ │ │ ├── api_def_SpaceToBatchND.pbtxt │ │ │ ├── api_def_SpaceToDepth.pbtxt │ │ │ ├── api_def_SparseAccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_SparseAccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_SparseAdd.pbtxt │ │ │ ├── api_def_SparseAddGrad.pbtxt │ │ │ ├── api_def_SparseApplyAdadelta.pbtxt │ │ │ ├── api_def_SparseApplyAdagrad.pbtxt │ │ │ ├── api_def_SparseApplyAdagradDA.pbtxt │ │ │ ├── api_def_SparseApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_SparseApplyFtrl.pbtxt │ │ │ ├── api_def_SparseApplyFtrlV2.pbtxt │ │ │ ├── api_def_SparseApplyMomentum.pbtxt │ │ │ ├── api_def_SparseApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_SparseApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_SparseApplyRMSProp.pbtxt │ │ │ ├── api_def_SparseConcat.pbtxt │ │ │ ├── api_def_SparseConditionalAccumulator.pbtxt │ │ │ ├── api_def_SparseCross.pbtxt │ │ │ ├── api_def_SparseDenseCwiseAdd.pbtxt │ │ │ ├── api_def_SparseDenseCwiseDiv.pbtxt │ │ │ ├── api_def_SparseDenseCwiseMul.pbtxt │ │ │ ├── api_def_SparseFillEmptyRows.pbtxt │ │ │ ├── api_def_SparseFillEmptyRowsGrad.pbtxt │ │ │ ├── api_def_SparseMatMul.pbtxt │ │ │ ├── api_def_SparseReduceMax.pbtxt │ │ │ ├── api_def_SparseReduceMaxSparse.pbtxt │ │ │ ├── api_def_SparseReduceSum.pbtxt │ │ │ ├── api_def_SparseReduceSumSparse.pbtxt │ │ │ ├── api_def_SparseReorder.pbtxt │ │ │ ├── api_def_SparseReshape.pbtxt │ │ │ ├── api_def_SparseSegmentMean.pbtxt │ │ │ ├── api_def_SparseSegmentMeanGrad.pbtxt │ │ │ ├── api_def_SparseSegmentMeanWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtN.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNGrad.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSegmentSum.pbtxt │ │ │ ├── api_def_SparseSegmentSumWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSlice.pbtxt │ │ │ ├── api_def_SparseSliceGrad.pbtxt │ │ │ ├── api_def_SparseSoftmax.pbtxt │ │ │ ├── api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt │ │ │ ├── api_def_SparseSparseMaximum.pbtxt │ │ │ ├── api_def_SparseSparseMinimum.pbtxt │ │ │ ├── api_def_SparseSplit.pbtxt │ │ │ ├── api_def_SparseTensorDenseAdd.pbtxt │ │ │ ├── api_def_SparseTensorDenseMatMul.pbtxt │ │ │ ├── api_def_SparseTensorSliceDataset.pbtxt │ │ │ ├── api_def_SparseToDense.pbtxt │ │ │ ├── api_def_SparseToSparseSetOperation.pbtxt │ │ │ ├── api_def_Split.pbtxt │ │ │ ├── api_def_SplitV.pbtxt │ │ │ ├── api_def_SqlDataset.pbtxt │ │ │ ├── api_def_Sqrt.pbtxt │ │ │ ├── api_def_SqrtGrad.pbtxt │ │ │ ├── api_def_Square.pbtxt │ │ │ ├── api_def_SquaredDifference.pbtxt │ │ │ ├── api_def_Squeeze.pbtxt │ │ │ ├── api_def_Stack.pbtxt │ │ │ ├── api_def_StackClose.pbtxt │ │ │ ├── api_def_StackCloseV2.pbtxt │ │ │ ├── api_def_StackPop.pbtxt │ │ │ ├── api_def_StackPopV2.pbtxt │ │ │ ├── api_def_StackPush.pbtxt │ │ │ ├── api_def_StackPushV2.pbtxt │ │ │ ├── api_def_StackV2.pbtxt │ │ │ ├── api_def_Stage.pbtxt │ │ │ ├── api_def_StageClear.pbtxt │ │ │ ├── api_def_StagePeek.pbtxt │ │ │ ├── api_def_StageSize.pbtxt │ │ │ ├── api_def_StatefulPartitionedCall.pbtxt │ │ │ ├── api_def_StatelessIf.pbtxt │ │ │ ├── api_def_StatelessMultinomial.pbtxt │ │ │ ├── api_def_StatelessRandomNormal.pbtxt │ │ │ ├── api_def_StatelessRandomUniform.pbtxt │ │ │ ├── api_def_StatelessRandomUniformInt.pbtxt │ │ │ ├── api_def_StatelessTruncatedNormal.pbtxt │ │ │ ├── api_def_StatelessWhile.pbtxt │ │ │ ├── api_def_StaticRegexFullMatch.pbtxt │ │ │ ├── api_def_StaticRegexReplace.pbtxt │ │ │ ├── api_def_StatsAggregatorHandle.pbtxt │ │ │ ├── api_def_StatsAggregatorSummary.pbtxt │ │ │ ├── api_def_StopGradient.pbtxt │ │ │ ├── api_def_StridedSlice.pbtxt │ │ │ ├── api_def_StridedSliceAssign.pbtxt │ │ │ ├── api_def_StridedSliceGrad.pbtxt │ │ │ ├── api_def_StringFormat.pbtxt │ │ │ ├── api_def_StringJoin.pbtxt │ │ │ ├── api_def_StringLength.pbtxt │ │ │ ├── api_def_StringSplit.pbtxt │ │ │ ├── api_def_StringSplitV2.pbtxt │ │ │ ├── api_def_StringStrip.pbtxt │ │ │ ├── api_def_StringToHashBucket.pbtxt │ │ │ ├── api_def_StringToHashBucketFast.pbtxt │ │ │ ├── api_def_StringToHashBucketStrong.pbtxt │ │ │ ├── api_def_StringToNumber.pbtxt │ │ │ ├── api_def_Sub.pbtxt │ │ │ ├── api_def_Substr.pbtxt │ │ │ ├── api_def_Sum.pbtxt │ │ │ ├── api_def_SummaryWriter.pbtxt │ │ │ ├── api_def_Svd.pbtxt │ │ │ ├── api_def_Switch.pbtxt │ │ │ ├── api_def_SymbolicGradient.pbtxt │ │ │ ├── api_def_TFRecordDataset.pbtxt │ │ │ ├── api_def_TFRecordReader.pbtxt │ │ │ ├── api_def_TFRecordReaderV2.pbtxt │ │ │ ├── api_def_TakeDataset.pbtxt │ │ │ ├── api_def_TakeManySparseFromTensorsMap.pbtxt │ │ │ ├── api_def_Tan.pbtxt │ │ │ ├── api_def_Tanh.pbtxt │ │ │ ├── api_def_TanhGrad.pbtxt │ │ │ ├── api_def_TemporaryVariable.pbtxt │ │ │ ├── api_def_TensorArray.pbtxt │ │ │ ├── api_def_TensorArrayClose.pbtxt │ │ │ ├── api_def_TensorArrayCloseV2.pbtxt │ │ │ ├── api_def_TensorArrayCloseV3.pbtxt │ │ │ ├── api_def_TensorArrayConcat.pbtxt │ │ │ ├── api_def_TensorArrayConcatV2.pbtxt │ │ │ ├── api_def_TensorArrayConcatV3.pbtxt │ │ │ ├── api_def_TensorArrayGather.pbtxt │ │ │ ├── api_def_TensorArrayGatherV2.pbtxt │ │ │ ├── api_def_TensorArrayGatherV3.pbtxt │ │ │ ├── api_def_TensorArrayGrad.pbtxt │ │ │ ├── api_def_TensorArrayGradV2.pbtxt │ │ │ ├── api_def_TensorArrayGradV3.pbtxt │ │ │ ├── api_def_TensorArrayGradWithShape.pbtxt │ │ │ ├── api_def_TensorArrayPack.pbtxt │ │ │ ├── api_def_TensorArrayRead.pbtxt │ │ │ ├── api_def_TensorArrayReadV2.pbtxt │ │ │ ├── api_def_TensorArrayReadV3.pbtxt │ │ │ ├── api_def_TensorArrayScatter.pbtxt │ │ │ ├── api_def_TensorArrayScatterV2.pbtxt │ │ │ ├── api_def_TensorArrayScatterV3.pbtxt │ │ │ ├── api_def_TensorArraySize.pbtxt │ │ │ ├── api_def_TensorArraySizeV2.pbtxt │ │ │ ├── api_def_TensorArraySizeV3.pbtxt │ │ │ ├── api_def_TensorArraySplit.pbtxt │ │ │ ├── api_def_TensorArraySplitV2.pbtxt │ │ │ ├── api_def_TensorArraySplitV3.pbtxt │ │ │ ├── api_def_TensorArrayUnpack.pbtxt │ │ │ ├── api_def_TensorArrayV2.pbtxt │ │ │ ├── api_def_TensorArrayV3.pbtxt │ │ │ ├── api_def_TensorArrayWrite.pbtxt │ │ │ ├── api_def_TensorArrayWriteV2.pbtxt │ │ │ ├── api_def_TensorArrayWriteV3.pbtxt │ │ │ ├── api_def_TensorDataset.pbtxt │ │ │ ├── api_def_TensorListConcatLists.pbtxt │ │ │ ├── api_def_TensorListElementShape.pbtxt │ │ │ ├── api_def_TensorListFromTensor.pbtxt │ │ │ ├── api_def_TensorListGather.pbtxt │ │ │ ├── api_def_TensorListGetItem.pbtxt │ │ │ ├── api_def_TensorListLength.pbtxt │ │ │ ├── api_def_TensorListPopBack.pbtxt │ │ │ ├── api_def_TensorListPushBack.pbtxt │ │ │ ├── api_def_TensorListPushBackBatch.pbtxt │ │ │ ├── api_def_TensorListReserve.pbtxt │ │ │ ├── api_def_TensorListScatter.pbtxt │ │ │ ├── api_def_TensorListSetItem.pbtxt │ │ │ ├── api_def_TensorListStack.pbtxt │ │ │ ├── api_def_TensorSliceDataset.pbtxt │ │ │ ├── api_def_TensorSummary.pbtxt │ │ │ ├── api_def_TensorSummaryV2.pbtxt │ │ │ ├── api_def_TextLineDataset.pbtxt │ │ │ ├── api_def_TextLineReader.pbtxt │ │ │ ├── api_def_TextLineReaderV2.pbtxt │ │ │ ├── api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_Tile.pbtxt │ │ │ ├── api_def_TileGrad.pbtxt │ │ │ ├── api_def_Timestamp.pbtxt │ │ │ ├── api_def_TopK.pbtxt │ │ │ ├── api_def_TopKV2.pbtxt │ │ │ ├── api_def_Transpose.pbtxt │ │ │ ├── api_def_TruncateDiv.pbtxt │ │ │ ├── api_def_TruncateMod.pbtxt │ │ │ ├── api_def_TruncatedNormal.pbtxt │ │ │ ├── api_def_Unbatch.pbtxt │ │ │ ├── api_def_UnbatchDataset.pbtxt │ │ │ ├── api_def_UnbatchGrad.pbtxt │ │ │ ├── api_def_UnicodeDecodeWithOffsets.pbtxt │ │ │ ├── api_def_UnicodeScript.pbtxt │ │ │ ├── api_def_UnicodeTranscode.pbtxt │ │ │ ├── api_def_UniformCandidateSampler.pbtxt │ │ │ ├── api_def_Unique.pbtxt │ │ │ ├── api_def_UniqueV2.pbtxt │ │ │ ├── api_def_UniqueWithCounts.pbtxt │ │ │ ├── api_def_UniqueWithCountsV2.pbtxt │ │ │ ├── api_def_Unpack.pbtxt │ │ │ ├── api_def_UnravelIndex.pbtxt │ │ │ ├── api_def_UnsortedSegmentMax.pbtxt │ │ │ ├── api_def_UnsortedSegmentMin.pbtxt │ │ │ ├── api_def_UnsortedSegmentProd.pbtxt │ │ │ ├── api_def_UnsortedSegmentSum.pbtxt │ │ │ ├── api_def_Unstage.pbtxt │ │ │ ├── api_def_UpperBound.pbtxt │ │ │ ├── api_def_VarHandleOp.pbtxt │ │ │ ├── api_def_VarIsInitializedOp.pbtxt │ │ │ ├── api_def_Variable.pbtxt │ │ │ ├── api_def_VariableShape.pbtxt │ │ │ ├── api_def_VariableV2.pbtxt │ │ │ ├── api_def_Where.pbtxt │ │ │ ├── api_def_While.pbtxt │ │ │ ├── api_def_WholeFileReader.pbtxt │ │ │ ├── api_def_WholeFileReaderV2.pbtxt │ │ │ ├── api_def_WindowDataset.pbtxt │ │ │ ├── api_def_WriteAudioSummary.pbtxt │ │ │ ├── api_def_WriteFile.pbtxt │ │ │ ├── api_def_WriteGraphSummary.pbtxt │ │ │ ├── api_def_WriteHistogramSummary.pbtxt │ │ │ ├── api_def_WriteImageSummary.pbtxt │ │ │ ├── api_def_WriteScalarSummary.pbtxt │ │ │ ├── api_def_WriteSummary.pbtxt │ │ │ ├── api_def_Xdivy.pbtxt │ │ │ ├── api_def_Xlogy.pbtxt │ │ │ ├── api_def_ZerosLike.pbtxt │ │ │ ├── api_def_Zeta.pbtxt │ │ │ └── api_def_ZipDataset.pbtxt │ │ ├── python_api │ │ │ ├── BUILD │ │ │ ├── api_def_Abort.pbtxt │ │ │ ├── api_def_Abs.pbtxt │ │ │ ├── api_def_AccumulateNV2.pbtxt │ │ │ ├── api_def_AccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_AccumulatorNumAccumulated.pbtxt │ │ │ ├── api_def_AccumulatorSetGlobalStep.pbtxt │ │ │ ├── api_def_AccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_Acos.pbtxt │ │ │ ├── api_def_Acosh.pbtxt │ │ │ ├── api_def_Add.pbtxt │ │ │ ├── api_def_AddManySparseToTensorsMap.pbtxt │ │ │ ├── api_def_AddN.pbtxt │ │ │ ├── api_def_AddSparseToTensorsMap.pbtxt │ │ │ ├── api_def_AddV2.pbtxt │ │ │ ├── api_def_AdjustContrast.pbtxt │ │ │ ├── api_def_AdjustContrastv2.pbtxt │ │ │ ├── api_def_AdjustHue.pbtxt │ │ │ ├── api_def_AdjustSaturation.pbtxt │ │ │ ├── api_def_All.pbtxt │ │ │ ├── api_def_AllCandidateSampler.pbtxt │ │ │ ├── api_def_Angle.pbtxt │ │ │ ├── api_def_AnonymousIterator.pbtxt │ │ │ ├── api_def_Any.pbtxt │ │ │ ├── api_def_ApplyAdaMax.pbtxt │ │ │ ├── api_def_ApplyAdadelta.pbtxt │ │ │ ├── api_def_ApplyAdagrad.pbtxt │ │ │ ├── api_def_ApplyAdagradDA.pbtxt │ │ │ ├── api_def_ApplyAdam.pbtxt │ │ │ ├── api_def_ApplyAddSign.pbtxt │ │ │ ├── api_def_ApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ApplyFtrl.pbtxt │ │ │ ├── api_def_ApplyFtrlV2.pbtxt │ │ │ ├── api_def_ApplyGradientDescent.pbtxt │ │ │ ├── api_def_ApplyMomentum.pbtxt │ │ │ ├── api_def_ApplyPowerSign.pbtxt │ │ │ ├── api_def_ApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ApplyRMSProp.pbtxt │ │ │ ├── api_def_ApproximateEqual.pbtxt │ │ │ ├── api_def_ArgMax.pbtxt │ │ │ ├── api_def_ArgMin.pbtxt │ │ │ ├── api_def_AsString.pbtxt │ │ │ ├── api_def_Asin.pbtxt │ │ │ ├── api_def_Asinh.pbtxt │ │ │ ├── api_def_Assert.pbtxt │ │ │ ├── api_def_Assign.pbtxt │ │ │ ├── api_def_AssignAdd.pbtxt │ │ │ ├── api_def_AssignAddVariableOp.pbtxt │ │ │ ├── api_def_AssignSub.pbtxt │ │ │ ├── api_def_AssignSubVariableOp.pbtxt │ │ │ ├── api_def_AssignVariableOp.pbtxt │ │ │ ├── api_def_Atan.pbtxt │ │ │ ├── api_def_Atan2.pbtxt │ │ │ ├── api_def_Atanh.pbtxt │ │ │ ├── api_def_AudioSpectrogram.pbtxt │ │ │ ├── api_def_AudioSummary.pbtxt │ │ │ ├── api_def_AudioSummaryV2.pbtxt │ │ │ ├── api_def_AvgPool.pbtxt │ │ │ ├── api_def_AvgPool3D.pbtxt │ │ │ ├── api_def_AvgPool3DGrad.pbtxt │ │ │ ├── api_def_BandedTriangularSolve.pbtxt │ │ │ ├── api_def_Barrier.pbtxt │ │ │ ├── api_def_BarrierClose.pbtxt │ │ │ ├── api_def_BarrierIncompleteSize.pbtxt │ │ │ ├── api_def_BarrierInsertMany.pbtxt │ │ │ ├── api_def_BarrierReadySize.pbtxt │ │ │ ├── api_def_BarrierTakeMany.pbtxt │ │ │ ├── api_def_Batch.pbtxt │ │ │ ├── api_def_BatchCholesky.pbtxt │ │ │ ├── api_def_BatchCholeskyGrad.pbtxt │ │ │ ├── api_def_BatchFFT.pbtxt │ │ │ ├── api_def_BatchFFT2D.pbtxt │ │ │ ├── api_def_BatchFFT3D.pbtxt │ │ │ ├── api_def_BatchFunction.pbtxt │ │ │ ├── api_def_BatchIFFT.pbtxt │ │ │ ├── api_def_BatchIFFT2D.pbtxt │ │ │ ├── api_def_BatchIFFT3D.pbtxt │ │ │ ├── api_def_BatchMatMul.pbtxt │ │ │ ├── api_def_BatchMatMulV2.pbtxt │ │ │ ├── api_def_BatchMatMulV3.pbtxt │ │ │ ├── api_def_BatchMatrixBandPart.pbtxt │ │ │ ├── api_def_BatchMatrixDeterminant.pbtxt │ │ │ ├── api_def_BatchMatrixDiag.pbtxt │ │ │ ├── api_def_BatchMatrixDiagPart.pbtxt │ │ │ ├── api_def_BatchMatrixInverse.pbtxt │ │ │ ├── api_def_BatchMatrixSetDiag.pbtxt │ │ │ ├── api_def_BatchMatrixSolve.pbtxt │ │ │ ├── api_def_BatchMatrixSolveLs.pbtxt │ │ │ ├── api_def_BatchMatrixTriangularSolve.pbtxt │ │ │ ├── api_def_BatchNormWithGlobalNormalization.pbtxt │ │ │ ├── api_def_BatchNormWithGlobalNormalizationGrad.pbtxt │ │ │ ├── api_def_BatchSelfAdjointEig.pbtxt │ │ │ ├── api_def_BatchSelfAdjointEigV2.pbtxt │ │ │ ├── api_def_BatchSvd.pbtxt │ │ │ ├── api_def_BatchToSpace.pbtxt │ │ │ ├── api_def_BatchToSpaceND.pbtxt │ │ │ ├── api_def_Betainc.pbtxt │ │ │ ├── api_def_BiasAdd.pbtxt │ │ │ ├── api_def_BiasAddGrad.pbtxt │ │ │ ├── api_def_BiasAddV1.pbtxt │ │ │ ├── api_def_Bincount.pbtxt │ │ │ ├── api_def_BitwiseAnd.pbtxt │ │ │ ├── api_def_BitwiseOr.pbtxt │ │ │ ├── api_def_BitwiseXor.pbtxt │ │ │ ├── api_def_BroadcastArgs.pbtxt │ │ │ ├── api_def_Bucketize.pbtxt │ │ │ ├── api_def_CTCBeamSearchDecoder.pbtxt │ │ │ ├── api_def_CTCGreedyDecoder.pbtxt │ │ │ ├── api_def_CTCLoss.pbtxt │ │ │ ├── api_def_Case.pbtxt │ │ │ ├── api_def_Cast.pbtxt │ │ │ ├── api_def_Ceil.pbtxt │ │ │ ├── api_def_CheckNumerics.pbtxt │ │ │ ├── api_def_CheckNumericsV2.pbtxt │ │ │ ├── api_def_Cholesky.pbtxt │ │ │ ├── api_def_CholeskyGrad.pbtxt │ │ │ ├── api_def_ClipByValue.pbtxt │ │ │ ├── api_def_CloseSummaryWriter.pbtxt │ │ │ ├── api_def_CollectiveBcastRecv.pbtxt │ │ │ ├── api_def_CollectiveBcastSend.pbtxt │ │ │ ├── api_def_CollectiveReduce.pbtxt │ │ │ ├── api_def_CombinedNonMaxSuppression.pbtxt │ │ │ ├── api_def_Complex.pbtxt │ │ │ ├── api_def_ComplexAbs.pbtxt │ │ │ ├── api_def_ComputeAccidentalHits.pbtxt │ │ │ ├── api_def_Concat.pbtxt │ │ │ ├── api_def_ConcatOffset.pbtxt │ │ │ ├── api_def_ConcatV2.pbtxt │ │ │ ├── api_def_ConditionalAccumulator.pbtxt │ │ │ ├── api_def_Conj.pbtxt │ │ │ ├── api_def_ConjugateTranspose.pbtxt │ │ │ ├── api_def_Const.pbtxt │ │ │ ├── api_def_ConsumeMutexLock.pbtxt │ │ │ ├── api_def_ControlTrigger.pbtxt │ │ │ ├── api_def_Conv2D.pbtxt │ │ │ ├── api_def_Conv2DBackpropFilter.pbtxt │ │ │ ├── api_def_Conv2DBackpropInput.pbtxt │ │ │ ├── api_def_Conv3D.pbtxt │ │ │ ├── api_def_Conv3DBackpropFilter.pbtxt │ │ │ ├── api_def_Conv3DBackpropFilterV2.pbtxt │ │ │ ├── api_def_Conv3DBackpropInput.pbtxt │ │ │ ├── api_def_Conv3DBackpropInputV2.pbtxt │ │ │ ├── api_def_ConvertToCooTensor.pbtxt │ │ │ ├── api_def_Copy.pbtxt │ │ │ ├── api_def_CopyHost.pbtxt │ │ │ ├── api_def_Cos.pbtxt │ │ │ ├── api_def_Cosh.pbtxt │ │ │ ├── api_def_CountUpTo.pbtxt │ │ │ ├── api_def_CreateSummaryDbWriter.pbtxt │ │ │ ├── api_def_CreateSummaryFileWriter.pbtxt │ │ │ ├── api_def_CropAndResize.pbtxt │ │ │ ├── api_def_CropAndResizeGradBoxes.pbtxt │ │ │ ├── api_def_CropAndResizeGradImage.pbtxt │ │ │ ├── api_def_Cross.pbtxt │ │ │ ├── api_def_CudnnRNN.pbtxt │ │ │ ├── api_def_CudnnRNNBackprop.pbtxt │ │ │ ├── api_def_CudnnRNNCanonicalToParams.pbtxt │ │ │ ├── api_def_CudnnRNNCanonicalToParamsV2.pbtxt │ │ │ ├── api_def_CudnnRNNParamsSize.pbtxt │ │ │ ├── api_def_CudnnRNNParamsToCanonical.pbtxt │ │ │ ├── api_def_CudnnRNNParamsToCanonicalV2.pbtxt │ │ │ ├── api_def_Cumprod.pbtxt │ │ │ ├── api_def_Cumsum.pbtxt │ │ │ ├── api_def_DataFormatDimMap.pbtxt │ │ │ ├── api_def_DataFormatVecPermute.pbtxt │ │ │ ├── api_def_DebugGradientIdentity.pbtxt │ │ │ ├── api_def_DebugGradientRefIdentity.pbtxt │ │ │ ├── api_def_DebugIdentity.pbtxt │ │ │ ├── api_def_DebugIdentityV2.pbtxt │ │ │ ├── api_def_DebugNanCount.pbtxt │ │ │ ├── api_def_DebugNumericSummary.pbtxt │ │ │ ├── api_def_DebugNumericSummaryV2.pbtxt │ │ │ ├── api_def_DecodeAndCropJpeg.pbtxt │ │ │ ├── api_def_DecodeBase64.pbtxt │ │ │ ├── api_def_DecodeBmp.pbtxt │ │ │ ├── api_def_DecodeCSV.pbtxt │ │ │ ├── api_def_DecodeCompressed.pbtxt │ │ │ ├── api_def_DecodeGif.pbtxt │ │ │ ├── api_def_DecodeImage.pbtxt │ │ │ ├── api_def_DecodeJSONExample.pbtxt │ │ │ ├── api_def_DecodeJpeg.pbtxt │ │ │ ├── api_def_DecodePaddedRaw.pbtxt │ │ │ ├── api_def_DecodePng.pbtxt │ │ │ ├── api_def_DecodeProtoV2.pbtxt │ │ │ ├── api_def_DecodeRaw.pbtxt │ │ │ ├── api_def_DecodeWav.pbtxt │ │ │ ├── api_def_DeepCopy.pbtxt │ │ │ ├── api_def_DeleteSessionTensor.pbtxt │ │ │ ├── api_def_DenseBincount.pbtxt │ │ │ ├── api_def_DenseToDenseSetOperation.pbtxt │ │ │ ├── api_def_DenseToSparseSetOperation.pbtxt │ │ │ ├── api_def_DepthToSpace.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNative.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt │ │ │ ├── api_def_DepthwiseConv2dNativeBackpropInput.pbtxt │ │ │ ├── api_def_Dequantize.pbtxt │ │ │ ├── api_def_DeserializeIterator.pbtxt │ │ │ ├── api_def_DeserializeManySparse.pbtxt │ │ │ ├── api_def_DeserializeSparse.pbtxt │ │ │ ├── api_def_DestroyResourceOp.pbtxt │ │ │ ├── api_def_DestroyTemporaryVariable.pbtxt │ │ │ ├── api_def_Diag.pbtxt │ │ │ ├── api_def_DiagPart.pbtxt │ │ │ ├── api_def_Digamma.pbtxt │ │ │ ├── api_def_Dilation2D.pbtxt │ │ │ ├── api_def_Dilation2DBackpropFilter.pbtxt │ │ │ ├── api_def_Dilation2DBackpropInput.pbtxt │ │ │ ├── api_def_Div.pbtxt │ │ │ ├── api_def_DivNoNan.pbtxt │ │ │ ├── api_def_DrawBoundingBoxes.pbtxt │ │ │ ├── api_def_DrawBoundingBoxesV2.pbtxt │ │ │ ├── api_def_EagerPyFunc.pbtxt │ │ │ ├── api_def_EditDistance.pbtxt │ │ │ ├── api_def_Eig.pbtxt │ │ │ ├── api_def_Einsum.pbtxt │ │ │ ├── api_def_Elu.pbtxt │ │ │ ├── api_def_Empty.pbtxt │ │ │ ├── api_def_EmptyTensorList.pbtxt │ │ │ ├── api_def_EmptyTensorMap.pbtxt │ │ │ ├── api_def_EncodeBase64.pbtxt │ │ │ ├── api_def_EncodeJpeg.pbtxt │ │ │ ├── api_def_EncodeJpegVariableQuality.pbtxt │ │ │ ├── api_def_EncodePng.pbtxt │ │ │ ├── api_def_EncodeProto.pbtxt │ │ │ ├── api_def_EncodeWav.pbtxt │ │ │ ├── api_def_EnsureShape.pbtxt │ │ │ ├── api_def_Equal.pbtxt │ │ │ ├── api_def_Erf.pbtxt │ │ │ ├── api_def_Erfc.pbtxt │ │ │ ├── api_def_Erfinv.pbtxt │ │ │ ├── api_def_EuclideanNorm.pbtxt │ │ │ ├── api_def_Exp.pbtxt │ │ │ ├── api_def_ExpandDims.pbtxt │ │ │ ├── api_def_Expm1.pbtxt │ │ │ ├── api_def_ExtractGlimpse.pbtxt │ │ │ ├── api_def_ExtractImagePatches.pbtxt │ │ │ ├── api_def_ExtractJpegShape.pbtxt │ │ │ ├── api_def_FFT.pbtxt │ │ │ ├── api_def_FFT2D.pbtxt │ │ │ ├── api_def_FFT3D.pbtxt │ │ │ ├── api_def_FIFOQueue.pbtxt │ │ │ ├── api_def_FIFOQueueV2.pbtxt │ │ │ ├── api_def_Fact.pbtxt │ │ │ ├── api_def_FakeParam.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxArgs.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxArgsGradient.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVars.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsGradient.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsPerChannel.pbtxt │ │ │ ├── api_def_FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt │ │ │ ├── api_def_FakeQueue.pbtxt │ │ │ ├── api_def_FileSystemSetConfiguration.pbtxt │ │ │ ├── api_def_Fill.pbtxt │ │ │ ├── api_def_Fingerprint.pbtxt │ │ │ ├── api_def_FixedLengthRecordReader.pbtxt │ │ │ ├── api_def_FixedLengthRecordReaderV2.pbtxt │ │ │ ├── api_def_FixedUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_Floor.pbtxt │ │ │ ├── api_def_FloorDiv.pbtxt │ │ │ ├── api_def_FloorMod.pbtxt │ │ │ ├── api_def_FlushSummaryWriter.pbtxt │ │ │ ├── api_def_For.pbtxt │ │ │ ├── api_def_FractionalAvgPool.pbtxt │ │ │ ├── api_def_FractionalMaxPool.pbtxt │ │ │ ├── api_def_FusedBatchNorm.pbtxt │ │ │ ├── api_def_FusedBatchNormGrad.pbtxt │ │ │ ├── api_def_FusedBatchNormGradV2.pbtxt │ │ │ ├── api_def_FusedBatchNormGradV3.pbtxt │ │ │ ├── api_def_FusedBatchNormV2.pbtxt │ │ │ ├── api_def_FusedBatchNormV3.pbtxt │ │ │ ├── api_def_FusedPadConv2D.pbtxt │ │ │ ├── api_def_FusedResizeAndPadConv2D.pbtxt │ │ │ ├── api_def_Gather.pbtxt │ │ │ ├── api_def_GatherNd.pbtxt │ │ │ ├── api_def_GatherV2.pbtxt │ │ │ ├── api_def_GenerateVocabRemapping.pbtxt │ │ │ ├── api_def_GetMinibatchSplitsWithPhysicalReplica.pbtxt │ │ │ ├── api_def_GetMinibatchesInCsrWithPhysicalReplica.pbtxt │ │ │ ├── api_def_GetSessionHandle.pbtxt │ │ │ ├── api_def_GetSessionHandleV2.pbtxt │ │ │ ├── api_def_GetSessionTensor.pbtxt │ │ │ ├── api_def_GlobalIterId.pbtxt │ │ │ ├── api_def_Greater.pbtxt │ │ │ ├── api_def_GreaterEqual.pbtxt │ │ │ ├── api_def_GuaranteeConst.pbtxt │ │ │ ├── api_def_HSVToRGB.pbtxt │ │ │ ├── api_def_HashTable.pbtxt │ │ │ ├── api_def_HashTableV2.pbtxt │ │ │ ├── api_def_HistogramFixedWidth.pbtxt │ │ │ ├── api_def_HistogramSummary.pbtxt │ │ │ ├── api_def_IFFT.pbtxt │ │ │ ├── api_def_IFFT2D.pbtxt │ │ │ ├── api_def_IFFT3D.pbtxt │ │ │ ├── api_def_IRFFT.pbtxt │ │ │ ├── api_def_IRFFT2D.pbtxt │ │ │ ├── api_def_IRFFT3D.pbtxt │ │ │ ├── api_def_Identity.pbtxt │ │ │ ├── api_def_IdentityReader.pbtxt │ │ │ ├── api_def_IdentityReaderV2.pbtxt │ │ │ ├── api_def_If.pbtxt │ │ │ ├── api_def_Igamma.pbtxt │ │ │ ├── api_def_Igammac.pbtxt │ │ │ ├── api_def_Imag.pbtxt │ │ │ ├── api_def_ImageSummary.pbtxt │ │ │ ├── api_def_ImmutableConst.pbtxt │ │ │ ├── api_def_ImportEvent.pbtxt │ │ │ ├── api_def_InTopK.pbtxt │ │ │ ├── api_def_InTopKV2.pbtxt │ │ │ ├── api_def_InitializeTable.pbtxt │ │ │ ├── api_def_InitializeTableFromTextFile.pbtxt │ │ │ ├── api_def_InitializeTableFromTextFileV2.pbtxt │ │ │ ├── api_def_InitializeTableV2.pbtxt │ │ │ ├── api_def_InplaceAdd.pbtxt │ │ │ ├── api_def_InplaceSub.pbtxt │ │ │ ├── api_def_InplaceUpdate.pbtxt │ │ │ ├── api_def_Inv.pbtxt │ │ │ ├── api_def_Invert.pbtxt │ │ │ ├── api_def_InvertPermutation.pbtxt │ │ │ ├── api_def_IsFinite.pbtxt │ │ │ ├── api_def_IsInf.pbtxt │ │ │ ├── api_def_IsNan.pbtxt │ │ │ ├── api_def_IsVariableInitialized.pbtxt │ │ │ ├── api_def_Iterator.pbtxt │ │ │ ├── api_def_IteratorFromStringHandle.pbtxt │ │ │ ├── api_def_IteratorGetNext.pbtxt │ │ │ ├── api_def_IteratorGetNextAsOptional.pbtxt │ │ │ ├── api_def_IteratorGetNextSync.pbtxt │ │ │ ├── api_def_IteratorToStringHandle.pbtxt │ │ │ ├── api_def_KMC2ChainInitialization.pbtxt │ │ │ ├── api_def_KmeansPlusPlusInitialization.pbtxt │ │ │ ├── api_def_L2Loss.pbtxt │ │ │ ├── api_def_LMDBReader.pbtxt │ │ │ ├── api_def_LRN.pbtxt │ │ │ ├── api_def_LearnedUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_LeftShift.pbtxt │ │ │ ├── api_def_Less.pbtxt │ │ │ ├── api_def_LessEqual.pbtxt │ │ │ ├── api_def_Lgamma.pbtxt │ │ │ ├── api_def_LinSpace.pbtxt │ │ │ ├── api_def_ListDiff.pbtxt │ │ │ ├── api_def_LoadAndRemapMatrix.pbtxt │ │ │ ├── api_def_Log.pbtxt │ │ │ ├── api_def_Log1p.pbtxt │ │ │ ├── api_def_LogMatrixDeterminant.pbtxt │ │ │ ├── api_def_LogSoftmax.pbtxt │ │ │ ├── api_def_LogUniformCandidateSampler.pbtxt │ │ │ ├── api_def_LogicalAnd.pbtxt │ │ │ ├── api_def_LogicalNot.pbtxt │ │ │ ├── api_def_LogicalOr.pbtxt │ │ │ ├── api_def_LookupTableExport.pbtxt │ │ │ ├── api_def_LookupTableExportV2.pbtxt │ │ │ ├── api_def_LookupTableFind.pbtxt │ │ │ ├── api_def_LookupTableFindV2.pbtxt │ │ │ ├── api_def_LookupTableImport.pbtxt │ │ │ ├── api_def_LookupTableImportV2.pbtxt │ │ │ ├── api_def_LookupTableInsert.pbtxt │ │ │ ├── api_def_LookupTableInsertV2.pbtxt │ │ │ ├── api_def_LookupTableSize.pbtxt │ │ │ ├── api_def_LookupTableSizeV2.pbtxt │ │ │ ├── api_def_LoopCond.pbtxt │ │ │ ├── api_def_Lu.pbtxt │ │ │ ├── api_def_MakeIterator.pbtxt │ │ │ ├── api_def_MapClear.pbtxt │ │ │ ├── api_def_MapIncompleteSize.pbtxt │ │ │ ├── api_def_MapPeek.pbtxt │ │ │ ├── api_def_MapSize.pbtxt │ │ │ ├── api_def_MapStage.pbtxt │ │ │ ├── api_def_MapUnstage.pbtxt │ │ │ ├── api_def_MapUnstageNoKey.pbtxt │ │ │ ├── api_def_MatMul.pbtxt │ │ │ ├── api_def_MatchingFiles.pbtxt │ │ │ ├── api_def_MatrixBandPart.pbtxt │ │ │ ├── api_def_MatrixDeterminant.pbtxt │ │ │ ├── api_def_MatrixDiag.pbtxt │ │ │ ├── api_def_MatrixDiagPart.pbtxt │ │ │ ├── api_def_MatrixDiagPartV2.pbtxt │ │ │ ├── api_def_MatrixDiagPartV3.pbtxt │ │ │ ├── api_def_MatrixDiagV2.pbtxt │ │ │ ├── api_def_MatrixDiagV3.pbtxt │ │ │ ├── api_def_MatrixExponential.pbtxt │ │ │ ├── api_def_MatrixInverse.pbtxt │ │ │ ├── api_def_MatrixSetDiag.pbtxt │ │ │ ├── api_def_MatrixSetDiagV2.pbtxt │ │ │ ├── api_def_MatrixSetDiagV3.pbtxt │ │ │ ├── api_def_MatrixSolve.pbtxt │ │ │ ├── api_def_MatrixSolveLs.pbtxt │ │ │ ├── api_def_MatrixSquareRoot.pbtxt │ │ │ ├── api_def_MatrixTriangularSolve.pbtxt │ │ │ ├── api_def_Max.pbtxt │ │ │ ├── api_def_MaxPool.pbtxt │ │ │ ├── api_def_MaxPool3D.pbtxt │ │ │ ├── api_def_MaxPool3DGrad.pbtxt │ │ │ ├── api_def_MaxPool3DGradGrad.pbtxt │ │ │ ├── api_def_MaxPoolGradGrad.pbtxt │ │ │ ├── api_def_MaxPoolGradGradV2.pbtxt │ │ │ ├── api_def_MaxPoolGradGradWithArgmax.pbtxt │ │ │ ├── api_def_MaxPoolGradV2.pbtxt │ │ │ ├── api_def_MaxPoolV2.pbtxt │ │ │ ├── api_def_MaxPoolWithArgmax.pbtxt │ │ │ ├── api_def_Maximum.pbtxt │ │ │ ├── api_def_Mean.pbtxt │ │ │ ├── api_def_Merge.pbtxt │ │ │ ├── api_def_MergeSummary.pbtxt │ │ │ ├── api_def_MergeV2Checkpoints.pbtxt │ │ │ ├── api_def_Mfcc.pbtxt │ │ │ ├── api_def_Min.pbtxt │ │ │ ├── api_def_Minimum.pbtxt │ │ │ ├── api_def_MirrorPad.pbtxt │ │ │ ├── api_def_Mod.pbtxt │ │ │ ├── api_def_Mul.pbtxt │ │ │ ├── api_def_MulNoNan.pbtxt │ │ │ ├── api_def_Multinomial.pbtxt │ │ │ ├── api_def_MutableDenseHashTable.pbtxt │ │ │ ├── api_def_MutableDenseHashTableV2.pbtxt │ │ │ ├── api_def_MutableHashTable.pbtxt │ │ │ ├── api_def_MutableHashTableOfTensors.pbtxt │ │ │ ├── api_def_MutableHashTableOfTensorsV2.pbtxt │ │ │ ├── api_def_MutableHashTableV2.pbtxt │ │ │ ├── api_def_MutexLock.pbtxt │ │ │ ├── api_def_MutexV2.pbtxt │ │ │ ├── api_def_Ndtri.pbtxt │ │ │ ├── api_def_NearestNeighbors.pbtxt │ │ │ ├── api_def_Neg.pbtxt │ │ │ ├── api_def_NegTrain.pbtxt │ │ │ ├── api_def_NextAfter.pbtxt │ │ │ ├── api_def_NextIteration.pbtxt │ │ │ ├── api_def_NonMaxSuppression.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV2.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV3.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV4.pbtxt │ │ │ ├── api_def_NonMaxSuppressionV5.pbtxt │ │ │ ├── api_def_NonMaxSuppressionWithOverlaps.pbtxt │ │ │ ├── api_def_NotEqual.pbtxt │ │ │ ├── api_def_NthElement.pbtxt │ │ │ ├── api_def_OneHot.pbtxt │ │ │ ├── api_def_OneShotIterator.pbtxt │ │ │ ├── api_def_OnesLike.pbtxt │ │ │ ├── api_def_OptionalFromValue.pbtxt │ │ │ ├── api_def_OptionalGetValue.pbtxt │ │ │ ├── api_def_OptionalHasValue.pbtxt │ │ │ ├── api_def_OptionalNone.pbtxt │ │ │ ├── api_def_OrderedMapClear.pbtxt │ │ │ ├── api_def_OrderedMapIncompleteSize.pbtxt │ │ │ ├── api_def_OrderedMapPeek.pbtxt │ │ │ ├── api_def_OrderedMapSize.pbtxt │ │ │ ├── api_def_OrderedMapStage.pbtxt │ │ │ ├── api_def_OrderedMapUnstage.pbtxt │ │ │ ├── api_def_OrderedMapUnstageNoKey.pbtxt │ │ │ ├── api_def_Pack.pbtxt │ │ │ ├── api_def_Pad.pbtxt │ │ │ ├── api_def_PadV2.pbtxt │ │ │ ├── api_def_PaddingFIFOQueue.pbtxt │ │ │ ├── api_def_PaddingFIFOQueueV2.pbtxt │ │ │ ├── api_def_ParallelConcat.pbtxt │ │ │ ├── api_def_ParallelDynamicStitch.pbtxt │ │ │ ├── api_def_ParameterizedTruncatedNormal.pbtxt │ │ │ ├── api_def_ParseExample.pbtxt │ │ │ ├── api_def_ParseExampleV2.pbtxt │ │ │ ├── api_def_ParseSequenceExample.pbtxt │ │ │ ├── api_def_ParseSequenceExampleV2.pbtxt │ │ │ ├── api_def_ParseSingleExample.pbtxt │ │ │ ├── api_def_ParseSingleSequenceExample.pbtxt │ │ │ ├── api_def_ParseTensor.pbtxt │ │ │ ├── api_def_PartitionedCall.pbtxt │ │ │ ├── api_def_Placeholder.pbtxt │ │ │ ├── api_def_PlaceholderV2.pbtxt │ │ │ ├── api_def_PlaceholderWithDefault.pbtxt │ │ │ ├── api_def_Polygamma.pbtxt │ │ │ ├── api_def_PopulationCount.pbtxt │ │ │ ├── api_def_Pow.pbtxt │ │ │ ├── api_def_PreventGradient.pbtxt │ │ │ ├── api_def_Print.pbtxt │ │ │ ├── api_def_PrintV2.pbtxt │ │ │ ├── api_def_PriorityQueue.pbtxt │ │ │ ├── api_def_PriorityQueueV2.pbtxt │ │ │ ├── api_def_Prod.pbtxt │ │ │ ├── api_def_PyFunc.pbtxt │ │ │ ├── api_def_PyFuncStateless.pbtxt │ │ │ ├── api_def_Qr.pbtxt │ │ │ ├── api_def_QuantizeAndDequantize.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV2.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV3.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV4.pbtxt │ │ │ ├── api_def_QuantizeAndDequantizeV4Grad.pbtxt │ │ │ ├── api_def_QuantizeDownAndShrinkRange.pbtxt │ │ │ ├── api_def_QuantizeV2.pbtxt │ │ │ ├── api_def_QuantizedAdd.pbtxt │ │ │ ├── api_def_QuantizedAvgPool.pbtxt │ │ │ ├── api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt │ │ │ ├── api_def_QuantizedBiasAdd.pbtxt │ │ │ ├── api_def_QuantizedConcat.pbtxt │ │ │ ├── api_def_QuantizedConv2D.pbtxt │ │ │ ├── api_def_QuantizedInstanceNorm.pbtxt │ │ │ ├── api_def_QuantizedMatMul.pbtxt │ │ │ ├── api_def_QuantizedMaxPool.pbtxt │ │ │ ├── api_def_QuantizedMul.pbtxt │ │ │ ├── api_def_QuantizedRelu.pbtxt │ │ │ ├── api_def_QuantizedRelu6.pbtxt │ │ │ ├── api_def_QuantizedReluX.pbtxt │ │ │ ├── api_def_QuantizedReshape.pbtxt │ │ │ ├── api_def_QuantizedResizeBilinear.pbtxt │ │ │ ├── api_def_QueueClose.pbtxt │ │ │ ├── api_def_QueueCloseV2.pbtxt │ │ │ ├── api_def_QueueDequeue.pbtxt │ │ │ ├── api_def_QueueDequeueMany.pbtxt │ │ │ ├── api_def_QueueDequeueManyV2.pbtxt │ │ │ ├── api_def_QueueDequeueUpTo.pbtxt │ │ │ ├── api_def_QueueDequeueUpToV2.pbtxt │ │ │ ├── api_def_QueueDequeueV2.pbtxt │ │ │ ├── api_def_QueueEnqueue.pbtxt │ │ │ ├── api_def_QueueEnqueueMany.pbtxt │ │ │ ├── api_def_QueueEnqueueManyV2.pbtxt │ │ │ ├── api_def_QueueEnqueueV2.pbtxt │ │ │ ├── api_def_QueueIsClosed.pbtxt │ │ │ ├── api_def_QueueIsClosedV2.pbtxt │ │ │ ├── api_def_QueueSize.pbtxt │ │ │ ├── api_def_QueueSizeV2.pbtxt │ │ │ ├── api_def_RFFT.pbtxt │ │ │ ├── api_def_RFFT2D.pbtxt │ │ │ ├── api_def_RFFT3D.pbtxt │ │ │ ├── api_def_RGBToHSV.pbtxt │ │ │ ├── api_def_RaggedBincount.pbtxt │ │ │ ├── api_def_RandomCrop.pbtxt │ │ │ ├── api_def_RandomGamma.pbtxt │ │ │ ├── api_def_RandomPoisson.pbtxt │ │ │ ├── api_def_RandomPoissonV2.pbtxt │ │ │ ├── api_def_RandomShuffle.pbtxt │ │ │ ├── api_def_RandomShuffleQueue.pbtxt │ │ │ ├── api_def_RandomShuffleQueueV2.pbtxt │ │ │ ├── api_def_RandomStandardNormal.pbtxt │ │ │ ├── api_def_RandomUniform.pbtxt │ │ │ ├── api_def_RandomUniformInt.pbtxt │ │ │ ├── api_def_Range.pbtxt │ │ │ ├── api_def_Rank.pbtxt │ │ │ ├── api_def_ReadFile.pbtxt │ │ │ ├── api_def_ReadVariableOp.pbtxt │ │ │ ├── api_def_ReaderNumRecordsProduced.pbtxt │ │ │ ├── api_def_ReaderNumRecordsProducedV2.pbtxt │ │ │ ├── api_def_ReaderNumWorkUnitsCompleted.pbtxt │ │ │ ├── api_def_ReaderNumWorkUnitsCompletedV2.pbtxt │ │ │ ├── api_def_ReaderRead.pbtxt │ │ │ ├── api_def_ReaderReadUpTo.pbtxt │ │ │ ├── api_def_ReaderReadUpToV2.pbtxt │ │ │ ├── api_def_ReaderReadV2.pbtxt │ │ │ ├── api_def_ReaderReset.pbtxt │ │ │ ├── api_def_ReaderResetV2.pbtxt │ │ │ ├── api_def_ReaderRestoreState.pbtxt │ │ │ ├── api_def_ReaderRestoreStateV2.pbtxt │ │ │ ├── api_def_ReaderSerializeState.pbtxt │ │ │ ├── api_def_ReaderSerializeStateV2.pbtxt │ │ │ ├── api_def_Real.pbtxt │ │ │ ├── api_def_RealDiv.pbtxt │ │ │ ├── api_def_Reciprocal.pbtxt │ │ │ ├── api_def_RecordInput.pbtxt │ │ │ ├── api_def_ReduceJoin.pbtxt │ │ │ ├── api_def_RefNextIteration.pbtxt │ │ │ ├── api_def_RefSelect.pbtxt │ │ │ ├── api_def_RefSwitch.pbtxt │ │ │ ├── api_def_RegexFullMatch.pbtxt │ │ │ ├── api_def_RegexReplace.pbtxt │ │ │ ├── api_def_Relu.pbtxt │ │ │ ├── api_def_Relu6.pbtxt │ │ │ ├── api_def_RemoteCall.pbtxt │ │ │ ├── api_def_RequantizationRange.pbtxt │ │ │ ├── api_def_Requantize.pbtxt │ │ │ ├── api_def_Reshape.pbtxt │ │ │ ├── api_def_ResizeArea.pbtxt │ │ │ ├── api_def_ResizeBicubic.pbtxt │ │ │ ├── api_def_ResizeBilinear.pbtxt │ │ │ ├── api_def_ResizeNearestNeighbor.pbtxt │ │ │ ├── api_def_ResourceApplyAdaMax.pbtxt │ │ │ ├── api_def_ResourceApplyAdadelta.pbtxt │ │ │ ├── api_def_ResourceApplyAdagrad.pbtxt │ │ │ ├── api_def_ResourceApplyAdagradDA.pbtxt │ │ │ ├── api_def_ResourceApplyAdam.pbtxt │ │ │ ├── api_def_ResourceApplyAdamWithAmsgrad.pbtxt │ │ │ ├── api_def_ResourceApplyAddSign.pbtxt │ │ │ ├── api_def_ResourceApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ResourceApplyFtrl.pbtxt │ │ │ ├── api_def_ResourceApplyFtrlV2.pbtxt │ │ │ ├── api_def_ResourceApplyGradientDescent.pbtxt │ │ │ ├── api_def_ResourceApplyKerasMomentum.pbtxt │ │ │ ├── api_def_ResourceApplyMomentum.pbtxt │ │ │ ├── api_def_ResourceApplyPowerSign.pbtxt │ │ │ ├── api_def_ResourceApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ResourceApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ResourceApplyRMSProp.pbtxt │ │ │ ├── api_def_ResourceCountUpTo.pbtxt │ │ │ ├── api_def_ResourceGather.pbtxt │ │ │ ├── api_def_ResourceScatterAdd.pbtxt │ │ │ ├── api_def_ResourceScatterDiv.pbtxt │ │ │ ├── api_def_ResourceScatterMax.pbtxt │ │ │ ├── api_def_ResourceScatterMin.pbtxt │ │ │ ├── api_def_ResourceScatterMul.pbtxt │ │ │ ├── api_def_ResourceScatterNdAdd.pbtxt │ │ │ ├── api_def_ResourceScatterNdMax.pbtxt │ │ │ ├── api_def_ResourceScatterNdMin.pbtxt │ │ │ ├── api_def_ResourceScatterNdSub.pbtxt │ │ │ ├── api_def_ResourceScatterNdUpdate.pbtxt │ │ │ ├── api_def_ResourceScatterSub.pbtxt │ │ │ ├── api_def_ResourceScatterUpdate.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdadelta.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagrad.pbtxt │ │ │ ├── api_def_ResourceSparseApplyAdagradDA.pbtxt │ │ │ ├── api_def_ResourceSparseApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_ResourceSparseApplyFtrl.pbtxt │ │ │ ├── api_def_ResourceSparseApplyFtrlV2.pbtxt │ │ │ ├── api_def_ResourceSparseApplyKerasMomentum.pbtxt │ │ │ ├── api_def_ResourceSparseApplyMomentum.pbtxt │ │ │ ├── api_def_ResourceSparseApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_ResourceSparseApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_ResourceSparseApplyRMSProp.pbtxt │ │ │ ├── api_def_ResourceStridedSliceAssign.pbtxt │ │ │ ├── api_def_ResrouceGatherNd.pbtxt │ │ │ ├── api_def_Restore.pbtxt │ │ │ ├── api_def_RestoreSlice.pbtxt │ │ │ ├── api_def_RestoreV2.pbtxt │ │ │ ├── api_def_Reverse.pbtxt │ │ │ ├── api_def_ReverseSequence.pbtxt │ │ │ ├── api_def_ReverseV2.pbtxt │ │ │ ├── api_def_RightShift.pbtxt │ │ │ ├── api_def_Rint.pbtxt │ │ │ ├── api_def_Roll.pbtxt │ │ │ ├── api_def_Round.pbtxt │ │ │ ├── api_def_Rsqrt.pbtxt │ │ │ ├── api_def_SampleDistortedBoundingBox.pbtxt │ │ │ ├── api_def_SampleDistortedBoundingBoxV2.pbtxt │ │ │ ├── api_def_Save.pbtxt │ │ │ ├── api_def_SaveSlices.pbtxt │ │ │ ├── api_def_SaveV2.pbtxt │ │ │ ├── api_def_ScalarSummary.pbtxt │ │ │ ├── api_def_ScaleAndTranslate.pbtxt │ │ │ ├── api_def_ScatterAdd.pbtxt │ │ │ ├── api_def_ScatterDiv.pbtxt │ │ │ ├── api_def_ScatterMax.pbtxt │ │ │ ├── api_def_ScatterMin.pbtxt │ │ │ ├── api_def_ScatterMul.pbtxt │ │ │ ├── api_def_ScatterNd.pbtxt │ │ │ ├── api_def_ScatterNdAdd.pbtxt │ │ │ ├── api_def_ScatterNdNonAliasingAdd.pbtxt │ │ │ ├── api_def_ScatterNdSub.pbtxt │ │ │ ├── api_def_ScatterNdUpdate.pbtxt │ │ │ ├── api_def_ScatterSub.pbtxt │ │ │ ├── api_def_ScatterUpdate.pbtxt │ │ │ ├── api_def_SdcaFprint.pbtxt │ │ │ ├── api_def_SdcaOptimizer.pbtxt │ │ │ ├── api_def_SdcaShrinkL1.pbtxt │ │ │ ├── api_def_SegmentMax.pbtxt │ │ │ ├── api_def_SegmentMaxV2.pbtxt │ │ │ ├── api_def_SegmentMean.pbtxt │ │ │ ├── api_def_SegmentMin.pbtxt │ │ │ ├── api_def_SegmentMinV2.pbtxt │ │ │ ├── api_def_SegmentProd.pbtxt │ │ │ ├── api_def_SegmentProdV2.pbtxt │ │ │ ├── api_def_SegmentSum.pbtxt │ │ │ ├── api_def_SegmentSumV2.pbtxt │ │ │ ├── api_def_Select.pbtxt │ │ │ ├── api_def_SelectV2.pbtxt │ │ │ ├── api_def_SelfAdjointEig.pbtxt │ │ │ ├── api_def_SelfAdjointEigV2.pbtxt │ │ │ ├── api_def_Selu.pbtxt │ │ │ ├── api_def_SerializeIterator.pbtxt │ │ │ ├── api_def_SerializeManySparse.pbtxt │ │ │ ├── api_def_SerializeSparse.pbtxt │ │ │ ├── api_def_SerializeTensor.pbtxt │ │ │ ├── api_def_SetSize.pbtxt │ │ │ ├── api_def_Shape.pbtxt │ │ │ ├── api_def_ShapeN.pbtxt │ │ │ ├── api_def_ShardedFilename.pbtxt │ │ │ ├── api_def_ShardedFilespec.pbtxt │ │ │ ├── api_def_Sigmoid.pbtxt │ │ │ ├── api_def_Sign.pbtxt │ │ │ ├── api_def_Sin.pbtxt │ │ │ ├── api_def_Sinh.pbtxt │ │ │ ├── api_def_Size.pbtxt │ │ │ ├── api_def_Skipgram.pbtxt │ │ │ ├── api_def_Slice.pbtxt │ │ │ ├── api_def_Snapshot.pbtxt │ │ │ ├── api_def_Softmax.pbtxt │ │ │ ├── api_def_SoftmaxCrossEntropyWithLogits.pbtxt │ │ │ ├── api_def_Softplus.pbtxt │ │ │ ├── api_def_Softsign.pbtxt │ │ │ ├── api_def_SpaceToBatch.pbtxt │ │ │ ├── api_def_SpaceToBatchND.pbtxt │ │ │ ├── api_def_SpaceToDepth.pbtxt │ │ │ ├── api_def_SparseAccumulatorApplyGradient.pbtxt │ │ │ ├── api_def_SparseAccumulatorTakeGradient.pbtxt │ │ │ ├── api_def_SparseAdd.pbtxt │ │ │ ├── api_def_SparseAddGrad.pbtxt │ │ │ ├── api_def_SparseApplyAdadelta.pbtxt │ │ │ ├── api_def_SparseApplyAdagrad.pbtxt │ │ │ ├── api_def_SparseApplyAdagradDA.pbtxt │ │ │ ├── api_def_SparseApplyCenteredRMSProp.pbtxt │ │ │ ├── api_def_SparseApplyFtrl.pbtxt │ │ │ ├── api_def_SparseApplyFtrlV2.pbtxt │ │ │ ├── api_def_SparseApplyMomentum.pbtxt │ │ │ ├── api_def_SparseApplyProximalAdagrad.pbtxt │ │ │ ├── api_def_SparseApplyProximalGradientDescent.pbtxt │ │ │ ├── api_def_SparseApplyRMSProp.pbtxt │ │ │ ├── api_def_SparseBincount.pbtxt │ │ │ ├── api_def_SparseConcat.pbtxt │ │ │ ├── api_def_SparseConditionalAccumulator.pbtxt │ │ │ ├── api_def_SparseCross.pbtxt │ │ │ ├── api_def_SparseCrossHashed.pbtxt │ │ │ ├── api_def_SparseCrossV2.pbtxt │ │ │ ├── api_def_SparseDenseCwiseAdd.pbtxt │ │ │ ├── api_def_SparseDenseCwiseDiv.pbtxt │ │ │ ├── api_def_SparseDenseCwiseMul.pbtxt │ │ │ ├── api_def_SparseFillEmptyRows.pbtxt │ │ │ ├── api_def_SparseFillEmptyRowsGrad.pbtxt │ │ │ ├── api_def_SparseMatMul.pbtxt │ │ │ ├── api_def_SparseReduceMax.pbtxt │ │ │ ├── api_def_SparseReduceMaxSparse.pbtxt │ │ │ ├── api_def_SparseReduceSum.pbtxt │ │ │ ├── api_def_SparseReduceSumSparse.pbtxt │ │ │ ├── api_def_SparseReorder.pbtxt │ │ │ ├── api_def_SparseReshape.pbtxt │ │ │ ├── api_def_SparseSegmentMean.pbtxt │ │ │ ├── api_def_SparseSegmentMeanGrad.pbtxt │ │ │ ├── api_def_SparseSegmentMeanGradV2.pbtxt │ │ │ ├── api_def_SparseSegmentMeanWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtN.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNGrad.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNGradV2.pbtxt │ │ │ ├── api_def_SparseSegmentSqrtNWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSegmentSum.pbtxt │ │ │ ├── api_def_SparseSegmentSumGrad.pbtxt │ │ │ ├── api_def_SparseSegmentSumGradV2.pbtxt │ │ │ ├── api_def_SparseSegmentSumWithNumSegments.pbtxt │ │ │ ├── api_def_SparseSlice.pbtxt │ │ │ ├── api_def_SparseSliceGrad.pbtxt │ │ │ ├── api_def_SparseSoftmax.pbtxt │ │ │ ├── api_def_SparseSoftmaxCrossEntropyWithLogits.pbtxt │ │ │ ├── api_def_SparseSparseMaximum.pbtxt │ │ │ ├── api_def_SparseSparseMinimum.pbtxt │ │ │ ├── api_def_SparseSplit.pbtxt │ │ │ ├── api_def_SparseTensorDenseAdd.pbtxt │ │ │ ├── api_def_SparseTensorDenseMatMul.pbtxt │ │ │ ├── api_def_SparseToDense.pbtxt │ │ │ ├── api_def_SparseToSparseSetOperation.pbtxt │ │ │ ├── api_def_Split.pbtxt │ │ │ ├── api_def_SplitV.pbtxt │ │ │ ├── api_def_Sqrt.pbtxt │ │ │ ├── api_def_Square.pbtxt │ │ │ ├── api_def_SquaredDifference.pbtxt │ │ │ ├── api_def_Squeeze.pbtxt │ │ │ ├── api_def_Stack.pbtxt │ │ │ ├── api_def_StackClose.pbtxt │ │ │ ├── api_def_StackCloseV2.pbtxt │ │ │ ├── api_def_StackPop.pbtxt │ │ │ ├── api_def_StackPopV2.pbtxt │ │ │ ├── api_def_StackPush.pbtxt │ │ │ ├── api_def_StackPushV2.pbtxt │ │ │ ├── api_def_StackV2.pbtxt │ │ │ ├── api_def_Stage.pbtxt │ │ │ ├── api_def_StageClear.pbtxt │ │ │ ├── api_def_StagePeek.pbtxt │ │ │ ├── api_def_StageSize.pbtxt │ │ │ ├── api_def_StatefulPartitionedCall.pbtxt │ │ │ ├── api_def_StatefulRandomBinomial.pbtxt │ │ │ ├── api_def_StatefulStandardNormal.pbtxt │ │ │ ├── api_def_StatefulStandardNormalV2.pbtxt │ │ │ ├── api_def_StatelessIf.pbtxt │ │ │ ├── api_def_StatelessMultinomial.pbtxt │ │ │ ├── api_def_StatelessRandomNormal.pbtxt │ │ │ ├── api_def_StatelessRandomUniform.pbtxt │ │ │ ├── api_def_StatelessSampleDistortedBoundingBox.pbtxt │ │ │ ├── api_def_StatelessTruncatedNormal.pbtxt │ │ │ ├── api_def_StatelessWhile.pbtxt │ │ │ ├── api_def_StopGradient.pbtxt │ │ │ ├── api_def_StoreMinibatchStatisticsInFdo.pbtxt │ │ │ ├── api_def_StridedSlice.pbtxt │ │ │ ├── api_def_StridedSliceAssign.pbtxt │ │ │ ├── api_def_StridedSliceGrad.pbtxt │ │ │ ├── api_def_StringFormat.pbtxt │ │ │ ├── api_def_StringJoin.pbtxt │ │ │ ├── api_def_StringLength.pbtxt │ │ │ ├── api_def_StringLower.pbtxt │ │ │ ├── api_def_StringNGrams.pbtxt │ │ │ ├── api_def_StringSplit.pbtxt │ │ │ ├── api_def_StringSplitV2.pbtxt │ │ │ ├── api_def_StringStrip.pbtxt │ │ │ ├── api_def_StringToHashBucket.pbtxt │ │ │ ├── api_def_StringToHashBucketFast.pbtxt │ │ │ ├── api_def_StringToHashBucketStrong.pbtxt │ │ │ ├── api_def_StringToNumber.pbtxt │ │ │ ├── api_def_StringUpper.pbtxt │ │ │ ├── api_def_Sub.pbtxt │ │ │ ├── api_def_Substr.pbtxt │ │ │ ├── api_def_Sum.pbtxt │ │ │ ├── api_def_SummaryWriter.pbtxt │ │ │ ├── api_def_Svd.pbtxt │ │ │ ├── api_def_Switch.pbtxt │ │ │ ├── api_def_SymbolicGradient.pbtxt │ │ │ ├── api_def_TFRecordReader.pbtxt │ │ │ ├── api_def_TFRecordReaderV2.pbtxt │ │ │ ├── api_def_TPUAnnotateTensorsWithDynamicShape.pbtxt │ │ │ ├── api_def_TPUCopyWithDynamicShape.pbtxt │ │ │ ├── api_def_TPUPartitionedInput.pbtxt │ │ │ ├── api_def_TPUPartitionedInputV2.pbtxt │ │ │ ├── api_def_TPUPartitionedOutput.pbtxt │ │ │ ├── api_def_TPUPartitionedOutputV2.pbtxt │ │ │ ├── api_def_TakeManySparseFromTensorsMap.pbtxt │ │ │ ├── api_def_Tan.pbtxt │ │ │ ├── api_def_Tanh.pbtxt │ │ │ ├── api_def_TemporaryVariable.pbtxt │ │ │ ├── api_def_TensorArray.pbtxt │ │ │ ├── api_def_TensorArrayClose.pbtxt │ │ │ ├── api_def_TensorArrayCloseV2.pbtxt │ │ │ ├── api_def_TensorArrayCloseV3.pbtxt │ │ │ ├── api_def_TensorArrayConcat.pbtxt │ │ │ ├── api_def_TensorArrayConcatV2.pbtxt │ │ │ ├── api_def_TensorArrayConcatV3.pbtxt │ │ │ ├── api_def_TensorArrayGather.pbtxt │ │ │ ├── api_def_TensorArrayGatherV2.pbtxt │ │ │ ├── api_def_TensorArrayGatherV3.pbtxt │ │ │ ├── api_def_TensorArrayGrad.pbtxt │ │ │ ├── api_def_TensorArrayGradV2.pbtxt │ │ │ ├── api_def_TensorArrayGradV3.pbtxt │ │ │ ├── api_def_TensorArrayGradWithShape.pbtxt │ │ │ ├── api_def_TensorArrayPack.pbtxt │ │ │ ├── api_def_TensorArrayRead.pbtxt │ │ │ ├── api_def_TensorArrayReadV2.pbtxt │ │ │ ├── api_def_TensorArrayReadV3.pbtxt │ │ │ ├── api_def_TensorArrayScatter.pbtxt │ │ │ ├── api_def_TensorArrayScatterV2.pbtxt │ │ │ ├── api_def_TensorArrayScatterV3.pbtxt │ │ │ ├── api_def_TensorArraySize.pbtxt │ │ │ ├── api_def_TensorArraySizeV2.pbtxt │ │ │ ├── api_def_TensorArraySizeV3.pbtxt │ │ │ ├── api_def_TensorArraySplit.pbtxt │ │ │ ├── api_def_TensorArraySplitV2.pbtxt │ │ │ ├── api_def_TensorArraySplitV3.pbtxt │ │ │ ├── api_def_TensorArrayUnpack.pbtxt │ │ │ ├── api_def_TensorArrayV2.pbtxt │ │ │ ├── api_def_TensorArrayV3.pbtxt │ │ │ ├── api_def_TensorArrayWrite.pbtxt │ │ │ ├── api_def_TensorArrayWriteV2.pbtxt │ │ │ ├── api_def_TensorArrayWriteV3.pbtxt │ │ │ ├── api_def_TensorListConcat.pbtxt │ │ │ ├── api_def_TensorListConcatLists.pbtxt │ │ │ ├── api_def_TensorListConcatV2.pbtxt │ │ │ ├── api_def_TensorListElementShape.pbtxt │ │ │ ├── api_def_TensorListFromTensor.pbtxt │ │ │ ├── api_def_TensorListGather.pbtxt │ │ │ ├── api_def_TensorListGetItem.pbtxt │ │ │ ├── api_def_TensorListLength.pbtxt │ │ │ ├── api_def_TensorListPopBack.pbtxt │ │ │ ├── api_def_TensorListPushBack.pbtxt │ │ │ ├── api_def_TensorListPushBackBatch.pbtxt │ │ │ ├── api_def_TensorListReserve.pbtxt │ │ │ ├── api_def_TensorListResize.pbtxt │ │ │ ├── api_def_TensorListScatter.pbtxt │ │ │ ├── api_def_TensorListScatterIntoExistingList.pbtxt │ │ │ ├── api_def_TensorListScatterV2.pbtxt │ │ │ ├── api_def_TensorListSetItem.pbtxt │ │ │ ├── api_def_TensorListSplit.pbtxt │ │ │ ├── api_def_TensorListStack.pbtxt │ │ │ ├── api_def_TensorMapErase.pbtxt │ │ │ ├── api_def_TensorMapHasKey.pbtxt │ │ │ ├── api_def_TensorMapInsert.pbtxt │ │ │ ├── api_def_TensorMapLookup.pbtxt │ │ │ ├── api_def_TensorMapSize.pbtxt │ │ │ ├── api_def_TensorMapStackKeys.pbtxt │ │ │ ├── api_def_TensorScatterAdd.pbtxt │ │ │ ├── api_def_TensorScatterMax.pbtxt │ │ │ ├── api_def_TensorScatterMin.pbtxt │ │ │ ├── api_def_TensorScatterSub.pbtxt │ │ │ ├── api_def_TensorScatterUpdate.pbtxt │ │ │ ├── api_def_TensorStridedSliceUpdate.pbtxt │ │ │ ├── api_def_TensorSummary.pbtxt │ │ │ ├── api_def_TensorSummaryV2.pbtxt │ │ │ ├── api_def_TextLineReader.pbtxt │ │ │ ├── api_def_TextLineReaderV2.pbtxt │ │ │ ├── api_def_ThreadUnsafeUnigramCandidateSampler.pbtxt │ │ │ ├── api_def_Tile.pbtxt │ │ │ ├── api_def_TileGrad.pbtxt │ │ │ ├── api_def_ToBool.pbtxt │ │ │ ├── api_def_TopK.pbtxt │ │ │ ├── api_def_TopKV2.pbtxt │ │ │ ├── api_def_Transpose.pbtxt │ │ │ ├── api_def_TruncateDiv.pbtxt │ │ │ ├── api_def_TruncateMod.pbtxt │ │ │ ├── api_def_TruncatedNormal.pbtxt │ │ │ ├── api_def_Unbatch.pbtxt │ │ │ ├── api_def_UnbatchGrad.pbtxt │ │ │ ├── api_def_UnicodeScript.pbtxt │ │ │ ├── api_def_UnicodeTranscode.pbtxt │ │ │ ├── api_def_UniformCandidateSampler.pbtxt │ │ │ ├── api_def_Unique.pbtxt │ │ │ ├── api_def_UniqueV2.pbtxt │ │ │ ├── api_def_UniqueWithCounts.pbtxt │ │ │ ├── api_def_UniqueWithCountsV2.pbtxt │ │ │ ├── api_def_Unpack.pbtxt │ │ │ ├── api_def_UnsortedSegmentJoin.pbtxt │ │ │ ├── api_def_UnsortedSegmentMax.pbtxt │ │ │ ├── api_def_UnsortedSegmentMin.pbtxt │ │ │ ├── api_def_UnsortedSegmentProd.pbtxt │ │ │ ├── api_def_UnsortedSegmentSum.pbtxt │ │ │ ├── api_def_Unstage.pbtxt │ │ │ ├── api_def_VarHandleOp.pbtxt │ │ │ ├── api_def_VarIsInitializedOp.pbtxt │ │ │ ├── api_def_Variable.pbtxt │ │ │ ├── api_def_VariableShape.pbtxt │ │ │ ├── api_def_VariableV2.pbtxt │ │ │ ├── api_def_Where.pbtxt │ │ │ ├── api_def_While.pbtxt │ │ │ ├── api_def_WholeFileReader.pbtxt │ │ │ ├── api_def_WholeFileReaderV2.pbtxt │ │ │ ├── api_def_WriteAudioSummary.pbtxt │ │ │ ├── api_def_WriteFile.pbtxt │ │ │ ├── api_def_WriteGraphSummary.pbtxt │ │ │ ├── api_def_WriteHistogramSummary.pbtxt │ │ │ ├── api_def_WriteImageSummary.pbtxt │ │ │ ├── api_def_WriteScalarSummary.pbtxt │ │ │ ├── api_def_WriteSummary.pbtxt │ │ │ ├── api_def_Xdivy.pbtxt │ │ │ ├── api_def_XlaSparseCoreAdagrad.pbtxt │ │ │ ├── api_def_XlaSparseCoreAdagradMomentum.pbtxt │ │ │ ├── api_def_XlaSparseCoreAdam.pbtxt │ │ │ ├── api_def_XlaSparseCoreFtrl.pbtxt │ │ │ ├── api_def_XlaSparseCoreSgd.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmul.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithAdagradAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithAdamAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithFtrlAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulGradWithSgdAndCsrInput.pbtxt │ │ │ ├── api_def_XlaSparseDenseMatmulWithCsrInput.pbtxt │ │ │ ├── api_def_Xlog1py.pbtxt │ │ │ ├── api_def_Xlogy.pbtxt │ │ │ ├── api_def_ZerosLike.pbtxt │ │ │ └── api_def_Zeta.pbtxt │ │ ├── update_api_def.cc │ │ ├── update_api_def.h │ │ ├── update_api_def.sh │ │ ├── update_api_def_main.cc │ │ └── update_api_def_test.cc │ ├── build_defs.bzl │ ├── common_runtime │ │ ├── BUILD │ │ ├── accumulate_n_optimizer.cc │ │ ├── all_to_all.cc │ │ ├── all_to_all.h │ │ ├── all_to_all_test.cc │ │ ├── allocator_retry.h │ │ ├── arg_ret_placement.cc │ │ ├── arg_ret_placement.h │ │ ├── arg_ret_placement_test.cc │ │ ├── base_collective_executor.cc │ │ ├── base_collective_executor.h │ │ ├── bfc_allocator.h │ │ ├── buf_rendezvous.cc │ │ ├── buf_rendezvous.h │ │ ├── buf_rendezvous_test.cc │ │ ├── build_graph_options.cc │ │ ├── build_graph_options.h │ │ ├── collective_executor_mgr.cc │ │ ├── collective_executor_mgr.h │ │ ├── collective_executor_mgr_test.cc │ │ ├── collective_param_resolver_local.cc │ │ ├── collective_param_resolver_local.h │ │ ├── collective_param_resolver_local_test.cc │ │ ├── collective_rma_local.cc │ │ ├── collective_rma_local.h │ │ ├── collective_rma_local_test.cc │ │ ├── collective_test_util.cc │ │ ├── collective_test_util.h │ │ ├── collective_util.cc │ │ ├── collective_util.h │ │ ├── colocation_graph.cc │ │ ├── colocation_graph.h │ │ ├── composite_device.cc │ │ ├── composite_device.h │ │ ├── composite_device_test.cc │ │ ├── constant_folding.cc │ │ ├── constant_folding.h │ │ ├── constant_folding_test.cc │ │ ├── copy_tensor.cc │ │ ├── copy_tensor.h │ │ ├── cost_constants.h │ │ ├── cost_measurement.h │ │ ├── cost_measurement_registry.cc │ │ ├── cost_measurement_registry.h │ │ ├── cost_measurement_registry_test.cc │ │ ├── cost_util.cc │ │ ├── cost_util.h │ │ ├── cost_util_test.cc │ │ ├── costmodel_manager.cc │ │ ├── costmodel_manager.h │ │ ├── debugger_state_interface.cc │ │ ├── debugger_state_interface.h │ │ ├── device.h │ │ ├── device │ │ │ ├── BUILD │ │ │ ├── device_event_mgr.cc │ │ │ ├── device_event_mgr.h │ │ │ ├── device_event_mgr_test.cc │ │ │ ├── device_host_allocator.h │ │ │ ├── device_id.h │ │ │ ├── device_id_manager.h │ │ │ ├── device_id_manager_test.cc │ │ │ ├── device_mem_allocator.h │ │ │ ├── device_utils.cc │ │ │ └── device_utils.h │ │ ├── device_factory.h │ │ ├── device_mgr.cc │ │ ├── device_mgr.h │ │ ├── device_mgr_test.cc │ │ ├── device_propagation.cc │ │ ├── device_propagation.h │ │ ├── device_propagation_test.cc │ │ ├── device_resolver_local.cc │ │ ├── device_resolver_local.h │ │ ├── device_resolver_local_test.cc │ │ ├── device_set.cc │ │ ├── device_set.h │ │ ├── device_set_test.cc │ │ ├── direct_session.cc │ │ ├── direct_session.h │ │ ├── direct_session_test.cc │ │ ├── direct_session_with_tracking_alloc_test.cc │ │ ├── dma_helper.h │ │ ├── dynamic_device_mgr.cc │ │ ├── dynamic_device_mgr_test.cc │ │ ├── eager │ │ │ ├── BUILD │ │ │ ├── attr_builder.cc │ │ │ ├── attr_builder.h │ │ │ ├── attr_builder_test.cc │ │ │ ├── context.cc │ │ │ ├── context.h │ │ │ ├── context_distributed_manager.cc │ │ │ ├── context_distributed_manager.h │ │ │ ├── context_test.cc │ │ │ ├── copy_to_device_node.h │ │ │ ├── core.cc │ │ │ ├── custom_device.cc │ │ │ ├── custom_device.h │ │ │ ├── custom_device_op_handler.cc │ │ │ ├── custom_device_op_handler.h │ │ │ ├── custom_device_test.cc │ │ │ ├── eager_executor.cc │ │ │ ├── eager_executor.h │ │ │ ├── eager_executor_test.cc │ │ │ ├── eager_op_rewrite_registry.cc │ │ │ ├── eager_op_rewrite_registry.h │ │ │ ├── eager_op_rewrite_registry_test.cc │ │ │ ├── eager_operation.cc │ │ │ ├── eager_operation.h │ │ │ ├── eager_operation_test.cc │ │ │ ├── execute.cc │ │ │ ├── execute.h │ │ │ ├── execute_node.cc │ │ │ ├── execute_node.h │ │ │ ├── execute_node_test.cc │ │ │ ├── execute_test.cc │ │ │ ├── kernel_and_device.cc │ │ │ ├── kernel_and_device.h │ │ │ ├── kernel_and_device_test.cc │ │ │ ├── mkl_eager_op_rewrite.cc │ │ │ ├── mkl_eager_op_rewrite_test.cc │ │ │ ├── placement_test.cc │ │ │ ├── placement_utils.cc │ │ │ ├── placement_utils.h │ │ │ ├── placement_utils_test.cc │ │ │ ├── rendezvous_cache.h │ │ │ ├── shape_inference.cc │ │ │ ├── shape_inference.h │ │ │ ├── small_constants_optimizer.cc │ │ │ ├── small_constants_optimizer.h │ │ │ ├── summary_optimizer.cc │ │ │ ├── summary_optimizer.h │ │ │ ├── summary_optimizer_test.cc │ │ │ ├── tensor_handle.cc │ │ │ ├── tensor_handle.h │ │ │ ├── tensor_handle_data.cc │ │ │ ├── tensor_handle_data.h │ │ │ ├── tensor_handle_data_test.cc │ │ │ ├── tensor_handle_test.cc │ │ │ └── zen_eager_op_rewrite.cc │ │ ├── entry.h │ │ ├── eval_const_tensor.cc │ │ ├── eval_const_tensor.h │ │ ├── eval_const_tensor_test.cc │ │ ├── executor.cc │ │ ├── executor.h │ │ ├── executor_factory.cc │ │ ├── executor_factory.h │ │ ├── executor_test.cc │ │ ├── function.cc │ │ ├── function.h │ │ ├── function_body.cc │ │ ├── function_body.h │ │ ├── function_def_utils.cc │ │ ├── function_def_utils.h │ │ ├── function_optimization_registration_test.cc │ │ ├── function_optimization_registry.cc │ │ ├── function_optimization_registry.h │ │ ├── function_optimization_registry_no_pass_test.cc │ │ ├── function_optimization_registry_pass_failure_test.cc │ │ ├── function_optimization_registry_test.cc │ │ ├── function_test.cc │ │ ├── function_testlib.cc │ │ ├── function_testlib.h │ │ ├── function_threadpool_test.cc │ │ ├── function_utils.cc │ │ ├── function_utils.h │ │ ├── gpu │ │ │ ├── BUILD │ │ │ ├── gpu_allocator_retry_test.cc │ │ │ ├── gpu_bfc_allocator.cc │ │ │ ├── gpu_bfc_allocator.h │ │ │ ├── gpu_bfc_allocator_test.cc │ │ │ ├── gpu_cudamalloc_allocator.cc │ │ │ ├── gpu_cudamalloc_allocator.h │ │ │ ├── gpu_debug_allocator.cc │ │ │ ├── gpu_debug_allocator.h │ │ │ ├── gpu_debug_allocator_test.cc │ │ │ ├── gpu_device.cc │ │ │ ├── gpu_device.h │ │ │ ├── gpu_device_factory.cc │ │ │ ├── gpu_device_on_non_gpu_machine_test.cc │ │ │ ├── gpu_device_test.cc │ │ │ ├── gpu_event_mgr.cc │ │ │ ├── gpu_event_mgr.h │ │ │ ├── gpu_id.h │ │ │ ├── gpu_id_manager.cc │ │ │ ├── gpu_id_manager.h │ │ │ ├── gpu_managed_allocator.cc │ │ │ ├── gpu_managed_allocator.h │ │ │ ├── gpu_process_state.cc │ │ │ ├── gpu_process_state.h │ │ │ ├── gpu_serving_device_selector.cc │ │ │ ├── gpu_serving_device_selector.h │ │ │ ├── gpu_serving_device_selector_test.cc │ │ │ ├── gpu_util.cc │ │ │ ├── gpu_util.h │ │ │ ├── gpu_util_platform_specific.cc │ │ │ ├── gpu_virtual_mem_allocator.cc │ │ │ ├── gpu_virtual_mem_allocator.h │ │ │ ├── gpu_virtual_mem_allocator_test.cc │ │ │ └── pool_allocator_test.cc │ │ ├── gpu_device_context.h │ │ ├── gradients.cc │ │ ├── gradients.h │ │ ├── graph_constructor.cc │ │ ├── graph_constructor.h │ │ ├── graph_constructor_fuzz.cc │ │ ├── graph_constructor_test.cc │ │ ├── graph_def_builder_util.cc │ │ ├── graph_def_builder_util.h │ │ ├── graph_execution_state.cc │ │ ├── graph_execution_state.h │ │ ├── graph_optimizer.cc │ │ ├── graph_optimizer.h │ │ ├── graph_runner.cc │ │ ├── graph_runner.h │ │ ├── graph_runner_test.cc │ │ ├── graph_view.cc │ │ ├── graph_view.h │ │ ├── hierarchical_tree_broadcaster.cc │ │ ├── hierarchical_tree_broadcaster.h │ │ ├── hierarchical_tree_broadcaster_test.cc │ │ ├── immutable_executor_state.cc │ │ ├── immutable_executor_state.h │ │ ├── inline_function_utils.cc │ │ ├── inline_function_utils.h │ │ ├── inline_function_utils_test.cc │ │ ├── input_colocation_exemption_registry.cc │ │ ├── input_colocation_exemption_registry.h │ │ ├── input_colocation_exemption_registry_test.cc │ │ ├── inspecting_placer.cc │ │ ├── inspecting_placer.h │ │ ├── int32_fulltype.cc │ │ ├── int32_fulltype.h │ │ ├── int32_fulltype_test.cc │ │ ├── isolate_placer_inspection_required_ops_pass.cc │ │ ├── isolate_placer_inspection_required_ops_pass.h │ │ ├── isolate_placer_inspection_required_ops_pass_test.cc │ │ ├── kernel_benchmark_testlib.cc │ │ ├── kernel_benchmark_testlib.h │ │ ├── layout_pass_util.cc │ │ ├── layout_pass_util.h │ │ ├── local_device.cc │ │ ├── local_device.h │ │ ├── local_executor_params.h │ │ ├── local_session_selection.cc │ │ ├── local_session_selection.h │ │ ├── lower_case_op.cc │ │ ├── lower_case_op.h │ │ ├── lower_case_op_test.cc │ │ ├── lower_function_call_inline_policy.cc │ │ ├── lower_function_call_inline_policy.h │ │ ├── lower_function_call_op.cc │ │ ├── lower_function_call_op.h │ │ ├── lower_function_call_op_test.cc │ │ ├── lower_functional_ops.cc │ │ ├── lower_functional_ops.h │ │ ├── lower_functional_ops_test.cc │ │ ├── lower_if_op.cc │ │ ├── lower_if_op.h │ │ ├── lower_if_op_test.cc │ │ ├── lower_while_op.cc │ │ ├── lower_while_op.h │ │ ├── lower_while_op_test.cc │ │ ├── memory_types.cc │ │ ├── memory_types.h │ │ ├── memory_types_test.cc │ │ ├── mkl_cpu_allocator.cc │ │ ├── mkl_cpu_allocator.h │ │ ├── mkl_cpu_allocator_test.cc │ │ ├── mkl_layout_pass.cc │ │ ├── mkl_layout_pass.h │ │ ├── mkl_threadpool_device_test.cc │ │ ├── next_pluggable_device │ │ │ ├── BUILD │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ ├── example_plugin.cc │ │ │ │ ├── example_plugin.h │ │ │ │ ├── outside_compilation_params.h │ │ │ │ ├── plugin_c_api.h │ │ │ │ ├── plugin_c_api_test.cc │ │ │ │ ├── test_next_pluggable_device_plugin.cc │ │ │ │ ├── tf_device_context_c_api.h │ │ │ │ ├── tf_device_context_c_api_helper.cc │ │ │ │ ├── tf_device_context_c_api_helper.h │ │ │ │ ├── tf_device_context_c_api_internal.cc │ │ │ │ ├── tf_device_context_c_api_internal.h │ │ │ │ ├── tf_rendezvous_c_api.h │ │ │ │ ├── tf_rendezvous_c_api_defn.h │ │ │ │ ├── tf_rendezvous_c_api_helper.cc │ │ │ │ ├── tf_rendezvous_c_api_helper.h │ │ │ │ ├── tf_rendezvous_c_api_internal.cc │ │ │ │ ├── tf_rendezvous_c_api_internal.h │ │ │ │ ├── tf_rendezvous_c_api_test.cc │ │ │ │ ├── tf_tensor_utils.cc │ │ │ │ └── tf_tensor_utils.h │ │ │ ├── c_plugin_coordination_service_agent.cc │ │ │ ├── c_plugin_coordination_service_agent.h │ │ │ ├── c_plugin_op_kernel.cc │ │ │ ├── c_plugin_op_kernel.h │ │ │ ├── c_plugin_variable.cc │ │ │ ├── c_plugin_variable.h │ │ │ ├── direct_plugin_coordination_service_agent.h │ │ │ ├── direct_plugin_op_kernel.cc │ │ │ ├── direct_plugin_op_kernel.h │ │ │ ├── direct_plugin_variable.cc │ │ │ ├── direct_plugin_variable.h │ │ │ ├── flags.cc │ │ │ ├── flags.h │ │ │ ├── next_pluggable_device.cc │ │ │ ├── next_pluggable_device.h │ │ │ ├── next_pluggable_device_allocator.cc │ │ │ ├── next_pluggable_device_allocator.h │ │ │ ├── next_pluggable_device_api.cc │ │ │ ├── next_pluggable_device_api.h │ │ │ ├── next_pluggable_device_context.cc │ │ │ ├── next_pluggable_device_context.h │ │ │ ├── next_pluggable_device_factory.cc │ │ │ ├── next_pluggable_device_factory.h │ │ │ ├── pjrt_compile_on_demand_op.cc │ │ │ ├── pjrt_compile_on_demand_op.h │ │ │ ├── plugin_coordination_service_agent.h │ │ │ ├── plugin_coordination_service_agent_helper.h │ │ │ ├── plugin_op_kernel.h │ │ │ ├── plugin_op_kernel_helper.h │ │ │ ├── plugin_resource.cc │ │ │ ├── plugin_resource.h │ │ │ ├── plugin_variable.h │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── no_op_cost_measurement.cc │ │ ├── no_op_cost_measurement.h │ │ ├── no_op_cost_measurement_test.cc │ │ ├── node_file_writer.cc │ │ ├── node_file_writer.h │ │ ├── null_request_cost_accessor.cc │ │ ├── null_request_cost_accessor.h │ │ ├── null_request_cost_accessor_test.cc │ │ ├── optimization_registry.cc │ │ ├── optimization_registry.h │ │ ├── optimization_registry_test.cc │ │ ├── optimize_cross_host_control_deps.cc │ │ ├── optimize_cross_host_control_deps.h │ │ ├── optimize_cross_host_control_deps_test.cc │ │ ├── optimize_function_graph_utils.cc │ │ ├── optimize_function_graph_utils.h │ │ ├── optimize_function_graph_utils_test.cc │ │ ├── optimized_function_graph_info.cc │ │ ├── optimized_function_graph_info.h │ │ ├── optimized_function_graph_info_test.cc │ │ ├── parallel_concat_optimizer.cc │ │ ├── partitioning_utils.cc │ │ ├── partitioning_utils.h │ │ ├── partitioning_utils_test.cc │ │ ├── pending_counts.h │ │ ├── pending_counts_test.cc │ │ ├── permuter.cc │ │ ├── permuter.h │ │ ├── permuter_test.cc │ │ ├── placer.cc │ │ ├── placer.h │ │ ├── placer_inspection_required_ops_utils.cc │ │ ├── placer_inspection_required_ops_utils.h │ │ ├── placer_inspection_required_ops_utils_test.cc │ │ ├── placer_test.cc │ │ ├── pluggable_device │ │ │ ├── BUILD │ │ │ ├── pluggable_device.cc │ │ │ ├── pluggable_device.h │ │ │ ├── pluggable_device_bfc_allocator.cc │ │ │ ├── pluggable_device_bfc_allocator.h │ │ │ ├── pluggable_device_context.cc │ │ │ ├── pluggable_device_context.h │ │ │ ├── pluggable_device_factory.cc │ │ │ ├── pluggable_device_factory.h │ │ │ ├── pluggable_device_init.cc │ │ │ ├── pluggable_device_init.h │ │ │ ├── pluggable_device_plugin_init.cc │ │ │ ├── pluggable_device_plugin_init.h │ │ │ ├── pluggable_device_process_state.cc │ │ │ ├── pluggable_device_process_state.h │ │ │ ├── pluggable_device_simple_allocator.cc │ │ │ ├── pluggable_device_simple_allocator.h │ │ │ ├── pluggable_device_util.cc │ │ │ └── pluggable_device_util.h │ │ ├── pool_allocator.cc │ │ ├── pool_allocator.h │ │ ├── process_function_library_runtime.cc │ │ ├── process_function_library_runtime.h │ │ ├── process_function_library_runtime_test.cc │ │ ├── process_state.cc │ │ ├── process_state.h │ │ ├── process_util.cc │ │ ├── process_util.h │ │ ├── process_util_test.cc │ │ ├── profile_handler.h │ │ ├── propagator_debug_utils.cc │ │ ├── propagator_debug_utils.h │ │ ├── propagator_state.cc │ │ ├── propagator_state.h │ │ ├── quantize_training.cc │ │ ├── quantize_training.h │ │ ├── quantize_training_test.cc │ │ ├── renamed_device.cc │ │ ├── renamed_device.h │ │ ├── rendezvous_mgr.cc │ │ ├── rendezvous_mgr.h │ │ ├── rendezvous_util.cc │ │ ├── rendezvous_util.h │ │ ├── rendezvous_util_test.cc │ │ ├── replicate_constants_pass.cc │ │ ├── replicate_constants_pass.h │ │ ├── replicate_constants_pass_test.cc │ │ ├── replicate_per_replica_nodes.cc │ │ ├── replicate_per_replica_nodes.h │ │ ├── replicate_per_replica_nodes_test.cc │ │ ├── request_cost.cc │ │ ├── request_cost.h │ │ ├── request_cost_accessor.h │ │ ├── request_cost_accessor_registry.cc │ │ ├── request_cost_accessor_registry.h │ │ ├── request_cost_accessor_registry_test.cc │ │ ├── request_cost_test.cc │ │ ├── ring_alg.cc │ │ ├── ring_alg.h │ │ ├── ring_gatherer.cc │ │ ├── ring_gatherer.h │ │ ├── ring_gatherer_test.cc │ │ ├── ring_reducer.cc │ │ ├── ring_reducer.h │ │ ├── ring_reducer_test.cc │ │ ├── scoped_allocator.cc │ │ ├── scoped_allocator.h │ │ ├── scoped_allocator_mgr.cc │ │ ├── scoped_allocator_mgr.h │ │ ├── scoped_allocator_mgr_test.cc │ │ ├── serving_device_selector.cc │ │ ├── serving_device_selector.h │ │ ├── serving_device_selector_policies.cc │ │ ├── serving_device_selector_policies.h │ │ ├── session.cc │ │ ├── session_factory.cc │ │ ├── session_factory.h │ │ ├── session_options.cc │ │ ├── session_state.cc │ │ ├── session_test.cc │ │ ├── shape_refiner.cc │ │ ├── shape_refiner.h │ │ ├── shape_refiner_test.cc │ │ ├── shared_counter.h │ │ ├── simple_propagator_state.cc │ │ ├── simple_propagator_state.h │ │ ├── single_threaded_cpu_device.cc │ │ ├── single_threaded_cpu_device.h │ │ ├── single_threaded_executor.cc │ │ ├── single_threaded_executor.h │ │ ├── single_threaded_executor_test.cc │ │ ├── stats_publisher_interface.cc │ │ ├── stats_publisher_interface.h │ │ ├── step_stats_collector.cc │ │ ├── step_stats_collector.h │ │ ├── test_collective_executor_mgr.h │ │ ├── testlib_ops.cc │ │ ├── threadpool_device.cc │ │ ├── threadpool_device.h │ │ ├── threadpool_device_factory.cc │ │ ├── threadpool_device_test.cc │ │ ├── type_inference.cc │ │ ├── type_inference.h │ │ ├── type_inference_test.cc │ │ └── zen_layout_pass.cc │ ├── config │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── flag_defs.cc │ │ ├── flag_defs.h │ │ ├── flags.cc │ │ ├── flags.h │ │ ├── flags.py │ │ ├── flags_api_wrapper.cc │ │ ├── flags_test.cc │ │ └── flags_test.py │ ├── data │ │ ├── BUILD │ │ ├── captured_function.cc │ │ ├── captured_function.h │ │ ├── compression_utils.cc │ │ ├── compression_utils.h │ │ ├── compression_utils_test.cc │ │ ├── dataset_test_base.cc │ │ ├── dataset_test_base.h │ │ ├── dataset_utils.cc │ │ ├── dataset_utils.h │ │ ├── dataset_utils_test.cc │ │ ├── file_logger_client_interface.h │ │ ├── file_logger_client_no_op.h │ │ ├── finalization_utils.cc │ │ ├── finalization_utils.h │ │ ├── hash_utils.cc │ │ ├── hash_utils.h │ │ ├── hash_utils_test.cc │ │ ├── metric_utils.cc │ │ ├── metric_utils.h │ │ ├── metric_utils_test.cc │ │ ├── name_utils.cc │ │ ├── name_utils.h │ │ ├── name_utils_test.cc │ │ ├── rewrite_utils.cc │ │ ├── rewrite_utils.h │ │ ├── rewrite_utils_test.cc │ │ ├── root_dataset.cc │ │ ├── root_dataset.h │ │ ├── serialization_utils.cc │ │ ├── serialization_utils.h │ │ ├── serialization_utils_test.cc │ │ ├── service │ │ │ ├── BUILD │ │ │ ├── auto_scaler.cc │ │ │ ├── auto_scaler.h │ │ │ ├── auto_scaler_test.cc │ │ │ ├── client │ │ │ │ ├── BUILD │ │ │ │ ├── common.h │ │ │ │ ├── data_service_client.cc │ │ │ │ ├── data_service_client.h │ │ │ │ ├── data_service_client_test.cc │ │ │ │ ├── utils.cc │ │ │ │ ├── utils.h │ │ │ │ ├── utils_test.cc │ │ │ │ ├── validate_utils.cc │ │ │ │ ├── validate_utils.h │ │ │ │ └── validate_utils_test.cc │ │ │ ├── common.cc │ │ │ ├── common.h │ │ │ ├── common.proto │ │ │ ├── common_test.cc │ │ │ ├── credentials_factory.cc │ │ │ ├── credentials_factory.h │ │ │ ├── credentials_factory_test.cc │ │ │ ├── cross_trainer_cache.h │ │ │ ├── cross_trainer_cache_test.cc │ │ │ ├── data_service_test.cc │ │ │ ├── data_transfer.cc │ │ │ ├── data_transfer.h │ │ │ ├── data_transfer_test.cc │ │ │ ├── dataset_store.cc │ │ │ ├── dataset_store.h │ │ │ ├── dataset_store_test.cc │ │ │ ├── dispatcher.proto │ │ │ ├── dispatcher_client.cc │ │ │ ├── dispatcher_client.h │ │ │ ├── dispatcher_client_test.cc │ │ │ ├── dispatcher_impl.cc │ │ │ ├── dispatcher_impl.h │ │ │ ├── dispatcher_state.cc │ │ │ ├── dispatcher_state.h │ │ │ ├── dispatcher_state_test.cc │ │ │ ├── export.proto │ │ │ ├── graph_rewriters.cc │ │ │ ├── graph_rewriters.h │ │ │ ├── graph_rewriters_test.cc │ │ │ ├── grpc_dispatcher_impl.cc │ │ │ ├── grpc_dispatcher_impl.h │ │ │ ├── grpc_dispatcher_impl_test.cc │ │ │ ├── grpc_util.cc │ │ │ ├── grpc_util.h │ │ │ ├── grpc_util_test.cc │ │ │ ├── grpc_worker_impl.cc │ │ │ ├── grpc_worker_impl.h │ │ │ ├── grpc_worker_impl_test.cc │ │ │ ├── journal.cc │ │ │ ├── journal.h │ │ │ ├── journal.proto │ │ │ ├── journal_test.cc │ │ │ ├── logging_utils.cc │ │ │ ├── logging_utils.h │ │ │ ├── logging_utils_test.cc │ │ │ ├── py_utils.cc │ │ │ ├── py_utils.h │ │ │ ├── server_lib.cc │ │ │ ├── server_lib.h │ │ │ ├── snapshot │ │ │ │ ├── BUILD │ │ │ │ ├── distributed_snapshot_test.cc │ │ │ │ ├── file_utils.cc │ │ │ │ ├── file_utils.h │ │ │ │ ├── file_utils_test.cc │ │ │ │ ├── path_utils.cc │ │ │ │ ├── path_utils.h │ │ │ │ ├── path_utils_test.cc │ │ │ │ ├── snapshot_chunk_dataset_op.cc │ │ │ │ ├── snapshot_manager.cc │ │ │ │ ├── snapshot_manager.h │ │ │ │ ├── snapshot_manager_test.cc │ │ │ │ ├── snapshot_split_provider.cc │ │ │ │ ├── snapshot_split_provider.h │ │ │ │ ├── snapshot_split_provider_test.cc │ │ │ │ ├── snapshot_stream_writer.cc │ │ │ │ ├── snapshot_stream_writer.h │ │ │ │ ├── snapshot_stream_writer_checkpoint_test.cc │ │ │ │ ├── snapshot_stream_writer_test.cc │ │ │ │ ├── test_utils.cc │ │ │ │ ├── test_utils.h │ │ │ │ ├── utils.cc │ │ │ │ ├── utils.h │ │ │ │ └── utils_test.cc │ │ │ ├── split_provider.cc │ │ │ ├── split_provider.h │ │ │ ├── task_remover.cc │ │ │ ├── task_remover.h │ │ │ ├── task_runner.cc │ │ │ ├── task_runner.h │ │ │ ├── task_runner_test.cc │ │ │ ├── test_cluster.cc │ │ │ ├── test_cluster.h │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ ├── test_util_test.cc │ │ │ ├── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── choose_from_datasets.pbtxt │ │ │ │ └── interleave_textline_dataset.pbtxt │ │ │ ├── thread_safe_buffer.h │ │ │ ├── thread_safe_buffer_test.cc │ │ │ ├── url.cc │ │ │ ├── url.h │ │ │ ├── url_test.cc │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ ├── utils_test.cc │ │ │ ├── validate_utils.cc │ │ │ ├── validate_utils.h │ │ │ ├── validate_utils_test.cc │ │ │ ├── worker.proto │ │ │ ├── worker_client.cc │ │ │ ├── worker_client.h │ │ │ ├── worker_client_test.cc │ │ │ ├── worker_impl.cc │ │ │ ├── worker_impl.h │ │ │ └── worker_impl_test.cc │ │ ├── snapshot_utils.cc │ │ ├── snapshot_utils.h │ │ ├── snapshot_utils_test.cc │ │ ├── split_utils.cc │ │ ├── split_utils.h │ │ ├── split_utils_test.cc │ │ ├── standalone.cc │ │ ├── standalone.h │ │ ├── standalone_save_restore_test.cc │ │ ├── standalone_test.cc │ │ ├── stats_utils.cc │ │ ├── stats_utils.h │ │ ├── test_utils.cc │ │ ├── test_utils.h │ │ ├── tfdataz_metrics.cc │ │ ├── tfdataz_metrics.h │ │ ├── tfdataz_metrics_test.cc │ │ ├── unbounded_thread_pool.cc │ │ ├── unbounded_thread_pool.h │ │ ├── unbounded_thread_pool_test.cc │ │ ├── utils.cc │ │ ├── utils.h │ │ └── utils_test.cc │ ├── debug │ │ ├── BUILD │ │ ├── bfc_dump_reader.cc │ │ ├── debug.cc │ │ ├── debug_callback_registry.cc │ │ ├── debug_callback_registry.h │ │ ├── debug_graph_utils.cc │ │ ├── debug_graph_utils.h │ │ ├── debug_graph_utils_test.cc │ │ ├── debug_grpc_io_utils_test.cc │ │ ├── debug_grpc_testlib.cc │ │ ├── debug_grpc_testlib.h │ │ ├── debug_io_utils.cc │ │ ├── debug_io_utils.h │ │ ├── debug_io_utils_test.cc │ │ ├── debug_node_key.cc │ │ ├── debug_node_key.h │ │ ├── debug_service.proto │ │ ├── debugger_event_metadata.proto │ │ ├── debugger_state_impl.cc │ │ ├── debugger_state_impl.h │ │ └── grpc_session_debug_test.cc │ ├── distributed_runtime │ │ ├── BUILD │ │ ├── README.md │ │ ├── base_rendezvous_mgr.cc │ │ ├── base_rendezvous_mgr.h │ │ ├── call_options.h │ │ ├── call_options_test.cc │ │ ├── cancellable_call.cc │ │ ├── cancellable_call.h │ │ ├── cluster_function_library_runtime.cc │ │ ├── cluster_function_library_runtime.h │ │ ├── cluster_function_library_runtime_test.cc │ │ ├── collective_param_resolver_distributed.cc │ │ ├── collective_param_resolver_distributed.h │ │ ├── collective_param_resolver_distributed_test.cc │ │ ├── collective_rma_distributed.cc │ │ ├── collective_rma_distributed.h │ │ ├── collective_rma_distributed_test.cc │ │ ├── coordination │ │ │ ├── BUILD │ │ │ ├── coordination_client.h │ │ │ ├── coordination_service_barrier_proxy.cc │ │ │ ├── coordination_service_barrier_proxy.h │ │ │ ├── coordination_service_barrier_proxy_test.cc │ │ │ ├── coordination_service_error_util.h │ │ │ └── coordination_service_rpc_handler.h │ │ ├── device_resolver_distributed.cc │ │ ├── device_resolver_distributed.h │ │ ├── device_resolver_distributed_test.cc │ │ ├── eager │ │ │ ├── BUILD │ │ │ ├── cluster_function_library_runtime.cc │ │ │ ├── cluster_function_library_runtime.h │ │ │ ├── destroy_tensor_handle_node.h │ │ │ ├── eager_client.h │ │ │ ├── eager_service_impl.cc │ │ │ ├── eager_service_impl.h │ │ │ ├── eager_service_impl_test.cc │ │ │ ├── remote_copy_node.cc │ │ │ ├── remote_copy_node.h │ │ │ ├── remote_execute_node.cc │ │ │ ├── remote_execute_node.h │ │ │ ├── remote_mgr.cc │ │ │ ├── remote_mgr.h │ │ │ ├── remote_mgr_test.cc │ │ │ ├── remote_tensor_handle.h │ │ │ ├── remote_tensor_handle_data.cc │ │ │ └── remote_tensor_handle_data.h │ │ ├── error_payloads.h │ │ ├── graph_mgr.cc │ │ ├── graph_mgr.h │ │ ├── integration_test │ │ │ ├── BUILD │ │ │ ├── c_api_coordination_test.cc │ │ │ ├── c_api_multi_client_function_test.cc │ │ │ ├── c_api_multi_client_test.cc │ │ │ ├── c_api_recoverable_jobs_test.cc │ │ │ ├── c_api_session_coordination_test.cc │ │ │ └── coordination_test_opkernel_registration.cc │ │ ├── local_master.cc │ │ ├── local_master.h │ │ ├── master.cc │ │ ├── master.h │ │ ├── master_env.h │ │ ├── master_interface.h │ │ ├── master_session.cc │ │ ├── master_session.h │ │ ├── master_test.cc │ │ ├── message_wrappers.cc │ │ ├── message_wrappers.h │ │ ├── message_wrappers_test.cc │ │ ├── partial_run_mgr.cc │ │ ├── partial_run_mgr.h │ │ ├── partial_run_mgr_test.cc │ │ ├── preemption │ │ │ ├── BUILD │ │ │ ├── check_preemption_op.cc │ │ │ ├── check_preemption_op_kernel.cc │ │ │ └── preemption_sync_manager.h │ │ ├── recent_request_ids.cc │ │ ├── recent_request_ids.h │ │ ├── recent_request_ids_test.cc │ │ ├── remote_device.cc │ │ ├── remote_device.h │ │ ├── remote_device_test.cc │ │ ├── rendezvous_mgr_interface.h │ │ ├── request_id.cc │ │ ├── request_id.h │ │ ├── request_id_test.cc │ │ ├── rpc │ │ │ ├── BUILD │ │ │ ├── coordination │ │ │ │ ├── BUILD │ │ │ │ ├── grpc_coordination_client.h │ │ │ │ └── grpc_coordination_service_impl.h │ │ │ ├── eager │ │ │ │ ├── BUILD │ │ │ │ ├── grpc_eager_client.cc │ │ │ │ ├── grpc_eager_client.h │ │ │ │ ├── grpc_eager_client_test.cc │ │ │ │ ├── grpc_eager_service.h │ │ │ │ ├── grpc_eager_service_impl.cc │ │ │ │ └── grpc_eager_service_impl.h │ │ │ ├── grpc_channel.h │ │ │ ├── grpc_client_cq_tag.h │ │ │ ├── grpc_master_service.cc │ │ │ ├── grpc_master_service.h │ │ │ ├── grpc_master_service_impl.cc │ │ │ ├── grpc_master_service_impl.h │ │ │ ├── grpc_remote_master.cc │ │ │ ├── grpc_remote_master.h │ │ │ ├── grpc_remote_worker.cc │ │ │ ├── grpc_remote_worker.h │ │ │ ├── grpc_server_lib.cc │ │ │ ├── grpc_server_lib.h │ │ │ ├── grpc_session.cc │ │ │ ├── grpc_session.h │ │ │ ├── grpc_session_test.cc │ │ │ ├── grpc_state.cc │ │ │ ├── grpc_state.h │ │ │ ├── grpc_tensor_coding.cc │ │ │ ├── grpc_tensor_coding.h │ │ │ ├── grpc_tensor_coding_test.cc │ │ │ ├── grpc_tensorflow_server.cc │ │ │ ├── grpc_testlib.cc │ │ │ ├── grpc_testlib.h │ │ │ ├── grpc_testlib_server.cc │ │ │ ├── grpc_util.cc │ │ │ ├── grpc_util.h │ │ │ ├── grpc_worker_cache.cc │ │ │ ├── grpc_worker_cache.h │ │ │ ├── grpc_worker_cache_test.cc │ │ │ ├── grpc_worker_service.cc │ │ │ ├── grpc_worker_service.h │ │ │ ├── grpc_worker_service_impl.cc │ │ │ ├── grpc_worker_service_impl.h │ │ │ ├── rpc_rendezvous_mgr.cc │ │ │ ├── rpc_rendezvous_mgr.h │ │ │ ├── rpc_rendezvous_mgr_test.cc │ │ │ ├── rpc_response_cache.cc │ │ │ └── rpc_response_cache.h │ │ ├── rpc_collective_executor_mgr.cc │ │ ├── rpc_collective_executor_mgr.h │ │ ├── rpc_collective_executor_mgr_test.cc │ │ ├── rpcbench_test.cc │ │ ├── scheduler.cc │ │ ├── scheduler.h │ │ ├── server_lib.cc │ │ ├── server_lib.h │ │ ├── server_lib_test.cc │ │ ├── session_mgr.cc │ │ ├── session_mgr.h │ │ ├── session_mgr_test.cc │ │ ├── tensor_coding.cc │ │ ├── tensor_coding.h │ │ ├── tensor_coding_test.cc │ │ ├── test_utils.h │ │ ├── worker.cc │ │ ├── worker.h │ │ ├── worker_cache.h │ │ ├── worker_cache_logger.cc │ │ ├── worker_cache_logger.h │ │ ├── worker_cache_partial.cc │ │ ├── worker_cache_partial.h │ │ ├── worker_cache_wrapper.h │ │ ├── worker_env.h │ │ ├── worker_interface.h │ │ ├── worker_session.cc │ │ └── worker_session.h │ ├── example │ │ ├── BUILD │ │ ├── example.proto │ │ ├── example_parser_configuration.cc │ │ ├── example_parser_configuration.h │ │ ├── example_parser_configuration.proto │ │ ├── example_parser_configuration_test.cc │ │ ├── feature.proto │ │ ├── feature_util.cc │ │ ├── feature_util.h │ │ ├── feature_util_test.cc │ │ └── testdata │ │ │ ├── BUILD │ │ │ └── parse_example_graph_def.pbtxt │ ├── framework │ │ ├── BUILD │ │ ├── allocation_description.proto │ │ ├── allocator.h │ │ ├── allocator_registry.h │ │ ├── allocator_test.cc │ │ ├── api_def.proto │ │ ├── attr_value.proto │ │ ├── attr_value_util.cc │ │ ├── attr_value_util.h │ │ ├── attr_value_util_test.cc │ │ ├── batch_util_test.cc │ │ ├── bfloat16.cc │ │ ├── bfloat16.h │ │ ├── bfloat16_test.cc │ │ ├── bounds_check.h │ │ ├── cancellation.h │ │ ├── collective.cc │ │ ├── collective.h │ │ ├── common_shape_fns.cc │ │ ├── common_shape_fns.h │ │ ├── common_shape_fns_test.cc │ │ ├── control_flow.h │ │ ├── cost_graph.proto │ │ ├── cpp_shape_inference.proto │ │ ├── dataset.cc │ │ ├── dataset.h │ │ ├── dataset.proto │ │ ├── dataset_metadata.proto │ │ ├── dataset_options.proto │ │ ├── dataset_stateful_op_allowlist.h │ │ ├── dataset_test.cc │ │ ├── device.cc │ │ ├── device.h │ │ ├── device_attributes.proto │ │ ├── device_base.cc │ │ ├── device_base.h │ │ ├── device_base_test.cc │ │ ├── device_factory.cc │ │ ├── device_factory.h │ │ ├── disable_jit_test.cc │ │ ├── fake_input.cc │ │ ├── fake_input.h │ │ ├── full_type.proto │ │ ├── full_type_inference_util.cc │ │ ├── full_type_inference_util.h │ │ ├── full_type_inference_util_test.cc │ │ ├── full_type_util.cc │ │ ├── full_type_util.h │ │ ├── full_type_util_test.cc │ │ ├── function.cc │ │ ├── function.h │ │ ├── function.proto │ │ ├── function_handle_cache.cc │ │ ├── function_handle_cache.h │ │ ├── function_test.cc │ │ ├── function_testlib.cc │ │ ├── function_testlib.h │ │ ├── graph.proto │ │ ├── graph_debug_info.proto │ │ ├── graph_def_util.cc │ │ ├── graph_def_util.h │ │ ├── graph_def_util_test.cc │ │ ├── graph_to_functiondef.cc │ │ ├── graph_to_functiondef.h │ │ ├── graph_to_functiondef_test.cc │ │ ├── graph_transfer_info.proto │ │ ├── kernel_def.proto │ │ ├── kernel_def_builder.cc │ │ ├── kernel_def_builder.h │ │ ├── kernel_def_builder_test.cc │ │ ├── kernel_def_util.cc │ │ ├── kernel_def_util.h │ │ ├── kernel_def_util_test.cc │ │ ├── kernel_shape_util.cc │ │ ├── kernel_shape_util.h │ │ ├── load_library.cc │ │ ├── local_rendezvous.cc │ │ ├── local_rendezvous.h │ │ ├── log_memory.cc │ │ ├── log_memory.h │ │ ├── log_memory.proto │ │ ├── logging.cc │ │ ├── logging.h │ │ ├── lookup_interface.cc │ │ ├── lookup_interface.h │ │ ├── memory_types.cc │ │ ├── memory_types.h │ │ ├── memory_types_test.cc │ │ ├── metrics.cc │ │ ├── metrics.h │ │ ├── metrics_test.cc │ │ ├── model.cc │ │ ├── model.h │ │ ├── model.proto │ │ ├── model_test.cc │ │ ├── node_def.proto │ │ ├── node_def_builder.cc │ │ ├── node_def_builder.h │ │ ├── node_def_builder_test.cc │ │ ├── node_def_util.cc │ │ ├── node_def_util.h │ │ ├── node_def_util_test.cc │ │ ├── node_properties.cc │ │ ├── node_properties.h │ │ ├── node_properties_test.cc │ │ ├── numeric_op.h │ │ ├── numeric_types.h │ │ ├── op.cc │ │ ├── op.h │ │ ├── op_compatibility_test.cc │ │ ├── op_def.proto │ │ ├── op_def_builder.cc │ │ ├── op_def_builder.h │ │ ├── op_def_builder_test.cc │ │ ├── op_def_util.cc │ │ ├── op_def_util.h │ │ ├── op_def_util_test.cc │ │ ├── op_gen_lib.cc │ │ ├── op_gen_lib.h │ │ ├── op_gen_lib_test.cc │ │ ├── op_kernel.cc │ │ ├── op_kernel.h │ │ ├── op_kernel_test.cc │ │ ├── op_kernel_test_base.h │ │ ├── op_registration_test.cc │ │ ├── op_requires.h │ │ ├── op_segment.cc │ │ ├── op_segment.h │ │ ├── op_segment_test.cc │ │ ├── ops_util.cc │ │ ├── ops_util.h │ │ ├── optimized_function_graph.proto │ │ ├── partial_tensor_shape.h │ │ ├── partial_tensor_shape_test.cc │ │ ├── queue_interface.h │ │ ├── reader_base.cc │ │ ├── reader_base.h │ │ ├── reader_base.proto │ │ ├── reader_interface.h │ │ ├── reader_op_kernel.h │ │ ├── ref_var.cc │ │ ├── ref_var.h │ │ ├── register_types.h │ │ ├── register_types_traits.h │ │ ├── registration │ │ │ ├── BUILD │ │ │ ├── options.h.tpl │ │ │ ├── registration.h │ │ │ └── registration_test.cc │ │ ├── rendezvous.cc │ │ ├── rendezvous.h │ │ ├── rendezvous_test.cc │ │ ├── resource_base.h │ │ ├── resource_handle.cc │ │ ├── resource_handle.h │ │ ├── resource_handle.proto │ │ ├── resource_handle_test.cc │ │ ├── resource_mgr.cc │ │ ├── resource_mgr.h │ │ ├── resource_mgr_test.cc │ │ ├── resource_op_kernel.h │ │ ├── resource_op_kernel_test.cc │ │ ├── resource_var.cc │ │ ├── resource_var.h │ │ ├── resource_var_test.cc │ │ ├── rng_alg.h │ │ ├── run_handler.cc │ │ ├── run_handler.h │ │ ├── run_handler_test.cc │ │ ├── run_handler_util.cc │ │ ├── run_handler_util.h │ │ ├── run_handler_util_test.cc │ │ ├── session_state.h │ │ ├── shape_inference.cc │ │ ├── shape_inference.h │ │ ├── shape_inference_test.cc │ │ ├── shape_inference_testutil.cc │ │ ├── shape_inference_testutil.h │ │ ├── shape_inference_testutil_test.cc │ │ ├── shared_ptr_variant.h │ │ ├── stats_aggregator.h │ │ ├── step_stats.proto │ │ ├── summary.proto │ │ ├── tensor.cc │ │ ├── tensor.h │ │ ├── tensor.proto │ │ ├── tensor_description.proto │ │ ├── tensor_fuzz.cc │ │ ├── tensor_key.h │ │ ├── tensor_matcher.cc │ │ ├── tensor_matcher.h │ │ ├── tensor_matcher_test.cc │ │ ├── tensor_reference.h │ │ ├── tensor_shape.cc │ │ ├── tensor_shape.h │ │ ├── tensor_shape.proto │ │ ├── tensor_shape_fuzz.cc │ │ ├── tensor_shape_test.cc │ │ ├── tensor_slice.cc │ │ ├── tensor_slice.h │ │ ├── tensor_slice.proto │ │ ├── tensor_slice_test.cc │ │ ├── tensor_test.cc │ │ ├── tensor_testutil.cc │ │ ├── tensor_testutil.h │ │ ├── tensor_testutil_test.cc │ │ ├── tensor_types.h │ │ ├── tensor_util.cc │ │ ├── tensor_util.h │ │ ├── tensor_util_test.cc │ │ ├── thread_factory.h │ │ ├── tracking_allocator.h │ │ ├── tracking_allocator_test.cc │ │ ├── type_index.h │ │ ├── type_traits.h │ │ ├── typed_allocator.cc │ │ ├── typed_allocator.h │ │ ├── types.cc │ │ ├── types.h │ │ ├── types.proto │ │ ├── types_test.cc │ │ ├── variable.proto │ │ ├── variant.cc │ │ ├── variant.h │ │ ├── variant_encode_decode.h │ │ ├── variant_op_copy_test.cc │ │ ├── variant_op_registry.cc │ │ ├── variant_op_registry.h │ │ ├── variant_op_registry_test.cc │ │ ├── variant_tensor_data.cc │ │ ├── variant_tensor_data.h │ │ ├── variant_test.cc │ │ ├── versions.cc │ │ ├── versions.h │ │ └── versions.proto │ ├── function │ │ ├── README.md │ │ ├── capture │ │ │ ├── BUILD │ │ │ ├── by_ref_capture_test.py │ │ │ ├── capture_container.py │ │ │ ├── capture_container_test.py │ │ │ ├── free_vars_detect.py │ │ │ ├── free_vars_detect_test.py │ │ │ └── restore_captures.py │ │ ├── integration_test │ │ │ ├── BUILD │ │ │ ├── side_inputs_manual_api_test.py │ │ │ └── side_inputs_test.py │ │ ├── polymorphism │ │ │ ├── BUILD │ │ │ ├── function_cache.py │ │ │ ├── function_cache_test.py │ │ │ ├── function_type.proto │ │ │ ├── function_type.py │ │ │ ├── function_type_test.py │ │ │ ├── type_dispatch.py │ │ │ └── type_dispatch_test.py │ │ ├── runtime_client │ │ │ ├── BUILD │ │ │ ├── runtime_client.cc │ │ │ ├── runtime_client.h │ │ │ ├── runtime_client.py │ │ │ ├── runtime_client_pybind.cc │ │ │ ├── runtime_client_pybind.pyi │ │ │ ├── runtime_client_test.cc │ │ │ └── runtime_client_test.py │ │ ├── testing │ │ │ ├── BUILD │ │ │ ├── test_pass.h │ │ │ ├── test_pass.py │ │ │ ├── test_pass_pybind.cc │ │ │ └── test_pass_pybind.pyi │ │ ├── trace_type │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── custom_nest_trace_type.py │ │ │ ├── custom_nest_trace_type_test.py │ │ │ ├── default_types.proto │ │ │ ├── default_types.py │ │ │ ├── default_types_test.py │ │ │ ├── serialization.proto │ │ │ ├── serialization.py │ │ │ ├── serialization_test.proto │ │ │ ├── serialization_test.py │ │ │ ├── trace_type_builder.py │ │ │ ├── trace_type_test.py │ │ │ └── util.py │ │ └── transform │ │ │ ├── BUILD │ │ │ ├── transform.py │ │ │ └── transform_test.py │ ├── graph │ │ ├── BUILD │ │ ├── algorithm.cc │ │ ├── algorithm.h │ │ ├── algorithm_test.cc │ │ ├── benchmark_testlib.h │ │ ├── collective_order.cc │ │ ├── collective_order.h │ │ ├── collective_order_test.cc │ │ ├── colors.cc │ │ ├── colors.h │ │ ├── control_flow.cc │ │ ├── control_flow.h │ │ ├── control_flow_test.cc │ │ ├── costmodel.cc │ │ ├── costmodel.h │ │ ├── costmodel_test.cc │ │ ├── default_device.h │ │ ├── edgeset.cc │ │ ├── edgeset.h │ │ ├── edgeset_test.cc │ │ ├── graph.cc │ │ ├── graph.h │ │ ├── graph_debug_info_builder.cc │ │ ├── graph_debug_info_builder.h │ │ ├── graph_debug_info_builder_test.cc │ │ ├── graph_def_builder.cc │ │ ├── graph_def_builder.h │ │ ├── graph_def_builder_test.cc │ │ ├── graph_node_util.cc │ │ ├── graph_node_util.h │ │ ├── graph_partition.cc │ │ ├── graph_partition.h │ │ ├── graph_partition_test.cc │ │ ├── graph_test.cc │ │ ├── mkl_graph_util.h │ │ ├── mkl_testlib.cc │ │ ├── mkl_testlib.h │ │ ├── node_builder.cc │ │ ├── node_builder.h │ │ ├── node_builder_test.cc │ │ ├── optimizer_cse.cc │ │ ├── optimizer_cse.h │ │ ├── optimizer_cse_test.cc │ │ ├── regularization │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── simple_delete.cc │ │ │ ├── simple_delete.h │ │ │ ├── simple_delete_test.cc │ │ │ ├── testdata │ │ │ │ ├── bert1 │ │ │ │ │ └── saved_model.pb │ │ │ │ └── bert2 │ │ │ │ │ └── saved_model.pb │ │ │ ├── util.cc │ │ │ ├── util.h │ │ │ └── util_test.cc │ │ ├── subgraph.cc │ │ ├── subgraph.h │ │ ├── subgraph_test.cc │ │ ├── tensor_id.cc │ │ ├── tensor_id.h │ │ ├── tensor_id_test.cc │ │ ├── testlib.cc │ │ ├── testlib.h │ │ ├── types.h │ │ ├── validate.cc │ │ ├── validate.h │ │ ├── validate_test.cc │ │ ├── while_context.cc │ │ ├── while_context.h │ │ └── zen_graph_util.h │ ├── grappler │ │ ├── BUILD │ │ ├── clusters │ │ │ ├── BUILD │ │ │ ├── cluster.cc │ │ │ ├── cluster.h │ │ │ ├── single_machine.cc │ │ │ ├── single_machine.h │ │ │ ├── single_machine_test.cc │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ ├── utils_test.cc │ │ │ ├── virtual_cluster.cc │ │ │ ├── virtual_cluster.h │ │ │ └── virtual_cluster_test.cc │ │ ├── costs │ │ │ ├── BUILD │ │ │ ├── analytical_cost_estimator.cc │ │ │ ├── analytical_cost_estimator.h │ │ │ ├── analytical_cost_estimator_test.cc │ │ │ ├── cost_estimator.cc │ │ │ ├── cost_estimator.h │ │ │ ├── cost_estimator_test.cc │ │ │ ├── graph_memory.cc │ │ │ ├── graph_memory.h │ │ │ ├── graph_memory_test.cc │ │ │ ├── graph_properties.cc │ │ │ ├── graph_properties.h │ │ │ ├── graph_properties_test.cc │ │ │ ├── graph_properties_testdata │ │ │ │ ├── BUILD │ │ │ │ ├── function_error.pbtxt │ │ │ │ ├── function_functional_while.pbtxt │ │ │ │ ├── function_switch.pbtxt │ │ │ │ ├── function_switch_2.pbtxt │ │ │ │ ├── function_switch_shapes.pbtxt │ │ │ │ ├── function_with_dt_resource_input.pbtxt │ │ │ │ ├── large_function_graph.pbtxt │ │ │ │ ├── large_graph.pbtxt.html │ │ │ │ ├── loops_and_queues.pbtxt │ │ │ │ ├── loops_and_resource_vars.pbtxt │ │ │ │ ├── merge_without_loops.pbtxt │ │ │ │ ├── nested_loop.pbtxt │ │ │ │ ├── queues_and_loops.pbtxt │ │ │ │ ├── simple_function.pbtxt │ │ │ │ ├── while_loop.pbtxt │ │ │ │ └── while_loop_var_handle_op.pbtxt │ │ │ ├── measuring_cost_estimator.cc │ │ │ ├── measuring_cost_estimator.h │ │ │ ├── op_context.h │ │ │ ├── op_level_cost_estimator.cc │ │ │ ├── op_level_cost_estimator.h │ │ │ ├── op_level_cost_estimator_test.cc │ │ │ ├── op_performance_data.proto │ │ │ ├── robust_stats.cc │ │ │ ├── robust_stats.h │ │ │ ├── robust_stats_test.cc │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ ├── utils_test.cc │ │ │ ├── virtual_placer.cc │ │ │ ├── virtual_placer.h │ │ │ ├── virtual_placer_test.cc │ │ │ ├── virtual_scheduler.cc │ │ │ ├── virtual_scheduler.h │ │ │ └── virtual_scheduler_test.cc │ │ ├── devices.cc │ │ ├── devices.h │ │ ├── graph_analyzer │ │ │ ├── BUILD │ │ │ ├── gen_node.cc │ │ │ ├── gen_node.h │ │ │ ├── gen_node_test.cc │ │ │ ├── graph_analyzer.cc │ │ │ ├── graph_analyzer.h │ │ │ ├── graph_analyzer_test.cc │ │ │ ├── graph_analyzer_tool.cc │ │ │ ├── graph_analyzer_tool.h │ │ │ ├── hash_tools.h │ │ │ ├── hash_tools_test.cc │ │ │ ├── map_tools.h │ │ │ ├── sig_node.cc │ │ │ ├── sig_node.h │ │ │ ├── sig_node_test.cc │ │ │ ├── subgraph.cc │ │ │ ├── subgraph.h │ │ │ ├── subgraph_test.cc │ │ │ ├── test_tools.cc │ │ │ └── test_tools.h │ │ ├── graph_topology_view.cc │ │ ├── graph_topology_view.h │ │ ├── graph_topology_view_test.cc │ │ ├── graph_view.cc │ │ ├── graph_view.h │ │ ├── graph_view_test.cc │ │ ├── grappler_item.cc │ │ ├── grappler_item.h │ │ ├── grappler_item_builder.cc │ │ ├── grappler_item_builder.h │ │ ├── grappler_item_builder_test.cc │ │ ├── grappler_item_test.cc │ │ ├── inputs │ │ │ ├── BUILD │ │ │ ├── file_input_yielder.cc │ │ │ ├── file_input_yielder.h │ │ │ ├── input_yielder.h │ │ │ ├── testdata │ │ │ │ ├── BUILD │ │ │ │ └── test_file.txt │ │ │ ├── trivial_test_graph_input_yielder.cc │ │ │ ├── trivial_test_graph_input_yielder.h │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ │ ├── mutable_graph_view.cc │ │ ├── mutable_graph_view.h │ │ ├── mutable_graph_view_test.cc │ │ ├── op_types.cc │ │ ├── op_types.h │ │ ├── optimizers │ │ │ ├── BUILD │ │ │ ├── arithmetic_optimizer.cc │ │ │ ├── arithmetic_optimizer.h │ │ │ ├── arithmetic_optimizer_test.cc │ │ │ ├── arithmetic_optimizer_test_utils.h │ │ │ ├── auto_mixed_precision.cc │ │ │ ├── auto_mixed_precision.h │ │ │ ├── auto_mixed_precision_lists.h │ │ │ ├── auto_mixed_precision_test.cc │ │ │ ├── auto_parallel.cc │ │ │ ├── auto_parallel.h │ │ │ ├── auto_parallel_test.cc │ │ │ ├── common_subgraph_elimination.cc │ │ │ ├── common_subgraph_elimination.h │ │ │ ├── common_subgraph_elimination_test.cc │ │ │ ├── constant_folding.cc │ │ │ ├── constant_folding.h │ │ │ ├── constant_folding_test.cc │ │ │ ├── custom_graph_optimizer.h │ │ │ ├── custom_graph_optimizer_registry.cc │ │ │ ├── custom_graph_optimizer_registry.h │ │ │ ├── custom_graph_optimizer_registry_test.cc │ │ │ ├── data │ │ │ │ ├── BUILD │ │ │ │ ├── auto_shard.cc │ │ │ │ ├── auto_shard.h │ │ │ │ ├── auto_shard_test.cc │ │ │ │ ├── autotune_buffer_sizes.cc │ │ │ │ ├── autotune_buffer_sizes.h │ │ │ │ ├── autotune_buffer_sizes_test.cc │ │ │ │ ├── batch_parallelization.cc │ │ │ │ ├── batch_parallelization.h │ │ │ │ ├── batch_parallelization_test.cc │ │ │ │ ├── disable_intra_op_parallelism.cc │ │ │ │ ├── disable_intra_op_parallelism.h │ │ │ │ ├── disable_intra_op_parallelism_test.cc │ │ │ │ ├── disable_prefetch_legacy_autotune.cc │ │ │ │ ├── disable_prefetch_legacy_autotune.h │ │ │ │ ├── disable_prefetch_legacy_autotune_test.cc │ │ │ │ ├── enable_gradient_descent.cc │ │ │ │ ├── enable_gradient_descent.h │ │ │ │ ├── enable_gradient_descent_test.cc │ │ │ │ ├── filter_fusion.cc │ │ │ │ ├── filter_fusion.h │ │ │ │ ├── filter_fusion_test.cc │ │ │ │ ├── filter_parallelization.cc │ │ │ │ ├── filter_parallelization.h │ │ │ │ ├── filter_parallelization_test.cc │ │ │ │ ├── function_utils.cc │ │ │ │ ├── function_utils.h │ │ │ │ ├── function_utils_test.cc │ │ │ │ ├── fusion_utils.cc │ │ │ │ ├── fusion_utils.h │ │ │ │ ├── fusion_utils_test.cc │ │ │ │ ├── graph_test_utils.cc │ │ │ │ ├── graph_test_utils.h │ │ │ │ ├── graph_utils.cc │ │ │ │ ├── graph_utils.h │ │ │ │ ├── graph_utils_test.cc │ │ │ │ ├── inject_io_prefetch.cc │ │ │ │ ├── inject_io_prefetch.h │ │ │ │ ├── inject_io_prefetch_test.cc │ │ │ │ ├── inject_prefetch.cc │ │ │ │ ├── inject_prefetch.h │ │ │ │ ├── inject_prefetch_test.cc │ │ │ │ ├── make_deterministic.cc │ │ │ │ ├── make_deterministic.h │ │ │ │ ├── make_deterministic_test.cc │ │ │ │ ├── make_sloppy.cc │ │ │ │ ├── make_sloppy.h │ │ │ │ ├── make_sloppy_test.cc │ │ │ │ ├── map_and_batch_fusion.cc │ │ │ │ ├── map_and_batch_fusion.h │ │ │ │ ├── map_and_batch_fusion_test.cc │ │ │ │ ├── map_and_filter_fusion.cc │ │ │ │ ├── map_and_filter_fusion.h │ │ │ │ ├── map_and_filter_fusion_test.cc │ │ │ │ ├── map_fusion.cc │ │ │ │ ├── map_fusion.h │ │ │ │ ├── map_fusion_test.cc │ │ │ │ ├── map_parallelization.cc │ │ │ │ ├── map_parallelization.h │ │ │ │ ├── map_parallelization_test.cc │ │ │ │ ├── meta_optimizer.cc │ │ │ │ ├── meta_optimizer.h │ │ │ │ ├── noop_elimination.cc │ │ │ │ ├── noop_elimination.h │ │ │ │ ├── noop_elimination_test.cc │ │ │ │ ├── optimizer_base.cc │ │ │ │ ├── optimizer_base.h │ │ │ │ ├── parallel_batch.cc │ │ │ │ ├── parallel_batch.h │ │ │ │ ├── parallel_batch_test.cc │ │ │ │ ├── remove_compression_map.cc │ │ │ │ ├── remove_compression_map.h │ │ │ │ ├── remove_compression_map_test.cc │ │ │ │ ├── replicate_on_split.cc │ │ │ │ ├── replicate_on_split.h │ │ │ │ ├── replicate_on_split_test.cc │ │ │ │ ├── shuffle_and_repeat_fusion.cc │ │ │ │ ├── shuffle_and_repeat_fusion.h │ │ │ │ ├── shuffle_and_repeat_fusion_test.cc │ │ │ │ ├── slack.cc │ │ │ │ ├── slack.h │ │ │ │ ├── slack_test.cc │ │ │ │ ├── split_utils.cc │ │ │ │ ├── split_utils.h │ │ │ │ ├── split_utils_test.cc │ │ │ │ ├── use_private_thread_pool.cc │ │ │ │ ├── use_private_thread_pool.h │ │ │ │ └── use_private_thread_pool_test.cc │ │ │ ├── debug_stripper.cc │ │ │ ├── debug_stripper.h │ │ │ ├── debug_stripper_test.cc │ │ │ ├── dependency_optimizer.cc │ │ │ ├── dependency_optimizer.h │ │ │ ├── dependency_optimizer_test.cc │ │ │ ├── evaluation_utils.cc │ │ │ ├── evaluation_utils.h │ │ │ ├── evaluation_utils_test.cc │ │ │ ├── function_api_info.cc │ │ │ ├── function_api_info.h │ │ │ ├── function_api_info_test.cc │ │ │ ├── function_optimizer.cc │ │ │ ├── function_optimizer.h │ │ │ ├── function_optimizer_test.cc │ │ │ ├── generic_layout_optimizer.cc │ │ │ ├── generic_layout_optimizer.h │ │ │ ├── generic_layout_optimizer_test.cc │ │ │ ├── generic_layout_optimizer_transposer.cc │ │ │ ├── generic_layout_optimizer_transposer.h │ │ │ ├── generic_layout_optimizer_transposer_factory.cc │ │ │ ├── generic_layout_optimizer_transposer_factory.h │ │ │ ├── generic_layout_optimizer_transposer_factory_test.cc │ │ │ ├── generic_layout_optimizer_transposer_test.cc │ │ │ ├── gpu_swapping_kernels.cc │ │ │ ├── gpu_swapping_ops.cc │ │ │ ├── graph_optimizer.h │ │ │ ├── graph_optimizer_stage.cc │ │ │ ├── graph_optimizer_stage.h │ │ │ ├── graph_optimizer_stage_test.cc │ │ │ ├── implementation_selector.cc │ │ │ ├── implementation_selector.h │ │ │ ├── implementation_selector_test.cc │ │ │ ├── inference │ │ │ │ ├── BUILD │ │ │ │ ├── batch_op_rewriter.cc │ │ │ │ ├── batch_op_rewriter.h │ │ │ │ ├── batch_op_rewriter.proto │ │ │ │ └── batch_op_rewriter_test.cc │ │ │ ├── loop_optimizer.cc │ │ │ ├── loop_optimizer.h │ │ │ ├── loop_optimizer_test.cc │ │ │ ├── memory_optimizer.cc │ │ │ ├── memory_optimizer.h │ │ │ ├── memory_optimizer_test.cc │ │ │ ├── meta_optimizer.cc │ │ │ ├── meta_optimizer.h │ │ │ ├── meta_optimizer_test.cc │ │ │ ├── mkl_remapper_test.cc │ │ │ ├── model_pruner.cc │ │ │ ├── model_pruner.h │ │ │ ├── model_pruner_test.cc │ │ │ ├── pin_to_host_optimizer.cc │ │ │ ├── pin_to_host_optimizer.h │ │ │ ├── pin_to_host_optimizer_test.cc │ │ │ ├── remapper.cc │ │ │ ├── remapper.h │ │ │ ├── remapper_test.cc │ │ │ ├── scoped_allocator_optimizer.cc │ │ │ ├── scoped_allocator_optimizer.h │ │ │ ├── scoped_allocator_optimizer_test.cc │ │ │ ├── shape_optimizer.cc │ │ │ ├── shape_optimizer.h │ │ │ ├── shape_optimizer_test.cc │ │ │ ├── static_schedule.cc │ │ │ ├── static_schedule.h │ │ │ ├── static_schedule_test.cc │ │ │ ├── tfg_optimizer_hook.cc │ │ │ ├── tfg_optimizer_hook.h │ │ │ ├── tfg_optimizer_hook_test.cc │ │ │ ├── tfg_passes_builder.cc │ │ │ └── tfg_passes_builder.h │ │ ├── utils.cc │ │ ├── utils.h │ │ ├── utils │ │ │ ├── BUILD │ │ │ ├── canonicalizer.cc │ │ │ ├── canonicalizer.h │ │ │ ├── canonicalizer_test.cc │ │ │ ├── colocation.cc │ │ │ ├── colocation.h │ │ │ ├── colocation_test.cc │ │ │ ├── frame.cc │ │ │ ├── frame.h │ │ │ ├── frame_test.cc │ │ │ ├── functions.cc │ │ │ ├── functions.h │ │ │ ├── functions_test.cc │ │ │ ├── graph_view.cc │ │ │ ├── graph_view.h │ │ │ ├── graph_view_internal.h │ │ │ ├── graph_view_internal_test.cc │ │ │ ├── graph_view_test.cc │ │ │ ├── grappler_test.cc │ │ │ ├── grappler_test.h │ │ │ ├── grappler_test_test.cc │ │ │ ├── pattern_utils.cc │ │ │ ├── pattern_utils.h │ │ │ ├── pattern_utils_test.cc │ │ │ ├── scc.cc │ │ │ ├── scc.h │ │ │ ├── scc_test.cc │ │ │ ├── symbolic_shapes.cc │ │ │ ├── symbolic_shapes.h │ │ │ ├── symbolic_shapes_test.cc │ │ │ ├── topological_sort.cc │ │ │ ├── topological_sort.h │ │ │ ├── topological_sort_test.cc │ │ │ ├── tpu.cc │ │ │ ├── tpu.h │ │ │ ├── tpu_test.cc │ │ │ ├── transitive_fanin.cc │ │ │ ├── transitive_fanin.h │ │ │ ├── transitive_fanin_test.cc │ │ │ ├── traversal.cc │ │ │ ├── traversal.h │ │ │ └── traversal_test.cc │ │ ├── utils_test.cc │ │ └── verifiers │ │ │ ├── BUILD │ │ │ ├── graph_verifier.h │ │ │ ├── structure_verifier.cc │ │ │ ├── structure_verifier.h │ │ │ └── structure_verifier_test.cc │ ├── ir │ │ ├── BUILD │ │ ├── README.md │ │ ├── dialect.h │ │ ├── dialect.td │ │ ├── importexport │ │ │ ├── BUILD │ │ │ ├── convert_attributes.cc │ │ │ ├── convert_attributes.h │ │ │ ├── convert_tensor.cc │ │ │ ├── convert_tensor.h │ │ │ ├── convert_types.cc │ │ │ ├── convert_types.h │ │ │ ├── functiondef_export.cc │ │ │ ├── functiondef_export.h │ │ │ ├── functiondef_import.cc │ │ │ ├── functiondef_import.h │ │ │ ├── graphdef_export.cc │ │ │ ├── graphdef_export.h │ │ │ ├── graphdef_import.cc │ │ │ ├── graphdef_import.h │ │ │ ├── load_proto.cc │ │ │ ├── load_proto.h │ │ │ ├── mangling.cc │ │ │ ├── mangling.h │ │ │ ├── parse_text_proto.cc │ │ │ ├── parse_text_proto.h │ │ │ ├── savedmodel_export.cc │ │ │ ├── savedmodel_export.h │ │ │ ├── savedmodel_import.cc │ │ │ ├── savedmodel_import.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── graphdef_to_mlir │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── arg_as_control_ret.pbtxt │ │ │ │ │ ├── backedge_segment.pbtxt │ │ │ │ │ ├── empty.pbtxt │ │ │ │ │ ├── error_during_backedge.pbtxt │ │ │ │ │ ├── import_case_with_attr_inference.pbtxt │ │ │ │ │ ├── import_if_with_attr_inference.pbtxt │ │ │ │ │ ├── import_iterator_get_next_attr_inference.pbtxt │ │ │ │ │ ├── import_underscore_output_shapes.pbtxt │ │ │ │ │ ├── import_while_with_attr_inference.pbtxt │ │ │ │ │ ├── infeed_dequeue.pbtxt │ │ │ │ │ ├── infer_arg_handle_type.pbtxt │ │ │ │ │ ├── infer_with_output_shapes.pbtxt │ │ │ │ │ ├── invalid_arg_name.pbtxt │ │ │ │ │ ├── invalid_backedge_input_size.pbtxt │ │ │ │ │ ├── invalid_duplicated_node_name.pbtxt │ │ │ │ │ ├── invalid_edge_index.pbtxt │ │ │ │ │ ├── invalid_edge_name.pbtxt │ │ │ │ │ ├── invalid_empty_attr_key.pbtxt │ │ │ │ │ ├── invalid_empty_func_attr_key.pbtxt │ │ │ │ │ ├── invalid_empty_func_attr_name.pbtxt │ │ │ │ │ ├── invalid_empty_op_type.pbtxt │ │ │ │ │ ├── invalid_func_with_empty_name.pbtxt │ │ │ │ │ ├── invalid_function_import.pbtxt │ │ │ │ │ ├── invalid_generic_func_with_empty_control_result.pbtxt │ │ │ │ │ ├── invalid_generic_func_with_empty_input.pbtxt │ │ │ │ │ ├── invalid_generic_func_with_empty_name.pbtxt │ │ │ │ │ ├── invalid_generic_func_with_empty_result.pbtxt │ │ │ │ │ ├── invalid_generic_function_attr_name.pbtxt │ │ │ │ │ ├── invalid_generic_function_named_edge_index.pbtxt │ │ │ │ │ ├── invalid_handle_data.pbtxt │ │ │ │ │ ├── invalid_missing_control_input.pbtxt │ │ │ │ │ ├── invalid_missing_control_result.pbtxt │ │ │ │ │ ├── invalid_missing_control_result_value.pbtxt │ │ │ │ │ ├── invalid_missing_data_result.pbtxt │ │ │ │ │ ├── invalid_missing_data_result_value.pbtxt │ │ │ │ │ ├── invalid_missing_input.pbtxt │ │ │ │ │ ├── invalid_missing_two_inputs.pbtxt │ │ │ │ │ ├── invalid_named_edge_index.pbtxt │ │ │ │ │ ├── invalid_op_name.pbtxt │ │ │ │ │ ├── invalid_type_list.pbtxt │ │ │ │ │ ├── legacy_call.pbtxt │ │ │ │ │ ├── negative_shape.pbtxt │ │ │ │ │ ├── negative_zero_constant.pbtxt │ │ │ │ │ ├── three_nodes_with_attrs.pbtxt │ │ │ │ │ └── version.pbtxt │ │ │ │ ├── mlir_to_graphdef │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── empty.mlir │ │ │ │ │ ├── fulltype.mlir │ │ │ │ │ ├── func_with_no_args_or_results.mlir │ │ │ │ │ ├── negative_zero_constant.mlir │ │ │ │ │ ├── nested_legacy_call.mlir │ │ │ │ │ ├── three_nodes_with_attrs.mlir │ │ │ │ │ └── version.mlir │ │ │ │ ├── roundtrip │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add.pbtxt │ │ │ │ │ ├── arg-as-fetch.pbtxt │ │ │ │ │ ├── arg-control-dep.pbtxt │ │ │ │ │ ├── arg-data-type.pbtxt │ │ │ │ │ ├── arg-retval-attrs.pbtxt │ │ │ │ │ ├── case_op.pbtxt │ │ │ │ │ ├── const-values.pbtxt │ │ │ │ │ ├── device-arg-retval-attr.pbtxt │ │ │ │ │ ├── empty-input-shapes.pbtxt │ │ │ │ │ ├── empty-value-attr.pbtxt │ │ │ │ │ ├── error-message-with-source-info.pbtxt │ │ │ │ │ ├── error-message-with-source-info.pbtxt.debug │ │ │ │ │ ├── error-message-with-source-info.pbtxt.fake_py.debug │ │ │ │ │ ├── feed-as-fetch.pbtxt │ │ │ │ │ ├── feed-control-dep.pbtxt │ │ │ │ │ ├── force_shared_name_for_resource_ops.pbtxt │ │ │ │ │ ├── fulltype.pbtxt │ │ │ │ │ ├── func_with_no_args_or_results.pbtxt │ │ │ │ │ ├── function-func-attr.pbtxt │ │ │ │ │ ├── functional-if-ops.pbtxt │ │ │ │ │ ├── functional-while-ops.pbtxt │ │ │ │ │ ├── graph-as-function-control-ret.pbtxt │ │ │ │ │ ├── graph-as-function-retval-of-arg.pbtxt │ │ │ │ │ ├── graph-as-function.pbtxt │ │ │ │ │ ├── graph-custom-operation.pbtxt │ │ │ │ │ ├── graph-default-attr.pbtxt │ │ │ │ │ ├── graph-device-retval.pbtxt │ │ │ │ │ ├── graph-empty-tensor-content.pbtxt │ │ │ │ │ ├── graph-func-attr.pbtxt │ │ │ │ │ ├── graph-function-call.pbtxt │ │ │ │ │ ├── graph-function-control-ret-diff-island.pbtxt │ │ │ │ │ ├── graph-function-control-ret-same-island.pbtxt │ │ │ │ │ ├── graph-function-defs.pbtxt │ │ │ │ │ ├── graph-function-input-shapes.pbtxt │ │ │ │ │ ├── graph-function-resource-args.pbtxt │ │ │ │ │ ├── graph-input-func-arg-name-collision.pbtxt │ │ │ │ │ ├── graph-library.pbtxt │ │ │ │ │ ├── graph-scalar-input.pbtxt │ │ │ │ │ ├── graph-uint8-return.pbtxt │ │ │ │ │ ├── graph-undefined-output.pbtxt │ │ │ │ │ ├── graph-version-info.pbtxt │ │ │ │ │ ├── graph-while-loop.pbtxt │ │ │ │ │ ├── import_legacy_call_as_generic_func.pbtxt │ │ │ │ │ ├── invalid-output-index.pbtxt │ │ │ │ │ ├── large_constants.pbtxt │ │ │ │ │ ├── merge_node_with_function.pbtxt │ │ │ │ │ ├── mlir_passthrough_op.pbtxt │ │ │ │ │ ├── multi-output-feeds.pbtxt │ │ │ │ │ ├── multiple-use-next-iteration.pbtxt │ │ │ │ │ ├── negative_zero_constant.pbtxt │ │ │ │ │ ├── node-locations.pbtxt │ │ │ │ │ ├── output-shapes-attr.pbtxt │ │ │ │ │ ├── parse_example.pbtxt │ │ │ │ │ ├── parse_example_v2.pbtxt │ │ │ │ │ ├── partial-device-name.pbtxt │ │ │ │ │ ├── prune_unused_nodes.pbtxt │ │ │ │ │ ├── quint8-const.pbtxt │ │ │ │ │ ├── roundtrip.bzl │ │ │ │ │ ├── roundtrip.cc │ │ │ │ │ ├── roundtrip.h │ │ │ │ │ ├── shape-attrs.pbtxt │ │ │ │ │ ├── stateful-attribute.pbtxt │ │ │ │ │ ├── string-attr.pbtxt │ │ │ │ │ ├── switch_n.pbtxt │ │ │ │ │ ├── target.pbtxt │ │ │ │ │ ├── tensor-list.pbtxt │ │ │ │ │ ├── test.pbtxt │ │ │ │ │ ├── test10.pbtxt │ │ │ │ │ ├── test11.pbtxt │ │ │ │ │ ├── test12.pbtxt │ │ │ │ │ ├── test13.pbtxt │ │ │ │ │ ├── test14.pbtxt │ │ │ │ │ ├── test15.pbtxt │ │ │ │ │ ├── test16.pbtxt │ │ │ │ │ ├── test17.pbtxt │ │ │ │ │ ├── test18.pbtxt │ │ │ │ │ ├── test19.pbtxt │ │ │ │ │ ├── test2.pbtxt │ │ │ │ │ ├── test20.pbtxt │ │ │ │ │ ├── test21.pbtxt │ │ │ │ │ ├── test22.pbtxt │ │ │ │ │ ├── test23.pbtxt │ │ │ │ │ ├── test24.pbtxt │ │ │ │ │ ├── test25.pbtxt │ │ │ │ │ ├── test26.pbtxt │ │ │ │ │ ├── test27.pbtxt │ │ │ │ │ ├── test3.pbtxt │ │ │ │ │ ├── test4.pbtxt │ │ │ │ │ ├── test5.pbtxt │ │ │ │ │ ├── test6.pbtxt │ │ │ │ │ ├── test7.pbtxt │ │ │ │ │ ├── test8.pbtxt │ │ │ │ │ ├── test9.pbtxt │ │ │ │ │ └── verify_roundtrip.cc │ │ │ │ └── saved_model │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── saved_model_roundtrip_test.cc │ │ │ │ │ ├── savedmodel_v1 │ │ │ │ │ └── saved_model.pb │ │ │ │ │ └── savedmodel_v2 │ │ │ │ │ └── saved_model.pb │ │ │ └── tfg-translate.cc │ │ ├── interfaces.cc │ │ ├── interfaces.h │ │ ├── interfaces.td │ │ ├── interfaces_test.cc │ │ ├── ops.cc │ │ ├── ops.h │ │ ├── ops.td │ │ ├── ops_test.cc │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── attributes.mlir │ │ │ ├── canonicalize.mlir │ │ │ ├── compatible_types.mlir │ │ │ ├── concrete-ops.mlir │ │ │ ├── generic_concrete_ops.mlir │ │ │ ├── invalid-concrete-ops.mlir │ │ │ ├── invalid-preserved-attrs.mlir │ │ │ ├── invalid.mlir │ │ │ ├── invalid_types.mlir │ │ │ ├── ops.mlir │ │ │ ├── region-invalid-ops.mlir │ │ │ ├── region-ops-graph.mlir │ │ │ ├── region-ops.mlir │ │ │ ├── tfg-opt-no-passes.cc │ │ │ └── types.mlir │ │ ├── tf_op_names.cc │ │ ├── tf_op_names.inc │ │ ├── tf_op_registry.cc │ │ ├── tf_op_registry.h │ │ ├── tf_op_registry_test.cc │ │ ├── tf_op_wrapper.cc │ │ ├── tf_op_wrapper.h │ │ ├── tf_op_wrapper_test.cc │ │ ├── types │ │ │ ├── BUILD │ │ │ ├── attributes.td │ │ │ ├── dialect.cc │ │ │ ├── dialect.h │ │ │ ├── dialect.td │ │ │ ├── dialect_test.cc │ │ │ ├── types.def │ │ │ └── types.td │ │ ├── utility.cc │ │ ├── utility.h │ │ ├── utility_test.cc │ │ └── utils │ │ │ ├── shape_inference_utils.cc │ │ │ ├── shape_inference_utils.h │ │ │ └── shape_inference_utils_test.cc │ ├── kernels │ │ ├── BUILD │ │ ├── aggregate_ops.cc │ │ ├── aggregate_ops.h │ │ ├── aggregate_ops_cpu.h │ │ ├── aggregate_ops_gpu.cu.cc │ │ ├── argmax_op.cc │ │ ├── argmax_op.h │ │ ├── argmax_op_gpu.cu.cc │ │ ├── as_string_op.cc │ │ ├── as_string_op_test.cc │ │ ├── assign_op.h │ │ ├── autotune_conv_impl.h │ │ ├── avgpooling_op.cc │ │ ├── avgpooling_op.h │ │ ├── avgpooling_op_gpu.cu.cc │ │ ├── barrier_ops.cc │ │ ├── base64_ops.cc │ │ ├── basic_ops_benchmark_test.cc │ │ ├── batch_kernel_test_util.cc │ │ ├── batch_kernel_test_util.h │ │ ├── batch_kernels.cc │ │ ├── batch_kernels.h │ │ ├── batch_kernels_env_test.cc │ │ ├── batch_kernels_test.cc │ │ ├── batch_norm_op.cc │ │ ├── batch_norm_op.h │ │ ├── batch_norm_op_gpu.cu.cc │ │ ├── batch_norm_op_test.cc │ │ ├── batch_util.h │ │ ├── batching_util │ │ │ ├── BUILD │ │ │ ├── adaptive_shared_batch_scheduler.h │ │ │ ├── adaptive_shared_batch_scheduler_test.cc │ │ │ ├── basic_batch_scheduler.h │ │ │ ├── basic_batch_scheduler_benchmark_test.cc │ │ │ ├── basic_batch_scheduler_test.cc │ │ │ ├── batch_input_task.h │ │ │ ├── batch_input_task_test.cc │ │ │ ├── batch_resource_base.cc │ │ │ ├── batch_resource_base.h │ │ │ ├── batch_resource_base_test.cc │ │ │ ├── batch_scheduler.h │ │ │ ├── batch_scheduler_test.cc │ │ │ ├── bounded_executor.cc │ │ │ ├── bounded_executor.h │ │ │ ├── bounded_executor_test.cc │ │ │ ├── concat_split_util.h │ │ │ ├── fake_clock_env.cc │ │ │ ├── fake_clock_env.h │ │ │ ├── input_split_metadata.cc │ │ │ ├── input_split_metadata.h │ │ │ ├── input_split_metadata_test.cc │ │ │ ├── periodic_function.cc │ │ │ ├── periodic_function.h │ │ │ ├── periodic_function_test.cc │ │ │ ├── serial_device_batch_scheduler.h │ │ │ ├── serial_device_batch_scheduler_test.cc │ │ │ ├── shared_batch_scheduler.h │ │ │ ├── shared_batch_scheduler_test.cc │ │ │ ├── threadsafe_status.cc │ │ │ ├── threadsafe_status.h │ │ │ ├── threadsafe_status_test.cc │ │ │ ├── warmup.cc │ │ │ └── warmup.h │ │ ├── batchtospace_op.cc │ │ ├── bcast_ops.cc │ │ ├── betainc_op.cc │ │ ├── betainc_op.h │ │ ├── betainc_op_gpu.cu.cc │ │ ├── bias_op.cc │ │ ├── bias_op.h │ │ ├── bias_op_gpu.cu.cc │ │ ├── bias_op_gpu.h │ │ ├── bias_op_test.cc │ │ ├── bincount_op.cc │ │ ├── bincount_op.h │ │ ├── bincount_op_gpu.cu.cc │ │ ├── bincount_op_test.cc │ │ ├── broadcast_to_op.cc │ │ ├── broadcast_to_op.h │ │ ├── broadcast_to_op_gpu.cu.cc │ │ ├── broadcast_to_op_test.cc │ │ ├── bucketize_op.cc │ │ ├── bucketize_op.h │ │ ├── bucketize_op_gpu.cu.cc │ │ ├── candidate_sampler_ops.cc │ │ ├── cast_op.cc │ │ ├── cast_op.h │ │ ├── cast_op_gpu.cu.cc │ │ ├── cast_op_impl.h │ │ ├── cast_op_impl_bfloat.cc │ │ ├── cast_op_impl_bool.cc │ │ ├── cast_op_impl_complex128.cc │ │ ├── cast_op_impl_complex64.cc │ │ ├── cast_op_impl_double.cc │ │ ├── cast_op_impl_float.cc │ │ ├── cast_op_impl_float8.cc │ │ ├── cast_op_impl_half.cc │ │ ├── cast_op_impl_int16.cc │ │ ├── cast_op_impl_int32.cc │ │ ├── cast_op_impl_int4.cc │ │ ├── cast_op_impl_int64.cc │ │ ├── cast_op_impl_int8.cc │ │ ├── cast_op_impl_uint16.cc │ │ ├── cast_op_impl_uint32.cc │ │ ├── cast_op_impl_uint4.cc │ │ ├── cast_op_impl_uint64.cc │ │ ├── cast_op_impl_uint8.cc │ │ ├── cast_op_test.cc │ │ ├── check_numerics_op.cc │ │ ├── check_numerics_op_gpu.cu.cc │ │ ├── checkpoint_callback_manager.cc │ │ ├── checkpoint_callback_manager.h │ │ ├── checkpoint_callback_manager_test.cc │ │ ├── clustering_ops.cc │ │ ├── clustering_ops_test.cc │ │ ├── collective_nccl.cc │ │ ├── collective_nccl.h │ │ ├── collective_nccl_all_to_all.cc │ │ ├── collective_nccl_all_to_all.h │ │ ├── collective_nccl_broadcaster.cc │ │ ├── collective_nccl_broadcaster.h │ │ ├── collective_nccl_gatherer.cc │ │ ├── collective_nccl_gatherer.h │ │ ├── collective_nccl_reducer.cc │ │ ├── collective_nccl_reducer.h │ │ ├── collective_nccl_test.cc │ │ ├── collective_ops.cc │ │ ├── composite_tensor_ops.cc │ │ ├── composite_tensor_variant.cc │ │ ├── composite_tensor_variant.h │ │ ├── composite_tensor_variant_test.cc │ │ ├── concat_lib.h │ │ ├── concat_lib_cpu.cc │ │ ├── concat_lib_cpu.h │ │ ├── concat_lib_gpu.cc │ │ ├── concat_lib_gpu.h │ │ ├── concat_lib_gpu_impl.cu.cc │ │ ├── concat_op.cc │ │ ├── concat_op_test.cc │ │ ├── conditional_accumulator.h │ │ ├── conditional_accumulator_base.cc │ │ ├── conditional_accumulator_base.h │ │ ├── conditional_accumulator_base_op.cc │ │ ├── conditional_accumulator_base_op.h │ │ ├── conditional_accumulator_op.cc │ │ ├── constant_op.cc │ │ ├── constant_op.h │ │ ├── constant_op_test.cc │ │ ├── control_flow_ops.cc │ │ ├── control_flow_ops.h │ │ ├── control_flow_ops_test.cc │ │ ├── conv_2d.h │ │ ├── conv_2d_gpu.h │ │ ├── conv_2d_gpu_bfloat16.cu.cc │ │ ├── conv_2d_gpu_double.cu.cc │ │ ├── conv_2d_gpu_float.cu.cc │ │ ├── conv_2d_gpu_half.cu.cc │ │ ├── conv_2d_gpu_int.cu.cc │ │ ├── conv_2d_gpu_int_spatial_convolution.cu.cc │ │ ├── conv_2d_gpu_int_spatial_convolution_backward.cu.cc │ │ ├── conv_2d_gpu_uint16.cu.cc │ │ ├── conv_2d_gpu_uint32.cu.cc │ │ ├── conv_2d_gpu_uint64.cu.cc │ │ ├── conv_2d_gpu_uint8.cu.cc │ │ ├── conv_3d.h │ │ ├── conv_grad_filter_ops.cc │ │ ├── conv_grad_filter_ops_3d.cc │ │ ├── conv_grad_filter_ops_benchmark_test.cc │ │ ├── conv_grad_filter_ops_launcher.cc │ │ ├── conv_grad_input_ops.cc │ │ ├── conv_grad_input_ops.h │ │ ├── conv_grad_input_ops_3d.cc │ │ ├── conv_grad_input_ops_benchmark_test.cc │ │ ├── conv_grad_input_ops_bfloat16.cc │ │ ├── conv_grad_input_ops_double.cc │ │ ├── conv_grad_input_ops_float.cc │ │ ├── conv_grad_input_ops_half.cc │ │ ├── conv_grad_input_ops_int32.cc │ │ ├── conv_grad_ops.h │ │ ├── conv_grad_shape_utils.cc │ │ ├── conv_grad_shape_utils.h │ │ ├── conv_ops.cc │ │ ├── conv_ops.h │ │ ├── conv_ops_3d.cc │ │ ├── conv_ops_benchmark_test.cc │ │ ├── conv_ops_bfloat16.cc │ │ ├── conv_ops_double.cc │ │ ├── conv_ops_float.cc │ │ ├── conv_ops_fused_double.cc │ │ ├── conv_ops_fused_float.cc │ │ ├── conv_ops_fused_half.cc │ │ ├── conv_ops_fused_image_transform.cc │ │ ├── conv_ops_fused_impl.h │ │ ├── conv_ops_fused_int8.cc │ │ ├── conv_ops_gpu.cc │ │ ├── conv_ops_gpu.h │ │ ├── conv_ops_half.cc │ │ ├── conv_ops_impl.h │ │ ├── conv_ops_int32.cc │ │ ├── conv_ops_test.cc │ │ ├── conv_ops_using_gemm.cc │ │ ├── count_ops.cc │ │ ├── count_ops_test.cc │ │ ├── count_up_to_op.cc │ │ ├── cross_op.cc │ │ ├── cross_op.h │ │ ├── cross_op_gpu.cu.cc │ │ ├── cross_op_test.cc │ │ ├── ctc_decoder_ops.cc │ │ ├── ctc_loss_op.cc │ │ ├── cudnn_pooling_gpu.cc │ │ ├── cudnn_pooling_gpu.h │ │ ├── cudnn_rnn_ops.cc │ │ ├── cwise_op_abs.cc │ │ ├── cwise_op_acos.cc │ │ ├── cwise_op_acosh.cc │ │ ├── cwise_op_add_1.cc │ │ ├── cwise_op_add_2.cc │ │ ├── cwise_op_arg.cc │ │ ├── cwise_op_asin.cc │ │ ├── cwise_op_asinh.cc │ │ ├── cwise_op_atan.cc │ │ ├── cwise_op_atan2.cc │ │ ├── cwise_op_atanh.cc │ │ ├── cwise_op_bitwise_and.cc │ │ ├── cwise_op_bitwise_or.cc │ │ ├── cwise_op_bitwise_xor.cc │ │ ├── cwise_op_ceil.cc │ │ ├── cwise_op_clip.cc │ │ ├── cwise_op_clip.h │ │ ├── cwise_op_clip_gpu.cu.cc │ │ ├── cwise_op_complex.cc │ │ ├── cwise_op_conj.cc │ │ ├── cwise_op_cos.cc │ │ ├── cwise_op_cosh.cc │ │ ├── cwise_op_digamma.cc │ │ ├── cwise_op_div.cc │ │ ├── cwise_op_equal_to_1.cc │ │ ├── cwise_op_equal_to_2.cc │ │ ├── cwise_op_erf.cc │ │ ├── cwise_op_erfc.cc │ │ ├── cwise_op_exp.cc │ │ ├── cwise_op_expm1.cc │ │ ├── cwise_op_floor.cc │ │ ├── cwise_op_floor_div.cc │ │ ├── cwise_op_floor_mod.cc │ │ ├── cwise_op_gpu_abs.cu.cc │ │ ├── cwise_op_gpu_acos.cu.cc │ │ ├── cwise_op_gpu_acosh.cu.cc │ │ ├── cwise_op_gpu_add.cu.cc │ │ ├── cwise_op_gpu_arg.cu.cc │ │ ├── cwise_op_gpu_asin.cu.cc │ │ ├── cwise_op_gpu_asinh.cu.cc │ │ ├── cwise_op_gpu_atan.cu.cc │ │ ├── cwise_op_gpu_atan2.cu.cc │ │ ├── cwise_op_gpu_atanh.cu.cc │ │ ├── cwise_op_gpu_bitwise_and.cu.cc │ │ ├── cwise_op_gpu_bitwise_or.cu.cc │ │ ├── cwise_op_gpu_bitwise_xor.cu.cc │ │ ├── cwise_op_gpu_ceil.cu.cc │ │ ├── cwise_op_gpu_complex.cu.cc │ │ ├── cwise_op_gpu_conj.cu.cc │ │ ├── cwise_op_gpu_cos.cu.cc │ │ ├── cwise_op_gpu_cosh.cu.cc │ │ ├── cwise_op_gpu_digamma.cu.cc │ │ ├── cwise_op_gpu_div_shard1.cu.cc │ │ ├── cwise_op_gpu_div_shard2.cu.cc │ │ ├── cwise_op_gpu_div_shard3.cu.cc │ │ ├── cwise_op_gpu_div_shard4.cu.cc │ │ ├── cwise_op_gpu_equal_to.cu.cc │ │ ├── cwise_op_gpu_erf.cu.cc │ │ ├── cwise_op_gpu_erfc.cu.cc │ │ ├── cwise_op_gpu_exp.cu.cc │ │ ├── cwise_op_gpu_expm1.cu.cc │ │ ├── cwise_op_gpu_floor.cu.cc │ │ ├── cwise_op_gpu_floor_div.cu.cc │ │ ├── cwise_op_gpu_floor_mod.cu.cc │ │ ├── cwise_op_gpu_greater.cu.cc │ │ ├── cwise_op_gpu_greater_equal.cu.cc │ │ ├── cwise_op_gpu_igammas_double.cu.cc │ │ ├── cwise_op_gpu_igammas_float.cu.cc │ │ ├── cwise_op_gpu_imag.cu.cc │ │ ├── cwise_op_gpu_inverse.cu.cc │ │ ├── cwise_op_gpu_invert.cu.cc │ │ ├── cwise_op_gpu_isfinite.cu.cc │ │ ├── cwise_op_gpu_isinf.cu.cc │ │ ├── cwise_op_gpu_isnan.cu.cc │ │ ├── cwise_op_gpu_left_shift.cu.cc │ │ ├── cwise_op_gpu_less.cu.cc │ │ ├── cwise_op_gpu_less_equal.cu.cc │ │ ├── cwise_op_gpu_lgamma.cu.cc │ │ ├── cwise_op_gpu_log.cu.cc │ │ ├── cwise_op_gpu_log1p.cu.cc │ │ ├── cwise_op_gpu_logical_and.cu.cc │ │ ├── cwise_op_gpu_logical_not.cu.cc │ │ ├── cwise_op_gpu_logical_or.cu.cc │ │ ├── cwise_op_gpu_maximum.cu.cc │ │ ├── cwise_op_gpu_minimum.cu.cc │ │ ├── cwise_op_gpu_mod.cu.cc │ │ ├── cwise_op_gpu_mul_shard0.cu.cc │ │ ├── cwise_op_gpu_mul_shard1.cu.cc │ │ ├── cwise_op_gpu_mul_shard2.cu.cc │ │ ├── cwise_op_gpu_neg.cu.cc │ │ ├── cwise_op_gpu_not_equal_to.cu.cc │ │ ├── cwise_op_gpu_polygamma.cu.cc │ │ ├── cwise_op_gpu_pow.cu.cc │ │ ├── cwise_op_gpu_random_grad.cu.cc │ │ ├── cwise_op_gpu_real.cu.cc │ │ ├── cwise_op_gpu_right_shift.cu.cc │ │ ├── cwise_op_gpu_rint.cu.cc │ │ ├── cwise_op_gpu_round.cu.cc │ │ ├── cwise_op_gpu_rsqrt.cu.cc │ │ ├── cwise_op_gpu_select.cu.cc │ │ ├── cwise_op_gpu_sigmoid.cu.cc │ │ ├── cwise_op_gpu_sign.cu.cc │ │ ├── cwise_op_gpu_sin.cu.cc │ │ ├── cwise_op_gpu_sinh.cu.cc │ │ ├── cwise_op_gpu_sqrt.cu.cc │ │ ├── cwise_op_gpu_square.cu.cc │ │ ├── cwise_op_gpu_squared_difference.cu.cc │ │ ├── cwise_op_gpu_sub.cu.cc │ │ ├── cwise_op_gpu_tan.cu.cc │ │ ├── cwise_op_gpu_tanh.cu.cc │ │ ├── cwise_op_gpu_xdivy.cu.cc │ │ ├── cwise_op_gpu_xlog1py.cu.cc │ │ ├── cwise_op_gpu_xlogy.cu.cc │ │ ├── cwise_op_gpu_zeta.cu.cc │ │ ├── cwise_op_greater.cc │ │ ├── cwise_op_greater_equal.cc │ │ ├── cwise_op_igammas.cc │ │ ├── cwise_op_imag.cc │ │ ├── cwise_op_invert.cc │ │ ├── cwise_op_isfinite.cc │ │ ├── cwise_op_isinf.cc │ │ ├── cwise_op_isnan.cc │ │ ├── cwise_op_leakyrelu.cc │ │ ├── cwise_op_left_shift.cc │ │ ├── cwise_op_less.cc │ │ ├── cwise_op_less_equal.cc │ │ ├── cwise_op_lgamma.cc │ │ ├── cwise_op_log.cc │ │ ├── cwise_op_log1p.cc │ │ ├── cwise_op_logical_and.cc │ │ ├── cwise_op_logical_not.cc │ │ ├── cwise_op_logical_or.cc │ │ ├── cwise_op_maximum.cc │ │ ├── cwise_op_minimum.cc │ │ ├── cwise_op_mod.cc │ │ ├── cwise_op_mul_1.cc │ │ ├── cwise_op_mul_2.cc │ │ ├── cwise_op_ndtri.cc │ │ ├── cwise_op_ndtri_double.cu.cc │ │ ├── cwise_op_ndtri_float.cu.cc │ │ ├── cwise_op_neg_1.cc │ │ ├── cwise_op_neg_2.cc │ │ ├── cwise_op_not_equal_to_1.cc │ │ ├── cwise_op_not_equal_to_2.cc │ │ ├── cwise_op_pow.cc │ │ ├── cwise_op_random_grad.cc │ │ ├── cwise_op_real.cc │ │ ├── cwise_op_reciprocal.cc │ │ ├── cwise_op_right_shift.cc │ │ ├── cwise_op_rint.cc │ │ ├── cwise_op_round.cc │ │ ├── cwise_op_rsqrt.cc │ │ ├── cwise_op_select.cc │ │ ├── cwise_op_sigmoid.cc │ │ ├── cwise_op_sign.cc │ │ ├── cwise_op_sin.cc │ │ ├── cwise_op_sinh.cc │ │ ├── cwise_op_sqrt.cc │ │ ├── cwise_op_square.cc │ │ ├── cwise_op_squared_difference.cc │ │ ├── cwise_op_sub.cc │ │ ├── cwise_op_tan.cc │ │ ├── cwise_op_tanh.cc │ │ ├── cwise_op_xdivy.cc │ │ ├── cwise_op_xlog1py.cc │ │ ├── cwise_op_xlogy.cc │ │ ├── cwise_op_zeta.cc │ │ ├── cwise_ops.h │ │ ├── cwise_ops_common.cc │ │ ├── cwise_ops_common.h │ │ ├── cwise_ops_gpu_common.cu.h │ │ ├── cwise_ops_gpu_gradients.cu.h │ │ ├── cwise_ops_gradients.h │ │ ├── cwise_ops_test.cc │ │ ├── data │ │ │ ├── BUILD │ │ │ ├── batch_dataset_op.cc │ │ │ ├── batch_dataset_op.h │ │ │ ├── batch_dataset_op_test.cc │ │ │ ├── cache_dataset_ops.cc │ │ │ ├── cache_dataset_ops.h │ │ │ ├── cache_dataset_ops_test.cc │ │ │ ├── cache_ops.cc │ │ │ ├── cache_ops.h │ │ │ ├── concatenate_dataset_op.cc │ │ │ ├── concatenate_dataset_op.h │ │ │ ├── concatenate_dataset_op_test.cc │ │ │ ├── dataset_ops.cc │ │ │ ├── dataset_ops.h │ │ │ ├── experimental │ │ │ │ ├── BUILD │ │ │ │ ├── assert_cardinality_dataset_op.cc │ │ │ │ ├── assert_cardinality_dataset_op.h │ │ │ │ ├── assert_next_dataset_op.cc │ │ │ │ ├── assert_next_dataset_op.h │ │ │ │ ├── assert_next_dataset_op_test.cc │ │ │ │ ├── assert_prev_dataset_op.cc │ │ │ │ ├── assert_prev_dataset_op.h │ │ │ │ ├── assert_prev_dataset_op_test.cc │ │ │ │ ├── auto_shard_dataset_op.cc │ │ │ │ ├── auto_shard_dataset_op.h │ │ │ │ ├── auto_shard_dataset_op_test.cc │ │ │ │ ├── choose_fastest_branch_dataset_op.cc │ │ │ │ ├── choose_fastest_dataset_op.cc │ │ │ │ ├── compression_ops.cc │ │ │ │ ├── compression_ops.h │ │ │ │ ├── compute_batch_size_op.cc │ │ │ │ ├── csv_dataset_op.cc │ │ │ │ ├── data_service_dataset_op.cc │ │ │ │ ├── data_service_dataset_op.h │ │ │ │ ├── data_service_ops.cc │ │ │ │ ├── data_service_ops.h │ │ │ │ ├── dense_to_sparse_batch_dataset_op.cc │ │ │ │ ├── directed_interleave_dataset_op.cc │ │ │ │ ├── directed_interleave_dataset_op.h │ │ │ │ ├── directed_interleave_dataset_op_test.cc │ │ │ │ ├── distributed_save_op.cc │ │ │ │ ├── distributed_save_op.h │ │ │ │ ├── group_by_reducer_dataset_op.cc │ │ │ │ ├── group_by_window_dataset_op.cc │ │ │ │ ├── ignore_errors_dataset_op.cc │ │ │ │ ├── list_dataset_op.cc │ │ │ │ ├── list_dataset_op.h │ │ │ │ ├── list_dataset_op_test.cc │ │ │ │ ├── lmdb_dataset_op.cc │ │ │ │ ├── lmdb_dataset_op.h │ │ │ │ ├── load_dataset_op.cc │ │ │ │ ├── load_dataset_op.h │ │ │ │ ├── lookup_ops.cc │ │ │ │ ├── map_and_batch_dataset_op.cc │ │ │ │ ├── map_and_batch_dataset_op.h │ │ │ │ ├── map_and_batch_dataset_op_test.cc │ │ │ │ ├── matching_files_dataset_op.cc │ │ │ │ ├── non_serializable_dataset_op.cc │ │ │ │ ├── parallel_interleave_dataset_op.cc │ │ │ │ ├── parallel_interleave_dataset_op.h │ │ │ │ ├── parallel_interleave_dataset_op_test.cc │ │ │ │ ├── parse_example_dataset_op.cc │ │ │ │ ├── prefetching_kernels.cc │ │ │ │ ├── random_access_ops.cc │ │ │ │ ├── random_access_ops.h │ │ │ │ ├── random_dataset_op.cc │ │ │ │ ├── random_dataset_op.h │ │ │ │ ├── random_dataset_op_test.cc │ │ │ │ ├── rebatch_dataset_op.cc │ │ │ │ ├── sampling_dataset_op.cc │ │ │ │ ├── sampling_dataset_op.h │ │ │ │ ├── sampling_dataset_op_test.cc │ │ │ │ ├── save_dataset_op.cc │ │ │ │ ├── save_dataset_op.h │ │ │ │ ├── save_dataset_op_test.cc │ │ │ │ ├── scan_dataset_op.cc │ │ │ │ ├── set_stats_aggregator_dataset_op.cc │ │ │ │ ├── sleep_dataset_op.cc │ │ │ │ ├── sliding_window_dataset_op.cc │ │ │ │ ├── snapshot_dataset_op.cc │ │ │ │ ├── snapshot_dataset_op.h │ │ │ │ ├── sql │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── driver_manager.cc │ │ │ │ │ ├── driver_manager.h │ │ │ │ │ ├── query_connection.h │ │ │ │ │ ├── sqlite_query_connection.cc │ │ │ │ │ └── sqlite_query_connection.h │ │ │ │ ├── sql_dataset_op.cc │ │ │ │ ├── stats_aggregator_ops.cc │ │ │ │ ├── stats_dataset_ops.cc │ │ │ │ ├── take_while_dataset_op.cc │ │ │ │ ├── threadpool_dataset_op.cc │ │ │ │ ├── threadpool_dataset_op.h │ │ │ │ ├── to_tf_record_op.cc │ │ │ │ ├── unbatch_dataset_op.cc │ │ │ │ ├── unique_dataset_op.cc │ │ │ │ ├── unique_dataset_op.h │ │ │ │ └── unique_dataset_op_test.cc │ │ │ ├── filter_dataset_op.cc │ │ │ ├── filter_dataset_op.h │ │ │ ├── filter_dataset_op_test.cc │ │ │ ├── finalize_dataset_op.cc │ │ │ ├── finalize_dataset_op.h │ │ │ ├── finalize_dataset_op_test.cc │ │ │ ├── fixed_length_record_dataset_op.cc │ │ │ ├── fixed_length_record_dataset_op.h │ │ │ ├── fixed_length_record_dataset_op_test.cc │ │ │ ├── flat_map_dataset_op.cc │ │ │ ├── flat_map_dataset_op.h │ │ │ ├── flat_map_dataset_op_test.cc │ │ │ ├── generator_dataset_op.cc │ │ │ ├── generator_dataset_op.h │ │ │ ├── get_options_op.cc │ │ │ ├── get_options_op.h │ │ │ ├── get_options_op_test.cc │ │ │ ├── interleave_dataset_op.cc │ │ │ ├── interleave_dataset_op.h │ │ │ ├── interleave_dataset_op_test.cc │ │ │ ├── iterator_ops.cc │ │ │ ├── iterator_ops.h │ │ │ ├── iterator_ops_test.cc │ │ │ ├── map_dataset_op.cc │ │ │ ├── map_dataset_op.h │ │ │ ├── map_dataset_op_test.cc │ │ │ ├── map_defun_op.cc │ │ │ ├── map_defun_op.h │ │ │ ├── map_defun_op_test.cc │ │ │ ├── model_dataset_op.cc │ │ │ ├── model_dataset_op.h │ │ │ ├── multi_device_iterator_ops.cc │ │ │ ├── optimize_dataset_op.cc │ │ │ ├── optimize_dataset_op.h │ │ │ ├── optimize_dataset_op_test.cc │ │ │ ├── optional_ops.cc │ │ │ ├── optional_ops.cu.cc │ │ │ ├── optional_ops.h │ │ │ ├── optional_ops_util.cc │ │ │ ├── optional_ops_util.h │ │ │ ├── options_dataset_op.cc │ │ │ ├── options_dataset_op.h │ │ │ ├── options_dataset_op_test.cc │ │ │ ├── padded_batch_dataset_op.cc │ │ │ ├── padded_batch_dataset_op.h │ │ │ ├── padded_batch_dataset_op_test.cc │ │ │ ├── parallel_batch_dataset_op.cc │ │ │ ├── parallel_batch_dataset_op.h │ │ │ ├── parallel_batch_dataset_op_test.cc │ │ │ ├── parallel_filter_dataset_op.cc │ │ │ ├── parallel_filter_dataset_op.h │ │ │ ├── parallel_filter_dataset_op_test.cc │ │ │ ├── parallel_interleave_dataset_op.cc │ │ │ ├── parallel_interleave_dataset_op.h │ │ │ ├── parallel_interleave_dataset_op_test.cc │ │ │ ├── parallel_map_dataset_op.cc │ │ │ ├── parallel_map_dataset_op.h │ │ │ ├── parallel_map_dataset_op_test.cc │ │ │ ├── prefetch_autotuner.cc │ │ │ ├── prefetch_autotuner.h │ │ │ ├── prefetch_autotuner_test.cc │ │ │ ├── prefetch_dataset_op.cc │ │ │ ├── prefetch_dataset_op.h │ │ │ ├── prefetch_dataset_op_test.cc │ │ │ ├── random_seed_ops.cc │ │ │ ├── random_seed_ops.h │ │ │ ├── range_dataset_op.cc │ │ │ ├── range_dataset_op.h │ │ │ ├── range_dataset_op_test.cc │ │ │ ├── reduce_dataset_op.cc │ │ │ ├── reduce_dataset_op.h │ │ │ ├── reduce_dataset_op_test.cc │ │ │ ├── repeat_dataset_op.cc │ │ │ ├── repeat_dataset_op.h │ │ │ ├── repeat_dataset_op_test.cc │ │ │ ├── rewrite_dataset_op.cc │ │ │ ├── rewrite_dataset_op.h │ │ │ ├── rewrite_dataset_op_test.cc │ │ │ ├── shard_dataset_op.cc │ │ │ ├── shard_dataset_op.h │ │ │ ├── shard_dataset_op_test.cc │ │ │ ├── shuffle_dataset_op.cc │ │ │ ├── shuffle_dataset_op.h │ │ │ ├── shuffle_dataset_op_test.cc │ │ │ ├── skip_dataset_op.cc │ │ │ ├── skip_dataset_op.h │ │ │ ├── skip_dataset_op_test.cc │ │ │ ├── sparse_tensor_slice_dataset_op.cc │ │ │ ├── sparse_tensor_slice_dataset_op_test.cc │ │ │ ├── take_dataset_op.cc │ │ │ ├── take_dataset_op.h │ │ │ ├── take_dataset_op_test.cc │ │ │ ├── tensor_dataset_op.cc │ │ │ ├── tensor_dataset_op.h │ │ │ ├── tensor_dataset_op_test.cc │ │ │ ├── tensor_slice_dataset_op.cc │ │ │ ├── tensor_slice_dataset_op.h │ │ │ ├── tensor_slice_dataset_op_test.cc │ │ │ ├── text_line_dataset_op.cc │ │ │ ├── text_line_dataset_op.h │ │ │ ├── text_line_dataset_op_test.cc │ │ │ ├── tf_record_dataset_op.cc │ │ │ ├── tf_record_dataset_op.h │ │ │ ├── tf_record_dataset_op_test.cc │ │ │ ├── window_dataset.cc │ │ │ ├── window_dataset.h │ │ │ ├── window_dataset_op.cc │ │ │ ├── window_dataset_op.h │ │ │ ├── window_dataset_op_test.cc │ │ │ ├── zip_dataset_op.cc │ │ │ ├── zip_dataset_op.h │ │ │ └── zip_dataset_op_test.cc │ │ ├── data_format_ops.cc │ │ ├── data_format_ops.h │ │ ├── data_format_ops_gpu.cu.cc │ │ ├── debug_ops.cc │ │ ├── debug_ops.h │ │ ├── debug_ops_gpu.cu.cc │ │ ├── debug_ops_test.cc │ │ ├── decode_compressed_op.cc │ │ ├── decode_csv_op.cc │ │ ├── decode_padded_raw_op.cc │ │ ├── decode_proto_op.cc │ │ ├── decode_raw_op.cc │ │ ├── decode_wav_op.cc │ │ ├── decode_wav_op_test.cc │ │ ├── deep_conv2d.cc │ │ ├── deep_conv2d.h │ │ ├── deep_conv2d_test.cc │ │ ├── dense_update_functor.cc │ │ ├── dense_update_functor.h │ │ ├── dense_update_functor_gpu.cu.cc │ │ ├── dense_update_ops.cc │ │ ├── depthtospace_op.cc │ │ ├── depthtospace_op.h │ │ ├── depthtospace_op_gpu.cu.cc │ │ ├── depthwise_conv_grad_op.cc │ │ ├── depthwise_conv_op.cc │ │ ├── depthwise_conv_op.h │ │ ├── depthwise_conv_op_gpu.h │ │ ├── depthwise_conv_op_gpu_bfloat16.cu.cc │ │ ├── depthwise_conv_op_gpu_double.cu.cc │ │ ├── depthwise_conv_op_gpu_float.cu.cc │ │ ├── depthwise_conv_op_gpu_half.cu.cc │ │ ├── depthwise_conv_ops_test.cc │ │ ├── dequantize_op.cc │ │ ├── dequantize_op_test.cc │ │ ├── deserialize_sparse_string_op.cc │ │ ├── deserialize_sparse_variant_op.cc │ │ ├── diag_op.cc │ │ ├── diag_op.h │ │ ├── diag_op_gpu.cu.cc │ │ ├── diag_op_test.cc │ │ ├── dilation_ops.cc │ │ ├── dilation_ops.h │ │ ├── dilation_ops_gpu.cu.cc │ │ ├── dynamic_partition_op.cc │ │ ├── dynamic_partition_op_gpu.cu.cc │ │ ├── dynamic_partition_op_test.cc │ │ ├── dynamic_stitch_op.cc │ │ ├── dynamic_stitch_op_gpu.cu.cc │ │ ├── dynamic_stitch_op_test.cc │ │ ├── edit_distance_op.cc │ │ ├── eigen_activations.h │ │ ├── eigen_activations_test.cc │ │ ├── eigen_attention.h │ │ ├── eigen_attention_test.cc │ │ ├── eigen_backward_cuboid_convolutions.h │ │ ├── eigen_backward_cuboid_convolutions_test.cc │ │ ├── eigen_backward_spatial_convolutions.h │ │ ├── eigen_backward_spatial_convolutions_test.cc │ │ ├── eigen_benchmark.h │ │ ├── eigen_benchmark_cpu_test.cc │ │ ├── eigen_cuboid_convolution.h │ │ ├── eigen_cuboid_convolutions_test.cc │ │ ├── eigen_mkldnn_contraction_kernel_test.cc │ │ ├── eigen_pooling.h │ │ ├── eigen_pooling_test.cc │ │ ├── encode_proto_op.cc │ │ ├── encode_wav_op.cc │ │ ├── encode_wav_op_test.cc │ │ ├── example_parsing_ops.cc │ │ ├── example_parsing_ops_test.cc │ │ ├── fact_op.cc │ │ ├── fake_quant_ops.cc │ │ ├── fake_quant_ops_functor.h │ │ ├── fake_quant_ops_gpu.cu.cc │ │ ├── fake_quant_ops_test.cc │ │ ├── fft_ops.cc │ │ ├── fifo_queue.cc │ │ ├── fifo_queue.h │ │ ├── fifo_queue_op.cc │ │ ├── filesystem_ops.cc │ │ ├── fill_empty_rows_functor.cc │ │ ├── fill_empty_rows_functor.h │ │ ├── fill_empty_rows_functor_gpu.cu.cc │ │ ├── fill_functor.cc │ │ ├── fill_functor.cu.cc │ │ ├── fill_functor.h │ │ ├── fingerprint_op.cc │ │ ├── fingerprint_op_test.cc │ │ ├── fixed_length_record_reader_op.cc │ │ ├── fractional_avg_pool_op.cc │ │ ├── fractional_max_pool_op.cc │ │ ├── fractional_pool_common.cc │ │ ├── fractional_pool_common.h │ │ ├── function_ops.cc │ │ ├── function_ops.h │ │ ├── functional_ops.cc │ │ ├── fused_batch_norm_ex_op_test.cc │ │ ├── fused_batch_norm_op.cc │ │ ├── fused_batch_norm_op.cu.cc │ │ ├── fused_batch_norm_op.h │ │ ├── fused_batch_norm_op_test.cc │ │ ├── fused_eigen_output_kernels.cc │ │ ├── fused_eigen_output_kernels.h │ │ ├── fuzzing │ │ │ ├── BUILD │ │ │ ├── check_numerics_fuzz.cc │ │ │ ├── corpus │ │ │ │ ├── decode_bmp │ │ │ │ │ ├── 012e3ad384a4a1165f8498b5c94ba0d32a73e187 │ │ │ │ │ ├── 055d77f7810048caa28323f6eb552a53d156040b │ │ │ │ │ ├── 131e251bfb82c681cb075d32b99f18fceaca115d │ │ │ │ │ ├── 1399ab0bd9f2c91d270cb43251bdc5729bef3526 │ │ │ │ │ ├── 16a6ce88f66d2e9686c8354cad8ba915cf0c11de │ │ │ │ │ ├── 185097ed0588195164619ea930ddd8274a5f32ad │ │ │ │ │ ├── 27711a87e06a50c81571c27c3aa403a6ad5dc55c │ │ │ │ │ ├── 298c3787ad1722b22569cbc405c464d2 │ │ │ │ │ ├── 2b95ba6d8141ce0d29ff279770903922 │ │ │ │ │ ├── 321fb3d758b86e37fc340ae2b09b8ed9fa73a4cb │ │ │ │ │ ├── 331a98b4e4c87840efea69223766ebd0e1736542 │ │ │ │ │ ├── 352d73f841223ecb630b5836585d2ba7b0f9d883 │ │ │ │ │ ├── 3a84f409d4c117edfdebc508cd23e8fc │ │ │ │ │ ├── 3ef5cc982c0b45f69a26fd0f7d376415fdebabd1 │ │ │ │ │ ├── 401c7de8e122018a0e17f57c93db7ee49ab0e906 │ │ │ │ │ ├── 52fee71bb8c9c79068e1fe580677ad739a2d0415 │ │ │ │ │ ├── 57b11507813d5727b7789354d888eda83d5f3d86 │ │ │ │ │ ├── 57dff0fa53ee0ef24a43cca6ab0523bfdc1f720d │ │ │ │ │ ├── 5c42d3df0dc400a7a4175b8d4eec6cc8ee2437b2 │ │ │ │ │ ├── 5cca20637ae75fddad9370ee930837baef8aeb43 │ │ │ │ │ ├── 5d34bc9cef0c844b9c5ebe948145c4ca11b5ca09 │ │ │ │ │ ├── 5e162fe883bd12fb1c4131d4e0c979a12bd15eac │ │ │ │ │ ├── 5e83f8faab9c1a51a33d5e29edbb9dcec23c6092 │ │ │ │ │ ├── 61b29dc2fcef7b6fbe3e0cc88769a7ef │ │ │ │ │ ├── 6361eca190157ece389665ee523ccc3aefcd957f │ │ │ │ │ ├── 65150515ab3b11d657519b22bb887d74e94b2d7f │ │ │ │ │ ├── 656f38ef6dcd58c6a909d61db11f777def69c394 │ │ │ │ │ ├── 66e0d2cafd592bf9d61ad900fade8ee530d5f3d7 │ │ │ │ │ ├── 6b5b42cb105a2c4c5fd6034e9885cbe457f1b50c │ │ │ │ │ ├── 722ed0197cb92ecbf9745edb38275e7a9aaf322f │ │ │ │ │ ├── 77bdd2efdf328366cbbf3c5688768dc0a88d02b1 │ │ │ │ │ ├── 7841bfa002c05c61d5a5d9241f214cc17a336166 │ │ │ │ │ ├── 7899e22fc83f6be28e9130c4a1c91a48 │ │ │ │ │ ├── 7dddccaebd16ae0c26daeffc42df50f529891119 │ │ │ │ │ ├── 8157442eee4bbfdd9716e264b11085d61a9955b7 │ │ │ │ │ ├── 81ff28ed63d5435ddc4c8771dd5d40aa658cbbe0 │ │ │ │ │ ├── 820c8c0d33c18f6c4d9edd314e91289186931ad0 │ │ │ │ │ ├── 849e9d7cee1c52105242327086997296e452b981 │ │ │ │ │ ├── 84ddb92c63e0fad7018f6069daf8779ce11501e2 │ │ │ │ │ ├── 86bc3d5dbb9313137502080e58551edd2e649c70 │ │ │ │ │ ├── 87d94d88fe29d277c76e1a52042b02c092d5ae14 │ │ │ │ │ ├── 8c4646f3357945c4e19a59ff79fffe3c874dbf16 │ │ │ │ │ ├── 90632bc6dee4eb836f3d7db1d16446a9c8510080 │ │ │ │ │ ├── 94d06016aa949e8e7203217e4cc6625ded7f4244 │ │ │ │ │ ├── 9875819b9e5783e7489c29a81cc9d4279209956a │ │ │ │ │ ├── 9c1cc734114b29aac6c51782d5c17e9dbe1faca2 │ │ │ │ │ ├── 9d2961871eeb201ef8a6f5503d8a8b62 │ │ │ │ │ ├── 9f39e11cdd88344a4894b678e5a04a810880064d │ │ │ │ │ ├── a350588a6dabe4376a066aed44ef8786d8e752e7 │ │ │ │ │ ├── a6101a79919d444e1fc50aefab5837c39e3f4a19 │ │ │ │ │ ├── a9c8793f8fb063bec839ee1280406fe5396545e5 │ │ │ │ │ ├── ad4e9d2234e8599bdf12607c6b8cab4edae82c4e │ │ │ │ │ ├── b90b6830917919e94186d312f06481bd │ │ │ │ │ ├── b98fd4cb1d7031240414301c19b03097c0035c6b │ │ │ │ │ ├── ba976fcdb4daf092ef17ce43bf2b78d9d8bc2aeb │ │ │ │ │ ├── bc112b571eafee0f5a031f3c9cce6244216d128d │ │ │ │ │ ├── c42b981c28a1715c375050f6fcf53f1d │ │ │ │ │ ├── c6049874b33eadb016fccf0c5fa66e556ae069b9 │ │ │ │ │ ├── c8697bf2369f6ab85f501376c4d93bb8a56974a3 │ │ │ │ │ ├── c8daf283e0aef2fd7b630c0430e05dc28f24ecf6 │ │ │ │ │ ├── cacff56e1af4b8fde912822da06b10fb8c545a19 │ │ │ │ │ ├── ce4dcc22b1d595c49a25121c0b580104 │ │ │ │ │ ├── d0cd71dbf039fd64cf42eff30da92a71a919226a │ │ │ │ │ ├── d5ce626ac3264bed6af5580e341a89406857cbb9 │ │ │ │ │ ├── d77ada02e9bc8c24b2711eca6a8f52ae356bfc21 │ │ │ │ │ ├── d7eb9c5a0f9803df4c00390793b8ab57bd7c9484 │ │ │ │ │ ├── dc1efccdeec17e151a1ec8228c09ab61c3040b33 │ │ │ │ │ ├── dcea22c66c60088165a2f1772036473f │ │ │ │ │ ├── de539ae7442fa05dafcfe1a021f0186ef74a2b0e │ │ │ │ │ ├── e2306b1d6b88d0ccc4e2c3a9edb07462a5a32215 │ │ │ │ │ ├── e2778da0240fdd15ef5844905d81c4e05f34a8bd │ │ │ │ │ ├── e6642e9266875f9d908942e534bf898103a2c794 │ │ │ │ │ ├── ec6cdb929c08d8daf2bd7fc185fbf4d787b45120 │ │ │ │ │ ├── ed8636357f79439b6a03eb14469b686cc401a1c9 │ │ │ │ │ ├── ee313e9acecb5c688ce8c9bb10e70e136fbb9c6d │ │ │ │ │ ├── ef689af320e7d9e22231109faae2e8149cb86e1c │ │ │ │ │ ├── fda6b9a9f6ffdf4765c00465619c7ceb3f7db2e4 │ │ │ │ │ └── ffe829bb0adac20d9c0756f68a22d1255e4fdb54 │ │ │ │ ├── decode_csv │ │ │ │ │ ├── 283a87da22c3e0e7cdf53a6828f629254fe53ebb │ │ │ │ │ ├── 44acc2aba022cada4f6f6745232b6d76749e54a9 │ │ │ │ │ ├── 61317d2434a1af0da4e8ae7cdbd213b23572f9a5 │ │ │ │ │ ├── 7331d2616685f209d090e1ccb83e97a779a96f75 │ │ │ │ │ ├── a6ef373bd6b4fa6eb22ef1bd2fe48b932c94a688 │ │ │ │ │ └── eaf97dfae78a3f435c7f376be52e5c7be203571d │ │ │ │ ├── decode_json_example │ │ │ │ │ ├── 013a29ea098a178f8a36741c9fd91144 │ │ │ │ │ ├── 0875575fb76d630ccb19c5da8aab66b2 │ │ │ │ │ ├── 7e7f58fc443a11a0a2c5d9b643b7e99b │ │ │ │ │ ├── 849a23936269a261c0370b5e9abe2416 │ │ │ │ │ ├── 85282c1696d98b9843ce3e8bd1cd899f │ │ │ │ │ ├── 90388b9c8093d8adedad0644b618da87 │ │ │ │ │ ├── 9fa2f86ea6d3ade36e961247c3026f8d │ │ │ │ │ ├── c4f18ca60a84e9869a28faf6f65dc758 │ │ │ │ │ ├── d456ee029700adef5d28438593010223 │ │ │ │ │ └── e9f0ff6ee8d691ae69d2ecb4710030a2 │ │ │ │ ├── decode_png │ │ │ │ │ ├── 010dc3d4b05288fcc40de2721052b3dc699f1cb3 │ │ │ │ │ ├── 0555cd5e9d99629819cc985285f80da0f00be1e9 │ │ │ │ │ ├── 0a0352aa168803ff65455792d9f6ee555c3e7c3f │ │ │ │ │ ├── 0ed54162df93ef8d00f993ce6b59ba422903d381 │ │ │ │ │ ├── 1547b448171c700613c3946d730de496c9b9863f │ │ │ │ │ ├── 17859046cbe4ac598a645173d679ce2a52c6afba │ │ │ │ │ ├── 1b0384bc2d5ac42b7425ac51c374f60c │ │ │ │ │ ├── 1df76c07817fbc3653a26f34d97658e9973627c2 │ │ │ │ │ ├── 1f0717f8856d7782e3ab7992d3a72d783a018443 │ │ │ │ │ ├── 23b911e4ce936def88bc9a46b8b433c0e83fba2a │ │ │ │ │ ├── 25592201c3edff0578dbdac6b0e4f2be109ce151 │ │ │ │ │ ├── 266fd8495e0b8eb64387c1a62264185e061fee73 │ │ │ │ │ ├── 27f178cf415b4ff8671131ddf1d042dafac2fb3e │ │ │ │ │ ├── 2a0bdc4d9cc5ea5bb21dd256d6ac96075376a94f │ │ │ │ │ ├── 2e5d25add6adc68e0457b358c7a34abf3d41c938 │ │ │ │ │ ├── 2e6c5b6a766dd5e9bd41eacfd0a36572bd2f7544 │ │ │ │ │ ├── 2e9c935cf82f6ca640e9a9abc3c30a578ad46176 │ │ │ │ │ ├── 2fcf1ed4477f7eaee028f5b3f9edeb5f1a737826 │ │ │ │ │ ├── 3480713774f590908ca5dba16d121cdfb8fba62b │ │ │ │ │ ├── 38bd2bd767d0c4ddd531b3893080b952 │ │ │ │ │ ├── 39289afcec60d98802b333e0fbb1da4d7aed4ce5 │ │ │ │ │ ├── 3adc488e21d4aca7bed9422f0241a42d0f93e7d9 │ │ │ │ │ ├── 3cbf274da522483dc991fad9df43a22ac4fb3173 │ │ │ │ │ ├── 3d840cdff7f5ad16fe8bcb985ed4946c03459432 │ │ │ │ │ ├── 3f1e6753c1fca958e859189857449746592158ea │ │ │ │ │ ├── 3fa4075993cb0f9bfa8eea785174a2038a69aa1b │ │ │ │ │ ├── 4023a373e977be58413e55350380310c5dd1fd6a │ │ │ │ │ ├── 40caba69dce1cfc48e0e43184d2bfbc6daa4399a │ │ │ │ │ ├── 41438a3c1c77c64a2f0840a2427f8834 │ │ │ │ │ ├── 41841e9561d8135945c1c1e55ab9e9a1e933653b │ │ │ │ │ ├── 41d40f2d66fa43e34537385594ee9911e65deadf │ │ │ │ │ ├── 421bd39810b50309a71adb2dadc3b19f01a52312 │ │ │ │ │ ├── 446c305b2c0665736f94fb2b62dbdef445eff0cf │ │ │ │ │ ├── 449cee952bb645f6f4241a6665d3c6028c073c7a │ │ │ │ │ ├── 45520b07609978c5aa3516d803527438b93fbadb │ │ │ │ │ ├── 4da74a34bcede234b0415f77fbd87d70bf9a777e │ │ │ │ │ ├── 51db5d31d2c5300d34831d9f23bcdd0aff9a998b │ │ │ │ │ ├── 5cde2a9167798cb77f10abbfb2640a5c357f99fc │ │ │ │ │ ├── 5e352fc10ac476cfbe1d755f092e069820223249 │ │ │ │ │ ├── 63661677dd1306cec4b5a565190e65adf2446e52 │ │ │ │ │ ├── 65887ed3db382aab1d9485c500f4401318d303b9 │ │ │ │ │ ├── 67b5181f8f0644597e9bde539e8f083b5cacd0e7 │ │ │ │ │ ├── 74c9dcf7afee2a6cb1ab3a2c0de744d1b03c1466 │ │ │ │ │ ├── 792181ca19e6ded261434e588bb7fc2a4816d4ce │ │ │ │ │ ├── 79f0e2a475487f8fa69e68c1cc947c5851bda741 │ │ │ │ │ ├── 7e5fcdfeb557ce379ed96925c68505eaac0112db │ │ │ │ │ ├── 7eec7530acf34b3a96fa9189783453999f7b6838 │ │ │ │ │ ├── 80114bf9781bffc9db411413d83541d8deaaf7c1 │ │ │ │ │ ├── 80425fb92bb86627e854892f23823fa804e5fdc3 │ │ │ │ │ ├── 821cdd6eeb919a8dd7f35289abbd583828dd4945 │ │ │ │ │ ├── 83e1a31785285338b0ddb3334b0ed098e63dedde │ │ │ │ │ ├── 8a4c8100dedd0fb5f2a8b468c678f7ad8269deeb │ │ │ │ │ ├── 8ae8268c24dc866c1edb3826b93a1c75dbf74ff4 │ │ │ │ │ ├── 90f72038cc627f34f074ea72eadbba87a5e3e288 │ │ │ │ │ ├── 92b67faee4a49df2cdbed785e27b4a1cddcfffa3 │ │ │ │ │ ├── 9463810467aacdc9923b2b20a2236116b760d75b │ │ │ │ │ ├── 94d7c96aea32ad41ce643d35b951a6d8990b81d6 │ │ │ │ │ ├── 98cc7e9fe87df914d89a0aef008930f27b3c26f5 │ │ │ │ │ ├── 99172dfdb4f59aaced29c7681ac6e6ce8356e814 │ │ │ │ │ ├── 9ae3b647d895af97fe872c0b1442df7b5b767160 │ │ │ │ │ ├── 9d2b1d2121b0508a4fa8d1508adb9d05633fdac3 │ │ │ │ │ ├── a335af37917ccf0c8b11bb884a3a74f3f1d2a7c6 │ │ │ │ │ ├── a738609112d3a6772c50a71e2c3504ebc515b709 │ │ │ │ │ ├── a8cecab5d917da5a4729632a7a18c564d7e1607d │ │ │ │ │ ├── ade919ab2b4a458e806575c941dfe50ae3fd3621 │ │ │ │ │ ├── b1251621a5eb5e7fda9cac9baead1c993a285c36 │ │ │ │ │ ├── b1516b78c3dfe77eeb554985fd7344c0478fbbcb │ │ │ │ │ ├── b41241740f5f8ad2c1d408f7bb6a313bd863c158 │ │ │ │ │ ├── b799c8596523a7ebeb8e11ada08818c10f7eabfc │ │ │ │ │ ├── ba48d0521a111222dc95a3a997c7c92dea5f4443 │ │ │ │ │ ├── c01457c6889fb1b597d308363a36412c0b7f90e7 │ │ │ │ │ ├── c82ebc0d6688d104af04fd20d6d3da591dc391f7 │ │ │ │ │ ├── c9a03eb758dd84e954e3d70916e2311e8fd21f3c │ │ │ │ │ ├── cf892756b33578a54ab20044514e573328d2f1d7 │ │ │ │ │ ├── d3bc3f158a63f1d50b474addd3f7b3d17f23e8e9 │ │ │ │ │ ├── d4906950aa9d60ad09dc0f5413c3d88080c3bc37 │ │ │ │ │ ├── da31578a8068bad65e1c7a3d06e8f543a2a0bc65 │ │ │ │ │ ├── dd4a9b5d0740679c249fc884efc499433b29436b │ │ │ │ │ ├── deea4ecc6f0b2a6d89fd25ff76762299f21602fb │ │ │ │ │ ├── e1040c7ffcb39915e0f539018c81f9798924cba6 │ │ │ │ │ ├── e381dc85682cc33ad99f622b89d145b47f7d6392 │ │ │ │ │ ├── ea24498fc7a144fccc6f1665ebf7020df803dd1a │ │ │ │ │ ├── eaa5d677e797c07bac98c3c7051abad91852e7c6 │ │ │ │ │ ├── ed7871269315725535d8bffec7836c45a3fc5c26 │ │ │ │ │ ├── ee8460f4077064c5a2137075b48eba7d3db5c570 │ │ │ │ │ ├── ef09f26e0ee61329f84a9f589629a865ae9ee0a6 │ │ │ │ │ ├── f477da4d7d8ff2066041e1dd5ee4e833b7111a1a │ │ │ │ │ ├── f8a379b2498a4eb452a85791a49adf065dab59ae │ │ │ │ │ ├── fe67bccb06f2174523943cc684518fcf1f7f8046 │ │ │ │ │ └── ff1e67d17c1c27ef0d97900d0ea276b563a64628 │ │ │ │ ├── decode_wav │ │ │ │ │ ├── 02cc44cdfec1d9d0d0c66c5a5f40d3d20e4c4c3a │ │ │ │ │ ├── 087e1d7fae1c1ddcbaa3b5f822a171ad15498186 │ │ │ │ │ ├── 0f61c33027394a0f14d29dcd22f405cad943b7cf │ │ │ │ │ ├── 10cdebea1659c21a0248f88654ae41f62786abf1 │ │ │ │ │ ├── 126e68def9fd973a100e0f66cadf09448a716b57 │ │ │ │ │ ├── 1275d41ebf8788ce3a949352e4bc654b04012da3 │ │ │ │ │ ├── 1a7f1c407fb3864ddb559f88f373a21d1be51584 │ │ │ │ │ ├── 1c3e1c91f187f6bcea86f172ff5bbbd955a9654d │ │ │ │ │ ├── 300fe1e0a47543037cbf0243b6756c9aa48799c4 │ │ │ │ │ ├── 31ec5b0134bedcfe283f4978e6e65b7d35d5d4ad │ │ │ │ │ ├── 4e7cbb27667bcfca92838aa8020749990013a9b1 │ │ │ │ │ ├── 585e469231d202812bfba8285fb30c8e31c857b9 │ │ │ │ │ ├── 58eab6bc2386e2ef43fe4f55cb6ad3611399d5de │ │ │ │ │ ├── 63448c6a9feb8c72b3e82af4d735ec2e62ddd328 │ │ │ │ │ ├── 6874d5b1c7a64b596c61f24877d422e89bebe58b │ │ │ │ │ ├── 7501f79cb067da108020579ed654349c7933d22f │ │ │ │ │ ├── 782051f8120182b860c7fe1b265179cfa2fe03fd │ │ │ │ │ ├── 793feab2deb35e284a975f6527d76a8be5540fe6 │ │ │ │ │ ├── 7f41ec3a9805c6b8f3656c4f9f6d0ff7dbf8a329 │ │ │ │ │ ├── 8210dc595a2652f2f812093b01e239e7918ea065 │ │ │ │ │ ├── 8dffe4c5c26d891b578fd2ea4b9adfc0c96ad5f7 │ │ │ │ │ ├── 91d787a9298ddc015efa783a92c4bdba8af0d7de │ │ │ │ │ ├── 92c065286f956f086e977556358f6b54b12bcacc │ │ │ │ │ ├── a35c9bb71792b60a13dea23a41b41847ad4b93d6 │ │ │ │ │ ├── a6ea960c7b4d42772888280277b26e645ceee904 │ │ │ │ │ ├── aa526aa853333f0bb11804b5243df411452cecd2 │ │ │ │ │ ├── ca533cd26c7ca6bf69e62351b265ded496fdf1d9 │ │ │ │ │ ├── f38c61da15f2cb7a39ff02e69f0b00e99f37ec86 │ │ │ │ │ ├── f88f1012473e6cfcc9b39b2552f682b2f73eff8c │ │ │ │ │ ├── fa79819c5de04bc06c69bec3fa7f2e982826ea2f │ │ │ │ │ └── fce08de222896ac3a20657a3b4f42d5b6c54a96a │ │ │ │ ├── scatter_nd │ │ │ │ │ ├── 5b0e5f8d2990c3cac80fa792ba141c43 │ │ │ │ │ ├── 5b61fa3a30dd267828f12d9ea2b2a191 │ │ │ │ │ ├── 8bc8b7d8beb3483c48158739791e56b0 │ │ │ │ │ ├── d2ef31d47578e9de8323bb0e4806f1be │ │ │ │ │ └── e2791edcf2c8d9f4af3678a75d43a3e4 │ │ │ │ ├── string_split │ │ │ │ │ ├── 4c01a1504da9de2216894743ecc44424 │ │ │ │ │ ├── 5bf16424630b5afbcffe711fb9834440 │ │ │ │ │ ├── a7185605aef0a8fd682fcb4656e4a736 │ │ │ │ │ ├── d5606def44fdbb9385dd764612069db0 │ │ │ │ │ ├── dbac766f3160de65894bf5153f478146 │ │ │ │ │ └── e85ff62f6d457666f54a37a19a115a24 │ │ │ │ ├── string_split_v2 │ │ │ │ │ ├── 00fd47bf73afcb72e7ed51bffd5f5fec │ │ │ │ │ ├── 14908973e6720513a5f37676cb9fcc29 │ │ │ │ │ ├── 2779ba7c4d23eee9f79efa3660084c5d │ │ │ │ │ ├── 5bf16424630b5afbcffe711fb9834440 │ │ │ │ │ ├── 89734a96b93275e495a9498b806fafe1 │ │ │ │ │ └── d5606def44fdbb9385dd764612069db0 │ │ │ │ └── string_to_number │ │ │ │ │ ├── 2db83ea58639b6d7d585fa12e3947a82 │ │ │ │ │ ├── 36b4a931886b941dc41180050d12ca94 │ │ │ │ │ ├── 50a2fabfdd276f573ff97ace8b11c5f4 │ │ │ │ │ ├── 62edb2a1eee34b001652cd86584becf2 │ │ │ │ │ ├── 90013d1ec28c46a5c00574e60c70b6fc │ │ │ │ │ ├── 94f3e3cee6957ce5815326d6788c85f4 │ │ │ │ │ ├── 96f547bc04bb913da0bc08915238ebd8 │ │ │ │ │ ├── d3a903d18fc11e1f35c572ad4da690ed │ │ │ │ │ ├── e3b629c92af44260c189deb32d6f06f3 │ │ │ │ │ ├── f03eecf3bcfe4967a1888156a3115c8d │ │ │ │ │ └── fa54ca9186f77122ae2a82684a062e16 │ │ │ ├── decode_base64_fuzz.cc │ │ │ ├── decode_bmp_fuzz.cc │ │ │ ├── decode_compressed_fuzz.cc │ │ │ ├── decode_csv_fuzz.cc │ │ │ ├── decode_json_example_fuzz.cc │ │ │ ├── decode_png_fuzz.cc │ │ │ ├── decode_wav_fuzz.cc │ │ │ ├── dictionaries │ │ │ │ ├── decode_json_example.dict │ │ │ │ ├── decode_png.dict │ │ │ │ └── decode_wav.dict │ │ │ ├── encode_base64_fuzz.cc │ │ │ ├── encode_jpeg_fuzz.cc │ │ │ ├── example_proto_fast_parsing_fuzz.cc │ │ │ ├── fuzz_session.h │ │ │ ├── one_hot_fuzz.cc │ │ │ ├── parse_tensor_op_fuzz.cc │ │ │ ├── scatter_nd_fuzz.cc │ │ │ ├── string_to_number_fuzz.cc │ │ │ └── tf_ops_fuzz_target_lib.bzl │ │ ├── gather_functor.cc │ │ ├── gather_functor.h │ │ ├── gather_functor_batched.cc │ │ ├── gather_functor_batched.h │ │ ├── gather_functor_batched_gpu.cu.cc │ │ ├── gather_functor_batched_gpu.cu.h │ │ ├── gather_functor_gpu.cu.cc │ │ ├── gather_functor_gpu.cu.h │ │ ├── gather_nd_op.cc │ │ ├── gather_nd_op.h │ │ ├── gather_nd_op_cpu_impl.h │ │ ├── gather_nd_op_cpu_impl_0.cc │ │ ├── gather_nd_op_cpu_impl_1.cc │ │ ├── gather_nd_op_cpu_impl_2.cc │ │ ├── gather_nd_op_cpu_impl_3.cc │ │ ├── gather_nd_op_cpu_impl_4.cc │ │ ├── gather_nd_op_cpu_impl_5.cc │ │ ├── gather_nd_op_cpu_impl_6.cc │ │ ├── gather_nd_op_cpu_impl_7.cc │ │ ├── gather_nd_op_gpu.cu.cc │ │ ├── gather_nd_op_test.cc │ │ ├── gather_op.cc │ │ ├── gather_op_test.cc │ │ ├── gemm_functors.h │ │ ├── generate_vocab_remapping_op.cc │ │ ├── gpu_device_array.h │ │ ├── gpu_device_array_gpu.h │ │ ├── gpu_prim.h │ │ ├── gpu_prim_helpers.h │ │ ├── gpu_prim_helpers_test.cu.cc │ │ ├── gpu_utils.cc │ │ ├── gpu_utils.h │ │ ├── guarantee_const_op.cc │ │ ├── guarantee_const_op_test.cc │ │ ├── hinge-loss.h │ │ ├── histogram_op.cc │ │ ├── histogram_op.h │ │ ├── histogram_op_gpu.cu.cc │ │ ├── host_constant_op.cc │ │ ├── host_constant_op.h │ │ ├── identity_n_op.cc │ │ ├── identity_n_op.h │ │ ├── identity_n_op_test.cc │ │ ├── identity_op.cc │ │ ├── identity_op.h │ │ ├── identity_op_test.cc │ │ ├── identity_reader_op.cc │ │ ├── image │ │ │ ├── BUILD │ │ │ ├── adjust_contrast_op.cc │ │ │ ├── adjust_contrast_op.h │ │ │ ├── adjust_contrast_op_benchmark_test.cc │ │ │ ├── adjust_contrast_op_gpu.cu.cc │ │ │ ├── adjust_contrast_op_test.cc │ │ │ ├── adjust_hsv_gpu.cu.h │ │ │ ├── adjust_hue_op.cc │ │ │ ├── adjust_hue_op.h │ │ │ ├── adjust_hue_op_gpu.cu.cc │ │ │ ├── adjust_saturation_op.cc │ │ │ ├── adjust_saturation_op.h │ │ │ ├── adjust_saturation_op_gpu.cu.cc │ │ │ ├── attention_ops.cc │ │ │ ├── colorspace_op.cc │ │ │ ├── colorspace_op.h │ │ │ ├── colorspace_op_gpu.cu.cc │ │ │ ├── colorspace_op_test.cc │ │ │ ├── crop_and_resize_op.cc │ │ │ ├── crop_and_resize_op.h │ │ │ ├── crop_and_resize_op_benchmark_test.cc │ │ │ ├── crop_and_resize_op_gpu.cu.cc │ │ │ ├── crop_and_resize_op_test.cc │ │ │ ├── decode_image_op.cc │ │ │ ├── draw_bounding_box_op.cc │ │ │ ├── encode_jpeg_op.cc │ │ │ ├── encode_jpeg_op_test.cc │ │ │ ├── encode_png_op.cc │ │ │ ├── extract_image_patches_op.cc │ │ │ ├── extract_image_patches_op.h │ │ │ ├── extract_image_patches_op_gpu.cu.cc │ │ │ ├── extract_jpeg_shape_op.cc │ │ │ ├── extract_volume_patches_op.cc │ │ │ ├── extract_volume_patches_op.h │ │ │ ├── extract_volume_patches_op_gpu.cu.cc │ │ │ ├── generate_box_proposals_op.cu.cc │ │ │ ├── image_ops.cc │ │ │ ├── image_ops.h │ │ │ ├── image_ops_gpu.cu.cc │ │ │ ├── mirror_pad_op.cc │ │ │ ├── mirror_pad_op.h │ │ │ ├── mirror_pad_op_benchmark_test.cc │ │ │ ├── mirror_pad_op_cpu_impl.h │ │ │ ├── mirror_pad_op_cpu_impl_1.cc │ │ │ ├── mirror_pad_op_cpu_impl_2.cc │ │ │ ├── mirror_pad_op_cpu_impl_3.cc │ │ │ ├── mirror_pad_op_cpu_impl_4.cc │ │ │ ├── mirror_pad_op_cpu_impl_5.cc │ │ │ ├── mirror_pad_op_gpu.cu.cc │ │ │ ├── mirror_pad_op_test.cc │ │ │ ├── non_max_suppression_op.cc │ │ │ ├── non_max_suppression_op.cu.cc │ │ │ ├── non_max_suppression_op.h │ │ │ ├── non_max_suppression_op_benchmark_test.cc │ │ │ ├── non_max_suppression_op_gpu_test.cc │ │ │ ├── non_max_suppression_op_test.cc │ │ │ ├── random_crop_op.cc │ │ │ ├── resize_area_op.cc │ │ │ ├── resize_area_op_test.cc │ │ │ ├── resize_bicubic_op.cc │ │ │ ├── resize_bicubic_op_test.cc │ │ │ ├── resize_bilinear_op.cc │ │ │ ├── resize_bilinear_op.h │ │ │ ├── resize_bilinear_op_gpu.cu.cc │ │ │ ├── resize_bilinear_op_test.cc │ │ │ ├── resize_nearest_neighbor_op.cc │ │ │ ├── resize_nearest_neighbor_op.h │ │ │ ├── resize_nearest_neighbor_op_gpu.cu.cc │ │ │ ├── resize_nearest_neighbor_op_test.cc │ │ │ ├── resize_op_benchmark_test.cc │ │ │ ├── sample_distorted_bounding_box_op.cc │ │ │ ├── sampling_kernels.cc │ │ │ ├── sampling_kernels.h │ │ │ ├── sampling_kernels_test.cc │ │ │ ├── scale_and_translate_op.cc │ │ │ ├── scale_and_translate_op.h │ │ │ └── scale_and_translate_op_test.cc │ │ ├── immutable_constant_op.cc │ │ ├── immutable_constant_op.h │ │ ├── immutable_constant_op_test.cc │ │ ├── in_topk_op.cc │ │ ├── in_topk_op.h │ │ ├── in_topk_op_gpu.cu.cc │ │ ├── in_topk_op_test.cc │ │ ├── initializable_lookup_table.cc │ │ ├── initializable_lookup_table.h │ │ ├── inplace_ops.cc │ │ ├── inplace_ops_functor.h │ │ ├── inplace_ops_functor_gpu.cu.cc │ │ ├── isotonic_regression_op.cc │ │ ├── isotonic_regression_op_test.cc │ │ ├── kernel_platform_strings.h │ │ ├── l2loss_op.cc │ │ ├── l2loss_op.h │ │ ├── l2loss_op_gpu.cu.cc │ │ ├── linalg │ │ │ ├── BUILD │ │ │ ├── banded_triangular_solve_op.cc │ │ │ ├── banded_triangular_solve_op_test.cc │ │ │ ├── cholesky_grad.cc │ │ │ ├── cholesky_op.cc │ │ │ ├── cholesky_op_gpu.cu.cc │ │ │ ├── determinant_op.cc │ │ │ ├── determinant_op.h │ │ │ ├── determinant_op_gpu.cu.cc │ │ │ ├── eig_op_complex128.cc │ │ │ ├── eig_op_complex64.cc │ │ │ ├── eig_op_double.cc │ │ │ ├── eig_op_float.cc │ │ │ ├── eig_op_impl.h │ │ │ ├── einsum_op.h │ │ │ ├── einsum_op_gpu.cu.cc │ │ │ ├── einsum_op_impl.h │ │ │ ├── einsum_op_impl_bfloat16.cc │ │ │ ├── einsum_op_impl_complex128.cc │ │ │ ├── einsum_op_impl_complex64.cc │ │ │ ├── einsum_op_impl_double.cc │ │ │ ├── einsum_op_impl_float.cc │ │ │ ├── einsum_op_impl_half.cc │ │ │ ├── einsum_op_impl_int32.cc │ │ │ ├── einsum_op_impl_int64.cc │ │ │ ├── eye_functor.h │ │ │ ├── eye_functor_gpu.cu.cc │ │ │ ├── linalg_ops_common.cc │ │ │ ├── linalg_ops_common.h │ │ │ ├── lu_op.cc │ │ │ ├── lu_op_gpu.cu.cc │ │ │ ├── matrix_band_part_op.cc │ │ │ ├── matrix_band_part_op.h │ │ │ ├── matrix_band_part_op_gpu.cu.cc │ │ │ ├── matrix_diag_op.cc │ │ │ ├── matrix_diag_op.h │ │ │ ├── matrix_diag_op_gpu.cu.cc │ │ │ ├── matrix_exponential_op.cc │ │ │ ├── matrix_inverse_op.cc │ │ │ ├── matrix_logarithm_op.cc │ │ │ ├── matrix_set_diag_op.cc │ │ │ ├── matrix_set_diag_op.h │ │ │ ├── matrix_set_diag_op_gpu.cu.cc │ │ │ ├── matrix_solve_ls_op_complex128.cc │ │ │ ├── matrix_solve_ls_op_complex64.cc │ │ │ ├── matrix_solve_ls_op_double.cc │ │ │ ├── matrix_solve_ls_op_float.cc │ │ │ ├── matrix_solve_ls_op_impl.h │ │ │ ├── matrix_solve_op.cc │ │ │ ├── matrix_square_root_op.cc │ │ │ ├── matrix_triangular_solve_op_complex.cc │ │ │ ├── matrix_triangular_solve_op_impl.h │ │ │ ├── matrix_triangular_solve_op_real.cc │ │ │ ├── matrix_triangular_solve_op_test.cc │ │ │ ├── qr_op_complex128.cc │ │ │ ├── qr_op_complex64.cc │ │ │ ├── qr_op_double.cc │ │ │ ├── qr_op_float.cc │ │ │ ├── qr_op_half.cc │ │ │ ├── qr_op_impl.h │ │ │ ├── self_adjoint_eig_op.cc │ │ │ ├── self_adjoint_eig_v2_op_complex128.cc │ │ │ ├── self_adjoint_eig_v2_op_complex64.cc │ │ │ ├── self_adjoint_eig_v2_op_double.cc │ │ │ ├── self_adjoint_eig_v2_op_float.cc │ │ │ ├── self_adjoint_eig_v2_op_gpu.cc │ │ │ ├── self_adjoint_eig_v2_op_impl.h │ │ │ ├── svd_op_complex128.cc │ │ │ ├── svd_op_complex64.cc │ │ │ ├── svd_op_double.cc │ │ │ ├── svd_op_float.cc │ │ │ ├── svd_op_gpu.cu.cc │ │ │ ├── svd_op_impl.h │ │ │ ├── tridiagonal_matmul_op.cc │ │ │ ├── tridiagonal_matmul_op_gpu.cu.cc │ │ │ ├── tridiagonal_solve_op.cc │ │ │ └── tridiagonal_solve_op_gpu.cu.cc │ │ ├── linalg_ops_common.h │ │ ├── list_kernels.cc │ │ ├── list_kernels.cu.cc │ │ ├── list_kernels.h │ │ ├── listdiff_op.cc │ │ ├── lmdb_reader_op.cc │ │ ├── load_and_remap_matrix_op.cc │ │ ├── logging_ops.cc │ │ ├── logging_ops.h │ │ ├── logging_ops_test.cc │ │ ├── logistic-loss.h │ │ ├── lookup_ops_test.cc │ │ ├── lookup_table_init_op.cc │ │ ├── lookup_table_init_op.h │ │ ├── lookup_table_op.cc │ │ ├── lookup_table_op.h │ │ ├── lookup_util.cc │ │ ├── lookup_util.h │ │ ├── loss.h │ │ ├── loss_test.cc │ │ ├── lrn_op.cc │ │ ├── lrn_op_test.cc │ │ ├── map_kernels.cc │ │ ├── map_kernels.h │ │ ├── map_stage_op.cc │ │ ├── matching_files_op.cc │ │ ├── matmul_op.h │ │ ├── matmul_op_complex.cc │ │ ├── matmul_op_fused.cc │ │ ├── matmul_op_impl.h │ │ ├── matmul_op_real.cc │ │ ├── matmul_op_test.cc │ │ ├── matmul_util.cc │ │ ├── matmul_util.h │ │ ├── maxpooling_op.cc │ │ ├── maxpooling_op.h │ │ ├── maxpooling_op_gpu.cu.cc │ │ ├── maxpooling_op_gpu.h │ │ ├── merge_v2_checkpoints_op_test.cc │ │ ├── meta_support.cc │ │ ├── meta_support.h │ │ ├── mfcc.cc │ │ ├── mfcc.h │ │ ├── mfcc_dct.cc │ │ ├── mfcc_dct.h │ │ ├── mfcc_dct_test.cc │ │ ├── mfcc_mel_filterbank.cc │ │ ├── mfcc_mel_filterbank.h │ │ ├── mfcc_mel_filterbank_test.cc │ │ ├── mfcc_op.cc │ │ ├── mfcc_op_test.cc │ │ ├── mfcc_test.cc │ │ ├── mkl │ │ │ ├── BUILD │ │ │ ├── mkl_avgpooling_op.cc │ │ │ ├── mkl_batch_matmul_helper.h │ │ │ ├── mkl_batch_matmul_op.cc │ │ │ ├── mkl_concat_op.cc │ │ │ ├── mkl_conv_grad_filter_ops.cc │ │ │ ├── mkl_conv_grad_input_ops.cc │ │ │ ├── mkl_conv_ops.cc │ │ │ ├── mkl_conv_ops.h │ │ │ ├── mkl_conv_ops_test.cc │ │ │ ├── mkl_deprecated_ops.cc │ │ │ ├── mkl_dequantize_op.cc │ │ │ ├── mkl_dequantize_op_test.cc │ │ │ ├── mkl_einsum_op.cc │ │ │ ├── mkl_eltwise_activation_base_op.h │ │ │ ├── mkl_fused_batch_norm_op.cc │ │ │ ├── mkl_fused_batch_norm_op_test.cc │ │ │ ├── mkl_fused_instance_norm_op.cc │ │ │ ├── mkl_fused_mish_op.cc │ │ │ ├── mkl_fused_ops_test.cc │ │ │ ├── mkl_kernel_util.cc │ │ │ ├── mkl_kernel_util.h │ │ │ ├── mkl_layer_norm_op.cc │ │ │ ├── mkl_matmul_op.cc │ │ │ ├── mkl_matmul_op_benchmark.cc │ │ │ ├── mkl_matmul_op_fused.cc │ │ │ ├── mkl_matmul_ops_common.h │ │ │ ├── mkl_maxpooling_op.cc │ │ │ ├── mkl_pooling_ops_common.cc │ │ │ ├── mkl_pooling_ops_common.h │ │ │ ├── mkl_qmatmul_op.cc │ │ │ ├── mkl_qmatmul_op_test.cc │ │ │ ├── mkl_quantize_op.cc │ │ │ ├── mkl_quantize_op_test.cc │ │ │ ├── mkl_quantized_concat_op_test.cc │ │ │ ├── mkl_quantized_conv_ops.h │ │ │ ├── mkl_quantized_conv_ops_perchannel_test.cc │ │ │ ├── mkl_quantized_conv_ops_test.cc │ │ │ ├── mkl_quantized_pooling_ops_test.cc │ │ │ ├── mkl_relu_op.cc │ │ │ ├── mkl_relu_op_test.cc │ │ │ ├── mkl_requantization_range_per_channel_op.cc │ │ │ ├── mkl_requantize_ops_test.cc │ │ │ ├── mkl_requantize_per_channel_op.cc │ │ │ ├── mkl_softmax_op.cc │ │ │ ├── mkl_swish_op.cc │ │ │ ├── mkl_swish_op_test.cc │ │ │ ├── mkl_tmp_bf16_ops.cc │ │ │ ├── mkl_transpose_op.cc │ │ │ └── onednn_nn_ops_benchmark.cc │ │ ├── mlir_generated │ │ │ ├── BUILD │ │ │ ├── base_binary_ops_test.h │ │ │ ├── base_gpu_op.h │ │ │ ├── base_op.cc │ │ │ ├── base_op.h │ │ │ ├── base_ops_test.cc │ │ │ ├── base_ops_test.h │ │ │ ├── base_unary_ops_test.h │ │ │ ├── build_defs.bzl │ │ │ ├── build_test.sh │ │ │ ├── gpu_binary_ops_large_tensor_test.cc │ │ │ ├── gpu_binary_ops_test.cc │ │ │ ├── gpu_op_abs.cc │ │ │ ├── gpu_op_acos.cc │ │ │ ├── gpu_op_acosh.cc │ │ │ ├── gpu_op_add.cc │ │ │ ├── gpu_op_angle.cc │ │ │ ├── gpu_op_asin.cc │ │ │ ├── gpu_op_asinh.cc │ │ │ ├── gpu_op_atan.cc │ │ │ ├── gpu_op_atan2.cc │ │ │ ├── gpu_op_atanh.cc │ │ │ ├── gpu_op_bitwise_and.cc │ │ │ ├── gpu_op_bitwise_or.cc │ │ │ ├── gpu_op_bitwise_xor.cc │ │ │ ├── gpu_op_cast.cc │ │ │ ├── gpu_op_ceil.cc │ │ │ ├── gpu_op_complex.cc │ │ │ ├── gpu_op_complex_abs.cc │ │ │ ├── gpu_op_conj.cc │ │ │ ├── gpu_op_cos.cc │ │ │ ├── gpu_op_cosh.cc │ │ │ ├── gpu_op_digamma.cc │ │ │ ├── gpu_op_div.cc │ │ │ ├── gpu_op_div_no_nan.cc │ │ │ ├── gpu_op_elu.cc │ │ │ ├── gpu_op_equal.cc │ │ │ ├── gpu_op_erf.cc │ │ │ ├── gpu_op_erfc.cc │ │ │ ├── gpu_op_exp.cc │ │ │ ├── gpu_op_expm1.cc │ │ │ ├── gpu_op_floor.cc │ │ │ ├── gpu_op_floor_div.cc │ │ │ ├── gpu_op_floor_mod.cc │ │ │ ├── gpu_op_greater.cc │ │ │ ├── gpu_op_greater_equal.cc │ │ │ ├── gpu_op_imag.cc │ │ │ ├── gpu_op_invert.cc │ │ │ ├── gpu_op_is_finite.cc │ │ │ ├── gpu_op_is_inf.cc │ │ │ ├── gpu_op_is_nan.cc │ │ │ ├── gpu_op_left_shift.cc │ │ │ ├── gpu_op_less.cc │ │ │ ├── gpu_op_less_equal.cc │ │ │ ├── gpu_op_lgamma.cc │ │ │ ├── gpu_op_log.cc │ │ │ ├── gpu_op_log1p.cc │ │ │ ├── gpu_op_logical_and.cc │ │ │ ├── gpu_op_logical_not.cc │ │ │ ├── gpu_op_logical_or.cc │ │ │ ├── gpu_op_maximum.cc │ │ │ ├── gpu_op_minimum.cc │ │ │ ├── gpu_op_mul.cc │ │ │ ├── gpu_op_mul_no_nan.cc │ │ │ ├── gpu_op_neg.cc │ │ │ ├── gpu_op_next_after.cc │ │ │ ├── gpu_op_not_equal.cc │ │ │ ├── gpu_op_ones_like.cc │ │ │ ├── gpu_op_polygamma.cc │ │ │ ├── gpu_op_pow.cc │ │ │ ├── gpu_op_real.cc │ │ │ ├── gpu_op_reciprocal.cc │ │ │ ├── gpu_op_relu.cc │ │ │ ├── gpu_op_right_shift.cc │ │ │ ├── gpu_op_rint.cc │ │ │ ├── gpu_op_round.cc │ │ │ ├── gpu_op_rsqrt.cc │ │ │ ├── gpu_op_select.cc │ │ │ ├── gpu_op_selu.cc │ │ │ ├── gpu_op_sigmoid.cc │ │ │ ├── gpu_op_sign.cc │ │ │ ├── gpu_op_sin.cc │ │ │ ├── gpu_op_sinh.cc │ │ │ ├── gpu_op_softplus.cc │ │ │ ├── gpu_op_softsign.cc │ │ │ ├── gpu_op_sqrt.cc │ │ │ ├── gpu_op_square.cc │ │ │ ├── gpu_op_squared_difference.cc │ │ │ ├── gpu_op_sub.cc │ │ │ ├── gpu_op_tan.cc │ │ │ ├── gpu_op_tanh.cc │ │ │ ├── gpu_op_truncate_div.cc │ │ │ ├── gpu_op_xdivy.cc │ │ │ ├── gpu_op_xlog1py.cc │ │ │ ├── gpu_op_xlogy.cc │ │ │ ├── gpu_op_zeros_like.cc │ │ │ ├── gpu_op_zeta.cc │ │ │ ├── gpu_unary_ops_large_tensor_test.cc │ │ │ ├── gpu_unary_ops_test.cc │ │ │ └── op_definitions │ │ │ │ ├── abs.mlir.tmpl │ │ │ │ ├── acos.mlir.tmpl │ │ │ │ ├── acosh.mlir.tmpl │ │ │ │ ├── add_v2.mlir.tmpl │ │ │ │ ├── angle.mlir.tmpl │ │ │ │ ├── asin.mlir.tmpl │ │ │ │ ├── asinh.mlir.tmpl │ │ │ │ ├── atan.mlir.tmpl │ │ │ │ ├── atan2.mlir.tmpl │ │ │ │ ├── atanh.mlir.tmpl │ │ │ │ ├── bitwise_and.mlir.tmpl │ │ │ │ ├── bitwise_or.mlir.tmpl │ │ │ │ ├── bitwise_xor.mlir.tmpl │ │ │ │ ├── cast.mlir.tmpl │ │ │ │ ├── ceil.mlir.tmpl │ │ │ │ ├── complex.mlir.tmpl │ │ │ │ ├── complex_abs.mlir.tmpl │ │ │ │ ├── conj.mlir.tmpl │ │ │ │ ├── cos.mlir.tmpl │ │ │ │ ├── cosh.mlir.tmpl │ │ │ │ ├── digamma.mlir.tmpl │ │ │ │ ├── div.mlir.tmpl │ │ │ │ ├── div_no_nan.mlir.tmpl │ │ │ │ ├── div_no_nan_cmplx.mlir.tmpl │ │ │ │ ├── elu.mlir.tmpl │ │ │ │ ├── equal.mlir.tmpl │ │ │ │ ├── erf.mlir.tmpl │ │ │ │ ├── erfc.mlir.tmpl │ │ │ │ ├── exp.mlir.tmpl │ │ │ │ ├── expm1.mlir.tmpl │ │ │ │ ├── floor.mlir.tmpl │ │ │ │ ├── floor_div.mlir.tmpl │ │ │ │ ├── floor_div_float.mlir.tmpl │ │ │ │ ├── floor_mod.mlir.tmpl │ │ │ │ ├── floor_mod_float.mlir.tmpl │ │ │ │ ├── floor_mod_unsigned.mlir.tmpl │ │ │ │ ├── greater.mlir.tmpl │ │ │ │ ├── greater_equal.mlir.tmpl │ │ │ │ ├── imag.mlir.tmpl │ │ │ │ ├── invert.mlir.tmpl │ │ │ │ ├── is_finite.mlir.tmpl │ │ │ │ ├── is_inf.mlir.tmpl │ │ │ │ ├── is_nan.mlir.tmpl │ │ │ │ ├── left_shift.mlir.tmpl │ │ │ │ ├── less.mlir.tmpl │ │ │ │ ├── less_equal.mlir.tmpl │ │ │ │ ├── lgamma.mlir.tmpl │ │ │ │ ├── log.mlir.tmpl │ │ │ │ ├── log1p.mlir.tmpl │ │ │ │ ├── logical_and.mlir.tmpl │ │ │ │ ├── logical_not.mlir.tmpl │ │ │ │ ├── logical_or.mlir.tmpl │ │ │ │ ├── maximum.mlir.tmpl │ │ │ │ ├── minimum.mlir.tmpl │ │ │ │ ├── mul.mlir.tmpl │ │ │ │ ├── mul_no_nan.mlir.tmpl │ │ │ │ ├── mul_no_nan_cmplx.mlir.tmpl │ │ │ │ ├── neg.mlir.tmpl │ │ │ │ ├── next_after.mlir.tmpl │ │ │ │ ├── not_equal.mlir.tmpl │ │ │ │ ├── ones_like.mlir.tmpl │ │ │ │ ├── ones_like_cmplx.mlir.tmpl │ │ │ │ ├── ones_like_float.mlir.tmpl │ │ │ │ ├── polygamma.mlir.tmpl │ │ │ │ ├── pow.mlir.tmpl │ │ │ │ ├── real.mlir.tmpl │ │ │ │ ├── reciprocal.mlir.tmpl │ │ │ │ ├── reciprocal_cmplx.mlir.tmpl │ │ │ │ ├── reciprocal_float.mlir.tmpl │ │ │ │ ├── relu.mlir.tmpl │ │ │ │ ├── relu_float.mlir.tmpl │ │ │ │ ├── relu_grad.mlir.tmpl │ │ │ │ ├── right_shift.mlir.tmpl │ │ │ │ ├── right_shift_unsigned.mlir.tmpl │ │ │ │ ├── rint.mlir.tmpl │ │ │ │ ├── round.mlir.tmpl │ │ │ │ ├── round_float.mlir.tmpl │ │ │ │ ├── rsqrt.mlir.tmpl │ │ │ │ ├── select_v2.mlir.tmpl │ │ │ │ ├── selu.mlir.tmpl │ │ │ │ ├── sigmoid.mlir.tmpl │ │ │ │ ├── sign.mlir.tmpl │ │ │ │ ├── sin.mlir.tmpl │ │ │ │ ├── sinh.mlir.tmpl │ │ │ │ ├── softplus_f16.mlir.tmpl │ │ │ │ ├── softplus_f32.mlir.tmpl │ │ │ │ ├── softplus_f64.mlir.tmpl │ │ │ │ ├── softsign.mlir.tmpl │ │ │ │ ├── sqrt.mlir.tmpl │ │ │ │ ├── square.mlir.tmpl │ │ │ │ ├── squared_difference.mlir.tmpl │ │ │ │ ├── sub.mlir.tmpl │ │ │ │ ├── tan.mlir.tmpl │ │ │ │ ├── tanh.mlir.tmpl │ │ │ │ ├── truncate_div.mlir.tmpl │ │ │ │ ├── truncate_div_float.mlir.tmpl │ │ │ │ ├── xdivy.mlir.tmpl │ │ │ │ ├── xdivy_cmplx.mlir.tmpl │ │ │ │ ├── xlog1py.mlir.tmpl │ │ │ │ ├── xlog1py_cmplx.mlir.tmpl │ │ │ │ ├── xlogy.mlir.tmpl │ │ │ │ ├── xlogy_cmplx.mlir.tmpl │ │ │ │ ├── zeros_like.mlir.tmpl │ │ │ │ ├── zeros_like_cmplx.mlir.tmpl │ │ │ │ ├── zeros_like_float.mlir.tmpl │ │ │ │ └── zeta.mlir.tmpl │ │ ├── multinomial_op.cc │ │ ├── multinomial_op.h │ │ ├── multinomial_op_gpu.cu.cc │ │ ├── multinomial_op_test.cc │ │ ├── mutex_ops.cc │ │ ├── nccl_ops.cc │ │ ├── nextafter_op.cc │ │ ├── nextafter_op.h │ │ ├── nextafter_op_gpu.cu.cc │ │ ├── nn_ops_test.cc │ │ ├── no_op.cc │ │ ├── no_op.h │ │ ├── nth_element_op.cc │ │ ├── nth_element_op.h │ │ ├── numeric_options_utils.h │ │ ├── one_hot_op.cc │ │ ├── one_hot_op.h │ │ ├── one_hot_op_gpu.cu.cc │ │ ├── one_hot_op_test.cc │ │ ├── ops_testutil.cc │ │ ├── ops_testutil.h │ │ ├── ops_testutil_test.cc │ │ ├── ops_util.h │ │ ├── ops_util_test.cc │ │ ├── pack_op.cc │ │ ├── pad_op.cc │ │ ├── pad_op.h │ │ ├── pad_op_gpu.cu.cc │ │ ├── padding_fifo_queue.cc │ │ ├── padding_fifo_queue.h │ │ ├── padding_fifo_queue_op.cc │ │ ├── parameterized_truncated_normal_op.cc │ │ ├── parameterized_truncated_normal_op.h │ │ ├── parameterized_truncated_normal_op_gpu.cu.cc │ │ ├── parameterized_truncated_normal_op_test.cc │ │ ├── parse_tensor_op.cc │ │ ├── parse_tensor_test.cc │ │ ├── partitioned_function_ops.cc │ │ ├── partitioned_function_ops.h │ │ ├── poisson-loss.h │ │ ├── pooling_ops_3d.cc │ │ ├── pooling_ops_3d.h │ │ ├── pooling_ops_3d_gpu.cu.cc │ │ ├── pooling_ops_3d_gpu.h │ │ ├── pooling_ops_common.cc │ │ ├── pooling_ops_common.h │ │ ├── pooling_ops_common_gpu.h │ │ ├── population_count_op.cc │ │ ├── population_count_op.h │ │ ├── population_count_op_gpu.cu.cc │ │ ├── priority_queue.cc │ │ ├── priority_queue.h │ │ ├── priority_queue_op.cc │ │ ├── quantization_utils.cc │ │ ├── quantization_utils.h │ │ ├── quantization_utils_test.cc │ │ ├── quantize_and_dequantize_op.cc │ │ ├── quantize_and_dequantize_op.h │ │ ├── quantize_and_dequantize_op_gpu.cu.cc │ │ ├── quantize_and_dequantize_op_test.cc │ │ ├── quantize_down_and_shrink_range.cc │ │ ├── quantize_down_and_shrink_range_op_test.cc │ │ ├── quantize_op.cc │ │ ├── quantize_op_test.cc │ │ ├── quantized_activation_ops.cc │ │ ├── quantized_activation_ops_test.cc │ │ ├── quantized_add_op.cc │ │ ├── quantized_add_op_test.cc │ │ ├── quantized_batch_norm_op.cc │ │ ├── quantized_batch_norm_op_test.cc │ │ ├── quantized_bias_add_op.cc │ │ ├── quantized_bias_add_op_test.cc │ │ ├── quantized_concat_op.cc │ │ ├── quantized_concat_op_test.cc │ │ ├── quantized_conv_ops.cc │ │ ├── quantized_conv_ops_test.cc │ │ ├── quantized_instance_norm.cc │ │ ├── quantized_instance_norm_test.cc │ │ ├── quantized_matmul_op.cc │ │ ├── quantized_matmul_op_test.cc │ │ ├── quantized_mul_op.cc │ │ ├── quantized_mul_op_test.cc │ │ ├── quantized_pooling_ops.cc │ │ ├── quantized_pooling_ops_test.cc │ │ ├── quantized_reshape_op.cc │ │ ├── quantized_reshape_op_test.cc │ │ ├── quantized_resize_bilinear_op.cc │ │ ├── quantized_resize_bilinear_op_test.cc │ │ ├── queue_base.cc │ │ ├── queue_base.h │ │ ├── queue_op.cc │ │ ├── queue_op.h │ │ ├── queue_ops.cc │ │ ├── ragged_cross_op.cc │ │ ├── ragged_fill_empty_rows_op.cc │ │ ├── ragged_fill_empty_rows_op_test.cc │ │ ├── ragged_gather_op.cc │ │ ├── ragged_gather_op_test.cc │ │ ├── ragged_range_op.cc │ │ ├── ragged_range_op_test.cc │ │ ├── ragged_tensor_from_variant_op.cc │ │ ├── ragged_tensor_from_variant_op_test.cc │ │ ├── ragged_tensor_to_sparse_kernel.cc │ │ ├── ragged_tensor_to_sparse_kernel_test.cc │ │ ├── ragged_tensor_to_tensor_op.cc │ │ ├── ragged_tensor_to_tensor_op_test.cc │ │ ├── ragged_tensor_to_variant_op.cc │ │ ├── ragged_tensor_to_variant_op_large_data_test.cc │ │ ├── ragged_tensor_to_variant_op_test.cc │ │ ├── ragged_tensor_to_variant_op_test.h │ │ ├── ragged_tensor_variant.cc │ │ ├── ragged_tensor_variant.h │ │ ├── ragged_utils.h │ │ ├── random_binomial_op.cc │ │ ├── random_binomial_op.h │ │ ├── random_binomial_op_test.cc │ │ ├── random_index_shuffle.cc │ │ ├── random_index_shuffle.h │ │ ├── random_index_shuffle_ops.cc │ │ ├── random_index_shuffle_test.cc │ │ ├── random_op.cc │ │ ├── random_op.h │ │ ├── random_op_cpu.h │ │ ├── random_op_gpu.cu.cc │ │ ├── random_op_gpu.h │ │ ├── random_op_test.cc │ │ ├── random_ops_util.h │ │ ├── random_poisson_op.cc │ │ ├── random_poisson_op.h │ │ ├── random_poisson_op_test.cc │ │ ├── random_shuffle_op.cc │ │ ├── random_shuffle_queue_op.cc │ │ ├── range_sampler.cc │ │ ├── range_sampler.h │ │ ├── range_sampler_test.cc │ │ ├── reader_ops.cc │ │ ├── record_input_op.cc │ │ ├── record_yielder.cc │ │ ├── record_yielder.h │ │ ├── reduce_join_op.cc │ │ ├── reduction_gpu_kernels.cu.h │ │ ├── reduction_ops.h │ │ ├── reduction_ops_all.cc │ │ ├── reduction_ops_any.cc │ │ ├── reduction_ops_common.cc │ │ ├── reduction_ops_common.h │ │ ├── reduction_ops_common_gpu.h │ │ ├── reduction_ops_euclidean.cc │ │ ├── reduction_ops_gpu_bfloat16.cu.cc │ │ ├── reduction_ops_gpu_bool.cu.cc │ │ ├── reduction_ops_gpu_complex128.cu.cc │ │ ├── reduction_ops_gpu_complex64.cu.cc │ │ ├── reduction_ops_gpu_double.cu.cc │ │ ├── reduction_ops_gpu_float.cu.cc │ │ ├── reduction_ops_gpu_int.cu.cc │ │ ├── reduction_ops_half_mean_sum.cu.cc │ │ ├── reduction_ops_half_prod_max_min.cu.cc │ │ ├── reduction_ops_max.cc │ │ ├── reduction_ops_mean.cc │ │ ├── reduction_ops_min.cc │ │ ├── reduction_ops_prod.cc │ │ ├── reduction_ops_sum.cc │ │ ├── reduction_ops_test.cc │ │ ├── redux_functor.h │ │ ├── reference_gemm.h │ │ ├── regex_full_match_op.cc │ │ ├── regex_replace_op.cc │ │ ├── regex_replace_op_test.cc │ │ ├── relu_op.cc │ │ ├── relu_op.h │ │ ├── relu_op_functor.h │ │ ├── relu_op_gpu.cu.cc │ │ ├── requantization_range_op.cc │ │ ├── requantization_range_op_test.cc │ │ ├── requantize.cc │ │ ├── requantize_op_test.cc │ │ ├── reshape_op.cc │ │ ├── reshape_op.h │ │ ├── reshape_util.cc │ │ ├── reshape_util.h │ │ ├── reshape_util_gpu.cu.cc │ │ ├── resource_ops_test.cc │ │ ├── resource_variable_ops.cc │ │ ├── resource_variable_ops.h │ │ ├── resource_variable_util.cc │ │ ├── resource_variable_util.h │ │ ├── restore_op.cc │ │ ├── restore_op_test.cc │ │ ├── restore_v2_op_test.cc │ │ ├── reverse_op.cc │ │ ├── reverse_op.h │ │ ├── reverse_op_gpu.cu.cc │ │ ├── reverse_op_test.cc │ │ ├── reverse_sequence_op.cc │ │ ├── reverse_sequence_op.h │ │ ├── reverse_sequence_op_gpu.cu.cc │ │ ├── risc │ │ │ ├── BUILD │ │ │ └── experimental │ │ │ │ ├── BUILD │ │ │ │ ├── risc_abs_op.cc │ │ │ │ ├── risc_add_op.cc │ │ │ │ ├── risc_binary_arithmetic_op.cc │ │ │ │ ├── risc_binary_comparison_op.cc │ │ │ │ ├── risc_bitcast_op.cc │ │ │ │ ├── risc_broadcast_op.cc │ │ │ │ ├── risc_cast_op.cc │ │ │ │ ├── risc_ceil_op.cc │ │ │ │ ├── risc_cholesky_op.cc │ │ │ │ ├── risc_concat_op.cc │ │ │ │ ├── risc_condition_op.cc │ │ │ │ ├── risc_conv_op.cc │ │ │ │ ├── risc_cos_op.cc │ │ │ │ ├── risc_div_op.cc │ │ │ │ ├── risc_dot_op.cc │ │ │ │ ├── risc_exp_op.cc │ │ │ │ ├── risc_fft_op.cc │ │ │ │ ├── risc_floor_op.cc │ │ │ │ ├── risc_gather_op.cc │ │ │ │ ├── risc_imag_op.cc │ │ │ │ ├── risc_is_finite_op.cc │ │ │ │ ├── risc_log_op.cc │ │ │ │ ├── risc_logical_and_op.cc │ │ │ │ ├── risc_logical_not_op.cc │ │ │ │ ├── risc_logical_or_op.cc │ │ │ │ ├── risc_max_op.cc │ │ │ │ ├── risc_min_op.cc │ │ │ │ ├── risc_mul_op.cc │ │ │ │ ├── risc_neg_op.cc │ │ │ │ ├── risc_pad_op.cc │ │ │ │ ├── risc_pool_op.cc │ │ │ │ ├── risc_pow_op.cc │ │ │ │ ├── risc_random_uniform_op.cc │ │ │ │ ├── risc_real_op.cc │ │ │ │ ├── risc_reduce_op.cc │ │ │ │ ├── risc_rem_op.cc │ │ │ │ ├── risc_reshape_op.cc │ │ │ │ ├── risc_reverse_op.cc │ │ │ │ ├── risc_scatter_op.cc │ │ │ │ ├── risc_shape_op.cc │ │ │ │ ├── risc_sign_op.cc │ │ │ │ ├── risc_slice_op.cc │ │ │ │ ├── risc_sort_op.cc │ │ │ │ ├── risc_squeeze_op.cc │ │ │ │ ├── risc_sub_op.cc │ │ │ │ ├── risc_transpose_op.cc │ │ │ │ ├── risc_triangular_solve_op.cc │ │ │ │ ├── risc_unary_op.cc │ │ │ │ └── risc_while_op.cc │ │ ├── rnn │ │ │ ├── BUILD │ │ │ ├── blas_gemm.cc │ │ │ ├── blas_gemm.h │ │ │ ├── gru_ops.cc │ │ │ ├── gru_ops.h │ │ │ ├── gru_ops_gpu.cu.cc │ │ │ ├── lstm_ops.cc │ │ │ ├── lstm_ops.h │ │ │ └── lstm_ops_gpu.cu.cc │ │ ├── roll_op.cc │ │ ├── roll_op.h │ │ ├── roll_op_gpu.cu.cc │ │ ├── roll_op_test.cc │ │ ├── save_op.cc │ │ ├── save_op_test.cc │ │ ├── save_restore_tensor.cc │ │ ├── save_restore_tensor.h │ │ ├── save_restore_v2_ops.cc │ │ ├── save_v2_op_test.cc │ │ ├── scan_ops.cc │ │ ├── scan_ops.h │ │ ├── scan_ops_gpu.h │ │ ├── scan_ops_gpu_bfloat16.cu.cc │ │ ├── scan_ops_gpu_double.cu.cc │ │ ├── scan_ops_gpu_float.cu.cc │ │ ├── scan_ops_gpu_half.cu.cc │ │ ├── scan_ops_gpu_int.cu.cc │ │ ├── scan_ops_test.cc │ │ ├── scatter_functor.cc │ │ ├── scatter_functor.h │ │ ├── scatter_functor_gpu.cu.cc │ │ ├── scatter_functor_gpu.cu.h │ │ ├── scatter_nd_op.cc │ │ ├── scatter_nd_op.h │ │ ├── scatter_nd_op_cpu_impl.h │ │ ├── scatter_nd_op_cpu_impl_0.cc │ │ ├── scatter_nd_op_cpu_impl_1.cc │ │ ├── scatter_nd_op_cpu_impl_2.cc │ │ ├── scatter_nd_op_cpu_impl_3.cc │ │ ├── scatter_nd_op_cpu_impl_4.cc │ │ ├── scatter_nd_op_cpu_impl_5.cc │ │ ├── scatter_nd_op_cpu_impl_6.cc │ │ ├── scatter_nd_op_cpu_impl_7.cc │ │ ├── scatter_nd_op_gpu.cu.cc │ │ ├── scatter_nd_op_test.cc │ │ ├── scatter_nd_util.cc │ │ ├── scatter_nd_util.h │ │ ├── scatter_op.cc │ │ ├── scatter_op_gpu.cu.cc │ │ ├── scatter_op_test.cc │ │ ├── scoped_allocator_ops.cc │ │ ├── scoped_allocator_ops_test.cc │ │ ├── sdca_internal.cc │ │ ├── sdca_internal.h │ │ ├── sdca_ops.cc │ │ ├── sdca_ops_test.cc │ │ ├── searchsorted_op.cc │ │ ├── searchsorted_op.h │ │ ├── searchsorted_op_gpu.cu.cc │ │ ├── segment_reduction_ops.h │ │ ├── segment_reduction_ops_gpu.cu.h │ │ ├── segment_reduction_ops_gpu_0.cu.cc │ │ ├── segment_reduction_ops_gpu_1.cu.cc │ │ ├── segment_reduction_ops_gpu_2.cu.cc │ │ ├── segment_reduction_ops_impl.h │ │ ├── segment_reduction_ops_impl_1.cc │ │ ├── segment_reduction_ops_impl_2.cc │ │ ├── segment_reduction_ops_impl_3.cc │ │ ├── segment_reduction_ops_impl_3b.cc │ │ ├── segment_reduction_ops_impl_4.cc │ │ ├── segment_reduction_ops_impl_5.cc │ │ ├── segment_reduction_ops_test.cc │ │ ├── sendrecv_ops.cc │ │ ├── sendrecv_ops.h │ │ ├── sendrecv_ops_test.cc │ │ ├── sequence_ops.cc │ │ ├── sequence_ops.h │ │ ├── sequence_ops_gpu.cu.cc │ │ ├── sequence_ops_test.cc │ │ ├── serialize_sparse_op.cc │ │ ├── session_ops.cc │ │ ├── set_kernels.cc │ │ ├── shape_ops.cc │ │ ├── shape_ops.h │ │ ├── shape_ops_test.cc │ │ ├── shuffle_common.h │ │ ├── slice_op.cc │ │ ├── slice_op.h │ │ ├── slice_op_cpu_impl.h │ │ ├── slice_op_cpu_impl_1.cc │ │ ├── slice_op_cpu_impl_2.cc │ │ ├── slice_op_cpu_impl_3.cc │ │ ├── slice_op_cpu_impl_4.cc │ │ ├── slice_op_cpu_impl_5.cc │ │ ├── slice_op_cpu_impl_6.cc │ │ ├── slice_op_cpu_impl_7.cc │ │ ├── slice_op_cpu_impl_8.cc │ │ ├── slice_op_gpu.cu.cc │ │ ├── slice_op_test.cc │ │ ├── smooth-hinge-loss.h │ │ ├── snapshot_op.cc │ │ ├── snapshot_op.h │ │ ├── snapshot_op_gpu.cu.cc │ │ ├── sobol_op.cc │ │ ├── softmax_op.cc │ │ ├── softmax_op_functor.h │ │ ├── softmax_op_gpu.cu.cc │ │ ├── softplus_op.cc │ │ ├── softplus_op.h │ │ ├── softplus_op_gpu.cu.cc │ │ ├── softsign_op.cc │ │ ├── softsign_op.h │ │ ├── softsign_op_gpu.cu.cc │ │ ├── spacetobatch_benchmark_test.cc │ │ ├── spacetobatch_functor.cc │ │ ├── spacetobatch_functor.h │ │ ├── spacetobatch_functor_gpu.cu.cc │ │ ├── spacetobatch_op.cc │ │ ├── spacetodepth_op.cc │ │ ├── spacetodepth_op.h │ │ ├── spacetodepth_op_gpu.cu.cc │ │ ├── sparse │ │ │ ├── BUILD │ │ │ ├── add_op.cc │ │ │ ├── conj_op.cc │ │ │ ├── csr_sparse_matrix_to_dense_op.cc │ │ │ ├── csr_sparse_matrix_to_sparse_tensor_op.cc │ │ │ ├── dense_to_csr_sparse_matrix_op.cc │ │ │ ├── kernels.cc │ │ │ ├── kernels.h │ │ │ ├── kernels_gpu.cu.cc │ │ │ ├── kernels_test.cc │ │ │ ├── mat_mul_op.cc │ │ │ ├── mul_op.cc │ │ │ ├── nnz_op.cc │ │ │ ├── softmax_op.cc │ │ │ ├── sparse_cholesky_op.cc │ │ │ ├── sparse_mat_mul_op.cc │ │ │ ├── sparse_matrix.cc │ │ │ ├── sparse_matrix.h │ │ │ ├── sparse_matrix_components_op.cc │ │ │ ├── sparse_ordering_amd_op.cc │ │ │ ├── sparse_tensor_to_csr_sparse_matrix_op.cc │ │ │ ├── transpose_op.cc │ │ │ ├── transpose_op.h │ │ │ ├── zeros_op.cc │ │ │ └── zeros_op.h │ │ ├── sparse_add_grad_op.cc │ │ ├── sparse_add_op.cc │ │ ├── sparse_add_op_test.cc │ │ ├── sparse_concat_op.cc │ │ ├── sparse_concat_op.h │ │ ├── sparse_concat_op_gpu.cu.cc │ │ ├── sparse_conditional_accumulator.h │ │ ├── sparse_conditional_accumulator_op.cc │ │ ├── sparse_cross_op.cc │ │ ├── sparse_dense_binary_op_shared.cc │ │ ├── sparse_dense_binary_op_shared_test.cc │ │ ├── sparse_fill_empty_rows_op.cc │ │ ├── sparse_fill_empty_rows_op_test.cc │ │ ├── sparse_matmul_op.cc │ │ ├── sparse_matmul_op.h │ │ ├── sparse_matmul_op_test.cc │ │ ├── sparse_reduce_op.cc │ │ ├── sparse_reduce_sum_op_test.cc │ │ ├── sparse_reorder_op.cc │ │ ├── sparse_reorder_op.h │ │ ├── sparse_reorder_op_gpu.cu.cc │ │ ├── sparse_reshape_op.cc │ │ ├── sparse_slice_grad_op.cc │ │ ├── sparse_slice_grad_op.h │ │ ├── sparse_slice_grad_op_gpu.cu.cc │ │ ├── sparse_slice_op.cc │ │ ├── sparse_slice_op.h │ │ ├── sparse_slice_op_gpu.cu.cc │ │ ├── sparse_softmax_op.cc │ │ ├── sparse_sparse_binary_op_shared.cc │ │ ├── sparse_split_op.cc │ │ ├── sparse_split_op.h │ │ ├── sparse_split_op_gpu.cu.cc │ │ ├── sparse_tensor_dense_add_op.cc │ │ ├── sparse_tensor_dense_add_op.h │ │ ├── sparse_tensor_dense_matmul_op.cc │ │ ├── sparse_tensor_dense_matmul_op.h │ │ ├── sparse_tensor_dense_matmul_op_gpu.cu.cc │ │ ├── sparse_tensor_dense_matmul_op_test.cc │ │ ├── sparse_tensors_map_ops.cc │ │ ├── sparse_to_dense_op.cc │ │ ├── sparse_to_dense_op_gpu.cu.cc │ │ ├── sparse_to_dense_op_gpu.h │ │ ├── sparse_to_dense_op_test.cc │ │ ├── sparse_utils.cc │ │ ├── sparse_utils.h │ │ ├── sparse_utils_test.cc │ │ ├── sparse_xent_op.cc │ │ ├── sparse_xent_op.h │ │ ├── sparse_xent_op_gpu.cu.cc │ │ ├── sparse_xent_op_test.cc │ │ ├── special_math │ │ │ ├── BUILD │ │ │ ├── special_math_op_bessel.cc │ │ │ ├── special_math_op_dawsn.cc │ │ │ ├── special_math_op_expint.cc │ │ │ ├── special_math_op_fresnel_cos.cc │ │ │ ├── special_math_op_fresnel_sin.cc │ │ │ ├── special_math_op_gpu_bessel.cu.cc │ │ │ ├── special_math_op_gpu_dawsn.cu.cc │ │ │ ├── special_math_op_gpu_expint.cu.cc │ │ │ ├── special_math_op_gpu_fresnel.cu.cc │ │ │ ├── special_math_op_gpu_spence.cu.cc │ │ │ ├── special_math_op_misc_impl.h │ │ │ └── special_math_op_spence.cc │ │ ├── spectrogram.cc │ │ ├── spectrogram.h │ │ ├── spectrogram_convert_test_data.cc │ │ ├── spectrogram_op.cc │ │ ├── spectrogram_op_test.cc │ │ ├── spectrogram_test.cc │ │ ├── spectrogram_test_data │ │ │ ├── BUILD │ │ │ ├── README │ │ │ ├── short_test_segment.wav │ │ │ ├── short_test_segment_spectrogram.csv.bin │ │ │ └── short_test_segment_spectrogram_400_200.csv.bin │ │ ├── spectrogram_test_utils.cc │ │ ├── spectrogram_test_utils.h │ │ ├── split_lib.h │ │ ├── split_lib_cpu.cc │ │ ├── split_lib_gpu.cu.cc │ │ ├── split_lib_gpu.h │ │ ├── split_op.cc │ │ ├── split_op_test.cc │ │ ├── split_v_op.cc │ │ ├── split_v_op_test.cc │ │ ├── squared-loss.h │ │ ├── stack.cc │ │ ├── stack.h │ │ ├── stack_ops.cc │ │ ├── stage_op.cc │ │ ├── stateful_random_ops.cc │ │ ├── stateful_random_ops.h │ │ ├── stateful_random_ops_cpu_gpu.h │ │ ├── stateful_random_ops_gpu.cu.cc │ │ ├── stateless_random_gamma_op.cc │ │ ├── stateless_random_gamma_op.h │ │ ├── stateless_random_gamma_op_gpu.cu.cc │ │ ├── stateless_random_ops.cc │ │ ├── stateless_random_ops.h │ │ ├── stateless_random_ops_v2.cc │ │ ├── stateless_random_ops_v2.h │ │ ├── stateless_random_ops_v2_util.h │ │ ├── stateless_shuffle.cc │ │ ├── stochastic_cast_op.cc │ │ ├── stochastic_cast_op.h │ │ ├── stochastic_cast_op_test.cc │ │ ├── strided_slice_op.cc │ │ ├── strided_slice_op.h │ │ ├── strided_slice_op_define_grad.cc │ │ ├── strided_slice_op_gpu_bool.cu.cc │ │ ├── strided_slice_op_gpu_complex.cu.cc │ │ ├── strided_slice_op_gpu_impl.h │ │ ├── strided_slice_op_gpu_int.cu.cc │ │ ├── strided_slice_op_gpu_number_types.cu.cc │ │ ├── strided_slice_op_impl.h │ │ ├── strided_slice_op_inst_0.cc │ │ ├── strided_slice_op_inst_1.cc │ │ ├── strided_slice_op_inst_2.cc │ │ ├── strided_slice_op_inst_3.cc │ │ ├── strided_slice_op_inst_4.cc │ │ ├── strided_slice_op_inst_5.cc │ │ ├── strided_slice_op_inst_6.cc │ │ ├── strided_slice_op_inst_7.cc │ │ ├── strided_slice_op_inst_8.cc │ │ ├── strided_slice_op_test.cc │ │ ├── string_format_op.cc │ │ ├── string_format_op_test.cc │ │ ├── string_join_op.cc │ │ ├── string_length_op.cc │ │ ├── string_lower_op.cc │ │ ├── string_ngrams_op.cc │ │ ├── string_ngrams_op_test.cc │ │ ├── string_split_op.cc │ │ ├── string_split_op_test.cc │ │ ├── string_strip_op.cc │ │ ├── string_to_hash_bucket_fast_op.cc │ │ ├── string_to_hash_bucket_fast_op.h │ │ ├── string_to_hash_bucket_op.cc │ │ ├── string_to_hash_bucket_op.h │ │ ├── string_to_number_op.cc │ │ ├── string_upper_op.cc │ │ ├── string_util.cc │ │ ├── string_util.h │ │ ├── substr_op.cc │ │ ├── substr_op_test.cc │ │ ├── summary_audio_op.cc │ │ ├── summary_audio_op_test.cc │ │ ├── summary_image_op.cc │ │ ├── summary_image_op_test.cc │ │ ├── summary_interface.h │ │ ├── summary_kernels.cc │ │ ├── summary_op_test.cc │ │ ├── summary_tensor_op.cc │ │ ├── summary_tensor_op_test.cc │ │ ├── sync_ops.cc │ │ ├── tensor_array.cc │ │ ├── tensor_array.h │ │ ├── tensor_array_ops.cc │ │ ├── tensor_cord.cc │ │ ├── tensor_cord.h │ │ ├── tensor_cord_test.cc │ │ ├── tensor_flag_utils.cc │ │ ├── tensor_flag_utils.h │ │ ├── tensor_flag_utils_test.cc │ │ ├── tensor_list.cc │ │ ├── tensor_list.h │ │ ├── tensor_list_util.cc │ │ ├── tensor_list_util.h │ │ ├── tensor_map.cc │ │ ├── tensor_map.h │ │ ├── tensor_map_test.cc │ │ ├── tensor_to_hash_bucket_op.cc │ │ ├── tensor_to_hash_bucket_op.h │ │ ├── tensor_to_hash_bucket_op_gpu.cu.cc │ │ ├── text_line_reader_op.cc │ │ ├── tf_record_reader_op.cc │ │ ├── tile_functor.h │ │ ├── tile_functor_cpu.h │ │ ├── tile_functor_cpu_bfloat16.cc │ │ ├── tile_functor_cpu_bool.cc │ │ ├── tile_functor_cpu_complex128.cc │ │ ├── tile_functor_cpu_complex64.cc │ │ ├── tile_functor_cpu_double.cc │ │ ├── tile_functor_cpu_float.cc │ │ ├── tile_functor_cpu_half.cc │ │ ├── tile_functor_cpu_int16.cc │ │ ├── tile_functor_cpu_int32.cc │ │ ├── tile_functor_cpu_int64.cc │ │ ├── tile_functor_cpu_int8.cc │ │ ├── tile_functor_cpu_tstring.cc │ │ ├── tile_functor_cpu_uint32.cc │ │ ├── tile_functor_cpu_uint64.cc │ │ ├── tile_functor_cpu_uint8.cc │ │ ├── tile_functor_cpu_variant.cc │ │ ├── tile_functor_gpu.h │ │ ├── tile_functor_gpu_bfloat16.cu.cc │ │ ├── tile_functor_gpu_bool.cu.cc │ │ ├── tile_functor_gpu_complex128.cu.cc │ │ ├── tile_functor_gpu_complex64.cu.cc │ │ ├── tile_functor_gpu_double.cu.cc │ │ ├── tile_functor_gpu_float.cu.cc │ │ ├── tile_functor_gpu_half.cu.cc │ │ ├── tile_functor_gpu_int16.cu.cc │ │ ├── tile_functor_gpu_int32.cu.cc │ │ ├── tile_functor_gpu_int64.cu.cc │ │ ├── tile_functor_sycl.cc │ │ ├── tile_ops.cc │ │ ├── tile_ops_cpu_impl.h │ │ ├── tile_ops_cpu_impl_1.cc │ │ ├── tile_ops_cpu_impl_2.cc │ │ ├── tile_ops_cpu_impl_3.cc │ │ ├── tile_ops_cpu_impl_4.cc │ │ ├── tile_ops_cpu_impl_5.cc │ │ ├── tile_ops_cpu_impl_6.cc │ │ ├── tile_ops_cpu_impl_7.cc │ │ ├── tile_ops_gpu_impl.h │ │ ├── tile_ops_gpu_impl_1.cu.cc │ │ ├── tile_ops_gpu_impl_2.cu.cc │ │ ├── tile_ops_gpu_impl_3.cu.cc │ │ ├── tile_ops_gpu_impl_4.cu.cc │ │ ├── tile_ops_gpu_impl_5.cu.cc │ │ ├── tile_ops_gpu_impl_6.cu.cc │ │ ├── tile_ops_gpu_impl_7.cu.cc │ │ ├── tile_ops_gpu_impl_8.cu.cc │ │ ├── tile_ops_impl.h │ │ ├── topk_op.cc │ │ ├── topk_op.h │ │ ├── topk_op_gpu.h │ │ ├── topk_op_gpu_bfloat16.cu.cc │ │ ├── topk_op_gpu_double.cu.cc │ │ ├── topk_op_gpu_float.cu.cc │ │ ├── topk_op_gpu_half.cu.cc │ │ ├── topk_op_gpu_int16.cu.cc │ │ ├── topk_op_gpu_int32.cu.cc │ │ ├── topk_op_gpu_int64.cu.cc │ │ ├── topk_op_gpu_int8.cu.cc │ │ ├── topk_op_gpu_uint16.cu.cc │ │ ├── topk_op_gpu_uint32.cu.cc │ │ ├── topk_op_gpu_uint64.cu.cc │ │ ├── topk_op_gpu_uint8.cu.cc │ │ ├── training_op_helpers.cc │ │ ├── training_op_helpers.h │ │ ├── training_ops.cc │ │ ├── training_ops.h │ │ ├── training_ops_gpu.cu.cc │ │ ├── training_ops_test.cc │ │ ├── transpose_functor.h │ │ ├── transpose_functor_cpu.cc │ │ ├── transpose_functor_gpu.cu.cc │ │ ├── transpose_op.cc │ │ ├── transpose_op.h │ │ ├── transpose_util_test.cc │ │ ├── typed_conditional_accumulator_base.h │ │ ├── typed_queue.h │ │ ├── unary_ops_composition.cc │ │ ├── unary_ops_composition_test.cc │ │ ├── unicode_ops.cc │ │ ├── unicode_script_op.cc │ │ ├── uniform_quant_ops │ │ │ ├── BUILD │ │ │ ├── math_utils.cc │ │ │ ├── math_utils.h │ │ │ ├── math_utils_test.cc │ │ │ ├── tensor_utils.cc │ │ │ ├── tensor_utils.h │ │ │ ├── tensor_utils_test.cc │ │ │ ├── uniform_dequantize_op.cc │ │ │ ├── uniform_dequantize_op_test.cc │ │ │ ├── uniform_quantize_op.cc │ │ │ ├── uniform_quantize_op_test.cc │ │ │ ├── uniform_quantized_add_op.cc │ │ │ ├── uniform_quantized_add_op_test.cc │ │ │ ├── uniform_quantized_clip_by_value_op.cc │ │ │ ├── uniform_quantized_clip_by_value_op_test.cc │ │ │ ├── uniform_quantized_convolution_ops.cc │ │ │ ├── uniform_quantized_convolution_ops_test.cc │ │ │ ├── uniform_quantized_dot_ops.cc │ │ │ ├── uniform_quantized_dot_ops_test.cc │ │ │ ├── uniform_requantize_op.cc │ │ │ └── uniform_requantize_op_test.cc │ │ ├── unique_op.cc │ │ ├── unique_op_gpu.cu.h │ │ ├── unique_op_gpu_0.cu.cc │ │ ├── unique_op_gpu_1.cu.cc │ │ ├── unique_op_gpu_2.cu.cc │ │ ├── unique_op_gpu_3.cu.cc │ │ ├── unique_op_test.cc │ │ ├── unpack_op.cc │ │ ├── unravel_index_op.cc │ │ ├── unsorted_segment_join_op.cc │ │ ├── variable_ops.cc │ │ ├── variable_ops.h │ │ ├── variable_ops_test.cc │ │ ├── variant_ops_util.cc │ │ ├── variant_ops_util.h │ │ ├── where_op.cc │ │ ├── where_op.h │ │ ├── where_op_gpu.cu.h │ │ ├── where_op_gpu_impl_1.cu.cc │ │ ├── where_op_gpu_impl_2.cu.cc │ │ ├── where_op_gpu_impl_3.cu.cc │ │ ├── where_op_gpu_impl_4.cu.cc │ │ ├── where_op_gpu_impl_5.cu.cc │ │ ├── where_op_gpu_impl_6.cu.cc │ │ ├── where_op_gpu_impl_7.cu.cc │ │ ├── where_op_gpu_impl_8.cu.cc │ │ ├── while_op_test.cc │ │ ├── whole_file_read_ops.cc │ │ ├── winograd_transform.h │ │ ├── word2vec_kernels.cc │ │ ├── xent_op.cc │ │ ├── xent_op.h │ │ ├── xent_op_gpu.cu.cc │ │ └── xent_op_test.cc │ ├── lib │ │ ├── bfloat16 │ │ │ ├── BUILD │ │ │ └── bfloat16.h │ │ ├── bmp │ │ │ ├── BUILD │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── grayscale_small.bmp │ │ │ │ ├── grayscale_small_3channels.bmp │ │ │ │ ├── grayscale_small_4channels.bmp │ │ │ │ ├── lena.bmp │ │ │ │ ├── rgb_small.bmp │ │ │ │ ├── rgb_small_255.bmp │ │ │ │ ├── rgba_small.bmp │ │ │ │ └── rgba_small_255.bmp │ │ ├── core │ │ │ ├── BUILD │ │ │ ├── arena.cc │ │ │ ├── arena.h │ │ │ ├── arena_test.cc │ │ │ ├── bitmap.h │ │ │ ├── bits.h │ │ │ ├── coding.h │ │ │ ├── coding_test.cc │ │ │ ├── error_codes.proto │ │ │ ├── errors.h │ │ │ ├── notification.h │ │ │ ├── notification_test.cc │ │ │ ├── raw_coding.h │ │ │ ├── refcount.h │ │ │ ├── status.h │ │ │ ├── status_test.cc │ │ │ ├── status_test_util.h │ │ │ ├── stringpiece.h │ │ │ ├── threadpool.h │ │ │ ├── threadpool_interface.h │ │ │ ├── threadpool_options.h │ │ │ └── threadpool_test.cc │ │ ├── db │ │ │ ├── BUILD │ │ │ ├── snapfn.cc │ │ │ ├── sqlite.cc │ │ │ ├── sqlite.h │ │ │ └── sqlite_test.cc │ │ ├── gif │ │ │ ├── BUILD │ │ │ ├── gif_io.cc │ │ │ ├── gif_io.h │ │ │ ├── gif_io_test.cc │ │ │ └── testdata │ │ │ │ ├── 3g_multiframe.gif │ │ │ │ ├── BUILD │ │ │ │ ├── lena.gif │ │ │ │ ├── optimized.gif │ │ │ │ ├── pendulum_sm.gif │ │ │ │ ├── pendulum_sm_frame0.png │ │ │ │ ├── pendulum_sm_frame1.png │ │ │ │ ├── pendulum_sm_frame2.png │ │ │ │ ├── red_black.gif │ │ │ │ ├── scan.gif │ │ │ │ └── squares.gif │ │ ├── gtl │ │ │ ├── BUILD │ │ │ ├── array_slice.h │ │ │ ├── cleanup.h │ │ │ ├── cleanup_test.cc │ │ │ ├── compactptrset.h │ │ │ ├── edit_distance.h │ │ │ ├── edit_distance_test.cc │ │ │ ├── flatmap.h │ │ │ ├── flatrep.h │ │ │ ├── flatset.h │ │ │ ├── inlined_vector.h │ │ │ ├── int_type.h │ │ │ ├── iterator_range.h │ │ │ ├── manual_constructor.h │ │ │ ├── manual_constructor_test.cc │ │ │ ├── map_util.h │ │ │ ├── priority_queue_util.h │ │ │ ├── subtle │ │ │ │ ├── BUILD │ │ │ │ └── map_traits.h │ │ │ ├── top_n.h │ │ │ └── top_n_test.cc │ │ ├── hash │ │ │ ├── BUILD │ │ │ ├── crc32c.h │ │ │ └── hash.h │ │ ├── histogram │ │ │ ├── BUILD │ │ │ └── histogram.h │ │ ├── io │ │ │ ├── BUILD │ │ │ ├── block.h │ │ │ ├── block_builder.h │ │ │ ├── buffered_inputstream.h │ │ │ ├── cache.h │ │ │ ├── compression.h │ │ │ ├── format.h │ │ │ ├── inputbuffer.h │ │ │ ├── inputstream_interface.h │ │ │ ├── iterator.h │ │ │ ├── path.h │ │ │ ├── proto_encode_helper.h │ │ │ ├── random_inputstream.h │ │ │ ├── record_reader.h │ │ │ ├── record_writer.h │ │ │ ├── table.h │ │ │ ├── table_builder.h │ │ │ ├── table_options.h │ │ │ ├── two_level_iterator.h │ │ │ ├── zlib_compression_options.h │ │ │ ├── zlib_inputstream.h │ │ │ └── zlib_outputbuffer.h │ │ ├── jpeg │ │ │ ├── BUILD │ │ │ ├── jpeg_handle.cc │ │ │ ├── jpeg_handle.h │ │ │ ├── jpeg_mem.cc │ │ │ ├── jpeg_mem.h │ │ │ ├── jpeg_mem_unittest.cc │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── bad_huffman.jpg │ │ │ │ ├── corrupt.jpg │ │ │ │ ├── corrupt34_2.jpg │ │ │ │ ├── corrupt34_3.jpg │ │ │ │ ├── corrupt34_4.jpg │ │ │ │ ├── jpeg_merge_test1.jpg │ │ │ │ ├── jpeg_merge_test1_cmyk.jpg │ │ │ │ ├── medium.jpg │ │ │ │ └── small.jpg │ │ ├── llvm_rtti │ │ │ ├── BUILD │ │ │ └── llvm_rtti.h │ │ ├── math │ │ │ ├── BUILD │ │ │ └── math_util.h │ │ ├── monitoring │ │ │ ├── BUILD │ │ │ ├── cell_reader-inl.h │ │ │ ├── cell_reader.h │ │ │ ├── cell_reader_test.cc │ │ │ ├── collected_metrics.h │ │ │ ├── collection_registry.h │ │ │ ├── collection_registry_test.cc │ │ │ ├── counter.h │ │ │ ├── counter_test.cc │ │ │ ├── gauge.h │ │ │ ├── gauge_test.cc │ │ │ ├── metric_def.h │ │ │ ├── metric_def_test.cc │ │ │ ├── percentile_sampler.h │ │ │ ├── percentile_sampler_test.cc │ │ │ ├── sampler.h │ │ │ ├── sampler_test.cc │ │ │ ├── test_utils.h │ │ │ ├── test_utils_test.cc │ │ │ ├── timed.h │ │ │ └── types.h │ │ ├── png │ │ │ ├── BUILD │ │ │ ├── png_io.cc │ │ │ ├── png_io.h │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── lena_gray.png │ │ │ │ ├── lena_palette.png │ │ │ │ ├── lena_palette_trns.png │ │ │ │ ├── lena_rgba.png │ │ │ │ └── palette_only.png │ │ ├── psnr │ │ │ ├── BUILD │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── cat_q20.jpg │ │ │ │ ├── cat_q72.jpg │ │ │ │ └── cat_q95.jpg │ │ ├── random │ │ │ ├── BUILD │ │ │ ├── distribution_sampler.h │ │ │ ├── exact_uniform_int.h │ │ │ ├── philox_random.h │ │ │ ├── random.h │ │ │ ├── random_distributions.h │ │ │ ├── random_distributions_utils.h │ │ │ ├── simple_philox.h │ │ │ └── weighted_picker.h │ │ ├── ssim │ │ │ ├── BUILD │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── checkerboard1.png │ │ │ │ ├── checkerboard2.png │ │ │ │ └── checkerboard3.png │ │ ├── strings │ │ │ ├── BUILD │ │ │ ├── base64.h │ │ │ ├── base64_test.cc │ │ │ ├── numbers.h │ │ │ ├── ordered_code.cc │ │ │ ├── ordered_code.h │ │ │ ├── ordered_code_test.cc │ │ │ ├── proto_serialization.h │ │ │ ├── proto_serialization_test.cc │ │ │ ├── proto_text_util.cc │ │ │ ├── proto_text_util.h │ │ │ ├── scanner.h │ │ │ ├── str_util.h │ │ │ ├── strcat.h │ │ │ └── stringprintf.h │ │ └── wav │ │ │ ├── BUILD │ │ │ ├── wav_io.cc │ │ │ ├── wav_io.h │ │ │ └── wav_io_test.cc │ ├── nccl │ │ ├── BUILD │ │ ├── collective_communicator.cc │ │ ├── collective_communicator.h │ │ ├── nccl_manager.cc │ │ ├── nccl_manager.h │ │ ├── nccl_manager_test.cc │ │ └── nccl_rewrite.cc │ ├── ops │ │ ├── BUILD │ │ ├── array_grad.cc │ │ ├── array_grad_test.cc │ │ ├── array_ops.cc │ │ ├── array_ops_test.cc │ │ ├── audio_ops.cc │ │ ├── batch_ops.cc │ │ ├── bitwise_ops.cc │ │ ├── boosted_trees_ops.cc │ │ ├── candidate_sampling_ops.cc │ │ ├── candidate_sampling_ops_test.cc │ │ ├── checkpoint_ops.cc │ │ ├── clustering_ops.cc │ │ ├── collective_ops.cc │ │ ├── compat │ │ │ ├── BUILD │ │ │ ├── backwards_compatibility_test.cc │ │ │ ├── op_compatibility_lib.cc │ │ │ ├── op_compatibility_lib.h │ │ │ ├── ops_history.v0.pbtxt │ │ │ ├── ops_history.v2.pbtxt │ │ │ ├── ops_history_v1 │ │ │ │ ├── Abort.pbtxt │ │ │ │ ├── Abs.pbtxt │ │ │ │ ├── AccumulateNV2.pbtxt │ │ │ │ ├── AccumulatorApplyGradient.pbtxt │ │ │ │ ├── AccumulatorNumAccumulated.pbtxt │ │ │ │ ├── AccumulatorSetGlobalStep.pbtxt │ │ │ │ ├── AccumulatorTakeGradient.pbtxt │ │ │ │ ├── Acos.pbtxt │ │ │ │ ├── Acosh.pbtxt │ │ │ │ ├── Add.pbtxt │ │ │ │ ├── AddManySparseToTensorsMap.pbtxt │ │ │ │ ├── AddN.pbtxt │ │ │ │ ├── AddSparseToTensorsMap.pbtxt │ │ │ │ ├── AddV2.pbtxt │ │ │ │ ├── AdjustContrast.pbtxt │ │ │ │ ├── AdjustContrastv2.pbtxt │ │ │ │ ├── AdjustHue.pbtxt │ │ │ │ ├── AdjustSaturation.pbtxt │ │ │ │ ├── All.pbtxt │ │ │ │ ├── AllCandidateSampler.pbtxt │ │ │ │ ├── AllToAll.pbtxt │ │ │ │ ├── Angle.pbtxt │ │ │ │ ├── AnonymousIterator.pbtxt │ │ │ │ ├── AnonymousIteratorV2.pbtxt │ │ │ │ ├── AnonymousMemoryCache.pbtxt │ │ │ │ ├── AnonymousMultiDeviceIterator.pbtxt │ │ │ │ ├── AnonymousRandomSeedGenerator.pbtxt │ │ │ │ ├── AnonymousSeedGenerator.pbtxt │ │ │ │ ├── Any.pbtxt │ │ │ │ ├── ApplyAdaMax.pbtxt │ │ │ │ ├── ApplyAdadelta.pbtxt │ │ │ │ ├── ApplyAdagrad.pbtxt │ │ │ │ ├── ApplyAdagradDA.pbtxt │ │ │ │ ├── ApplyAdagradV2.pbtxt │ │ │ │ ├── ApplyAdam.pbtxt │ │ │ │ ├── ApplyAddSign.pbtxt │ │ │ │ ├── ApplyCenteredRMSProp.pbtxt │ │ │ │ ├── ApplyFtrl.pbtxt │ │ │ │ ├── ApplyFtrlV2.pbtxt │ │ │ │ ├── ApplyGradientDescent.pbtxt │ │ │ │ ├── ApplyMomentum.pbtxt │ │ │ │ ├── ApplyPowerSign.pbtxt │ │ │ │ ├── ApplyProximalAdagrad.pbtxt │ │ │ │ ├── ApplyProximalGradientDescent.pbtxt │ │ │ │ ├── ApplyRMSProp.pbtxt │ │ │ │ ├── ApproximateEqual.pbtxt │ │ │ │ ├── ArgMax.pbtxt │ │ │ │ ├── ArgMin.pbtxt │ │ │ │ ├── AsString.pbtxt │ │ │ │ ├── Asin.pbtxt │ │ │ │ ├── Asinh.pbtxt │ │ │ │ ├── Assert.pbtxt │ │ │ │ ├── AssertCardinalityDataset.pbtxt │ │ │ │ ├── AssertNextDataset.pbtxt │ │ │ │ ├── Assign.pbtxt │ │ │ │ ├── AssignAdd.pbtxt │ │ │ │ ├── AssignAddVariableOp.pbtxt │ │ │ │ ├── AssignSub.pbtxt │ │ │ │ ├── AssignSubVariableOp.pbtxt │ │ │ │ ├── AssignVariableOp.pbtxt │ │ │ │ ├── Atan.pbtxt │ │ │ │ ├── Atan2.pbtxt │ │ │ │ ├── Atanh.pbtxt │ │ │ │ ├── AudioSpectrogram.pbtxt │ │ │ │ ├── AudioSummary.pbtxt │ │ │ │ ├── AudioSummaryV2.pbtxt │ │ │ │ ├── AutoShardDataset.pbtxt │ │ │ │ ├── AvgPool.pbtxt │ │ │ │ ├── AvgPool3D.pbtxt │ │ │ │ ├── AvgPool3DGrad.pbtxt │ │ │ │ ├── AvgPoolGrad.pbtxt │ │ │ │ ├── BUILD │ │ │ │ ├── Barrier.pbtxt │ │ │ │ ├── BarrierClose.pbtxt │ │ │ │ ├── BarrierIncompleteSize.pbtxt │ │ │ │ ├── BarrierInsertMany.pbtxt │ │ │ │ ├── BarrierReadySize.pbtxt │ │ │ │ ├── BarrierTakeMany.pbtxt │ │ │ │ ├── Batch.pbtxt │ │ │ │ ├── BatchCholesky.pbtxt │ │ │ │ ├── BatchCholeskyGrad.pbtxt │ │ │ │ ├── BatchDataset.pbtxt │ │ │ │ ├── BatchDatasetV2.pbtxt │ │ │ │ ├── BatchFFT.pbtxt │ │ │ │ ├── BatchFFT2D.pbtxt │ │ │ │ ├── BatchFFT3D.pbtxt │ │ │ │ ├── BatchFunction.pbtxt │ │ │ │ ├── BatchIFFT.pbtxt │ │ │ │ ├── BatchIFFT2D.pbtxt │ │ │ │ ├── BatchIFFT3D.pbtxt │ │ │ │ ├── BatchMatMul.pbtxt │ │ │ │ ├── BatchMatMulV2.pbtxt │ │ │ │ ├── BatchMatrixBandPart.pbtxt │ │ │ │ ├── BatchMatrixDeterminant.pbtxt │ │ │ │ ├── BatchMatrixDiag.pbtxt │ │ │ │ ├── BatchMatrixDiagPart.pbtxt │ │ │ │ ├── BatchMatrixInverse.pbtxt │ │ │ │ ├── BatchMatrixSetDiag.pbtxt │ │ │ │ ├── BatchMatrixSolve.pbtxt │ │ │ │ ├── BatchMatrixSolveLs.pbtxt │ │ │ │ ├── BatchMatrixTriangularSolve.pbtxt │ │ │ │ ├── BatchNormWithGlobalNormalization.pbtxt │ │ │ │ ├── BatchNormWithGlobalNormalizationGrad.pbtxt │ │ │ │ ├── BatchSelfAdjointEig.pbtxt │ │ │ │ ├── BatchSelfAdjointEigV2.pbtxt │ │ │ │ ├── BatchSvd.pbtxt │ │ │ │ ├── BatchToSpace.pbtxt │ │ │ │ ├── BatchToSpaceND.pbtxt │ │ │ │ ├── BesselI0e.pbtxt │ │ │ │ ├── BesselI1e.pbtxt │ │ │ │ ├── Betainc.pbtxt │ │ │ │ ├── BiasAdd.pbtxt │ │ │ │ ├── BiasAddGrad.pbtxt │ │ │ │ ├── BiasAddV1.pbtxt │ │ │ │ ├── Bincount.pbtxt │ │ │ │ ├── Bitcast.pbtxt │ │ │ │ ├── BitwiseAnd.pbtxt │ │ │ │ ├── BitwiseOr.pbtxt │ │ │ │ ├── BitwiseXor.pbtxt │ │ │ │ ├── BlockLSTM.pbtxt │ │ │ │ ├── BlockLSTMGrad.pbtxt │ │ │ │ ├── BlockLSTMGradV2.pbtxt │ │ │ │ ├── BlockLSTMV2.pbtxt │ │ │ │ ├── BoostedTreesAggregateStats.pbtxt │ │ │ │ ├── BoostedTreesBucketize.pbtxt │ │ │ │ ├── BoostedTreesCalculateBestFeatureSplit.pbtxt │ │ │ │ ├── BoostedTreesCalculateBestFeatureSplitV2.pbtxt │ │ │ │ ├── BoostedTreesCalculateBestGainsPerFeature.pbtxt │ │ │ │ ├── BoostedTreesCenterBias.pbtxt │ │ │ │ ├── BoostedTreesCreateEnsemble.pbtxt │ │ │ │ ├── BoostedTreesCreateQuantileStreamResource.pbtxt │ │ │ │ ├── BoostedTreesDeserializeEnsemble.pbtxt │ │ │ │ ├── BoostedTreesEnsembleResourceHandleOp.pbtxt │ │ │ │ ├── BoostedTreesExampleDebugOutputs.pbtxt │ │ │ │ ├── BoostedTreesFlushQuantileSummaries.pbtxt │ │ │ │ ├── BoostedTreesGetEnsembleStates.pbtxt │ │ │ │ ├── BoostedTreesMakeQuantileSummaries.pbtxt │ │ │ │ ├── BoostedTreesMakeStatsSummary.pbtxt │ │ │ │ ├── BoostedTreesPredict.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceAddSummaries.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceDeserialize.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceFlush.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceHandleOp.pbtxt │ │ │ │ ├── BoostedTreesSerializeEnsemble.pbtxt │ │ │ │ ├── BoostedTreesSparseAggregateStats.pbtxt │ │ │ │ ├── BoostedTreesSparseCalculateBestFeatureSplit.pbtxt │ │ │ │ ├── BoostedTreesTrainingPredict.pbtxt │ │ │ │ ├── BoostedTreesUpdateEnsemble.pbtxt │ │ │ │ ├── BoostedTreesUpdateEnsembleV2.pbtxt │ │ │ │ ├── BroadcastArgs.pbtxt │ │ │ │ ├── BroadcastGradientArgs.pbtxt │ │ │ │ ├── BroadcastTo.pbtxt │ │ │ │ ├── Bucketize.pbtxt │ │ │ │ ├── BytesProducedStatsDataset.pbtxt │ │ │ │ ├── CSRSparseMatrixComponents.pbtxt │ │ │ │ ├── CSRSparseMatrixToDense.pbtxt │ │ │ │ ├── CSRSparseMatrixToSparseTensor.pbtxt │ │ │ │ ├── CSVDataset.pbtxt │ │ │ │ ├── CTCBeamSearchDecoder.pbtxt │ │ │ │ ├── CTCGreedyDecoder.pbtxt │ │ │ │ ├── CTCLoss.pbtxt │ │ │ │ ├── CTCLossV2.pbtxt │ │ │ │ ├── CacheDataset.pbtxt │ │ │ │ ├── CacheDatasetV2.pbtxt │ │ │ │ ├── Case.pbtxt │ │ │ │ ├── Cast.pbtxt │ │ │ │ ├── Ceil.pbtxt │ │ │ │ ├── CheckNumerics.pbtxt │ │ │ │ ├── CheckNumericsV2.pbtxt │ │ │ │ ├── Cholesky.pbtxt │ │ │ │ ├── CholeskyGrad.pbtxt │ │ │ │ ├── ChooseFastestBranchDataset.pbtxt │ │ │ │ ├── ChooseFastestDataset.pbtxt │ │ │ │ ├── ClipByValue.pbtxt │ │ │ │ ├── CloseSummaryWriter.pbtxt │ │ │ │ ├── CollectiveBcastRecv.pbtxt │ │ │ │ ├── CollectiveBcastSend.pbtxt │ │ │ │ ├── CollectiveGather.pbtxt │ │ │ │ ├── CollectivePermute.pbtxt │ │ │ │ ├── CollectiveReduce.pbtxt │ │ │ │ ├── CombinedNonMaxSuppression.pbtxt │ │ │ │ ├── Complex.pbtxt │ │ │ │ ├── ComplexAbs.pbtxt │ │ │ │ ├── ComputeAccidentalHits.pbtxt │ │ │ │ ├── Concat.pbtxt │ │ │ │ ├── ConcatOffset.pbtxt │ │ │ │ ├── ConcatV2.pbtxt │ │ │ │ ├── ConcatenateDataset.pbtxt │ │ │ │ ├── ConditionalAccumulator.pbtxt │ │ │ │ ├── ConfigureDistributedTPU.pbtxt │ │ │ │ ├── ConfigureTPUEmbedding.pbtxt │ │ │ │ ├── Conj.pbtxt │ │ │ │ ├── ConjugateTranspose.pbtxt │ │ │ │ ├── Const.pbtxt │ │ │ │ ├── ConsumeMutexLock.pbtxt │ │ │ │ ├── ControlTrigger.pbtxt │ │ │ │ ├── Conv2D.pbtxt │ │ │ │ ├── Conv2DBackpropFilter.pbtxt │ │ │ │ ├── Conv2DBackpropInput.pbtxt │ │ │ │ ├── Conv3D.pbtxt │ │ │ │ ├── Conv3DBackpropFilter.pbtxt │ │ │ │ ├── Conv3DBackpropFilterV2.pbtxt │ │ │ │ ├── Conv3DBackpropInput.pbtxt │ │ │ │ ├── Conv3DBackpropInputV2.pbtxt │ │ │ │ ├── Copy.pbtxt │ │ │ │ ├── CopyHost.pbtxt │ │ │ │ ├── Cos.pbtxt │ │ │ │ ├── Cosh.pbtxt │ │ │ │ ├── CountUpTo.pbtxt │ │ │ │ ├── CreateSummaryDbWriter.pbtxt │ │ │ │ ├── CreateSummaryFileWriter.pbtxt │ │ │ │ ├── CropAndResize.pbtxt │ │ │ │ ├── CropAndResizeGradBoxes.pbtxt │ │ │ │ ├── CropAndResizeGradImage.pbtxt │ │ │ │ ├── Cross.pbtxt │ │ │ │ ├── CrossReplicaSum.pbtxt │ │ │ │ ├── CudnnRNN.pbtxt │ │ │ │ ├── CudnnRNNBackprop.pbtxt │ │ │ │ ├── CudnnRNNBackpropV2.pbtxt │ │ │ │ ├── CudnnRNNBackpropV3.pbtxt │ │ │ │ ├── CudnnRNNCanonicalToParams.pbtxt │ │ │ │ ├── CudnnRNNCanonicalToParamsV2.pbtxt │ │ │ │ ├── CudnnRNNParamsSize.pbtxt │ │ │ │ ├── CudnnRNNParamsToCanonical.pbtxt │ │ │ │ ├── CudnnRNNParamsToCanonicalV2.pbtxt │ │ │ │ ├── CudnnRNNV2.pbtxt │ │ │ │ ├── CudnnRNNV3.pbtxt │ │ │ │ ├── Cumprod.pbtxt │ │ │ │ ├── Cumsum.pbtxt │ │ │ │ ├── CumulativeLogsumexp.pbtxt │ │ │ │ ├── DataFormatDimMap.pbtxt │ │ │ │ ├── DataFormatVecPermute.pbtxt │ │ │ │ ├── DatasetCardinality.pbtxt │ │ │ │ ├── DatasetFromGraph.pbtxt │ │ │ │ ├── DatasetToGraph.pbtxt │ │ │ │ ├── DatasetToGraphV2.pbtxt │ │ │ │ ├── DatasetToSingleElement.pbtxt │ │ │ │ ├── DatasetToTFRecord.pbtxt │ │ │ │ ├── Dawsn.pbtxt │ │ │ │ ├── DebugGradientIdentity.pbtxt │ │ │ │ ├── DebugGradientRefIdentity.pbtxt │ │ │ │ ├── DebugIdentity.pbtxt │ │ │ │ ├── DebugIdentityV2.pbtxt │ │ │ │ ├── DebugNanCount.pbtxt │ │ │ │ ├── DebugNumericSummary.pbtxt │ │ │ │ ├── DebugNumericSummaryV2.pbtxt │ │ │ │ ├── DecodeAndCropJpeg.pbtxt │ │ │ │ ├── DecodeBase64.pbtxt │ │ │ │ ├── DecodeBmp.pbtxt │ │ │ │ ├── DecodeCSV.pbtxt │ │ │ │ ├── DecodeCompressed.pbtxt │ │ │ │ ├── DecodeGif.pbtxt │ │ │ │ ├── DecodeJSONExample.pbtxt │ │ │ │ ├── DecodeJpeg.pbtxt │ │ │ │ ├── DecodePaddedRaw.pbtxt │ │ │ │ ├── DecodePng.pbtxt │ │ │ │ ├── DecodeProtoV2.pbtxt │ │ │ │ ├── DecodeRaw.pbtxt │ │ │ │ ├── DecodeWav.pbtxt │ │ │ │ ├── DeepCopy.pbtxt │ │ │ │ ├── DeleteIterator.pbtxt │ │ │ │ ├── DeleteMemoryCache.pbtxt │ │ │ │ ├── DeleteMultiDeviceIterator.pbtxt │ │ │ │ ├── DeleteRandomSeedGenerator.pbtxt │ │ │ │ ├── DeleteSeedGenerator.pbtxt │ │ │ │ ├── DeleteSessionTensor.pbtxt │ │ │ │ ├── DenseToCSRSparseMatrix.pbtxt │ │ │ │ ├── DenseToDenseSetOperation.pbtxt │ │ │ │ ├── DenseToSparseBatchDataset.pbtxt │ │ │ │ ├── DenseToSparseSetOperation.pbtxt │ │ │ │ ├── DepthToSpace.pbtxt │ │ │ │ ├── DepthwiseConv2dNative.pbtxt │ │ │ │ ├── DepthwiseConv2dNativeBackpropFilter.pbtxt │ │ │ │ ├── DepthwiseConv2dNativeBackpropInput.pbtxt │ │ │ │ ├── Dequantize.pbtxt │ │ │ │ ├── DeserializeIterator.pbtxt │ │ │ │ ├── DeserializeManySparse.pbtxt │ │ │ │ ├── DeserializeSparse.pbtxt │ │ │ │ ├── DestroyResourceOp.pbtxt │ │ │ │ ├── DestroyTemporaryVariable.pbtxt │ │ │ │ ├── Diag.pbtxt │ │ │ │ ├── DiagPart.pbtxt │ │ │ │ ├── Digamma.pbtxt │ │ │ │ ├── Dilation2D.pbtxt │ │ │ │ ├── Dilation2DBackpropFilter.pbtxt │ │ │ │ ├── Dilation2DBackpropInput.pbtxt │ │ │ │ ├── DirectedInterleaveDataset.pbtxt │ │ │ │ ├── Div.pbtxt │ │ │ │ ├── DivNoNan.pbtxt │ │ │ │ ├── DrawBoundingBoxes.pbtxt │ │ │ │ ├── DrawBoundingBoxesV2.pbtxt │ │ │ │ ├── DynamicPartition.pbtxt │ │ │ │ ├── DynamicStitch.pbtxt │ │ │ │ ├── EagerPyFunc.pbtxt │ │ │ │ ├── EditDistance.pbtxt │ │ │ │ ├── Eig.pbtxt │ │ │ │ ├── Einsum.pbtxt │ │ │ │ ├── Elu.pbtxt │ │ │ │ ├── EluGrad.pbtxt │ │ │ │ ├── Empty.pbtxt │ │ │ │ ├── EmptyTensorList.pbtxt │ │ │ │ ├── EncodeBase64.pbtxt │ │ │ │ ├── EncodeJpeg.pbtxt │ │ │ │ ├── EncodeJpegVariableQuality.pbtxt │ │ │ │ ├── EncodePng.pbtxt │ │ │ │ ├── EncodeProto.pbtxt │ │ │ │ ├── EncodeWav.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingIntegerBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingSparseBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingSparseTensorBatch.pbtxt │ │ │ │ ├── EnsureShape.pbtxt │ │ │ │ ├── Enter.pbtxt │ │ │ │ ├── Equal.pbtxt │ │ │ │ ├── Erf.pbtxt │ │ │ │ ├── Erfc.pbtxt │ │ │ │ ├── Erfinv.pbtxt │ │ │ │ ├── EuclideanNorm.pbtxt │ │ │ │ ├── Exit.pbtxt │ │ │ │ ├── Exp.pbtxt │ │ │ │ ├── ExpandDims.pbtxt │ │ │ │ ├── ExperimentalAssertNextDataset.pbtxt │ │ │ │ ├── ExperimentalAutoShardDataset.pbtxt │ │ │ │ ├── ExperimentalBytesProducedStatsDataset.pbtxt │ │ │ │ ├── ExperimentalCSVDataset.pbtxt │ │ │ │ ├── ExperimentalChooseFastestDataset.pbtxt │ │ │ │ ├── ExperimentalDatasetCardinality.pbtxt │ │ │ │ ├── ExperimentalDatasetToTFRecord.pbtxt │ │ │ │ ├── ExperimentalDenseToSparseBatchDataset.pbtxt │ │ │ │ ├── ExperimentalDirectedInterleaveDataset.pbtxt │ │ │ │ ├── ExperimentalGroupByReducerDataset.pbtxt │ │ │ │ ├── ExperimentalGroupByWindowDataset.pbtxt │ │ │ │ ├── ExperimentalIgnoreErrorsDataset.pbtxt │ │ │ │ ├── ExperimentalIteratorGetDevice.pbtxt │ │ │ │ ├── ExperimentalLMDBDataset.pbtxt │ │ │ │ ├── ExperimentalLatencyStatsDataset.pbtxt │ │ │ │ ├── ExperimentalMapAndBatchDataset.pbtxt │ │ │ │ ├── ExperimentalMapDataset.pbtxt │ │ │ │ ├── ExperimentalMatchingFilesDataset.pbtxt │ │ │ │ ├── ExperimentalMaxIntraOpParallelismDataset.pbtxt │ │ │ │ ├── ExperimentalNonSerializableDataset.pbtxt │ │ │ │ ├── ExperimentalParallelInterleaveDataset.pbtxt │ │ │ │ ├── ExperimentalParseExampleDataset.pbtxt │ │ │ │ ├── ExperimentalPrivateThreadPoolDataset.pbtxt │ │ │ │ ├── ExperimentalRandomDataset.pbtxt │ │ │ │ ├── ExperimentalRebatchDataset.pbtxt │ │ │ │ ├── ExperimentalScanDataset.pbtxt │ │ │ │ ├── ExperimentalSetStatsAggregatorDataset.pbtxt │ │ │ │ ├── ExperimentalSleepDataset.pbtxt │ │ │ │ ├── ExperimentalSlidingWindowDataset.pbtxt │ │ │ │ ├── ExperimentalSqlDataset.pbtxt │ │ │ │ ├── ExperimentalStatsAggregatorHandle.pbtxt │ │ │ │ ├── ExperimentalStatsAggregatorSummary.pbtxt │ │ │ │ ├── ExperimentalTakeWhileDataset.pbtxt │ │ │ │ ├── ExperimentalThreadPoolDataset.pbtxt │ │ │ │ ├── ExperimentalThreadPoolHandle.pbtxt │ │ │ │ ├── ExperimentalUnbatchDataset.pbtxt │ │ │ │ ├── ExperimentalUniqueDataset.pbtxt │ │ │ │ ├── Expint.pbtxt │ │ │ │ ├── Expm1.pbtxt │ │ │ │ ├── ExtractGlimpse.pbtxt │ │ │ │ ├── ExtractImagePatches.pbtxt │ │ │ │ ├── ExtractJpegShape.pbtxt │ │ │ │ ├── ExtractVolumePatches.pbtxt │ │ │ │ ├── FFT.pbtxt │ │ │ │ ├── FFT2D.pbtxt │ │ │ │ ├── FFT3D.pbtxt │ │ │ │ ├── FIFOQueue.pbtxt │ │ │ │ ├── FIFOQueueV2.pbtxt │ │ │ │ ├── Fact.pbtxt │ │ │ │ ├── FakeParam.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxArgs.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxArgsGradient.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVars.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVarsGradient.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVarsPerChannel.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt │ │ │ │ ├── FakeQueue.pbtxt │ │ │ │ ├── Fill.pbtxt │ │ │ │ ├── FilterByLastComponentDataset.pbtxt │ │ │ │ ├── FilterDataset.pbtxt │ │ │ │ ├── Fingerprint.pbtxt │ │ │ │ ├── FixedLengthRecordDataset.pbtxt │ │ │ │ ├── FixedLengthRecordDatasetV2.pbtxt │ │ │ │ ├── FixedLengthRecordReader.pbtxt │ │ │ │ ├── FixedLengthRecordReaderV2.pbtxt │ │ │ │ ├── FixedUnigramCandidateSampler.pbtxt │ │ │ │ ├── FlatMapDataset.pbtxt │ │ │ │ ├── Floor.pbtxt │ │ │ │ ├── FloorDiv.pbtxt │ │ │ │ ├── FloorMod.pbtxt │ │ │ │ ├── FlushSummaryWriter.pbtxt │ │ │ │ ├── For.pbtxt │ │ │ │ ├── FractionalAvgPool.pbtxt │ │ │ │ ├── FractionalAvgPoolGrad.pbtxt │ │ │ │ ├── FractionalMaxPool.pbtxt │ │ │ │ ├── FractionalMaxPoolGrad.pbtxt │ │ │ │ ├── FresnelCos.pbtxt │ │ │ │ ├── FresnelSin.pbtxt │ │ │ │ ├── FusedBatchNorm.pbtxt │ │ │ │ ├── FusedBatchNormGrad.pbtxt │ │ │ │ ├── FusedBatchNormGradV2.pbtxt │ │ │ │ ├── FusedBatchNormGradV3.pbtxt │ │ │ │ ├── FusedBatchNormV2.pbtxt │ │ │ │ ├── FusedBatchNormV3.pbtxt │ │ │ │ ├── FusedPadConv2D.pbtxt │ │ │ │ ├── FusedResizeAndPadConv2D.pbtxt │ │ │ │ ├── GRUBlockCell.pbtxt │ │ │ │ ├── GRUBlockCellGrad.pbtxt │ │ │ │ ├── Gather.pbtxt │ │ │ │ ├── GatherNd.pbtxt │ │ │ │ ├── GatherV2.pbtxt │ │ │ │ ├── GenerateBoundingBoxProposals.pbtxt │ │ │ │ ├── GenerateVocabRemapping.pbtxt │ │ │ │ ├── GeneratorDataset.pbtxt │ │ │ │ ├── GetSessionHandle.pbtxt │ │ │ │ ├── GetSessionHandleV2.pbtxt │ │ │ │ ├── GetSessionTensor.pbtxt │ │ │ │ ├── Greater.pbtxt │ │ │ │ ├── GreaterEqual.pbtxt │ │ │ │ ├── GroupByReducerDataset.pbtxt │ │ │ │ ├── GroupByWindowDataset.pbtxt │ │ │ │ ├── GuaranteeConst.pbtxt │ │ │ │ ├── HSVToRGB.pbtxt │ │ │ │ ├── HashTable.pbtxt │ │ │ │ ├── HashTableV2.pbtxt │ │ │ │ ├── HistogramFixedWidth.pbtxt │ │ │ │ ├── HistogramSummary.pbtxt │ │ │ │ ├── HostConst.pbtxt │ │ │ │ ├── IFFT.pbtxt │ │ │ │ ├── IFFT2D.pbtxt │ │ │ │ ├── IFFT3D.pbtxt │ │ │ │ ├── IRFFT.pbtxt │ │ │ │ ├── IRFFT2D.pbtxt │ │ │ │ ├── IRFFT3D.pbtxt │ │ │ │ ├── Identity.pbtxt │ │ │ │ ├── IdentityN.pbtxt │ │ │ │ ├── IdentityReader.pbtxt │ │ │ │ ├── IdentityReaderV2.pbtxt │ │ │ │ ├── If.pbtxt │ │ │ │ ├── Igamma.pbtxt │ │ │ │ ├── IgammaGradA.pbtxt │ │ │ │ ├── Igammac.pbtxt │ │ │ │ ├── IgnoreErrorsDataset.pbtxt │ │ │ │ ├── Imag.pbtxt │ │ │ │ ├── ImageProjectiveTransformV2.pbtxt │ │ │ │ ├── ImageSummary.pbtxt │ │ │ │ ├── ImmutableConst.pbtxt │ │ │ │ ├── ImportEvent.pbtxt │ │ │ │ ├── InTopK.pbtxt │ │ │ │ ├── InTopKV2.pbtxt │ │ │ │ ├── InfeedDequeue.pbtxt │ │ │ │ ├── InfeedDequeueTuple.pbtxt │ │ │ │ ├── InfeedEnqueue.pbtxt │ │ │ │ ├── InfeedEnqueuePrelinearizedBuffer.pbtxt │ │ │ │ ├── InfeedEnqueueTuple.pbtxt │ │ │ │ ├── InitializeTable.pbtxt │ │ │ │ ├── InitializeTableFromDataset.pbtxt │ │ │ │ ├── InitializeTableFromTextFile.pbtxt │ │ │ │ ├── InitializeTableFromTextFileV2.pbtxt │ │ │ │ ├── InitializeTableV2.pbtxt │ │ │ │ ├── InplaceAdd.pbtxt │ │ │ │ ├── InplaceSub.pbtxt │ │ │ │ ├── InplaceUpdate.pbtxt │ │ │ │ ├── InterleaveDataset.pbtxt │ │ │ │ ├── Inv.pbtxt │ │ │ │ ├── InvGrad.pbtxt │ │ │ │ ├── Invert.pbtxt │ │ │ │ ├── InvertPermutation.pbtxt │ │ │ │ ├── IsBoostedTreesEnsembleInitialized.pbtxt │ │ │ │ ├── IsBoostedTreesQuantileStreamResourceInitialized.pbtxt │ │ │ │ ├── IsFinite.pbtxt │ │ │ │ ├── IsInf.pbtxt │ │ │ │ ├── IsNan.pbtxt │ │ │ │ ├── IsVariableInitialized.pbtxt │ │ │ │ ├── Iterator.pbtxt │ │ │ │ ├── IteratorFromStringHandle.pbtxt │ │ │ │ ├── IteratorFromStringHandleV2.pbtxt │ │ │ │ ├── IteratorGetDevice.pbtxt │ │ │ │ ├── IteratorGetNext.pbtxt │ │ │ │ ├── IteratorGetNextAsOptional.pbtxt │ │ │ │ ├── IteratorGetNextSync.pbtxt │ │ │ │ ├── IteratorToStringHandle.pbtxt │ │ │ │ ├── IteratorV2.pbtxt │ │ │ │ ├── KMC2ChainInitialization.pbtxt │ │ │ │ ├── KmeansPlusPlusInitialization.pbtxt │ │ │ │ ├── L2Loss.pbtxt │ │ │ │ ├── LMDBDataset.pbtxt │ │ │ │ ├── LMDBReader.pbtxt │ │ │ │ ├── LRN.pbtxt │ │ │ │ ├── LRNGrad.pbtxt │ │ │ │ ├── LSTMBlockCell.pbtxt │ │ │ │ ├── LSTMBlockCellGrad.pbtxt │ │ │ │ ├── LatencyStatsDataset.pbtxt │ │ │ │ ├── LeakyRelu.pbtxt │ │ │ │ ├── LeakyReluGrad.pbtxt │ │ │ │ ├── LearnedUnigramCandidateSampler.pbtxt │ │ │ │ ├── LeftShift.pbtxt │ │ │ │ ├── LegacyParallelInterleaveDatasetV2.pbtxt │ │ │ │ ├── Less.pbtxt │ │ │ │ ├── LessEqual.pbtxt │ │ │ │ ├── Lgamma.pbtxt │ │ │ │ ├── LinSpace.pbtxt │ │ │ │ ├── ListDiff.pbtxt │ │ │ │ ├── LoadAndRemapMatrix.pbtxt │ │ │ │ ├── LoadTPUEmbeddingADAMParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingAdadeltaParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingAdagradParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingFTRLParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingMomentumParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingProximalAdagradParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingProximalYogiParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingRMSPropParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt │ │ │ │ ├── Log.pbtxt │ │ │ │ ├── Log1p.pbtxt │ │ │ │ ├── LogMatrixDeterminant.pbtxt │ │ │ │ ├── LogSoftmax.pbtxt │ │ │ │ ├── LogUniformCandidateSampler.pbtxt │ │ │ │ ├── LogicalAnd.pbtxt │ │ │ │ ├── LogicalNot.pbtxt │ │ │ │ ├── LogicalOr.pbtxt │ │ │ │ ├── LookupTableExport.pbtxt │ │ │ │ ├── LookupTableExportV2.pbtxt │ │ │ │ ├── LookupTableFind.pbtxt │ │ │ │ ├── LookupTableFindV2.pbtxt │ │ │ │ ├── LookupTableImport.pbtxt │ │ │ │ ├── LookupTableImportV2.pbtxt │ │ │ │ ├── LookupTableInsert.pbtxt │ │ │ │ ├── LookupTableInsertV2.pbtxt │ │ │ │ ├── LookupTableRemoveV2.pbtxt │ │ │ │ ├── LookupTableSize.pbtxt │ │ │ │ ├── LookupTableSizeV2.pbtxt │ │ │ │ ├── LoopCond.pbtxt │ │ │ │ ├── LowerBound.pbtxt │ │ │ │ ├── Lu.pbtxt │ │ │ │ ├── MakeIterator.pbtxt │ │ │ │ ├── MapAndBatchDataset.pbtxt │ │ │ │ ├── MapClear.pbtxt │ │ │ │ ├── MapDataset.pbtxt │ │ │ │ ├── MapDefun.pbtxt │ │ │ │ ├── MapIncompleteSize.pbtxt │ │ │ │ ├── MapPeek.pbtxt │ │ │ │ ├── MapSize.pbtxt │ │ │ │ ├── MapStage.pbtxt │ │ │ │ ├── MapUnstage.pbtxt │ │ │ │ ├── MapUnstageNoKey.pbtxt │ │ │ │ ├── MatMul.pbtxt │ │ │ │ ├── MatchingFiles.pbtxt │ │ │ │ ├── MatchingFilesDataset.pbtxt │ │ │ │ ├── MatrixBandPart.pbtxt │ │ │ │ ├── MatrixDeterminant.pbtxt │ │ │ │ ├── MatrixDiag.pbtxt │ │ │ │ ├── MatrixDiagPart.pbtxt │ │ │ │ ├── MatrixDiagPartV2.pbtxt │ │ │ │ ├── MatrixDiagPartV3.pbtxt │ │ │ │ ├── MatrixDiagV2.pbtxt │ │ │ │ ├── MatrixDiagV3.pbtxt │ │ │ │ ├── MatrixExponential.pbtxt │ │ │ │ ├── MatrixInverse.pbtxt │ │ │ │ ├── MatrixLogarithm.pbtxt │ │ │ │ ├── MatrixSetDiag.pbtxt │ │ │ │ ├── MatrixSetDiagV2.pbtxt │ │ │ │ ├── MatrixSetDiagV3.pbtxt │ │ │ │ ├── MatrixSolve.pbtxt │ │ │ │ ├── MatrixSolveLs.pbtxt │ │ │ │ ├── MatrixSquareRoot.pbtxt │ │ │ │ ├── MatrixTriangularSolve.pbtxt │ │ │ │ ├── Max.pbtxt │ │ │ │ ├── MaxIntraOpParallelismDataset.pbtxt │ │ │ │ ├── MaxPool.pbtxt │ │ │ │ ├── MaxPool3D.pbtxt │ │ │ │ ├── MaxPool3DGrad.pbtxt │ │ │ │ ├── MaxPool3DGradGrad.pbtxt │ │ │ │ ├── MaxPoolGrad.pbtxt │ │ │ │ ├── MaxPoolGradGrad.pbtxt │ │ │ │ ├── MaxPoolGradGradV2.pbtxt │ │ │ │ ├── MaxPoolGradGradWithArgmax.pbtxt │ │ │ │ ├── MaxPoolGradV2.pbtxt │ │ │ │ ├── MaxPoolGradWithArgmax.pbtxt │ │ │ │ ├── MaxPoolV2.pbtxt │ │ │ │ ├── MaxPoolWithArgmax.pbtxt │ │ │ │ ├── Maximum.pbtxt │ │ │ │ ├── Mean.pbtxt │ │ │ │ ├── Merge.pbtxt │ │ │ │ ├── MergeSummary.pbtxt │ │ │ │ ├── MergeV2Checkpoints.pbtxt │ │ │ │ ├── Mfcc.pbtxt │ │ │ │ ├── Min.pbtxt │ │ │ │ ├── Minimum.pbtxt │ │ │ │ ├── MirrorPad.pbtxt │ │ │ │ ├── MirrorPadGrad.pbtxt │ │ │ │ ├── MlirPassthroughOp.pbtxt │ │ │ │ ├── Mod.pbtxt │ │ │ │ ├── ModelDataset.pbtxt │ │ │ │ ├── Mul.pbtxt │ │ │ │ ├── MulNoNan.pbtxt │ │ │ │ ├── MultiDeviceIterator.pbtxt │ │ │ │ ├── MultiDeviceIteratorFromStringHandle.pbtxt │ │ │ │ ├── MultiDeviceIteratorGetNextFromShard.pbtxt │ │ │ │ ├── MultiDeviceIteratorInit.pbtxt │ │ │ │ ├── MultiDeviceIteratorToStringHandle.pbtxt │ │ │ │ ├── Multinomial.pbtxt │ │ │ │ ├── MutableDenseHashTable.pbtxt │ │ │ │ ├── MutableDenseHashTableV2.pbtxt │ │ │ │ ├── MutableHashTable.pbtxt │ │ │ │ ├── MutableHashTableOfTensors.pbtxt │ │ │ │ ├── MutableHashTableOfTensorsV2.pbtxt │ │ │ │ ├── MutableHashTableV2.pbtxt │ │ │ │ ├── MutexLock.pbtxt │ │ │ │ ├── MutexV2.pbtxt │ │ │ │ ├── NcclAllReduce.pbtxt │ │ │ │ ├── NcclBroadcast.pbtxt │ │ │ │ ├── NcclReduce.pbtxt │ │ │ │ ├── Ndtri.pbtxt │ │ │ │ ├── NearestNeighbors.pbtxt │ │ │ │ ├── Neg.pbtxt │ │ │ │ ├── NegTrain.pbtxt │ │ │ │ ├── NextAfter.pbtxt │ │ │ │ ├── NextIteration.pbtxt │ │ │ │ ├── NoOp.pbtxt │ │ │ │ ├── NonDeterministicInts.pbtxt │ │ │ │ ├── NonMaxSuppression.pbtxt │ │ │ │ ├── NonMaxSuppressionV2.pbtxt │ │ │ │ ├── NonMaxSuppressionV3.pbtxt │ │ │ │ ├── NonMaxSuppressionV4.pbtxt │ │ │ │ ├── NonMaxSuppressionV5.pbtxt │ │ │ │ ├── NonMaxSuppressionWithOverlaps.pbtxt │ │ │ │ ├── NonSerializableDataset.pbtxt │ │ │ │ ├── NotEqual.pbtxt │ │ │ │ ├── NthElement.pbtxt │ │ │ │ ├── OneHot.pbtxt │ │ │ │ ├── OneShotIterator.pbtxt │ │ │ │ ├── OnesLike.pbtxt │ │ │ │ ├── OptimizeDataset.pbtxt │ │ │ │ ├── OptionalFromValue.pbtxt │ │ │ │ ├── OptionalGetValue.pbtxt │ │ │ │ ├── OptionalHasValue.pbtxt │ │ │ │ ├── OptionalNone.pbtxt │ │ │ │ ├── OrderedMapClear.pbtxt │ │ │ │ ├── OrderedMapIncompleteSize.pbtxt │ │ │ │ ├── OrderedMapPeek.pbtxt │ │ │ │ ├── OrderedMapSize.pbtxt │ │ │ │ ├── OrderedMapStage.pbtxt │ │ │ │ ├── OrderedMapUnstage.pbtxt │ │ │ │ ├── OrderedMapUnstageNoKey.pbtxt │ │ │ │ ├── OutfeedDequeue.pbtxt │ │ │ │ ├── OutfeedDequeueTuple.pbtxt │ │ │ │ ├── OutfeedEnqueue.pbtxt │ │ │ │ ├── OutfeedEnqueueTuple.pbtxt │ │ │ │ ├── Pack.pbtxt │ │ │ │ ├── Pad.pbtxt │ │ │ │ ├── PadV2.pbtxt │ │ │ │ ├── PaddedBatchDataset.pbtxt │ │ │ │ ├── PaddedBatchDatasetV2.pbtxt │ │ │ │ ├── PaddingFIFOQueue.pbtxt │ │ │ │ ├── PaddingFIFOQueueV2.pbtxt │ │ │ │ ├── ParallelConcat.pbtxt │ │ │ │ ├── ParallelDynamicStitch.pbtxt │ │ │ │ ├── ParallelInterleaveDataset.pbtxt │ │ │ │ ├── ParallelInterleaveDatasetV2.pbtxt │ │ │ │ ├── ParallelInterleaveDatasetV3.pbtxt │ │ │ │ ├── ParallelInterleaveDatasetV4.pbtxt │ │ │ │ ├── ParallelMapDataset.pbtxt │ │ │ │ ├── ParallelMapDatasetV2.pbtxt │ │ │ │ ├── ParameterizedTruncatedNormal.pbtxt │ │ │ │ ├── ParseExample.pbtxt │ │ │ │ ├── ParseExampleDataset.pbtxt │ │ │ │ ├── ParseExampleDatasetV2.pbtxt │ │ │ │ ├── ParseExampleV2.pbtxt │ │ │ │ ├── ParseSequenceExample.pbtxt │ │ │ │ ├── ParseSequenceExampleV2.pbtxt │ │ │ │ ├── ParseSingleExample.pbtxt │ │ │ │ ├── ParseSingleSequenceExample.pbtxt │ │ │ │ ├── ParseTensor.pbtxt │ │ │ │ ├── PartitionedCall.pbtxt │ │ │ │ ├── Placeholder.pbtxt │ │ │ │ ├── PlaceholderV2.pbtxt │ │ │ │ ├── PlaceholderWithDefault.pbtxt │ │ │ │ ├── Polygamma.pbtxt │ │ │ │ ├── PopulationCount.pbtxt │ │ │ │ ├── Pow.pbtxt │ │ │ │ ├── PrefetchDataset.pbtxt │ │ │ │ ├── Prelinearize.pbtxt │ │ │ │ ├── PrelinearizeTuple.pbtxt │ │ │ │ ├── PreventGradient.pbtxt │ │ │ │ ├── Print.pbtxt │ │ │ │ ├── PrintV2.pbtxt │ │ │ │ ├── PriorityQueue.pbtxt │ │ │ │ ├── PriorityQueueV2.pbtxt │ │ │ │ ├── PrivateThreadPoolDataset.pbtxt │ │ │ │ ├── Prod.pbtxt │ │ │ │ ├── PyFunc.pbtxt │ │ │ │ ├── PyFuncStateless.pbtxt │ │ │ │ ├── Qr.pbtxt │ │ │ │ ├── QuantizeAndDequantize.pbtxt │ │ │ │ ├── QuantizeAndDequantizeV2.pbtxt │ │ │ │ ├── QuantizeAndDequantizeV3.pbtxt │ │ │ │ ├── QuantizeDownAndShrinkRange.pbtxt │ │ │ │ ├── QuantizeV2.pbtxt │ │ │ │ ├── QuantizedAdd.pbtxt │ │ │ │ ├── QuantizedAvgPool.pbtxt │ │ │ │ ├── QuantizedBatchNormWithGlobalNormalization.pbtxt │ │ │ │ ├── QuantizedBiasAdd.pbtxt │ │ │ │ ├── QuantizedConcat.pbtxt │ │ │ │ ├── QuantizedConv2D.pbtxt │ │ │ │ ├── QuantizedConv2DAndRelu.pbtxt │ │ │ │ ├── QuantizedConv2DAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DPerChannel.pbtxt │ │ │ │ ├── QuantizedConv2DWithBias.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasAndRelu.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasSumAndRelu.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2D.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2DWithBias.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedInstanceNorm.pbtxt │ │ │ │ ├── QuantizedMatMul.pbtxt │ │ │ │ ├── QuantizedMatMulWithBias.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndDequantize.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndRelu.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndRequantize.pbtxt │ │ │ │ ├── QuantizedMaxPool.pbtxt │ │ │ │ ├── QuantizedMul.pbtxt │ │ │ │ ├── QuantizedRelu.pbtxt │ │ │ │ ├── QuantizedRelu6.pbtxt │ │ │ │ ├── QuantizedReluX.pbtxt │ │ │ │ ├── QuantizedReshape.pbtxt │ │ │ │ ├── QuantizedResizeBilinear.pbtxt │ │ │ │ ├── QueueClose.pbtxt │ │ │ │ ├── QueueCloseV2.pbtxt │ │ │ │ ├── QueueDequeue.pbtxt │ │ │ │ ├── QueueDequeueMany.pbtxt │ │ │ │ ├── QueueDequeueManyV2.pbtxt │ │ │ │ ├── QueueDequeueUpTo.pbtxt │ │ │ │ ├── QueueDequeueUpToV2.pbtxt │ │ │ │ ├── QueueDequeueV2.pbtxt │ │ │ │ ├── QueueEnqueue.pbtxt │ │ │ │ ├── QueueEnqueueMany.pbtxt │ │ │ │ ├── QueueEnqueueManyV2.pbtxt │ │ │ │ ├── QueueEnqueueV2.pbtxt │ │ │ │ ├── QueueIsClosed.pbtxt │ │ │ │ ├── QueueIsClosedV2.pbtxt │ │ │ │ ├── QueueSize.pbtxt │ │ │ │ ├── QueueSizeV2.pbtxt │ │ │ │ ├── RFFT.pbtxt │ │ │ │ ├── RFFT2D.pbtxt │ │ │ │ ├── RFFT3D.pbtxt │ │ │ │ ├── RGBToHSV.pbtxt │ │ │ │ ├── RaggedCross.pbtxt │ │ │ │ ├── RaggedGather.pbtxt │ │ │ │ ├── RaggedRange.pbtxt │ │ │ │ ├── RaggedTensorFromVariant.pbtxt │ │ │ │ ├── RaggedTensorToSparse.pbtxt │ │ │ │ ├── RaggedTensorToTensor.pbtxt │ │ │ │ ├── RaggedTensorToVariant.pbtxt │ │ │ │ ├── RandomCrop.pbtxt │ │ │ │ ├── RandomDataset.pbtxt │ │ │ │ ├── RandomGamma.pbtxt │ │ │ │ ├── RandomGammaGrad.pbtxt │ │ │ │ ├── RandomPoisson.pbtxt │ │ │ │ ├── RandomPoissonV2.pbtxt │ │ │ │ ├── RandomShuffle.pbtxt │ │ │ │ ├── RandomShuffleQueue.pbtxt │ │ │ │ ├── RandomShuffleQueueV2.pbtxt │ │ │ │ ├── RandomStandardNormal.pbtxt │ │ │ │ ├── RandomUniform.pbtxt │ │ │ │ ├── RandomUniformInt.pbtxt │ │ │ │ ├── Range.pbtxt │ │ │ │ ├── RangeDataset.pbtxt │ │ │ │ ├── Rank.pbtxt │ │ │ │ ├── ReadFile.pbtxt │ │ │ │ ├── ReadVariableOp.pbtxt │ │ │ │ ├── ReaderNumRecordsProduced.pbtxt │ │ │ │ ├── ReaderNumRecordsProducedV2.pbtxt │ │ │ │ ├── ReaderNumWorkUnitsCompleted.pbtxt │ │ │ │ ├── ReaderNumWorkUnitsCompletedV2.pbtxt │ │ │ │ ├── ReaderRead.pbtxt │ │ │ │ ├── ReaderReadUpTo.pbtxt │ │ │ │ ├── ReaderReadUpToV2.pbtxt │ │ │ │ ├── ReaderReadV2.pbtxt │ │ │ │ ├── ReaderReset.pbtxt │ │ │ │ ├── ReaderResetV2.pbtxt │ │ │ │ ├── ReaderRestoreState.pbtxt │ │ │ │ ├── ReaderRestoreStateV2.pbtxt │ │ │ │ ├── ReaderSerializeState.pbtxt │ │ │ │ ├── ReaderSerializeStateV2.pbtxt │ │ │ │ ├── Real.pbtxt │ │ │ │ ├── RealDiv.pbtxt │ │ │ │ ├── RebatchDataset.pbtxt │ │ │ │ ├── Reciprocal.pbtxt │ │ │ │ ├── ReciprocalGrad.pbtxt │ │ │ │ ├── RecordInput.pbtxt │ │ │ │ ├── Recv.pbtxt │ │ │ │ ├── RecvTPUEmbeddingActivations.pbtxt │ │ │ │ ├── ReduceDataset.pbtxt │ │ │ │ ├── ReduceJoin.pbtxt │ │ │ │ ├── RefEnter.pbtxt │ │ │ │ ├── RefExit.pbtxt │ │ │ │ ├── RefIdentity.pbtxt │ │ │ │ ├── RefMerge.pbtxt │ │ │ │ ├── RefNextIteration.pbtxt │ │ │ │ ├── RefSelect.pbtxt │ │ │ │ ├── RefSwitch.pbtxt │ │ │ │ ├── RegexFullMatch.pbtxt │ │ │ │ ├── RegexReplace.pbtxt │ │ │ │ ├── RegisterDataset.pbtxt │ │ │ │ ├── Relu.pbtxt │ │ │ │ ├── Relu6.pbtxt │ │ │ │ ├── Relu6Grad.pbtxt │ │ │ │ ├── ReluGrad.pbtxt │ │ │ │ ├── RemoteCall.pbtxt │ │ │ │ ├── RepeatDataset.pbtxt │ │ │ │ ├── RequantizationRange.pbtxt │ │ │ │ ├── RequantizationRangePerChannel.pbtxt │ │ │ │ ├── Requantize.pbtxt │ │ │ │ ├── RequantizePerChannel.pbtxt │ │ │ │ ├── Reshape.pbtxt │ │ │ │ ├── ResizeArea.pbtxt │ │ │ │ ├── ResizeBicubic.pbtxt │ │ │ │ ├── ResizeBicubicGrad.pbtxt │ │ │ │ ├── ResizeBilinear.pbtxt │ │ │ │ ├── ResizeBilinearGrad.pbtxt │ │ │ │ ├── ResizeNearestNeighbor.pbtxt │ │ │ │ ├── ResizeNearestNeighborGrad.pbtxt │ │ │ │ ├── ResourceAccumulatorApplyGradient.pbtxt │ │ │ │ ├── ResourceAccumulatorNumAccumulated.pbtxt │ │ │ │ ├── ResourceAccumulatorSetGlobalStep.pbtxt │ │ │ │ ├── ResourceAccumulatorTakeGradient.pbtxt │ │ │ │ ├── ResourceApplyAdaMax.pbtxt │ │ │ │ ├── ResourceApplyAdadelta.pbtxt │ │ │ │ ├── ResourceApplyAdagrad.pbtxt │ │ │ │ ├── ResourceApplyAdagradDA.pbtxt │ │ │ │ ├── ResourceApplyAdagradV2.pbtxt │ │ │ │ ├── ResourceApplyAdam.pbtxt │ │ │ │ ├── ResourceApplyAdamWithAmsgrad.pbtxt │ │ │ │ ├── ResourceApplyAddSign.pbtxt │ │ │ │ ├── ResourceApplyCenteredRMSProp.pbtxt │ │ │ │ ├── ResourceApplyFtrl.pbtxt │ │ │ │ ├── ResourceApplyFtrlV2.pbtxt │ │ │ │ ├── ResourceApplyGradientDescent.pbtxt │ │ │ │ ├── ResourceApplyKerasMomentum.pbtxt │ │ │ │ ├── ResourceApplyMomentum.pbtxt │ │ │ │ ├── ResourceApplyPowerSign.pbtxt │ │ │ │ ├── ResourceApplyProximalAdagrad.pbtxt │ │ │ │ ├── ResourceApplyProximalGradientDescent.pbtxt │ │ │ │ ├── ResourceApplyRMSProp.pbtxt │ │ │ │ ├── ResourceConditionalAccumulator.pbtxt │ │ │ │ ├── ResourceCountUpTo.pbtxt │ │ │ │ ├── ResourceGather.pbtxt │ │ │ │ ├── ResourceGatherNd.pbtxt │ │ │ │ ├── ResourceScatterAdd.pbtxt │ │ │ │ ├── ResourceScatterDiv.pbtxt │ │ │ │ ├── ResourceScatterMax.pbtxt │ │ │ │ ├── ResourceScatterMin.pbtxt │ │ │ │ ├── ResourceScatterMul.pbtxt │ │ │ │ ├── ResourceScatterNdAdd.pbtxt │ │ │ │ ├── ResourceScatterNdSub.pbtxt │ │ │ │ ├── ResourceScatterNdUpdate.pbtxt │ │ │ │ ├── ResourceScatterSub.pbtxt │ │ │ │ ├── ResourceScatterUpdate.pbtxt │ │ │ │ ├── ResourceSparseApplyAdadelta.pbtxt │ │ │ │ ├── ResourceSparseApplyAdagrad.pbtxt │ │ │ │ ├── ResourceSparseApplyAdagradDA.pbtxt │ │ │ │ ├── ResourceSparseApplyAdagradV2.pbtxt │ │ │ │ ├── ResourceSparseApplyCenteredRMSProp.pbtxt │ │ │ │ ├── ResourceSparseApplyFtrl.pbtxt │ │ │ │ ├── ResourceSparseApplyFtrlV2.pbtxt │ │ │ │ ├── ResourceSparseApplyKerasMomentum.pbtxt │ │ │ │ ├── ResourceSparseApplyMomentum.pbtxt │ │ │ │ ├── ResourceSparseApplyProximalAdagrad.pbtxt │ │ │ │ ├── ResourceSparseApplyProximalGradientDescent.pbtxt │ │ │ │ ├── ResourceSparseApplyRMSProp.pbtxt │ │ │ │ ├── ResourceStridedSliceAssign.pbtxt │ │ │ │ ├── Restore.pbtxt │ │ │ │ ├── RestoreSlice.pbtxt │ │ │ │ ├── RestoreV2.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingADAMParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingAdadeltaParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingAdagradParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingFTRLParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingMomentumParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingProximalYogiParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingRMSPropParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt │ │ │ │ ├── Reverse.pbtxt │ │ │ │ ├── ReverseSequence.pbtxt │ │ │ │ ├── ReverseV2.pbtxt │ │ │ │ ├── RightShift.pbtxt │ │ │ │ ├── Rint.pbtxt │ │ │ │ ├── RngSkip.pbtxt │ │ │ │ ├── Roll.pbtxt │ │ │ │ ├── Round.pbtxt │ │ │ │ ├── Rsqrt.pbtxt │ │ │ │ ├── RsqrtGrad.pbtxt │ │ │ │ ├── SampleDistortedBoundingBox.pbtxt │ │ │ │ ├── SampleDistortedBoundingBoxV2.pbtxt │ │ │ │ ├── SamplingDataset.pbtxt │ │ │ │ ├── Save.pbtxt │ │ │ │ ├── SaveSlices.pbtxt │ │ │ │ ├── SaveV2.pbtxt │ │ │ │ ├── ScalarSummary.pbtxt │ │ │ │ ├── ScaleAndTranslate.pbtxt │ │ │ │ ├── ScaleAndTranslateGrad.pbtxt │ │ │ │ ├── ScanDataset.pbtxt │ │ │ │ ├── ScatterAdd.pbtxt │ │ │ │ ├── ScatterDiv.pbtxt │ │ │ │ ├── ScatterMax.pbtxt │ │ │ │ ├── ScatterMin.pbtxt │ │ │ │ ├── ScatterMul.pbtxt │ │ │ │ ├── ScatterNd.pbtxt │ │ │ │ ├── ScatterNdAdd.pbtxt │ │ │ │ ├── ScatterNdNonAliasingAdd.pbtxt │ │ │ │ ├── ScatterNdSub.pbtxt │ │ │ │ ├── ScatterNdUpdate.pbtxt │ │ │ │ ├── ScatterSub.pbtxt │ │ │ │ ├── ScatterUpdate.pbtxt │ │ │ │ ├── SdcaFprint.pbtxt │ │ │ │ ├── SdcaOptimizer.pbtxt │ │ │ │ ├── SdcaOptimizerV2.pbtxt │ │ │ │ ├── SdcaShrinkL1.pbtxt │ │ │ │ ├── SegmentMax.pbtxt │ │ │ │ ├── SegmentMean.pbtxt │ │ │ │ ├── SegmentMin.pbtxt │ │ │ │ ├── SegmentProd.pbtxt │ │ │ │ ├── SegmentSum.pbtxt │ │ │ │ ├── Select.pbtxt │ │ │ │ ├── SelectV2.pbtxt │ │ │ │ ├── SelfAdjointEig.pbtxt │ │ │ │ ├── SelfAdjointEigV2.pbtxt │ │ │ │ ├── Selu.pbtxt │ │ │ │ ├── SeluGrad.pbtxt │ │ │ │ ├── Send.pbtxt │ │ │ │ ├── SendTPUEmbeddingGradients.pbtxt │ │ │ │ ├── SerializeIterator.pbtxt │ │ │ │ ├── SerializeManySparse.pbtxt │ │ │ │ ├── SerializeSparse.pbtxt │ │ │ │ ├── SerializeTensor.pbtxt │ │ │ │ ├── SetSize.pbtxt │ │ │ │ ├── SetStatsAggregatorDataset.pbtxt │ │ │ │ ├── Shape.pbtxt │ │ │ │ ├── ShapeN.pbtxt │ │ │ │ ├── ShardDataset.pbtxt │ │ │ │ ├── ShardedFilename.pbtxt │ │ │ │ ├── ShardedFilespec.pbtxt │ │ │ │ ├── ShuffleAndRepeatDataset.pbtxt │ │ │ │ ├── ShuffleDataset.pbtxt │ │ │ │ ├── ShuffleDatasetV2.pbtxt │ │ │ │ ├── ShutdownDistributedTPU.pbtxt │ │ │ │ ├── Sigmoid.pbtxt │ │ │ │ ├── SigmoidGrad.pbtxt │ │ │ │ ├── Sign.pbtxt │ │ │ │ ├── Sin.pbtxt │ │ │ │ ├── Sinh.pbtxt │ │ │ │ ├── Size.pbtxt │ │ │ │ ├── SkipDataset.pbtxt │ │ │ │ ├── Skipgram.pbtxt │ │ │ │ ├── SleepDataset.pbtxt │ │ │ │ ├── Slice.pbtxt │ │ │ │ ├── SlidingWindowDataset.pbtxt │ │ │ │ ├── Snapshot.pbtxt │ │ │ │ ├── SnapshotDataset.pbtxt │ │ │ │ ├── SobolSample.pbtxt │ │ │ │ ├── Softmax.pbtxt │ │ │ │ ├── SoftmaxCrossEntropyWithLogits.pbtxt │ │ │ │ ├── Softplus.pbtxt │ │ │ │ ├── SoftplusGrad.pbtxt │ │ │ │ ├── Softsign.pbtxt │ │ │ │ ├── SoftsignGrad.pbtxt │ │ │ │ ├── SpaceToBatch.pbtxt │ │ │ │ ├── SpaceToBatchND.pbtxt │ │ │ │ ├── SpaceToDepth.pbtxt │ │ │ │ ├── SparseAccumulatorApplyGradient.pbtxt │ │ │ │ ├── SparseAccumulatorTakeGradient.pbtxt │ │ │ │ ├── SparseAdd.pbtxt │ │ │ │ ├── SparseAddGrad.pbtxt │ │ │ │ ├── SparseApplyAdadelta.pbtxt │ │ │ │ ├── SparseApplyAdagrad.pbtxt │ │ │ │ ├── SparseApplyAdagradDA.pbtxt │ │ │ │ ├── SparseApplyAdagradV2.pbtxt │ │ │ │ ├── SparseApplyCenteredRMSProp.pbtxt │ │ │ │ ├── SparseApplyFtrl.pbtxt │ │ │ │ ├── SparseApplyFtrlV2.pbtxt │ │ │ │ ├── SparseApplyMomentum.pbtxt │ │ │ │ ├── SparseApplyProximalAdagrad.pbtxt │ │ │ │ ├── SparseApplyProximalGradientDescent.pbtxt │ │ │ │ ├── SparseApplyRMSProp.pbtxt │ │ │ │ ├── SparseConcat.pbtxt │ │ │ │ ├── SparseConditionalAccumulator.pbtxt │ │ │ │ ├── SparseCross.pbtxt │ │ │ │ ├── SparseDenseCwiseAdd.pbtxt │ │ │ │ ├── SparseDenseCwiseDiv.pbtxt │ │ │ │ ├── SparseDenseCwiseMul.pbtxt │ │ │ │ ├── SparseFillEmptyRows.pbtxt │ │ │ │ ├── SparseFillEmptyRowsGrad.pbtxt │ │ │ │ ├── SparseMatMul.pbtxt │ │ │ │ ├── SparseMatrixAdd.pbtxt │ │ │ │ ├── SparseMatrixMatMul.pbtxt │ │ │ │ ├── SparseMatrixMul.pbtxt │ │ │ │ ├── SparseMatrixNNZ.pbtxt │ │ │ │ ├── SparseMatrixOrderingAMD.pbtxt │ │ │ │ ├── SparseMatrixSoftmax.pbtxt │ │ │ │ ├── SparseMatrixSoftmaxGrad.pbtxt │ │ │ │ ├── SparseMatrixSparseCholesky.pbtxt │ │ │ │ ├── SparseMatrixSparseMatMul.pbtxt │ │ │ │ ├── SparseMatrixTranspose.pbtxt │ │ │ │ ├── SparseMatrixZeros.pbtxt │ │ │ │ ├── SparseReduceMax.pbtxt │ │ │ │ ├── SparseReduceMaxSparse.pbtxt │ │ │ │ ├── SparseReduceSum.pbtxt │ │ │ │ ├── SparseReduceSumSparse.pbtxt │ │ │ │ ├── SparseReorder.pbtxt │ │ │ │ ├── SparseReshape.pbtxt │ │ │ │ ├── SparseSegmentMean.pbtxt │ │ │ │ ├── SparseSegmentMeanGrad.pbtxt │ │ │ │ ├── SparseSegmentMeanWithNumSegments.pbtxt │ │ │ │ ├── SparseSegmentSqrtN.pbtxt │ │ │ │ ├── SparseSegmentSqrtNGrad.pbtxt │ │ │ │ ├── SparseSegmentSqrtNWithNumSegments.pbtxt │ │ │ │ ├── SparseSegmentSum.pbtxt │ │ │ │ ├── SparseSegmentSumWithNumSegments.pbtxt │ │ │ │ ├── SparseSlice.pbtxt │ │ │ │ ├── SparseSliceGrad.pbtxt │ │ │ │ ├── SparseSoftmax.pbtxt │ │ │ │ ├── SparseSoftmaxCrossEntropyWithLogits.pbtxt │ │ │ │ ├── SparseSparseMaximum.pbtxt │ │ │ │ ├── SparseSparseMinimum.pbtxt │ │ │ │ ├── SparseSplit.pbtxt │ │ │ │ ├── SparseTensorDenseAdd.pbtxt │ │ │ │ ├── SparseTensorDenseMatMul.pbtxt │ │ │ │ ├── SparseTensorSliceDataset.pbtxt │ │ │ │ ├── SparseTensorToCSRSparseMatrix.pbtxt │ │ │ │ ├── SparseToDense.pbtxt │ │ │ │ ├── SparseToSparseSetOperation.pbtxt │ │ │ │ ├── Spence.pbtxt │ │ │ │ ├── Split.pbtxt │ │ │ │ ├── SplitV.pbtxt │ │ │ │ ├── SqlDataset.pbtxt │ │ │ │ ├── Sqrt.pbtxt │ │ │ │ ├── SqrtGrad.pbtxt │ │ │ │ ├── Square.pbtxt │ │ │ │ ├── SquaredDifference.pbtxt │ │ │ │ ├── Squeeze.pbtxt │ │ │ │ ├── Stack.pbtxt │ │ │ │ ├── StackClose.pbtxt │ │ │ │ ├── StackCloseV2.pbtxt │ │ │ │ ├── StackPop.pbtxt │ │ │ │ ├── StackPopV2.pbtxt │ │ │ │ ├── StackPush.pbtxt │ │ │ │ ├── StackPushV2.pbtxt │ │ │ │ ├── StackV2.pbtxt │ │ │ │ ├── Stage.pbtxt │ │ │ │ ├── StageClear.pbtxt │ │ │ │ ├── StagePeek.pbtxt │ │ │ │ ├── StageSize.pbtxt │ │ │ │ ├── StatefulPartitionedCall.pbtxt │ │ │ │ ├── StatefulRandomBinomial.pbtxt │ │ │ │ ├── StatefulStandardNormal.pbtxt │ │ │ │ ├── StatefulStandardNormalV2.pbtxt │ │ │ │ ├── StatefulTruncatedNormal.pbtxt │ │ │ │ ├── StatefulUniform.pbtxt │ │ │ │ ├── StatefulUniformFullInt.pbtxt │ │ │ │ ├── StatefulUniformInt.pbtxt │ │ │ │ ├── StatelessIf.pbtxt │ │ │ │ ├── StatelessMultinomial.pbtxt │ │ │ │ ├── StatelessRandomBinomial.pbtxt │ │ │ │ ├── StatelessRandomGammaV2.pbtxt │ │ │ │ ├── StatelessRandomNormal.pbtxt │ │ │ │ ├── StatelessRandomPoisson.pbtxt │ │ │ │ ├── StatelessRandomUniform.pbtxt │ │ │ │ ├── StatelessRandomUniformFullInt.pbtxt │ │ │ │ ├── StatelessRandomUniformInt.pbtxt │ │ │ │ ├── StatelessTruncatedNormal.pbtxt │ │ │ │ ├── StatelessWhile.pbtxt │ │ │ │ ├── StaticRegexFullMatch.pbtxt │ │ │ │ ├── StaticRegexReplace.pbtxt │ │ │ │ ├── StatsAggregatorHandle.pbtxt │ │ │ │ ├── StatsAggregatorHandleV2.pbtxt │ │ │ │ ├── StatsAggregatorSetSummaryWriter.pbtxt │ │ │ │ ├── StatsAggregatorSummary.pbtxt │ │ │ │ ├── StopGradient.pbtxt │ │ │ │ ├── StridedSlice.pbtxt │ │ │ │ ├── StridedSliceAssign.pbtxt │ │ │ │ ├── StridedSliceGrad.pbtxt │ │ │ │ ├── StringFormat.pbtxt │ │ │ │ ├── StringJoin.pbtxt │ │ │ │ ├── StringLength.pbtxt │ │ │ │ ├── StringLower.pbtxt │ │ │ │ ├── StringNGrams.pbtxt │ │ │ │ ├── StringSplit.pbtxt │ │ │ │ ├── StringSplitV2.pbtxt │ │ │ │ ├── StringStrip.pbtxt │ │ │ │ ├── StringToHashBucket.pbtxt │ │ │ │ ├── StringToHashBucketFast.pbtxt │ │ │ │ ├── StringToHashBucketStrong.pbtxt │ │ │ │ ├── StringToNumber.pbtxt │ │ │ │ ├── StringUpper.pbtxt │ │ │ │ ├── Sub.pbtxt │ │ │ │ ├── Substr.pbtxt │ │ │ │ ├── Sum.pbtxt │ │ │ │ ├── SummaryWriter.pbtxt │ │ │ │ ├── Svd.pbtxt │ │ │ │ ├── Switch.pbtxt │ │ │ │ ├── SymbolicGradient.pbtxt │ │ │ │ ├── TFRecordDataset.pbtxt │ │ │ │ ├── TFRecordReader.pbtxt │ │ │ │ ├── TFRecordReaderV2.pbtxt │ │ │ │ ├── TPUCompilationResult.pbtxt │ │ │ │ ├── TPUEmbeddingActivations.pbtxt │ │ │ │ ├── TPUOrdinalSelector.pbtxt │ │ │ │ ├── TPUPartitionedCall.pbtxt │ │ │ │ ├── TPUReplicateMetadata.pbtxt │ │ │ │ ├── TPUReplicatedInput.pbtxt │ │ │ │ ├── TPUReplicatedOutput.pbtxt │ │ │ │ ├── TakeDataset.pbtxt │ │ │ │ ├── TakeManySparseFromTensorsMap.pbtxt │ │ │ │ ├── TakeWhileDataset.pbtxt │ │ │ │ ├── Tan.pbtxt │ │ │ │ ├── Tanh.pbtxt │ │ │ │ ├── TanhGrad.pbtxt │ │ │ │ ├── TemporaryVariable.pbtxt │ │ │ │ ├── TensorArray.pbtxt │ │ │ │ ├── TensorArrayClose.pbtxt │ │ │ │ ├── TensorArrayCloseV2.pbtxt │ │ │ │ ├── TensorArrayCloseV3.pbtxt │ │ │ │ ├── TensorArrayConcat.pbtxt │ │ │ │ ├── TensorArrayConcatV2.pbtxt │ │ │ │ ├── TensorArrayConcatV3.pbtxt │ │ │ │ ├── TensorArrayGather.pbtxt │ │ │ │ ├── TensorArrayGatherV2.pbtxt │ │ │ │ ├── TensorArrayGatherV3.pbtxt │ │ │ │ ├── TensorArrayGrad.pbtxt │ │ │ │ ├── TensorArrayGradV2.pbtxt │ │ │ │ ├── TensorArrayGradV3.pbtxt │ │ │ │ ├── TensorArrayGradWithShape.pbtxt │ │ │ │ ├── TensorArrayPack.pbtxt │ │ │ │ ├── TensorArrayRead.pbtxt │ │ │ │ ├── TensorArrayReadV2.pbtxt │ │ │ │ ├── TensorArrayReadV3.pbtxt │ │ │ │ ├── TensorArrayScatter.pbtxt │ │ │ │ ├── TensorArrayScatterV2.pbtxt │ │ │ │ ├── TensorArrayScatterV3.pbtxt │ │ │ │ ├── TensorArraySize.pbtxt │ │ │ │ ├── TensorArraySizeV2.pbtxt │ │ │ │ ├── TensorArraySizeV3.pbtxt │ │ │ │ ├── TensorArraySplit.pbtxt │ │ │ │ ├── TensorArraySplitV2.pbtxt │ │ │ │ ├── TensorArraySplitV3.pbtxt │ │ │ │ ├── TensorArrayUnpack.pbtxt │ │ │ │ ├── TensorArrayV2.pbtxt │ │ │ │ ├── TensorArrayV3.pbtxt │ │ │ │ ├── TensorArrayWrite.pbtxt │ │ │ │ ├── TensorArrayWriteV2.pbtxt │ │ │ │ ├── TensorArrayWriteV3.pbtxt │ │ │ │ ├── TensorDataset.pbtxt │ │ │ │ ├── TensorListConcat.pbtxt │ │ │ │ ├── TensorListConcatLists.pbtxt │ │ │ │ ├── TensorListConcatV2.pbtxt │ │ │ │ ├── TensorListElementShape.pbtxt │ │ │ │ ├── TensorListFromTensor.pbtxt │ │ │ │ ├── TensorListGather.pbtxt │ │ │ │ ├── TensorListGetItem.pbtxt │ │ │ │ ├── TensorListLength.pbtxt │ │ │ │ ├── TensorListPopBack.pbtxt │ │ │ │ ├── TensorListPushBack.pbtxt │ │ │ │ ├── TensorListPushBackBatch.pbtxt │ │ │ │ ├── TensorListReserve.pbtxt │ │ │ │ ├── TensorListResize.pbtxt │ │ │ │ ├── TensorListScatter.pbtxt │ │ │ │ ├── TensorListScatterIntoExistingList.pbtxt │ │ │ │ ├── TensorListScatterV2.pbtxt │ │ │ │ ├── TensorListSetItem.pbtxt │ │ │ │ ├── TensorListSplit.pbtxt │ │ │ │ ├── TensorListStack.pbtxt │ │ │ │ ├── TensorScatterAdd.pbtxt │ │ │ │ ├── TensorScatterSub.pbtxt │ │ │ │ ├── TensorScatterUpdate.pbtxt │ │ │ │ ├── TensorSliceDataset.pbtxt │ │ │ │ ├── TensorStridedSliceUpdate.pbtxt │ │ │ │ ├── TensorSummary.pbtxt │ │ │ │ ├── TensorSummaryV2.pbtxt │ │ │ │ ├── TextLineDataset.pbtxt │ │ │ │ ├── TextLineReader.pbtxt │ │ │ │ ├── TextLineReaderV2.pbtxt │ │ │ │ ├── ThreadPoolDataset.pbtxt │ │ │ │ ├── ThreadPoolHandle.pbtxt │ │ │ │ ├── ThreadUnsafeUnigramCandidateSampler.pbtxt │ │ │ │ ├── Tile.pbtxt │ │ │ │ ├── TileGrad.pbtxt │ │ │ │ ├── Timestamp.pbtxt │ │ │ │ ├── ToBool.pbtxt │ │ │ │ ├── TopK.pbtxt │ │ │ │ ├── TopKV2.pbtxt │ │ │ │ ├── Transpose.pbtxt │ │ │ │ ├── TridiagonalMatMul.pbtxt │ │ │ │ ├── TridiagonalSolve.pbtxt │ │ │ │ ├── TruncateDiv.pbtxt │ │ │ │ ├── TruncateMod.pbtxt │ │ │ │ ├── TruncatedNormal.pbtxt │ │ │ │ ├── Unbatch.pbtxt │ │ │ │ ├── UnbatchDataset.pbtxt │ │ │ │ ├── UnbatchGrad.pbtxt │ │ │ │ ├── UnicodeDecode.pbtxt │ │ │ │ ├── UnicodeDecodeWithOffsets.pbtxt │ │ │ │ ├── UnicodeEncode.pbtxt │ │ │ │ ├── UnicodeScript.pbtxt │ │ │ │ ├── UnicodeTranscode.pbtxt │ │ │ │ ├── UniformCandidateSampler.pbtxt │ │ │ │ ├── Unique.pbtxt │ │ │ │ ├── UniqueDataset.pbtxt │ │ │ │ ├── UniqueV2.pbtxt │ │ │ │ ├── UniqueWithCounts.pbtxt │ │ │ │ ├── UniqueWithCountsV2.pbtxt │ │ │ │ ├── Unpack.pbtxt │ │ │ │ ├── UnravelIndex.pbtxt │ │ │ │ ├── UnsortedSegmentJoin.pbtxt │ │ │ │ ├── UnsortedSegmentMax.pbtxt │ │ │ │ ├── UnsortedSegmentMin.pbtxt │ │ │ │ ├── UnsortedSegmentProd.pbtxt │ │ │ │ ├── UnsortedSegmentSum.pbtxt │ │ │ │ ├── Unstage.pbtxt │ │ │ │ ├── UnwrapDatasetVariant.pbtxt │ │ │ │ ├── UpperBound.pbtxt │ │ │ │ ├── VarHandleOp.pbtxt │ │ │ │ ├── VarIsInitializedOp.pbtxt │ │ │ │ ├── Variable.pbtxt │ │ │ │ ├── VariableShape.pbtxt │ │ │ │ ├── VariableV2.pbtxt │ │ │ │ ├── Where.pbtxt │ │ │ │ ├── While.pbtxt │ │ │ │ ├── WholeFileReader.pbtxt │ │ │ │ ├── WholeFileReaderV2.pbtxt │ │ │ │ ├── WindowDataset.pbtxt │ │ │ │ ├── WorkerHeartbeat.pbtxt │ │ │ │ ├── WrapDatasetVariant.pbtxt │ │ │ │ ├── WriteAudioSummary.pbtxt │ │ │ │ ├── WriteFile.pbtxt │ │ │ │ ├── WriteGraphSummary.pbtxt │ │ │ │ ├── WriteHistogramSummary.pbtxt │ │ │ │ ├── WriteImageSummary.pbtxt │ │ │ │ ├── WriteRawProtoSummary.pbtxt │ │ │ │ ├── WriteScalarSummary.pbtxt │ │ │ │ ├── WriteSummary.pbtxt │ │ │ │ ├── Xdivy.pbtxt │ │ │ │ ├── Xlog1py.pbtxt │ │ │ │ ├── Xlogy.pbtxt │ │ │ │ ├── ZerosLike.pbtxt │ │ │ │ ├── Zeta.pbtxt │ │ │ │ └── ZipDataset.pbtxt │ │ │ ├── ops_history_v2 │ │ │ │ ├── Abort.pbtxt │ │ │ │ ├── Abs.pbtxt │ │ │ │ ├── AccumulateNV2.pbtxt │ │ │ │ ├── AccumulatorApplyGradient.pbtxt │ │ │ │ ├── AccumulatorNumAccumulated.pbtxt │ │ │ │ ├── AccumulatorSetGlobalStep.pbtxt │ │ │ │ ├── AccumulatorTakeGradient.pbtxt │ │ │ │ ├── Acos.pbtxt │ │ │ │ ├── Acosh.pbtxt │ │ │ │ ├── Add.pbtxt │ │ │ │ ├── AddManySparseToTensorsMap.pbtxt │ │ │ │ ├── AddN.pbtxt │ │ │ │ ├── AddSparseToTensorsMap.pbtxt │ │ │ │ ├── AddV2.pbtxt │ │ │ │ ├── AdjustContrast.pbtxt │ │ │ │ ├── AdjustContrastv2.pbtxt │ │ │ │ ├── AdjustHue.pbtxt │ │ │ │ ├── AdjustSaturation.pbtxt │ │ │ │ ├── All.pbtxt │ │ │ │ ├── AllCandidateSampler.pbtxt │ │ │ │ ├── AllToAll.pbtxt │ │ │ │ ├── Angle.pbtxt │ │ │ │ ├── AnonymousHashTable.pbtxt │ │ │ │ ├── AnonymousIterator.pbtxt │ │ │ │ ├── AnonymousIteratorV2.pbtxt │ │ │ │ ├── AnonymousIteratorV3.pbtxt │ │ │ │ ├── AnonymousMemoryCache.pbtxt │ │ │ │ ├── AnonymousMultiDeviceIterator.pbtxt │ │ │ │ ├── AnonymousMultiDeviceIteratorV3.pbtxt │ │ │ │ ├── AnonymousMutableDenseHashTable.pbtxt │ │ │ │ ├── AnonymousMutableHashTable.pbtxt │ │ │ │ ├── AnonymousMutableHashTableOfTensors.pbtxt │ │ │ │ ├── AnonymousRandomSeedGenerator.pbtxt │ │ │ │ ├── AnonymousSeedGenerator.pbtxt │ │ │ │ ├── Any.pbtxt │ │ │ │ ├── ApplyAdaMax.pbtxt │ │ │ │ ├── ApplyAdadelta.pbtxt │ │ │ │ ├── ApplyAdagrad.pbtxt │ │ │ │ ├── ApplyAdagradDA.pbtxt │ │ │ │ ├── ApplyAdagradV2.pbtxt │ │ │ │ ├── ApplyAdam.pbtxt │ │ │ │ ├── ApplyAddSign.pbtxt │ │ │ │ ├── ApplyCenteredRMSProp.pbtxt │ │ │ │ ├── ApplyFtrl.pbtxt │ │ │ │ ├── ApplyFtrlV2.pbtxt │ │ │ │ ├── ApplyGradientDescent.pbtxt │ │ │ │ ├── ApplyMomentum.pbtxt │ │ │ │ ├── ApplyPowerSign.pbtxt │ │ │ │ ├── ApplyProximalAdagrad.pbtxt │ │ │ │ ├── ApplyProximalGradientDescent.pbtxt │ │ │ │ ├── ApplyRMSProp.pbtxt │ │ │ │ ├── ApproxTopK.pbtxt │ │ │ │ ├── ApproximateEqual.pbtxt │ │ │ │ ├── ArgMax.pbtxt │ │ │ │ ├── ArgMin.pbtxt │ │ │ │ ├── AsString.pbtxt │ │ │ │ ├── Asin.pbtxt │ │ │ │ ├── Asinh.pbtxt │ │ │ │ ├── Assert.pbtxt │ │ │ │ ├── AssertCardinalityDataset.pbtxt │ │ │ │ ├── AssertNextDataset.pbtxt │ │ │ │ ├── AssertPrevDataset.pbtxt │ │ │ │ ├── Assign.pbtxt │ │ │ │ ├── AssignAdd.pbtxt │ │ │ │ ├── AssignAddVariableOp.pbtxt │ │ │ │ ├── AssignSub.pbtxt │ │ │ │ ├── AssignSubVariableOp.pbtxt │ │ │ │ ├── AssignVariableOp.pbtxt │ │ │ │ ├── AssignVariableXlaConcatND.pbtxt │ │ │ │ ├── Atan.pbtxt │ │ │ │ ├── Atan2.pbtxt │ │ │ │ ├── Atanh.pbtxt │ │ │ │ ├── AudioSpectrogram.pbtxt │ │ │ │ ├── AudioSummary.pbtxt │ │ │ │ ├── AudioSummaryV2.pbtxt │ │ │ │ ├── AutoShardDataset.pbtxt │ │ │ │ ├── AvgPool.pbtxt │ │ │ │ ├── AvgPool3D.pbtxt │ │ │ │ ├── AvgPool3DGrad.pbtxt │ │ │ │ ├── AvgPoolGrad.pbtxt │ │ │ │ ├── BUILD │ │ │ │ ├── BandedTriangularSolve.pbtxt │ │ │ │ ├── Barrier.pbtxt │ │ │ │ ├── BarrierClose.pbtxt │ │ │ │ ├── BarrierIncompleteSize.pbtxt │ │ │ │ ├── BarrierInsertMany.pbtxt │ │ │ │ ├── BarrierReadySize.pbtxt │ │ │ │ ├── BarrierTakeMany.pbtxt │ │ │ │ ├── Batch.pbtxt │ │ │ │ ├── BatchCholesky.pbtxt │ │ │ │ ├── BatchCholeskyGrad.pbtxt │ │ │ │ ├── BatchDataset.pbtxt │ │ │ │ ├── BatchDatasetV2.pbtxt │ │ │ │ ├── BatchFFT.pbtxt │ │ │ │ ├── BatchFFT2D.pbtxt │ │ │ │ ├── BatchFFT3D.pbtxt │ │ │ │ ├── BatchFunction.pbtxt │ │ │ │ ├── BatchIFFT.pbtxt │ │ │ │ ├── BatchIFFT2D.pbtxt │ │ │ │ ├── BatchIFFT3D.pbtxt │ │ │ │ ├── BatchMatMul.pbtxt │ │ │ │ ├── BatchMatMulV2.pbtxt │ │ │ │ ├── BatchMatMulV3.pbtxt │ │ │ │ ├── BatchMatrixBandPart.pbtxt │ │ │ │ ├── BatchMatrixDeterminant.pbtxt │ │ │ │ ├── BatchMatrixDiag.pbtxt │ │ │ │ ├── BatchMatrixDiagPart.pbtxt │ │ │ │ ├── BatchMatrixInverse.pbtxt │ │ │ │ ├── BatchMatrixSetDiag.pbtxt │ │ │ │ ├── BatchMatrixSolve.pbtxt │ │ │ │ ├── BatchMatrixSolveLs.pbtxt │ │ │ │ ├── BatchMatrixTriangularSolve.pbtxt │ │ │ │ ├── BatchNormWithGlobalNormalization.pbtxt │ │ │ │ ├── BatchNormWithGlobalNormalizationGrad.pbtxt │ │ │ │ ├── BatchSelfAdjointEig.pbtxt │ │ │ │ ├── BatchSelfAdjointEigV2.pbtxt │ │ │ │ ├── BatchSvd.pbtxt │ │ │ │ ├── BatchToSpace.pbtxt │ │ │ │ ├── BatchToSpaceND.pbtxt │ │ │ │ ├── BesselI0.pbtxt │ │ │ │ ├── BesselI0e.pbtxt │ │ │ │ ├── BesselI1.pbtxt │ │ │ │ ├── BesselI1e.pbtxt │ │ │ │ ├── BesselJ0.pbtxt │ │ │ │ ├── BesselJ1.pbtxt │ │ │ │ ├── BesselK0.pbtxt │ │ │ │ ├── BesselK0e.pbtxt │ │ │ │ ├── BesselK1.pbtxt │ │ │ │ ├── BesselK1e.pbtxt │ │ │ │ ├── BesselY0.pbtxt │ │ │ │ ├── BesselY1.pbtxt │ │ │ │ ├── Betainc.pbtxt │ │ │ │ ├── BiasAdd.pbtxt │ │ │ │ ├── BiasAddGrad.pbtxt │ │ │ │ ├── BiasAddV1.pbtxt │ │ │ │ ├── Bincount.pbtxt │ │ │ │ ├── Bitcast.pbtxt │ │ │ │ ├── BitwiseAnd.pbtxt │ │ │ │ ├── BitwiseOr.pbtxt │ │ │ │ ├── BitwiseXor.pbtxt │ │ │ │ ├── BlockLSTM.pbtxt │ │ │ │ ├── BlockLSTMGrad.pbtxt │ │ │ │ ├── BlockLSTMGradV2.pbtxt │ │ │ │ ├── BlockLSTMV2.pbtxt │ │ │ │ ├── BoostedTreesAggregateStats.pbtxt │ │ │ │ ├── BoostedTreesBucketize.pbtxt │ │ │ │ ├── BoostedTreesCalculateBestFeatureSplit.pbtxt │ │ │ │ ├── BoostedTreesCalculateBestFeatureSplitV2.pbtxt │ │ │ │ ├── BoostedTreesCalculateBestGainsPerFeature.pbtxt │ │ │ │ ├── BoostedTreesCenterBias.pbtxt │ │ │ │ ├── BoostedTreesCreateEnsemble.pbtxt │ │ │ │ ├── BoostedTreesCreateQuantileStreamResource.pbtxt │ │ │ │ ├── BoostedTreesDeserializeEnsemble.pbtxt │ │ │ │ ├── BoostedTreesEnsembleResourceHandleOp.pbtxt │ │ │ │ ├── BoostedTreesExampleDebugOutputs.pbtxt │ │ │ │ ├── BoostedTreesFlushQuantileSummaries.pbtxt │ │ │ │ ├── BoostedTreesGetEnsembleStates.pbtxt │ │ │ │ ├── BoostedTreesMakeQuantileSummaries.pbtxt │ │ │ │ ├── BoostedTreesMakeStatsSummary.pbtxt │ │ │ │ ├── BoostedTreesPredict.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceAddSummaries.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceDeserialize.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceFlush.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceGetBucketBoundaries.pbtxt │ │ │ │ ├── BoostedTreesQuantileStreamResourceHandleOp.pbtxt │ │ │ │ ├── BoostedTreesSerializeEnsemble.pbtxt │ │ │ │ ├── BoostedTreesSparseAggregateStats.pbtxt │ │ │ │ ├── BoostedTreesSparseCalculateBestFeatureSplit.pbtxt │ │ │ │ ├── BoostedTreesTrainingPredict.pbtxt │ │ │ │ ├── BoostedTreesUpdateEnsemble.pbtxt │ │ │ │ ├── BoostedTreesUpdateEnsembleV2.pbtxt │ │ │ │ ├── BroadcastArgs.pbtxt │ │ │ │ ├── BroadcastGradientArgs.pbtxt │ │ │ │ ├── BroadcastTo.pbtxt │ │ │ │ ├── Bucketize.pbtxt │ │ │ │ ├── BytesProducedStatsDataset.pbtxt │ │ │ │ ├── CSRSparseMatrixComponents.pbtxt │ │ │ │ ├── CSRSparseMatrixToDense.pbtxt │ │ │ │ ├── CSRSparseMatrixToSparseTensor.pbtxt │ │ │ │ ├── CSVDataset.pbtxt │ │ │ │ ├── CSVDatasetV2.pbtxt │ │ │ │ ├── CTCBeamSearchDecoder.pbtxt │ │ │ │ ├── CTCGreedyDecoder.pbtxt │ │ │ │ ├── CTCLoss.pbtxt │ │ │ │ ├── CTCLossV2.pbtxt │ │ │ │ ├── CacheDataset.pbtxt │ │ │ │ ├── CacheDatasetV2.pbtxt │ │ │ │ ├── Case.pbtxt │ │ │ │ ├── Cast.pbtxt │ │ │ │ ├── Ceil.pbtxt │ │ │ │ ├── CheckNumerics.pbtxt │ │ │ │ ├── CheckNumericsV2.pbtxt │ │ │ │ ├── Cholesky.pbtxt │ │ │ │ ├── CholeskyGrad.pbtxt │ │ │ │ ├── ChooseFastestBranchDataset.pbtxt │ │ │ │ ├── ChooseFastestDataset.pbtxt │ │ │ │ ├── ClipByValue.pbtxt │ │ │ │ ├── CloseSummaryWriter.pbtxt │ │ │ │ ├── CollateTPUEmbeddingMemory.pbtxt │ │ │ │ ├── CollectiveAllToAllV2.pbtxt │ │ │ │ ├── CollectiveAllToAllV3.pbtxt │ │ │ │ ├── CollectiveAssignGroupV2.pbtxt │ │ │ │ ├── CollectiveBcastRecv.pbtxt │ │ │ │ ├── CollectiveBcastRecvV2.pbtxt │ │ │ │ ├── CollectiveBcastSend.pbtxt │ │ │ │ ├── CollectiveBcastSendV2.pbtxt │ │ │ │ ├── CollectiveGather.pbtxt │ │ │ │ ├── CollectiveGatherV2.pbtxt │ │ │ │ ├── CollectiveInitializeCommunicator.pbtxt │ │ │ │ ├── CollectivePermute.pbtxt │ │ │ │ ├── CollectiveReduce.pbtxt │ │ │ │ ├── CollectiveReduceScatterV2.pbtxt │ │ │ │ ├── CollectiveReduceV2.pbtxt │ │ │ │ ├── CollectiveReduceV3.pbtxt │ │ │ │ ├── CombinedNonMaxSuppression.pbtxt │ │ │ │ ├── Complex.pbtxt │ │ │ │ ├── ComplexAbs.pbtxt │ │ │ │ ├── CompositeTensorVariantFromComponents.pbtxt │ │ │ │ ├── CompositeTensorVariantToComponents.pbtxt │ │ │ │ ├── CompressElement.pbtxt │ │ │ │ ├── ComputeAccidentalHits.pbtxt │ │ │ │ ├── ComputeBatchSize.pbtxt │ │ │ │ ├── ComputeDedupDataSize.pbtxt │ │ │ │ ├── ComputeDedupDataTupleMask.pbtxt │ │ │ │ ├── Concat.pbtxt │ │ │ │ ├── ConcatOffset.pbtxt │ │ │ │ ├── ConcatV2.pbtxt │ │ │ │ ├── ConcatenateDataset.pbtxt │ │ │ │ ├── ConditionalAccumulator.pbtxt │ │ │ │ ├── ConfigureAndInitializeGlobalTPU.pbtxt │ │ │ │ ├── ConfigureDistributedTPU.pbtxt │ │ │ │ ├── ConfigureTPUEmbedding.pbtxt │ │ │ │ ├── ConfigureTPUEmbeddingHost.pbtxt │ │ │ │ ├── ConfigureTPUEmbeddingMemory.pbtxt │ │ │ │ ├── Conj.pbtxt │ │ │ │ ├── ConjugateTranspose.pbtxt │ │ │ │ ├── ConnectTPUEmbeddingHosts.pbtxt │ │ │ │ ├── Const.pbtxt │ │ │ │ ├── ConsumeMutexLock.pbtxt │ │ │ │ ├── ControlTrigger.pbtxt │ │ │ │ ├── Conv.pbtxt │ │ │ │ ├── Conv2D.pbtxt │ │ │ │ ├── Conv2DBackpropFilter.pbtxt │ │ │ │ ├── Conv2DBackpropFilterV2.pbtxt │ │ │ │ ├── Conv2DBackpropInput.pbtxt │ │ │ │ ├── Conv2DBackpropInputV2.pbtxt │ │ │ │ ├── Conv3D.pbtxt │ │ │ │ ├── Conv3DBackpropFilter.pbtxt │ │ │ │ ├── Conv3DBackpropFilterV2.pbtxt │ │ │ │ ├── Conv3DBackpropInput.pbtxt │ │ │ │ ├── Conv3DBackpropInputV2.pbtxt │ │ │ │ ├── ConvertToCooTensor.pbtxt │ │ │ │ ├── Copy.pbtxt │ │ │ │ ├── CopyHost.pbtxt │ │ │ │ ├── CopyToMesh.pbtxt │ │ │ │ ├── CopyToMeshGrad.pbtxt │ │ │ │ ├── Cos.pbtxt │ │ │ │ ├── Cosh.pbtxt │ │ │ │ ├── CountUpTo.pbtxt │ │ │ │ ├── CreateSummaryDbWriter.pbtxt │ │ │ │ ├── CreateSummaryFileWriter.pbtxt │ │ │ │ ├── CropAndResize.pbtxt │ │ │ │ ├── CropAndResizeGradBoxes.pbtxt │ │ │ │ ├── CropAndResizeGradImage.pbtxt │ │ │ │ ├── Cross.pbtxt │ │ │ │ ├── CrossReplicaSum.pbtxt │ │ │ │ ├── CudnnRNN.pbtxt │ │ │ │ ├── CudnnRNNBackprop.pbtxt │ │ │ │ ├── CudnnRNNBackpropV2.pbtxt │ │ │ │ ├── CudnnRNNBackpropV3.pbtxt │ │ │ │ ├── CudnnRNNCanonicalToParams.pbtxt │ │ │ │ ├── CudnnRNNCanonicalToParamsV2.pbtxt │ │ │ │ ├── CudnnRNNParamsSize.pbtxt │ │ │ │ ├── CudnnRNNParamsToCanonical.pbtxt │ │ │ │ ├── CudnnRNNParamsToCanonicalV2.pbtxt │ │ │ │ ├── CudnnRNNV2.pbtxt │ │ │ │ ├── CudnnRNNV3.pbtxt │ │ │ │ ├── Cumprod.pbtxt │ │ │ │ ├── Cumsum.pbtxt │ │ │ │ ├── CumulativeLogsumexp.pbtxt │ │ │ │ ├── DTensorRestoreV2.pbtxt │ │ │ │ ├── DTensorSetGlobalTPUArray.pbtxt │ │ │ │ ├── DataFormatDimMap.pbtxt │ │ │ │ ├── DataFormatVecPermute.pbtxt │ │ │ │ ├── DataServiceDataset.pbtxt │ │ │ │ ├── DataServiceDatasetV2.pbtxt │ │ │ │ ├── DataServiceDatasetV3.pbtxt │ │ │ │ ├── DataServiceDatasetV4.pbtxt │ │ │ │ ├── DatasetCardinality.pbtxt │ │ │ │ ├── DatasetFromGraph.pbtxt │ │ │ │ ├── DatasetToGraph.pbtxt │ │ │ │ ├── DatasetToGraphV2.pbtxt │ │ │ │ ├── DatasetToSingleElement.pbtxt │ │ │ │ ├── DatasetToTFRecord.pbtxt │ │ │ │ ├── Dawsn.pbtxt │ │ │ │ ├── DebugGradientIdentity.pbtxt │ │ │ │ ├── DebugGradientRefIdentity.pbtxt │ │ │ │ ├── DebugIdentity.pbtxt │ │ │ │ ├── DebugIdentityV2.pbtxt │ │ │ │ ├── DebugIdentityV3.pbtxt │ │ │ │ ├── DebugNanCount.pbtxt │ │ │ │ ├── DebugNumericSummary.pbtxt │ │ │ │ ├── DebugNumericSummaryV2.pbtxt │ │ │ │ ├── DecodeAndCropJpeg.pbtxt │ │ │ │ ├── DecodeBase64.pbtxt │ │ │ │ ├── DecodeBmp.pbtxt │ │ │ │ ├── DecodeCSV.pbtxt │ │ │ │ ├── DecodeCompressed.pbtxt │ │ │ │ ├── DecodeGif.pbtxt │ │ │ │ ├── DecodeImage.pbtxt │ │ │ │ ├── DecodeJSONExample.pbtxt │ │ │ │ ├── DecodeJpeg.pbtxt │ │ │ │ ├── DecodePaddedRaw.pbtxt │ │ │ │ ├── DecodePng.pbtxt │ │ │ │ ├── DecodeProtoV2.pbtxt │ │ │ │ ├── DecodeRaw.pbtxt │ │ │ │ ├── DecodeWav.pbtxt │ │ │ │ ├── DeepCopy.pbtxt │ │ │ │ ├── DeleteIterator.pbtxt │ │ │ │ ├── DeleteMemoryCache.pbtxt │ │ │ │ ├── DeleteMultiDeviceIterator.pbtxt │ │ │ │ ├── DeleteRandomSeedGenerator.pbtxt │ │ │ │ ├── DeleteSeedGenerator.pbtxt │ │ │ │ ├── DeleteSessionTensor.pbtxt │ │ │ │ ├── DenseBincount.pbtxt │ │ │ │ ├── DenseCountSparseOutput.pbtxt │ │ │ │ ├── DenseToCSRSparseMatrix.pbtxt │ │ │ │ ├── DenseToDenseSetOperation.pbtxt │ │ │ │ ├── DenseToSparseBatchDataset.pbtxt │ │ │ │ ├── DenseToSparseSetOperation.pbtxt │ │ │ │ ├── DepthToSpace.pbtxt │ │ │ │ ├── DepthwiseConv2dNative.pbtxt │ │ │ │ ├── DepthwiseConv2dNativeBackpropFilter.pbtxt │ │ │ │ ├── DepthwiseConv2dNativeBackpropInput.pbtxt │ │ │ │ ├── Dequantize.pbtxt │ │ │ │ ├── DeserializeIterator.pbtxt │ │ │ │ ├── DeserializeManySparse.pbtxt │ │ │ │ ├── DeserializeSparse.pbtxt │ │ │ │ ├── DestroyResourceOp.pbtxt │ │ │ │ ├── DestroyTemporaryVariable.pbtxt │ │ │ │ ├── DeviceIndex.pbtxt │ │ │ │ ├── Diag.pbtxt │ │ │ │ ├── DiagPart.pbtxt │ │ │ │ ├── Digamma.pbtxt │ │ │ │ ├── Dilation2D.pbtxt │ │ │ │ ├── Dilation2DBackpropFilter.pbtxt │ │ │ │ ├── Dilation2DBackpropInput.pbtxt │ │ │ │ ├── DirectedInterleaveDataset.pbtxt │ │ │ │ ├── DisableCopyOnRead.pbtxt │ │ │ │ ├── DistributedSave.pbtxt │ │ │ │ ├── Div.pbtxt │ │ │ │ ├── DivNoNan.pbtxt │ │ │ │ ├── DrawBoundingBoxes.pbtxt │ │ │ │ ├── DrawBoundingBoxesV2.pbtxt │ │ │ │ ├── DummyIterationCounter.pbtxt │ │ │ │ ├── DummyMemoryCache.pbtxt │ │ │ │ ├── DummySeedGenerator.pbtxt │ │ │ │ ├── DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt │ │ │ │ ├── DynamicEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt │ │ │ │ ├── DynamicPartition.pbtxt │ │ │ │ ├── DynamicStitch.pbtxt │ │ │ │ ├── EagerPyFunc.pbtxt │ │ │ │ ├── EditDistance.pbtxt │ │ │ │ ├── Eig.pbtxt │ │ │ │ ├── Einsum.pbtxt │ │ │ │ ├── Elu.pbtxt │ │ │ │ ├── EluGrad.pbtxt │ │ │ │ ├── Empty.pbtxt │ │ │ │ ├── EmptyTensorList.pbtxt │ │ │ │ ├── EmptyTensorMap.pbtxt │ │ │ │ ├── EncodeBase64.pbtxt │ │ │ │ ├── EncodeJpeg.pbtxt │ │ │ │ ├── EncodeJpegVariableQuality.pbtxt │ │ │ │ ├── EncodePng.pbtxt │ │ │ │ ├── EncodeProto.pbtxt │ │ │ │ ├── EncodeWav.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingIntegerBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingSparseBatch.pbtxt │ │ │ │ ├── EnqueueTPUEmbeddingSparseTensorBatch.pbtxt │ │ │ │ ├── EnsureShape.pbtxt │ │ │ │ ├── Enter.pbtxt │ │ │ │ ├── Equal.pbtxt │ │ │ │ ├── Erf.pbtxt │ │ │ │ ├── Erfc.pbtxt │ │ │ │ ├── Erfinv.pbtxt │ │ │ │ ├── EuclideanNorm.pbtxt │ │ │ │ ├── ExecuteTPUEmbeddingPartitioner.pbtxt │ │ │ │ ├── Exit.pbtxt │ │ │ │ ├── Exp.pbtxt │ │ │ │ ├── ExpandDims.pbtxt │ │ │ │ ├── ExperimentalAssertNextDataset.pbtxt │ │ │ │ ├── ExperimentalAutoShardDataset.pbtxt │ │ │ │ ├── ExperimentalBytesProducedStatsDataset.pbtxt │ │ │ │ ├── ExperimentalCSVDataset.pbtxt │ │ │ │ ├── ExperimentalChooseFastestDataset.pbtxt │ │ │ │ ├── ExperimentalDatasetCardinality.pbtxt │ │ │ │ ├── ExperimentalDatasetToTFRecord.pbtxt │ │ │ │ ├── ExperimentalDenseToSparseBatchDataset.pbtxt │ │ │ │ ├── ExperimentalDirectedInterleaveDataset.pbtxt │ │ │ │ ├── ExperimentalGroupByReducerDataset.pbtxt │ │ │ │ ├── ExperimentalGroupByWindowDataset.pbtxt │ │ │ │ ├── ExperimentalIgnoreErrorsDataset.pbtxt │ │ │ │ ├── ExperimentalIteratorGetDevice.pbtxt │ │ │ │ ├── ExperimentalLMDBDataset.pbtxt │ │ │ │ ├── ExperimentalLatencyStatsDataset.pbtxt │ │ │ │ ├── ExperimentalMapAndBatchDataset.pbtxt │ │ │ │ ├── ExperimentalMapDataset.pbtxt │ │ │ │ ├── ExperimentalMatchingFilesDataset.pbtxt │ │ │ │ ├── ExperimentalMaxIntraOpParallelismDataset.pbtxt │ │ │ │ ├── ExperimentalNonSerializableDataset.pbtxt │ │ │ │ ├── ExperimentalParallelInterleaveDataset.pbtxt │ │ │ │ ├── ExperimentalParseExampleDataset.pbtxt │ │ │ │ ├── ExperimentalPrivateThreadPoolDataset.pbtxt │ │ │ │ ├── ExperimentalRandomDataset.pbtxt │ │ │ │ ├── ExperimentalRebatchDataset.pbtxt │ │ │ │ ├── ExperimentalScanDataset.pbtxt │ │ │ │ ├── ExperimentalSetStatsAggregatorDataset.pbtxt │ │ │ │ ├── ExperimentalSleepDataset.pbtxt │ │ │ │ ├── ExperimentalSlidingWindowDataset.pbtxt │ │ │ │ ├── ExperimentalSqlDataset.pbtxt │ │ │ │ ├── ExperimentalStatsAggregatorHandle.pbtxt │ │ │ │ ├── ExperimentalStatsAggregatorSummary.pbtxt │ │ │ │ ├── ExperimentalTakeWhileDataset.pbtxt │ │ │ │ ├── ExperimentalThreadPoolDataset.pbtxt │ │ │ │ ├── ExperimentalThreadPoolHandle.pbtxt │ │ │ │ ├── ExperimentalUnbatchDataset.pbtxt │ │ │ │ ├── ExperimentalUniqueDataset.pbtxt │ │ │ │ ├── Expint.pbtxt │ │ │ │ ├── Expm1.pbtxt │ │ │ │ ├── ExtractGlimpse.pbtxt │ │ │ │ ├── ExtractGlimpseV2.pbtxt │ │ │ │ ├── ExtractImagePatches.pbtxt │ │ │ │ ├── ExtractJpegShape.pbtxt │ │ │ │ ├── ExtractVolumePatches.pbtxt │ │ │ │ ├── FFT.pbtxt │ │ │ │ ├── FFT2D.pbtxt │ │ │ │ ├── FFT3D.pbtxt │ │ │ │ ├── FFTND.pbtxt │ │ │ │ ├── FIFOQueue.pbtxt │ │ │ │ ├── FIFOQueueV2.pbtxt │ │ │ │ ├── Fact.pbtxt │ │ │ │ ├── FakeParam.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxArgs.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxArgsGradient.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVars.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVarsGradient.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVarsPerChannel.pbtxt │ │ │ │ ├── FakeQuantWithMinMaxVarsPerChannelGradient.pbtxt │ │ │ │ ├── FakeQueue.pbtxt │ │ │ │ ├── FileSystemSetConfiguration.pbtxt │ │ │ │ ├── Fill.pbtxt │ │ │ │ ├── FilterByLastComponentDataset.pbtxt │ │ │ │ ├── FilterDataset.pbtxt │ │ │ │ ├── FinalizeDataset.pbtxt │ │ │ │ ├── FinalizeTPUEmbedding.pbtxt │ │ │ │ ├── Fingerprint.pbtxt │ │ │ │ ├── FixedLengthRecordDataset.pbtxt │ │ │ │ ├── FixedLengthRecordDatasetV2.pbtxt │ │ │ │ ├── FixedLengthRecordReader.pbtxt │ │ │ │ ├── FixedLengthRecordReaderV2.pbtxt │ │ │ │ ├── FixedUnigramCandidateSampler.pbtxt │ │ │ │ ├── FlatMapDataset.pbtxt │ │ │ │ ├── Floor.pbtxt │ │ │ │ ├── FloorDiv.pbtxt │ │ │ │ ├── FloorMod.pbtxt │ │ │ │ ├── FlushSummaryWriter.pbtxt │ │ │ │ ├── For.pbtxt │ │ │ │ ├── FractionalAvgPool.pbtxt │ │ │ │ ├── FractionalAvgPoolGrad.pbtxt │ │ │ │ ├── FractionalMaxPool.pbtxt │ │ │ │ ├── FractionalMaxPoolGrad.pbtxt │ │ │ │ ├── FresnelCos.pbtxt │ │ │ │ ├── FresnelSin.pbtxt │ │ │ │ ├── FusedBatchNorm.pbtxt │ │ │ │ ├── FusedBatchNormGrad.pbtxt │ │ │ │ ├── FusedBatchNormGradV2.pbtxt │ │ │ │ ├── FusedBatchNormGradV3.pbtxt │ │ │ │ ├── FusedBatchNormV2.pbtxt │ │ │ │ ├── FusedBatchNormV3.pbtxt │ │ │ │ ├── FusedPadConv2D.pbtxt │ │ │ │ ├── FusedResizeAndPadConv2D.pbtxt │ │ │ │ ├── GRUBlockCell.pbtxt │ │ │ │ ├── GRUBlockCellGrad.pbtxt │ │ │ │ ├── Gather.pbtxt │ │ │ │ ├── GatherNd.pbtxt │ │ │ │ ├── GatherV2.pbtxt │ │ │ │ ├── GenerateBoundingBoxProposals.pbtxt │ │ │ │ ├── GenerateVocabRemapping.pbtxt │ │ │ │ ├── GeneratorDataset.pbtxt │ │ │ │ ├── GetElementAtIndex.pbtxt │ │ │ │ ├── GetMinibatchSplitsWithPhysicalReplica.pbtxt │ │ │ │ ├── GetMinibatchesInCsrWithPhysicalReplica.pbtxt │ │ │ │ ├── GetOptions.pbtxt │ │ │ │ ├── GetSessionHandle.pbtxt │ │ │ │ ├── GetSessionHandleV2.pbtxt │ │ │ │ ├── GetSessionTensor.pbtxt │ │ │ │ ├── GlobalIterId.pbtxt │ │ │ │ ├── Greater.pbtxt │ │ │ │ ├── GreaterEqual.pbtxt │ │ │ │ ├── GroupByReducerDataset.pbtxt │ │ │ │ ├── GroupByWindowDataset.pbtxt │ │ │ │ ├── GuaranteeConst.pbtxt │ │ │ │ ├── HSVToRGB.pbtxt │ │ │ │ ├── HashTable.pbtxt │ │ │ │ ├── HashTableV2.pbtxt │ │ │ │ ├── HistogramFixedWidth.pbtxt │ │ │ │ ├── HistogramSummary.pbtxt │ │ │ │ ├── HostConst.pbtxt │ │ │ │ ├── IFFT.pbtxt │ │ │ │ ├── IFFT2D.pbtxt │ │ │ │ ├── IFFT3D.pbtxt │ │ │ │ ├── IFFTND.pbtxt │ │ │ │ ├── IRFFT.pbtxt │ │ │ │ ├── IRFFT2D.pbtxt │ │ │ │ ├── IRFFT3D.pbtxt │ │ │ │ ├── IRFFTND.pbtxt │ │ │ │ ├── Identity.pbtxt │ │ │ │ ├── IdentityN.pbtxt │ │ │ │ ├── IdentityReader.pbtxt │ │ │ │ ├── IdentityReaderV2.pbtxt │ │ │ │ ├── If.pbtxt │ │ │ │ ├── Igamma.pbtxt │ │ │ │ ├── IgammaGradA.pbtxt │ │ │ │ ├── Igammac.pbtxt │ │ │ │ ├── IgnoreErrorsDataset.pbtxt │ │ │ │ ├── Imag.pbtxt │ │ │ │ ├── ImageProjectiveTransformV2.pbtxt │ │ │ │ ├── ImageProjectiveTransformV3.pbtxt │ │ │ │ ├── ImageSummary.pbtxt │ │ │ │ ├── ImmutableConst.pbtxt │ │ │ │ ├── ImportEvent.pbtxt │ │ │ │ ├── InTopK.pbtxt │ │ │ │ ├── InTopKV2.pbtxt │ │ │ │ ├── InfeedDequeue.pbtxt │ │ │ │ ├── InfeedDequeueTuple.pbtxt │ │ │ │ ├── InfeedEnqueue.pbtxt │ │ │ │ ├── InfeedEnqueuePrelinearizedBuffer.pbtxt │ │ │ │ ├── InfeedEnqueueTuple.pbtxt │ │ │ │ ├── InitializeTable.pbtxt │ │ │ │ ├── InitializeTableFromDataset.pbtxt │ │ │ │ ├── InitializeTableFromTextFile.pbtxt │ │ │ │ ├── InitializeTableFromTextFileV2.pbtxt │ │ │ │ ├── InitializeTableV2.pbtxt │ │ │ │ ├── InplaceAdd.pbtxt │ │ │ │ ├── InplaceSub.pbtxt │ │ │ │ ├── InplaceUpdate.pbtxt │ │ │ │ ├── InterleaveDataset.pbtxt │ │ │ │ ├── Inv.pbtxt │ │ │ │ ├── InvGrad.pbtxt │ │ │ │ ├── Invert.pbtxt │ │ │ │ ├── InvertPermutation.pbtxt │ │ │ │ ├── IsBoostedTreesEnsembleInitialized.pbtxt │ │ │ │ ├── IsBoostedTreesQuantileStreamResourceInitialized.pbtxt │ │ │ │ ├── IsFinite.pbtxt │ │ │ │ ├── IsInf.pbtxt │ │ │ │ ├── IsNan.pbtxt │ │ │ │ ├── IsTPUEmbeddingInitialized.pbtxt │ │ │ │ ├── IsVariableInitialized.pbtxt │ │ │ │ ├── IsotonicRegression.pbtxt │ │ │ │ ├── Iterator.pbtxt │ │ │ │ ├── IteratorFromStringHandle.pbtxt │ │ │ │ ├── IteratorFromStringHandleV2.pbtxt │ │ │ │ ├── IteratorGetDevice.pbtxt │ │ │ │ ├── IteratorGetNext.pbtxt │ │ │ │ ├── IteratorGetNextAsOptional.pbtxt │ │ │ │ ├── IteratorGetNextSync.pbtxt │ │ │ │ ├── IteratorToStringHandle.pbtxt │ │ │ │ ├── IteratorV2.pbtxt │ │ │ │ ├── KMC2ChainInitialization.pbtxt │ │ │ │ ├── KmeansPlusPlusInitialization.pbtxt │ │ │ │ ├── KthOrderStatistic.pbtxt │ │ │ │ ├── L2Loss.pbtxt │ │ │ │ ├── LMDBDataset.pbtxt │ │ │ │ ├── LMDBReader.pbtxt │ │ │ │ ├── LRN.pbtxt │ │ │ │ ├── LRNGrad.pbtxt │ │ │ │ ├── LSTMBlockCell.pbtxt │ │ │ │ ├── LSTMBlockCellGrad.pbtxt │ │ │ │ ├── LatencyStatsDataset.pbtxt │ │ │ │ ├── LeakyRelu.pbtxt │ │ │ │ ├── LeakyReluGrad.pbtxt │ │ │ │ ├── LearnedUnigramCandidateSampler.pbtxt │ │ │ │ ├── LeftShift.pbtxt │ │ │ │ ├── LegacyParallelInterleaveDatasetV2.pbtxt │ │ │ │ ├── Less.pbtxt │ │ │ │ ├── LessEqual.pbtxt │ │ │ │ ├── Lgamma.pbtxt │ │ │ │ ├── LinSpace.pbtxt │ │ │ │ ├── ListDataset.pbtxt │ │ │ │ ├── ListDiff.pbtxt │ │ │ │ ├── LoadAllTPUEmbeddingParameters.pbtxt │ │ │ │ ├── LoadAndRemapMatrix.pbtxt │ │ │ │ ├── LoadDataset.pbtxt │ │ │ │ ├── LoadTPUEmbeddingADAMParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingAdadeltaParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingAdagradMomentumParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingAdagradParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingCenteredRMSPropParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingFTRLParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingMDLAdagradLightParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingMomentumParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingProximalAdagradParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingProximalYogiParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingRMSPropParameters.pbtxt │ │ │ │ ├── LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt │ │ │ │ ├── Log.pbtxt │ │ │ │ ├── Log1p.pbtxt │ │ │ │ ├── LogMatrixDeterminant.pbtxt │ │ │ │ ├── LogSoftmax.pbtxt │ │ │ │ ├── LogUniformCandidateSampler.pbtxt │ │ │ │ ├── LogicalAnd.pbtxt │ │ │ │ ├── LogicalNot.pbtxt │ │ │ │ ├── LogicalOr.pbtxt │ │ │ │ ├── LookupTableExport.pbtxt │ │ │ │ ├── LookupTableExportV2.pbtxt │ │ │ │ ├── LookupTableFind.pbtxt │ │ │ │ ├── LookupTableFindV2.pbtxt │ │ │ │ ├── LookupTableImport.pbtxt │ │ │ │ ├── LookupTableImportV2.pbtxt │ │ │ │ ├── LookupTableInsert.pbtxt │ │ │ │ ├── LookupTableInsertV2.pbtxt │ │ │ │ ├── LookupTableRemoveV2.pbtxt │ │ │ │ ├── LookupTableSize.pbtxt │ │ │ │ ├── LookupTableSizeV2.pbtxt │ │ │ │ ├── LoopCond.pbtxt │ │ │ │ ├── LowerBound.pbtxt │ │ │ │ ├── Lu.pbtxt │ │ │ │ ├── MakeIterator.pbtxt │ │ │ │ ├── MakeUnique.pbtxt │ │ │ │ ├── MapAndBatchDataset.pbtxt │ │ │ │ ├── MapClear.pbtxt │ │ │ │ ├── MapDataset.pbtxt │ │ │ │ ├── MapDefun.pbtxt │ │ │ │ ├── MapIncompleteSize.pbtxt │ │ │ │ ├── MapPeek.pbtxt │ │ │ │ ├── MapSize.pbtxt │ │ │ │ ├── MapStage.pbtxt │ │ │ │ ├── MapUnstage.pbtxt │ │ │ │ ├── MapUnstageNoKey.pbtxt │ │ │ │ ├── MatMul.pbtxt │ │ │ │ ├── MatchingFiles.pbtxt │ │ │ │ ├── MatchingFilesDataset.pbtxt │ │ │ │ ├── MatrixBandPart.pbtxt │ │ │ │ ├── MatrixDeterminant.pbtxt │ │ │ │ ├── MatrixDiag.pbtxt │ │ │ │ ├── MatrixDiagPart.pbtxt │ │ │ │ ├── MatrixDiagPartV2.pbtxt │ │ │ │ ├── MatrixDiagPartV3.pbtxt │ │ │ │ ├── MatrixDiagV2.pbtxt │ │ │ │ ├── MatrixDiagV3.pbtxt │ │ │ │ ├── MatrixExponential.pbtxt │ │ │ │ ├── MatrixInverse.pbtxt │ │ │ │ ├── MatrixLogarithm.pbtxt │ │ │ │ ├── MatrixSetDiag.pbtxt │ │ │ │ ├── MatrixSetDiagV2.pbtxt │ │ │ │ ├── MatrixSetDiagV3.pbtxt │ │ │ │ ├── MatrixSolve.pbtxt │ │ │ │ ├── MatrixSolveLs.pbtxt │ │ │ │ ├── MatrixSquareRoot.pbtxt │ │ │ │ ├── MatrixTriangularSolve.pbtxt │ │ │ │ ├── Max.pbtxt │ │ │ │ ├── MaxIntraOpParallelismDataset.pbtxt │ │ │ │ ├── MaxPool.pbtxt │ │ │ │ ├── MaxPool3D.pbtxt │ │ │ │ ├── MaxPool3DGrad.pbtxt │ │ │ │ ├── MaxPool3DGradGrad.pbtxt │ │ │ │ ├── MaxPoolGrad.pbtxt │ │ │ │ ├── MaxPoolGradGrad.pbtxt │ │ │ │ ├── MaxPoolGradGradV2.pbtxt │ │ │ │ ├── MaxPoolGradGradWithArgmax.pbtxt │ │ │ │ ├── MaxPoolGradV2.pbtxt │ │ │ │ ├── MaxPoolGradWithArgmax.pbtxt │ │ │ │ ├── MaxPoolV2.pbtxt │ │ │ │ ├── MaxPoolWithArgmax.pbtxt │ │ │ │ ├── Maximum.pbtxt │ │ │ │ ├── Mean.pbtxt │ │ │ │ ├── Merge.pbtxt │ │ │ │ ├── MergeDedupData.pbtxt │ │ │ │ ├── MergeSummary.pbtxt │ │ │ │ ├── MergeV2Checkpoints.pbtxt │ │ │ │ ├── Mfcc.pbtxt │ │ │ │ ├── Min.pbtxt │ │ │ │ ├── Minimum.pbtxt │ │ │ │ ├── MirrorPad.pbtxt │ │ │ │ ├── MirrorPadGrad.pbtxt │ │ │ │ ├── MlirPassthroughOp.pbtxt │ │ │ │ ├── Mod.pbtxt │ │ │ │ ├── ModelDataset.pbtxt │ │ │ │ ├── Mul.pbtxt │ │ │ │ ├── MulNoNan.pbtxt │ │ │ │ ├── MultiDeviceIterator.pbtxt │ │ │ │ ├── MultiDeviceIteratorFromStringHandle.pbtxt │ │ │ │ ├── MultiDeviceIteratorGetNextFromShard.pbtxt │ │ │ │ ├── MultiDeviceIteratorInit.pbtxt │ │ │ │ ├── MultiDeviceIteratorToStringHandle.pbtxt │ │ │ │ ├── Multinomial.pbtxt │ │ │ │ ├── MutableDenseHashTable.pbtxt │ │ │ │ ├── MutableDenseHashTableV2.pbtxt │ │ │ │ ├── MutableHashTable.pbtxt │ │ │ │ ├── MutableHashTableOfTensors.pbtxt │ │ │ │ ├── MutableHashTableOfTensorsV2.pbtxt │ │ │ │ ├── MutableHashTableV2.pbtxt │ │ │ │ ├── MutexLock.pbtxt │ │ │ │ ├── MutexV2.pbtxt │ │ │ │ ├── NcclAllReduce.pbtxt │ │ │ │ ├── NcclBroadcast.pbtxt │ │ │ │ ├── NcclReduce.pbtxt │ │ │ │ ├── Ndtri.pbtxt │ │ │ │ ├── NearestNeighbors.pbtxt │ │ │ │ ├── Neg.pbtxt │ │ │ │ ├── NegTrain.pbtxt │ │ │ │ ├── NextAfter.pbtxt │ │ │ │ ├── NextIteration.pbtxt │ │ │ │ ├── NoOp.pbtxt │ │ │ │ ├── NonDeterministicInts.pbtxt │ │ │ │ ├── NonMaxSuppression.pbtxt │ │ │ │ ├── NonMaxSuppressionV2.pbtxt │ │ │ │ ├── NonMaxSuppressionV3.pbtxt │ │ │ │ ├── NonMaxSuppressionV4.pbtxt │ │ │ │ ├── NonMaxSuppressionV5.pbtxt │ │ │ │ ├── NonMaxSuppressionWithOverlaps.pbtxt │ │ │ │ ├── NonSerializableDataset.pbtxt │ │ │ │ ├── NotEqual.pbtxt │ │ │ │ ├── NthElement.pbtxt │ │ │ │ ├── OneHot.pbtxt │ │ │ │ ├── OneShotIterator.pbtxt │ │ │ │ ├── OnesLike.pbtxt │ │ │ │ ├── OptimizeDataset.pbtxt │ │ │ │ ├── OptimizeDatasetV2.pbtxt │ │ │ │ ├── OptionalFromValue.pbtxt │ │ │ │ ├── OptionalGetValue.pbtxt │ │ │ │ ├── OptionalHasValue.pbtxt │ │ │ │ ├── OptionalNone.pbtxt │ │ │ │ ├── OptionsDataset.pbtxt │ │ │ │ ├── OrderedMapClear.pbtxt │ │ │ │ ├── OrderedMapIncompleteSize.pbtxt │ │ │ │ ├── OrderedMapPeek.pbtxt │ │ │ │ ├── OrderedMapSize.pbtxt │ │ │ │ ├── OrderedMapStage.pbtxt │ │ │ │ ├── OrderedMapUnstage.pbtxt │ │ │ │ ├── OrderedMapUnstageNoKey.pbtxt │ │ │ │ ├── OutfeedDequeue.pbtxt │ │ │ │ ├── OutfeedDequeueTuple.pbtxt │ │ │ │ ├── OutfeedDequeueTupleV2.pbtxt │ │ │ │ ├── OutfeedDequeueV2.pbtxt │ │ │ │ ├── OutfeedEnqueue.pbtxt │ │ │ │ ├── OutfeedEnqueueTuple.pbtxt │ │ │ │ ├── Pack.pbtxt │ │ │ │ ├── Pad.pbtxt │ │ │ │ ├── PadV2.pbtxt │ │ │ │ ├── PaddedBatchDataset.pbtxt │ │ │ │ ├── PaddedBatchDatasetV2.pbtxt │ │ │ │ ├── PaddingFIFOQueue.pbtxt │ │ │ │ ├── PaddingFIFOQueueV2.pbtxt │ │ │ │ ├── ParallelBatchDataset.pbtxt │ │ │ │ ├── ParallelConcat.pbtxt │ │ │ │ ├── ParallelDynamicStitch.pbtxt │ │ │ │ ├── ParallelFilterDataset.pbtxt │ │ │ │ ├── ParallelInterleaveDataset.pbtxt │ │ │ │ ├── ParallelInterleaveDatasetV2.pbtxt │ │ │ │ ├── ParallelInterleaveDatasetV3.pbtxt │ │ │ │ ├── ParallelInterleaveDatasetV4.pbtxt │ │ │ │ ├── ParallelMapDataset.pbtxt │ │ │ │ ├── ParallelMapDatasetV2.pbtxt │ │ │ │ ├── ParameterizedTruncatedNormal.pbtxt │ │ │ │ ├── ParseExample.pbtxt │ │ │ │ ├── ParseExampleDataset.pbtxt │ │ │ │ ├── ParseExampleDatasetV2.pbtxt │ │ │ │ ├── ParseExampleV2.pbtxt │ │ │ │ ├── ParseSequenceExample.pbtxt │ │ │ │ ├── ParseSequenceExampleV2.pbtxt │ │ │ │ ├── ParseSingleExample.pbtxt │ │ │ │ ├── ParseSingleSequenceExample.pbtxt │ │ │ │ ├── ParseTensor.pbtxt │ │ │ │ ├── PartitionedCall.pbtxt │ │ │ │ ├── Placeholder.pbtxt │ │ │ │ ├── PlaceholderV2.pbtxt │ │ │ │ ├── PlaceholderWithDefault.pbtxt │ │ │ │ ├── Polygamma.pbtxt │ │ │ │ ├── PopulationCount.pbtxt │ │ │ │ ├── Pow.pbtxt │ │ │ │ ├── PrefetchDataset.pbtxt │ │ │ │ ├── Prelinearize.pbtxt │ │ │ │ ├── PrelinearizeTuple.pbtxt │ │ │ │ ├── PreventGradient.pbtxt │ │ │ │ ├── Print.pbtxt │ │ │ │ ├── PrintV2.pbtxt │ │ │ │ ├── PriorityQueue.pbtxt │ │ │ │ ├── PriorityQueueV2.pbtxt │ │ │ │ ├── PrivateThreadPoolDataset.pbtxt │ │ │ │ ├── Prod.pbtxt │ │ │ │ ├── PyFunc.pbtxt │ │ │ │ ├── PyFuncStateless.pbtxt │ │ │ │ ├── Qr.pbtxt │ │ │ │ ├── QuantizeAndDequantize.pbtxt │ │ │ │ ├── QuantizeAndDequantizeV2.pbtxt │ │ │ │ ├── QuantizeAndDequantizeV3.pbtxt │ │ │ │ ├── QuantizeAndDequantizeV4.pbtxt │ │ │ │ ├── QuantizeAndDequantizeV4Grad.pbtxt │ │ │ │ ├── QuantizeDownAndShrinkRange.pbtxt │ │ │ │ ├── QuantizeV2.pbtxt │ │ │ │ ├── QuantizedAdd.pbtxt │ │ │ │ ├── QuantizedAvgPool.pbtxt │ │ │ │ ├── QuantizedBatchNormWithGlobalNormalization.pbtxt │ │ │ │ ├── QuantizedBiasAdd.pbtxt │ │ │ │ ├── QuantizedConcat.pbtxt │ │ │ │ ├── QuantizedConv2D.pbtxt │ │ │ │ ├── QuantizedConv2DAndRelu.pbtxt │ │ │ │ ├── QuantizedConv2DAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DPerChannel.pbtxt │ │ │ │ ├── QuantizedConv2DWithBias.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasAndRelu.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasSumAndRelu.pbtxt │ │ │ │ ├── QuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2D.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2DWithBias.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt │ │ │ │ ├── QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedInstanceNorm.pbtxt │ │ │ │ ├── QuantizedMatMul.pbtxt │ │ │ │ ├── QuantizedMatMulWithBias.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndDequantize.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndRelu.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt │ │ │ │ ├── QuantizedMatMulWithBiasAndRequantize.pbtxt │ │ │ │ ├── QuantizedMaxPool.pbtxt │ │ │ │ ├── QuantizedMul.pbtxt │ │ │ │ ├── QuantizedRelu.pbtxt │ │ │ │ ├── QuantizedRelu6.pbtxt │ │ │ │ ├── QuantizedReluX.pbtxt │ │ │ │ ├── QuantizedReshape.pbtxt │ │ │ │ ├── QuantizedResizeBilinear.pbtxt │ │ │ │ ├── QueueClose.pbtxt │ │ │ │ ├── QueueCloseV2.pbtxt │ │ │ │ ├── QueueDequeue.pbtxt │ │ │ │ ├── QueueDequeueMany.pbtxt │ │ │ │ ├── QueueDequeueManyV2.pbtxt │ │ │ │ ├── QueueDequeueUpTo.pbtxt │ │ │ │ ├── QueueDequeueUpToV2.pbtxt │ │ │ │ ├── QueueDequeueV2.pbtxt │ │ │ │ ├── QueueEnqueue.pbtxt │ │ │ │ ├── QueueEnqueueMany.pbtxt │ │ │ │ ├── QueueEnqueueManyV2.pbtxt │ │ │ │ ├── QueueEnqueueV2.pbtxt │ │ │ │ ├── QueueIsClosed.pbtxt │ │ │ │ ├── QueueIsClosedV2.pbtxt │ │ │ │ ├── QueueSize.pbtxt │ │ │ │ ├── QueueSizeV2.pbtxt │ │ │ │ ├── RFFT.pbtxt │ │ │ │ ├── RFFT2D.pbtxt │ │ │ │ ├── RFFT3D.pbtxt │ │ │ │ ├── RFFTND.pbtxt │ │ │ │ ├── RGBToHSV.pbtxt │ │ │ │ ├── RaggedBincount.pbtxt │ │ │ │ ├── RaggedCountSparseOutput.pbtxt │ │ │ │ ├── RaggedCross.pbtxt │ │ │ │ ├── RaggedFillEmptyRows.pbtxt │ │ │ │ ├── RaggedFillEmptyRowsGrad.pbtxt │ │ │ │ ├── RaggedGather.pbtxt │ │ │ │ ├── RaggedRange.pbtxt │ │ │ │ ├── RaggedTensorFromVariant.pbtxt │ │ │ │ ├── RaggedTensorToSparse.pbtxt │ │ │ │ ├── RaggedTensorToTensor.pbtxt │ │ │ │ ├── RaggedTensorToVariant.pbtxt │ │ │ │ ├── RaggedTensorToVariantGradient.pbtxt │ │ │ │ ├── RandomCrop.pbtxt │ │ │ │ ├── RandomDataset.pbtxt │ │ │ │ ├── RandomDatasetV2.pbtxt │ │ │ │ ├── RandomGamma.pbtxt │ │ │ │ ├── RandomGammaGrad.pbtxt │ │ │ │ ├── RandomIndexShuffle.pbtxt │ │ │ │ ├── RandomPoisson.pbtxt │ │ │ │ ├── RandomPoissonV2.pbtxt │ │ │ │ ├── RandomShuffle.pbtxt │ │ │ │ ├── RandomShuffleQueue.pbtxt │ │ │ │ ├── RandomShuffleQueueV2.pbtxt │ │ │ │ ├── RandomStandardNormal.pbtxt │ │ │ │ ├── RandomUniform.pbtxt │ │ │ │ ├── RandomUniformInt.pbtxt │ │ │ │ ├── Range.pbtxt │ │ │ │ ├── RangeDataset.pbtxt │ │ │ │ ├── Rank.pbtxt │ │ │ │ ├── ReadFile.pbtxt │ │ │ │ ├── ReadVariableOp.pbtxt │ │ │ │ ├── ReadVariableXlaSplitND.pbtxt │ │ │ │ ├── ReaderNumRecordsProduced.pbtxt │ │ │ │ ├── ReaderNumRecordsProducedV2.pbtxt │ │ │ │ ├── ReaderNumWorkUnitsCompleted.pbtxt │ │ │ │ ├── ReaderNumWorkUnitsCompletedV2.pbtxt │ │ │ │ ├── ReaderRead.pbtxt │ │ │ │ ├── ReaderReadUpTo.pbtxt │ │ │ │ ├── ReaderReadUpToV2.pbtxt │ │ │ │ ├── ReaderReadV2.pbtxt │ │ │ │ ├── ReaderReset.pbtxt │ │ │ │ ├── ReaderResetV2.pbtxt │ │ │ │ ├── ReaderRestoreState.pbtxt │ │ │ │ ├── ReaderRestoreStateV2.pbtxt │ │ │ │ ├── ReaderSerializeState.pbtxt │ │ │ │ ├── ReaderSerializeStateV2.pbtxt │ │ │ │ ├── Real.pbtxt │ │ │ │ ├── RealDiv.pbtxt │ │ │ │ ├── RebatchDataset.pbtxt │ │ │ │ ├── RebatchDatasetV2.pbtxt │ │ │ │ ├── Reciprocal.pbtxt │ │ │ │ ├── ReciprocalGrad.pbtxt │ │ │ │ ├── RecordInput.pbtxt │ │ │ │ ├── Recv.pbtxt │ │ │ │ ├── RecvTPUEmbeddingActivations.pbtxt │ │ │ │ ├── ReduceDataset.pbtxt │ │ │ │ ├── ReduceJoin.pbtxt │ │ │ │ ├── RefEnter.pbtxt │ │ │ │ ├── RefExit.pbtxt │ │ │ │ ├── RefIdentity.pbtxt │ │ │ │ ├── RefMerge.pbtxt │ │ │ │ ├── RefNextIteration.pbtxt │ │ │ │ ├── RefSelect.pbtxt │ │ │ │ ├── RefSwitch.pbtxt │ │ │ │ ├── RegexFullMatch.pbtxt │ │ │ │ ├── RegexReplace.pbtxt │ │ │ │ ├── RegisterDataset.pbtxt │ │ │ │ ├── RegisterDatasetV2.pbtxt │ │ │ │ ├── Relayout.pbtxt │ │ │ │ ├── RelayoutLike.pbtxt │ │ │ │ ├── Relu.pbtxt │ │ │ │ ├── Relu6.pbtxt │ │ │ │ ├── Relu6Grad.pbtxt │ │ │ │ ├── ReluGrad.pbtxt │ │ │ │ ├── RemoteCall.pbtxt │ │ │ │ ├── RepeatDataset.pbtxt │ │ │ │ ├── RequantizationRange.pbtxt │ │ │ │ ├── RequantizationRangePerChannel.pbtxt │ │ │ │ ├── Requantize.pbtxt │ │ │ │ ├── RequantizePerChannel.pbtxt │ │ │ │ ├── Reshape.pbtxt │ │ │ │ ├── ResizeArea.pbtxt │ │ │ │ ├── ResizeBicubic.pbtxt │ │ │ │ ├── ResizeBicubicGrad.pbtxt │ │ │ │ ├── ResizeBilinear.pbtxt │ │ │ │ ├── ResizeBilinearGrad.pbtxt │ │ │ │ ├── ResizeNearestNeighbor.pbtxt │ │ │ │ ├── ResizeNearestNeighborGrad.pbtxt │ │ │ │ ├── ResourceAccumulatorApplyGradient.pbtxt │ │ │ │ ├── ResourceAccumulatorNumAccumulated.pbtxt │ │ │ │ ├── ResourceAccumulatorSetGlobalStep.pbtxt │ │ │ │ ├── ResourceAccumulatorTakeGradient.pbtxt │ │ │ │ ├── ResourceApplyAdaMax.pbtxt │ │ │ │ ├── ResourceApplyAdadelta.pbtxt │ │ │ │ ├── ResourceApplyAdagrad.pbtxt │ │ │ │ ├── ResourceApplyAdagradDA.pbtxt │ │ │ │ ├── ResourceApplyAdagradV2.pbtxt │ │ │ │ ├── ResourceApplyAdam.pbtxt │ │ │ │ ├── ResourceApplyAdamWithAmsgrad.pbtxt │ │ │ │ ├── ResourceApplyAddSign.pbtxt │ │ │ │ ├── ResourceApplyCenteredRMSProp.pbtxt │ │ │ │ ├── ResourceApplyFtrl.pbtxt │ │ │ │ ├── ResourceApplyFtrlV2.pbtxt │ │ │ │ ├── ResourceApplyGradientDescent.pbtxt │ │ │ │ ├── ResourceApplyKerasMomentum.pbtxt │ │ │ │ ├── ResourceApplyMomentum.pbtxt │ │ │ │ ├── ResourceApplyPowerSign.pbtxt │ │ │ │ ├── ResourceApplyProximalAdagrad.pbtxt │ │ │ │ ├── ResourceApplyProximalGradientDescent.pbtxt │ │ │ │ ├── ResourceApplyRMSProp.pbtxt │ │ │ │ ├── ResourceConditionalAccumulator.pbtxt │ │ │ │ ├── ResourceCountUpTo.pbtxt │ │ │ │ ├── ResourceGather.pbtxt │ │ │ │ ├── ResourceGatherNd.pbtxt │ │ │ │ ├── ResourceScatterAdd.pbtxt │ │ │ │ ├── ResourceScatterDiv.pbtxt │ │ │ │ ├── ResourceScatterMax.pbtxt │ │ │ │ ├── ResourceScatterMin.pbtxt │ │ │ │ ├── ResourceScatterMul.pbtxt │ │ │ │ ├── ResourceScatterNdAdd.pbtxt │ │ │ │ ├── ResourceScatterNdMax.pbtxt │ │ │ │ ├── ResourceScatterNdMin.pbtxt │ │ │ │ ├── ResourceScatterNdSub.pbtxt │ │ │ │ ├── ResourceScatterNdUpdate.pbtxt │ │ │ │ ├── ResourceScatterSub.pbtxt │ │ │ │ ├── ResourceScatterUpdate.pbtxt │ │ │ │ ├── ResourceSparseApplyAdadelta.pbtxt │ │ │ │ ├── ResourceSparseApplyAdagrad.pbtxt │ │ │ │ ├── ResourceSparseApplyAdagradDA.pbtxt │ │ │ │ ├── ResourceSparseApplyAdagradV2.pbtxt │ │ │ │ ├── ResourceSparseApplyCenteredRMSProp.pbtxt │ │ │ │ ├── ResourceSparseApplyFtrl.pbtxt │ │ │ │ ├── ResourceSparseApplyFtrlV2.pbtxt │ │ │ │ ├── ResourceSparseApplyKerasMomentum.pbtxt │ │ │ │ ├── ResourceSparseApplyMomentum.pbtxt │ │ │ │ ├── ResourceSparseApplyProximalAdagrad.pbtxt │ │ │ │ ├── ResourceSparseApplyProximalGradientDescent.pbtxt │ │ │ │ ├── ResourceSparseApplyRMSProp.pbtxt │ │ │ │ ├── ResourceStridedSliceAssign.pbtxt │ │ │ │ ├── Restore.pbtxt │ │ │ │ ├── RestoreSlice.pbtxt │ │ │ │ ├── RestoreV2.pbtxt │ │ │ │ ├── RetrieveAllTPUEmbeddingParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingADAMParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingAdadeltaParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingAdagradMomentumParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingAdagradParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingCenteredRMSPropParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingFTRLParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingMDLAdagradLightParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingMomentumParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingProximalYogiParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingRMSPropParameters.pbtxt │ │ │ │ ├── RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt │ │ │ │ ├── Reverse.pbtxt │ │ │ │ ├── ReverseSequence.pbtxt │ │ │ │ ├── ReverseV2.pbtxt │ │ │ │ ├── RewriteDataset.pbtxt │ │ │ │ ├── RightShift.pbtxt │ │ │ │ ├── Rint.pbtxt │ │ │ │ ├── RiscAbs.pbtxt │ │ │ │ ├── RiscAdd.pbtxt │ │ │ │ ├── RiscBinaryArithmetic.pbtxt │ │ │ │ ├── RiscBinaryComparison.pbtxt │ │ │ │ ├── RiscBitcast.pbtxt │ │ │ │ ├── RiscBroadcast.pbtxt │ │ │ │ ├── RiscCast.pbtxt │ │ │ │ ├── RiscCeil.pbtxt │ │ │ │ ├── RiscCholesky.pbtxt │ │ │ │ ├── RiscConcat.pbtxt │ │ │ │ ├── RiscCondition.pbtxt │ │ │ │ ├── RiscConv.pbtxt │ │ │ │ ├── RiscCos.pbtxt │ │ │ │ ├── RiscDiv.pbtxt │ │ │ │ ├── RiscDot.pbtxt │ │ │ │ ├── RiscExp.pbtxt │ │ │ │ ├── RiscFft.pbtxt │ │ │ │ ├── RiscFloor.pbtxt │ │ │ │ ├── RiscGather.pbtxt │ │ │ │ ├── RiscImag.pbtxt │ │ │ │ ├── RiscIsFinite.pbtxt │ │ │ │ ├── RiscLog.pbtxt │ │ │ │ ├── RiscLogicalAnd.pbtxt │ │ │ │ ├── RiscLogicalNot.pbtxt │ │ │ │ ├── RiscLogicalOr.pbtxt │ │ │ │ ├── RiscMax.pbtxt │ │ │ │ ├── RiscMin.pbtxt │ │ │ │ ├── RiscMul.pbtxt │ │ │ │ ├── RiscNeg.pbtxt │ │ │ │ ├── RiscPad.pbtxt │ │ │ │ ├── RiscPool.pbtxt │ │ │ │ ├── RiscPow.pbtxt │ │ │ │ ├── RiscRandomUniform.pbtxt │ │ │ │ ├── RiscReal.pbtxt │ │ │ │ ├── RiscReduce.pbtxt │ │ │ │ ├── RiscRem.pbtxt │ │ │ │ ├── RiscReshape.pbtxt │ │ │ │ ├── RiscReverse.pbtxt │ │ │ │ ├── RiscScatter.pbtxt │ │ │ │ ├── RiscShape.pbtxt │ │ │ │ ├── RiscSign.pbtxt │ │ │ │ ├── RiscSlice.pbtxt │ │ │ │ ├── RiscSort.pbtxt │ │ │ │ ├── RiscSqueeze.pbtxt │ │ │ │ ├── RiscSub.pbtxt │ │ │ │ ├── RiscTranspose.pbtxt │ │ │ │ ├── RiscTriangularSolve.pbtxt │ │ │ │ ├── RiscUnary.pbtxt │ │ │ │ ├── RiscWhile.pbtxt │ │ │ │ ├── RngReadAndSkip.pbtxt │ │ │ │ ├── RngSkip.pbtxt │ │ │ │ ├── Roll.pbtxt │ │ │ │ ├── Round.pbtxt │ │ │ │ ├── Rsqrt.pbtxt │ │ │ │ ├── RsqrtGrad.pbtxt │ │ │ │ ├── SampleDistortedBoundingBox.pbtxt │ │ │ │ ├── SampleDistortedBoundingBoxV2.pbtxt │ │ │ │ ├── SamplingDataset.pbtxt │ │ │ │ ├── Save.pbtxt │ │ │ │ ├── SaveDataset.pbtxt │ │ │ │ ├── SaveDatasetV2.pbtxt │ │ │ │ ├── SaveSlices.pbtxt │ │ │ │ ├── SaveV2.pbtxt │ │ │ │ ├── ScalarSummary.pbtxt │ │ │ │ ├── ScaleAndTranslate.pbtxt │ │ │ │ ├── ScaleAndTranslateGrad.pbtxt │ │ │ │ ├── ScanDataset.pbtxt │ │ │ │ ├── ScatterAdd.pbtxt │ │ │ │ ├── ScatterDiv.pbtxt │ │ │ │ ├── ScatterMax.pbtxt │ │ │ │ ├── ScatterMin.pbtxt │ │ │ │ ├── ScatterMul.pbtxt │ │ │ │ ├── ScatterNd.pbtxt │ │ │ │ ├── ScatterNdAdd.pbtxt │ │ │ │ ├── ScatterNdMax.pbtxt │ │ │ │ ├── ScatterNdMin.pbtxt │ │ │ │ ├── ScatterNdNonAliasingAdd.pbtxt │ │ │ │ ├── ScatterNdSub.pbtxt │ │ │ │ ├── ScatterNdUpdate.pbtxt │ │ │ │ ├── ScatterSub.pbtxt │ │ │ │ ├── ScatterUpdate.pbtxt │ │ │ │ ├── SdcaFprint.pbtxt │ │ │ │ ├── SdcaOptimizer.pbtxt │ │ │ │ ├── SdcaOptimizerV2.pbtxt │ │ │ │ ├── SdcaShrinkL1.pbtxt │ │ │ │ ├── SegmentMax.pbtxt │ │ │ │ ├── SegmentMaxV2.pbtxt │ │ │ │ ├── SegmentMean.pbtxt │ │ │ │ ├── SegmentMin.pbtxt │ │ │ │ ├── SegmentMinV2.pbtxt │ │ │ │ ├── SegmentProd.pbtxt │ │ │ │ ├── SegmentProdV2.pbtxt │ │ │ │ ├── SegmentSum.pbtxt │ │ │ │ ├── SegmentSumV2.pbtxt │ │ │ │ ├── Select.pbtxt │ │ │ │ ├── SelectV2.pbtxt │ │ │ │ ├── SelfAdjointEig.pbtxt │ │ │ │ ├── SelfAdjointEigV2.pbtxt │ │ │ │ ├── Selu.pbtxt │ │ │ │ ├── SeluGrad.pbtxt │ │ │ │ ├── Send.pbtxt │ │ │ │ ├── SendTPUEmbeddingGradients.pbtxt │ │ │ │ ├── SerializeIterator.pbtxt │ │ │ │ ├── SerializeManySparse.pbtxt │ │ │ │ ├── SerializeSparse.pbtxt │ │ │ │ ├── SerializeTensor.pbtxt │ │ │ │ ├── SetSize.pbtxt │ │ │ │ ├── SetStatsAggregatorDataset.pbtxt │ │ │ │ ├── Shape.pbtxt │ │ │ │ ├── ShapeN.pbtxt │ │ │ │ ├── ShardDataset.pbtxt │ │ │ │ ├── ShardedFilename.pbtxt │ │ │ │ ├── ShardedFilespec.pbtxt │ │ │ │ ├── ShuffleAndRepeatDataset.pbtxt │ │ │ │ ├── ShuffleAndRepeatDatasetV2.pbtxt │ │ │ │ ├── ShuffleDataset.pbtxt │ │ │ │ ├── ShuffleDatasetV2.pbtxt │ │ │ │ ├── ShuffleDatasetV3.pbtxt │ │ │ │ ├── ShutdownDistributedTPU.pbtxt │ │ │ │ ├── ShutdownTPUSystem.pbtxt │ │ │ │ ├── Sigmoid.pbtxt │ │ │ │ ├── SigmoidGrad.pbtxt │ │ │ │ ├── Sign.pbtxt │ │ │ │ ├── Sin.pbtxt │ │ │ │ ├── Sinh.pbtxt │ │ │ │ ├── Size.pbtxt │ │ │ │ ├── SkipDataset.pbtxt │ │ │ │ ├── Skipgram.pbtxt │ │ │ │ ├── SleepDataset.pbtxt │ │ │ │ ├── Slice.pbtxt │ │ │ │ ├── SlidingWindowDataset.pbtxt │ │ │ │ ├── Snapshot.pbtxt │ │ │ │ ├── SnapshotChunkDataset.pbtxt │ │ │ │ ├── SnapshotDataset.pbtxt │ │ │ │ ├── SnapshotDatasetReader.pbtxt │ │ │ │ ├── SnapshotDatasetV2.pbtxt │ │ │ │ ├── SnapshotNestedDatasetReader.pbtxt │ │ │ │ ├── SobolSample.pbtxt │ │ │ │ ├── Softmax.pbtxt │ │ │ │ ├── SoftmaxCrossEntropyWithLogits.pbtxt │ │ │ │ ├── Softplus.pbtxt │ │ │ │ ├── SoftplusGrad.pbtxt │ │ │ │ ├── Softsign.pbtxt │ │ │ │ ├── SoftsignGrad.pbtxt │ │ │ │ ├── SpaceToBatch.pbtxt │ │ │ │ ├── SpaceToBatchND.pbtxt │ │ │ │ ├── SpaceToDepth.pbtxt │ │ │ │ ├── SparseAccumulatorApplyGradient.pbtxt │ │ │ │ ├── SparseAccumulatorTakeGradient.pbtxt │ │ │ │ ├── SparseAdd.pbtxt │ │ │ │ ├── SparseAddGrad.pbtxt │ │ │ │ ├── SparseApplyAdadelta.pbtxt │ │ │ │ ├── SparseApplyAdagrad.pbtxt │ │ │ │ ├── SparseApplyAdagradDA.pbtxt │ │ │ │ ├── SparseApplyAdagradV2.pbtxt │ │ │ │ ├── SparseApplyCenteredRMSProp.pbtxt │ │ │ │ ├── SparseApplyFtrl.pbtxt │ │ │ │ ├── SparseApplyFtrlV2.pbtxt │ │ │ │ ├── SparseApplyMomentum.pbtxt │ │ │ │ ├── SparseApplyProximalAdagrad.pbtxt │ │ │ │ ├── SparseApplyProximalGradientDescent.pbtxt │ │ │ │ ├── SparseApplyRMSProp.pbtxt │ │ │ │ ├── SparseBincount.pbtxt │ │ │ │ ├── SparseConcat.pbtxt │ │ │ │ ├── SparseConditionalAccumulator.pbtxt │ │ │ │ ├── SparseCountSparseOutput.pbtxt │ │ │ │ ├── SparseCross.pbtxt │ │ │ │ ├── SparseCrossHashed.pbtxt │ │ │ │ ├── SparseCrossV2.pbtxt │ │ │ │ ├── SparseDenseCwiseAdd.pbtxt │ │ │ │ ├── SparseDenseCwiseDiv.pbtxt │ │ │ │ ├── SparseDenseCwiseMul.pbtxt │ │ │ │ ├── SparseFillEmptyRows.pbtxt │ │ │ │ ├── SparseFillEmptyRowsGrad.pbtxt │ │ │ │ ├── SparseMatMul.pbtxt │ │ │ │ ├── SparseMatrixAdd.pbtxt │ │ │ │ ├── SparseMatrixMatMul.pbtxt │ │ │ │ ├── SparseMatrixMul.pbtxt │ │ │ │ ├── SparseMatrixNNZ.pbtxt │ │ │ │ ├── SparseMatrixOrderingAMD.pbtxt │ │ │ │ ├── SparseMatrixSoftmax.pbtxt │ │ │ │ ├── SparseMatrixSoftmaxGrad.pbtxt │ │ │ │ ├── SparseMatrixSparseCholesky.pbtxt │ │ │ │ ├── SparseMatrixSparseMatMul.pbtxt │ │ │ │ ├── SparseMatrixTranspose.pbtxt │ │ │ │ ├── SparseMatrixZeros.pbtxt │ │ │ │ ├── SparseReduceMax.pbtxt │ │ │ │ ├── SparseReduceMaxSparse.pbtxt │ │ │ │ ├── SparseReduceSum.pbtxt │ │ │ │ ├── SparseReduceSumSparse.pbtxt │ │ │ │ ├── SparseReorder.pbtxt │ │ │ │ ├── SparseReshape.pbtxt │ │ │ │ ├── SparseSegmentMean.pbtxt │ │ │ │ ├── SparseSegmentMeanGrad.pbtxt │ │ │ │ ├── SparseSegmentMeanGradV2.pbtxt │ │ │ │ ├── SparseSegmentMeanWithNumSegments.pbtxt │ │ │ │ ├── SparseSegmentSqrtN.pbtxt │ │ │ │ ├── SparseSegmentSqrtNGrad.pbtxt │ │ │ │ ├── SparseSegmentSqrtNGradV2.pbtxt │ │ │ │ ├── SparseSegmentSqrtNWithNumSegments.pbtxt │ │ │ │ ├── SparseSegmentSum.pbtxt │ │ │ │ ├── SparseSegmentSumGrad.pbtxt │ │ │ │ ├── SparseSegmentSumGradV2.pbtxt │ │ │ │ ├── SparseSegmentSumWithNumSegments.pbtxt │ │ │ │ ├── SparseSlice.pbtxt │ │ │ │ ├── SparseSliceGrad.pbtxt │ │ │ │ ├── SparseSoftmax.pbtxt │ │ │ │ ├── SparseSoftmaxCrossEntropyWithLogits.pbtxt │ │ │ │ ├── SparseSparseMaximum.pbtxt │ │ │ │ ├── SparseSparseMinimum.pbtxt │ │ │ │ ├── SparseSplit.pbtxt │ │ │ │ ├── SparseTensorDenseAdd.pbtxt │ │ │ │ ├── SparseTensorDenseMatMul.pbtxt │ │ │ │ ├── SparseTensorSliceDataset.pbtxt │ │ │ │ ├── SparseTensorToCSRSparseMatrix.pbtxt │ │ │ │ ├── SparseToDense.pbtxt │ │ │ │ ├── SparseToSparseSetOperation.pbtxt │ │ │ │ ├── Spence.pbtxt │ │ │ │ ├── Split.pbtxt │ │ │ │ ├── SplitDedupData.pbtxt │ │ │ │ ├── SplitV.pbtxt │ │ │ │ ├── SqlDataset.pbtxt │ │ │ │ ├── Sqrt.pbtxt │ │ │ │ ├── SqrtGrad.pbtxt │ │ │ │ ├── Square.pbtxt │ │ │ │ ├── SquaredDifference.pbtxt │ │ │ │ ├── Squeeze.pbtxt │ │ │ │ ├── Stack.pbtxt │ │ │ │ ├── StackClose.pbtxt │ │ │ │ ├── StackCloseV2.pbtxt │ │ │ │ ├── StackPop.pbtxt │ │ │ │ ├── StackPopV2.pbtxt │ │ │ │ ├── StackPush.pbtxt │ │ │ │ ├── StackPushV2.pbtxt │ │ │ │ ├── StackV2.pbtxt │ │ │ │ ├── Stage.pbtxt │ │ │ │ ├── StageClear.pbtxt │ │ │ │ ├── StagePeek.pbtxt │ │ │ │ ├── StageSize.pbtxt │ │ │ │ ├── StatefulPartitionedCall.pbtxt │ │ │ │ ├── StatefulRandomBinomial.pbtxt │ │ │ │ ├── StatefulStandardNormal.pbtxt │ │ │ │ ├── StatefulStandardNormalV2.pbtxt │ │ │ │ ├── StatefulTruncatedNormal.pbtxt │ │ │ │ ├── StatefulUniform.pbtxt │ │ │ │ ├── StatefulUniformFullInt.pbtxt │ │ │ │ ├── StatefulUniformInt.pbtxt │ │ │ │ ├── StatelessCase.pbtxt │ │ │ │ ├── StatelessIf.pbtxt │ │ │ │ ├── StatelessMultinomial.pbtxt │ │ │ │ ├── StatelessParameterizedTruncatedNormal.pbtxt │ │ │ │ ├── StatelessRandomBinomial.pbtxt │ │ │ │ ├── StatelessRandomGammaV2.pbtxt │ │ │ │ ├── StatelessRandomGammaV3.pbtxt │ │ │ │ ├── StatelessRandomGetAlg.pbtxt │ │ │ │ ├── StatelessRandomGetKeyCounter.pbtxt │ │ │ │ ├── StatelessRandomGetKeyCounterAlg.pbtxt │ │ │ │ ├── StatelessRandomNormal.pbtxt │ │ │ │ ├── StatelessRandomNormalV2.pbtxt │ │ │ │ ├── StatelessRandomPoisson.pbtxt │ │ │ │ ├── StatelessRandomUniform.pbtxt │ │ │ │ ├── StatelessRandomUniformFullInt.pbtxt │ │ │ │ ├── StatelessRandomUniformFullIntV2.pbtxt │ │ │ │ ├── StatelessRandomUniformInt.pbtxt │ │ │ │ ├── StatelessRandomUniformIntV2.pbtxt │ │ │ │ ├── StatelessRandomUniformV2.pbtxt │ │ │ │ ├── StatelessSampleDistortedBoundingBox.pbtxt │ │ │ │ ├── StatelessShuffle.pbtxt │ │ │ │ ├── StatelessTruncatedNormal.pbtxt │ │ │ │ ├── StatelessTruncatedNormalV2.pbtxt │ │ │ │ ├── StatelessWhile.pbtxt │ │ │ │ ├── StaticRegexFullMatch.pbtxt │ │ │ │ ├── StaticRegexReplace.pbtxt │ │ │ │ ├── StatsAggregatorHandle.pbtxt │ │ │ │ ├── StatsAggregatorHandleV2.pbtxt │ │ │ │ ├── StatsAggregatorSetSummaryWriter.pbtxt │ │ │ │ ├── StatsAggregatorSummary.pbtxt │ │ │ │ ├── StochasticCastToInt.pbtxt │ │ │ │ ├── StopGradient.pbtxt │ │ │ │ ├── StoreMinibatchStatisticsInFdo.pbtxt │ │ │ │ ├── StridedSlice.pbtxt │ │ │ │ ├── StridedSliceAssign.pbtxt │ │ │ │ ├── StridedSliceGrad.pbtxt │ │ │ │ ├── StringFormat.pbtxt │ │ │ │ ├── StringJoin.pbtxt │ │ │ │ ├── StringLength.pbtxt │ │ │ │ ├── StringLower.pbtxt │ │ │ │ ├── StringNGrams.pbtxt │ │ │ │ ├── StringSplit.pbtxt │ │ │ │ ├── StringSplitV2.pbtxt │ │ │ │ ├── StringStrip.pbtxt │ │ │ │ ├── StringToHashBucket.pbtxt │ │ │ │ ├── StringToHashBucketFast.pbtxt │ │ │ │ ├── StringToHashBucketStrong.pbtxt │ │ │ │ ├── StringToNumber.pbtxt │ │ │ │ ├── StringUpper.pbtxt │ │ │ │ ├── Sub.pbtxt │ │ │ │ ├── Substr.pbtxt │ │ │ │ ├── Sum.pbtxt │ │ │ │ ├── SummaryWriter.pbtxt │ │ │ │ ├── Svd.pbtxt │ │ │ │ ├── Switch.pbtxt │ │ │ │ ├── SymbolicGradient.pbtxt │ │ │ │ ├── SyncDevice.pbtxt │ │ │ │ ├── TFRecordDataset.pbtxt │ │ │ │ ├── TFRecordDatasetV2.pbtxt │ │ │ │ ├── TFRecordReader.pbtxt │ │ │ │ ├── TFRecordReaderV2.pbtxt │ │ │ │ ├── TPUAnnotateTensorsWithDynamicShape.pbtxt │ │ │ │ ├── TPUCompilationResult.pbtxt │ │ │ │ ├── TPUCompile.pbtxt │ │ │ │ ├── TPUCompileSucceededAssert.pbtxt │ │ │ │ ├── TPUCopyWithDynamicShape.pbtxt │ │ │ │ ├── TPUEmbeddingActivations.pbtxt │ │ │ │ ├── TPUExecute.pbtxt │ │ │ │ ├── TPUExecuteAndUpdateVariables.pbtxt │ │ │ │ ├── TPUOrdinalSelector.pbtxt │ │ │ │ ├── TPUPartitionedCall.pbtxt │ │ │ │ ├── TPUPartitionedInput.pbtxt │ │ │ │ ├── TPUPartitionedInputV2.pbtxt │ │ │ │ ├── TPUPartitionedOutput.pbtxt │ │ │ │ ├── TPUPartitionedOutputV2.pbtxt │ │ │ │ ├── TPUReplicateMetadata.pbtxt │ │ │ │ ├── TPUReplicatedInput.pbtxt │ │ │ │ ├── TPUReplicatedOutput.pbtxt │ │ │ │ ├── TPUReshardVariables.pbtxt │ │ │ │ ├── TPURoundRobin.pbtxt │ │ │ │ ├── TakeDataset.pbtxt │ │ │ │ ├── TakeManySparseFromTensorsMap.pbtxt │ │ │ │ ├── TakeWhileDataset.pbtxt │ │ │ │ ├── Tan.pbtxt │ │ │ │ ├── Tanh.pbtxt │ │ │ │ ├── TanhGrad.pbtxt │ │ │ │ ├── TemporaryVariable.pbtxt │ │ │ │ ├── TensorArray.pbtxt │ │ │ │ ├── TensorArrayClose.pbtxt │ │ │ │ ├── TensorArrayCloseV2.pbtxt │ │ │ │ ├── TensorArrayCloseV3.pbtxt │ │ │ │ ├── TensorArrayConcat.pbtxt │ │ │ │ ├── TensorArrayConcatV2.pbtxt │ │ │ │ ├── TensorArrayConcatV3.pbtxt │ │ │ │ ├── TensorArrayGather.pbtxt │ │ │ │ ├── TensorArrayGatherV2.pbtxt │ │ │ │ ├── TensorArrayGatherV3.pbtxt │ │ │ │ ├── TensorArrayGrad.pbtxt │ │ │ │ ├── TensorArrayGradV2.pbtxt │ │ │ │ ├── TensorArrayGradV3.pbtxt │ │ │ │ ├── TensorArrayGradWithShape.pbtxt │ │ │ │ ├── TensorArrayPack.pbtxt │ │ │ │ ├── TensorArrayRead.pbtxt │ │ │ │ ├── TensorArrayReadV2.pbtxt │ │ │ │ ├── TensorArrayReadV3.pbtxt │ │ │ │ ├── TensorArrayScatter.pbtxt │ │ │ │ ├── TensorArrayScatterV2.pbtxt │ │ │ │ ├── TensorArrayScatterV3.pbtxt │ │ │ │ ├── TensorArraySize.pbtxt │ │ │ │ ├── TensorArraySizeV2.pbtxt │ │ │ │ ├── TensorArraySizeV3.pbtxt │ │ │ │ ├── TensorArraySplit.pbtxt │ │ │ │ ├── TensorArraySplitV2.pbtxt │ │ │ │ ├── TensorArraySplitV3.pbtxt │ │ │ │ ├── TensorArrayUnpack.pbtxt │ │ │ │ ├── TensorArrayV2.pbtxt │ │ │ │ ├── TensorArrayV3.pbtxt │ │ │ │ ├── TensorArrayWrite.pbtxt │ │ │ │ ├── TensorArrayWriteV2.pbtxt │ │ │ │ ├── TensorArrayWriteV3.pbtxt │ │ │ │ ├── TensorDataset.pbtxt │ │ │ │ ├── TensorListConcat.pbtxt │ │ │ │ ├── TensorListConcatLists.pbtxt │ │ │ │ ├── TensorListConcatV2.pbtxt │ │ │ │ ├── TensorListElementShape.pbtxt │ │ │ │ ├── TensorListFromTensor.pbtxt │ │ │ │ ├── TensorListGather.pbtxt │ │ │ │ ├── TensorListGetItem.pbtxt │ │ │ │ ├── TensorListLength.pbtxt │ │ │ │ ├── TensorListPopBack.pbtxt │ │ │ │ ├── TensorListPushBack.pbtxt │ │ │ │ ├── TensorListPushBackBatch.pbtxt │ │ │ │ ├── TensorListReserve.pbtxt │ │ │ │ ├── TensorListResize.pbtxt │ │ │ │ ├── TensorListScatter.pbtxt │ │ │ │ ├── TensorListScatterIntoExistingList.pbtxt │ │ │ │ ├── TensorListScatterV2.pbtxt │ │ │ │ ├── TensorListSetItem.pbtxt │ │ │ │ ├── TensorListSplit.pbtxt │ │ │ │ ├── TensorListStack.pbtxt │ │ │ │ ├── TensorMapErase.pbtxt │ │ │ │ ├── TensorMapHasKey.pbtxt │ │ │ │ ├── TensorMapInsert.pbtxt │ │ │ │ ├── TensorMapLookup.pbtxt │ │ │ │ ├── TensorMapSize.pbtxt │ │ │ │ ├── TensorMapStackKeys.pbtxt │ │ │ │ ├── TensorScatterAdd.pbtxt │ │ │ │ ├── TensorScatterMax.pbtxt │ │ │ │ ├── TensorScatterMin.pbtxt │ │ │ │ ├── TensorScatterSub.pbtxt │ │ │ │ ├── TensorScatterUpdate.pbtxt │ │ │ │ ├── TensorSliceDataset.pbtxt │ │ │ │ ├── TensorStridedSliceUpdate.pbtxt │ │ │ │ ├── TensorSummary.pbtxt │ │ │ │ ├── TensorSummaryV2.pbtxt │ │ │ │ ├── TextLineDataset.pbtxt │ │ │ │ ├── TextLineReader.pbtxt │ │ │ │ ├── TextLineReaderV2.pbtxt │ │ │ │ ├── ThreadPoolDataset.pbtxt │ │ │ │ ├── ThreadPoolHandle.pbtxt │ │ │ │ ├── ThreadUnsafeUnigramCandidateSampler.pbtxt │ │ │ │ ├── Tile.pbtxt │ │ │ │ ├── TileGrad.pbtxt │ │ │ │ ├── Timestamp.pbtxt │ │ │ │ ├── ToBool.pbtxt │ │ │ │ ├── TopK.pbtxt │ │ │ │ ├── TopKUnique.pbtxt │ │ │ │ ├── TopKV2.pbtxt │ │ │ │ ├── TopKWithUnique.pbtxt │ │ │ │ ├── TpuHandleToProtoKey.pbtxt │ │ │ │ ├── Transpose.pbtxt │ │ │ │ ├── TridiagonalMatMul.pbtxt │ │ │ │ ├── TridiagonalSolve.pbtxt │ │ │ │ ├── TruncateDiv.pbtxt │ │ │ │ ├── TruncateMod.pbtxt │ │ │ │ ├── TruncatedNormal.pbtxt │ │ │ │ ├── Unbatch.pbtxt │ │ │ │ ├── UnbatchDataset.pbtxt │ │ │ │ ├── UnbatchGrad.pbtxt │ │ │ │ ├── UncompressElement.pbtxt │ │ │ │ ├── UnicodeDecode.pbtxt │ │ │ │ ├── UnicodeDecodeWithOffsets.pbtxt │ │ │ │ ├── UnicodeEncode.pbtxt │ │ │ │ ├── UnicodeScript.pbtxt │ │ │ │ ├── UnicodeTranscode.pbtxt │ │ │ │ ├── UniformCandidateSampler.pbtxt │ │ │ │ ├── UniformDequantize.pbtxt │ │ │ │ ├── UniformQuantize.pbtxt │ │ │ │ ├── UniformQuantizedAdd.pbtxt │ │ │ │ ├── UniformQuantizedClipByValue.pbtxt │ │ │ │ ├── UniformQuantizedConvolution.pbtxt │ │ │ │ ├── UniformQuantizedConvolutionHybrid.pbtxt │ │ │ │ ├── UniformQuantizedDot.pbtxt │ │ │ │ ├── UniformQuantizedDotHybrid.pbtxt │ │ │ │ ├── UniformRequantize.pbtxt │ │ │ │ ├── Unique.pbtxt │ │ │ │ ├── UniqueDataset.pbtxt │ │ │ │ ├── UniqueV2.pbtxt │ │ │ │ ├── UniqueWithCounts.pbtxt │ │ │ │ ├── UniqueWithCountsV2.pbtxt │ │ │ │ ├── Unpack.pbtxt │ │ │ │ ├── UnravelIndex.pbtxt │ │ │ │ ├── UnsortedSegmentJoin.pbtxt │ │ │ │ ├── UnsortedSegmentMax.pbtxt │ │ │ │ ├── UnsortedSegmentMin.pbtxt │ │ │ │ ├── UnsortedSegmentProd.pbtxt │ │ │ │ ├── UnsortedSegmentSum.pbtxt │ │ │ │ ├── Unstage.pbtxt │ │ │ │ ├── UnwrapDatasetVariant.pbtxt │ │ │ │ ├── UpperBound.pbtxt │ │ │ │ ├── VarHandleOp.pbtxt │ │ │ │ ├── VarIsInitializedOp.pbtxt │ │ │ │ ├── Variable.pbtxt │ │ │ │ ├── VariableShape.pbtxt │ │ │ │ ├── VariableV2.pbtxt │ │ │ │ ├── Where.pbtxt │ │ │ │ ├── While.pbtxt │ │ │ │ ├── WholeFileReader.pbtxt │ │ │ │ ├── WholeFileReaderV2.pbtxt │ │ │ │ ├── WindowDataset.pbtxt │ │ │ │ ├── WindowOp.pbtxt │ │ │ │ ├── WorkerHeartbeat.pbtxt │ │ │ │ ├── WrapDatasetVariant.pbtxt │ │ │ │ ├── WriteAudioSummary.pbtxt │ │ │ │ ├── WriteFile.pbtxt │ │ │ │ ├── WriteGraphSummary.pbtxt │ │ │ │ ├── WriteHistogramSummary.pbtxt │ │ │ │ ├── WriteImageSummary.pbtxt │ │ │ │ ├── WriteRawProtoSummary.pbtxt │ │ │ │ ├── WriteScalarSummary.pbtxt │ │ │ │ ├── WriteSummary.pbtxt │ │ │ │ ├── Xdivy.pbtxt │ │ │ │ ├── XlaConcatND.pbtxt │ │ │ │ ├── XlaHostCompute.pbtxt │ │ │ │ ├── XlaRecvFromHost.pbtxt │ │ │ │ ├── XlaRecvTPUEmbeddingActivations.pbtxt │ │ │ │ ├── XlaRecvTPUEmbeddingDeduplicationData.pbtxt │ │ │ │ ├── XlaSendTPUEmbeddingGradients.pbtxt │ │ │ │ ├── XlaSendToHost.pbtxt │ │ │ │ ├── XlaSparseCoreAdagrad.pbtxt │ │ │ │ ├── XlaSparseCoreAdagradMomentum.pbtxt │ │ │ │ ├── XlaSparseCoreAdam.pbtxt │ │ │ │ ├── XlaSparseCoreFtrl.pbtxt │ │ │ │ ├── XlaSparseCoreSgd.pbtxt │ │ │ │ ├── XlaSparseDenseMatmul.pbtxt │ │ │ │ ├── XlaSparseDenseMatmulGradWithAdagradAndCsrInput.pbtxt │ │ │ │ ├── XlaSparseDenseMatmulGradWithAdagradMomentumAndCsrInput.pbtxt │ │ │ │ ├── XlaSparseDenseMatmulGradWithAdamAndCsrInput.pbtxt │ │ │ │ ├── XlaSparseDenseMatmulGradWithFtrlAndCsrInput.pbtxt │ │ │ │ ├── XlaSparseDenseMatmulGradWithSgdAndCsrInput.pbtxt │ │ │ │ ├── XlaSparseDenseMatmulWithCsrInput.pbtxt │ │ │ │ ├── XlaSplitND.pbtxt │ │ │ │ ├── Xlog1py.pbtxt │ │ │ │ ├── Xlogy.pbtxt │ │ │ │ ├── ZerosLike.pbtxt │ │ │ │ ├── Zeta.pbtxt │ │ │ │ └── ZipDataset.pbtxt │ │ │ └── update_ops_main.cc │ │ ├── composite_tensor_ops.cc │ │ ├── control_flow_ops.cc │ │ ├── control_flow_ops_test.cc │ │ ├── count_ops.cc │ │ ├── ctc_ops.cc │ │ ├── ctc_ops_test.cc │ │ ├── cudnn_rnn_ops.cc │ │ ├── cudnn_rnn_ops_test.cc │ │ ├── data_flow_ops.cc │ │ ├── data_flow_ops_test.cc │ │ ├── dataset_ops.cc │ │ ├── debug_ops.cc │ │ ├── decode_proto_ops.cc │ │ ├── encode_proto_ops.cc │ │ ├── experimental_dataset_ops.cc │ │ ├── filesystem_ops.cc │ │ ├── function_ops.cc │ │ ├── functional_grad.cc │ │ ├── functional_ops.cc │ │ ├── functional_ops_test.cc │ │ ├── image_ops.cc │ │ ├── image_ops_test.cc │ │ ├── io_ops.cc │ │ ├── io_ops_test.cc │ │ ├── linalg_ops.cc │ │ ├── linalg_ops_test.cc │ │ ├── list_ops.cc │ │ ├── logging_ops.cc │ │ ├── lookup_ops.cc │ │ ├── lookup_ops_test.cc │ │ ├── manip_ops.cc │ │ ├── map_ops.cc │ │ ├── math_grad.cc │ │ ├── math_grad_test.cc │ │ ├── math_ops.cc │ │ ├── math_ops_test.cc │ │ ├── mkl_array_ops.cc │ │ ├── mkl_nn_ops.cc │ │ ├── nccl_ops.cc │ │ ├── nn_grad.cc │ │ ├── nn_ops.cc │ │ ├── nn_ops_test.cc │ │ ├── no_op.cc │ │ ├── ops.pbtxt │ │ ├── optional_ops.cc │ │ ├── parsing_ops.cc │ │ ├── parsing_ops_test.cc │ │ ├── ragged_array_ops.cc │ │ ├── ragged_conversion_ops.cc │ │ ├── ragged_math_ops.cc │ │ ├── random_grad.cc │ │ ├── random_index_shuffle_ops.cc │ │ ├── random_ops.cc │ │ ├── random_ops_test.cc │ │ ├── resource_variable_ops.cc │ │ ├── risc_ops.cc │ │ ├── rnn_ops.cc │ │ ├── rnn_ops_test.cc │ │ ├── scoped_allocator_ops.cc │ │ ├── script_ops.cc │ │ ├── sdca_ops.cc │ │ ├── sendrecv_ops.cc │ │ ├── set_ops.cc │ │ ├── set_ops_test.cc │ │ ├── shape_function_test.cc │ │ ├── sparse_csr_matrix_ops.cc │ │ ├── sparse_csr_matrix_ops_test.cc │ │ ├── sparse_ops.cc │ │ ├── sparse_ops_test.cc │ │ ├── special_math_ops.cc │ │ ├── spectral_ops.cc │ │ ├── spectral_ops_test.cc │ │ ├── state_ops.cc │ │ ├── state_ops_test.cc │ │ ├── stateful_random_ops.cc │ │ ├── stateless_random_grad.cc │ │ ├── stateless_random_ops.cc │ │ ├── stateless_random_ops_v2.cc │ │ ├── stochastic_cast_op.cc │ │ ├── stochastic_cast_op_test.cc │ │ ├── string_ops.cc │ │ ├── string_ops_test.cc │ │ ├── summary_ops.cc │ │ ├── sync_ops.cc │ │ ├── tpu_configuration_ops.cc │ │ ├── tpu_cross_replica_ops.cc │ │ ├── tpu_cross_replica_ops_test.cc │ │ ├── tpu_embedding_load_retrieve_ops.cc │ │ ├── tpu_embedding_ops.cc │ │ ├── tpu_functional_ops.cc │ │ ├── tpu_heartbeat_ops.cc │ │ ├── tpu_host_compute_ops.cc │ │ ├── tpu_infeed_ops.cc │ │ ├── tpu_ordinal_selector_ops.cc │ │ ├── tpu_outfeed_ops.cc │ │ ├── tpu_replication_ops.cc │ │ ├── tpu_sharding_util_ops.cc │ │ ├── training_ops.cc │ │ ├── training_ops_test.cc │ │ ├── uniform_quant_ops.cc │ │ ├── uniform_quant_ops_test.cc │ │ └── word2vec_ops.cc │ ├── platform │ │ ├── BUILD │ │ ├── README.md │ │ ├── abi.h │ │ ├── base64.h │ │ ├── bfloat16.h │ │ ├── blocking_counter.h │ │ ├── build_config.bzl │ │ ├── build_config.default.bzl │ │ ├── build_config_root.bzl │ │ ├── build_config_root.default.bzl │ │ ├── build_test.cc │ │ ├── byte_order.h │ │ ├── casts.h │ │ ├── cloud │ │ │ ├── BUILD │ │ │ ├── auth_provider.h │ │ │ ├── compute_engine_metadata_client.h │ │ │ ├── compute_engine_zone_provider.h │ │ │ ├── curl_http_request.h │ │ │ ├── expiring_lru_cache.h │ │ │ ├── file_block_cache.h │ │ │ ├── gcs_dns_cache.h │ │ │ ├── gcs_file_system.h │ │ │ ├── gcs_throttle.h │ │ │ ├── google_auth_provider.h │ │ │ ├── http_request.h │ │ │ ├── http_request_fake.h │ │ │ ├── now_seconds_env.h │ │ │ ├── oauth_client.h │ │ │ ├── ram_file_block_cache.h │ │ │ ├── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── application_default_credentials.json │ │ │ │ ├── service_account_credentials.json │ │ │ │ └── service_account_public_key.txt │ │ │ ├── time_util.h │ │ │ └── zone_provider.h │ │ ├── coding.h │ │ ├── context.h │ │ ├── cord.h │ │ ├── cpu_feature_guard.cc │ │ ├── cpu_feature_guard.h │ │ ├── cpu_info.h │ │ ├── crash_analysis.h │ │ ├── ctstring.h │ │ ├── ctstring_internal.h │ │ ├── cuda.h │ │ ├── default │ │ │ └── build_config │ │ │ │ └── BUILD │ │ ├── demangle.h │ │ ├── denormal.h │ │ ├── distribute.bzl │ │ ├── dynamic_annotations.h │ │ ├── enable_tf2_utils.cc │ │ ├── enable_tf2_utils.h │ │ ├── enable_tf2_utils_test.cc │ │ ├── env.h │ │ ├── env_test.cc │ │ ├── env_time.h │ │ ├── error_logging.h │ │ ├── error_payloads.cc │ │ ├── error_payloads.h │ │ ├── errors.h │ │ ├── fake_python_env_test.cc │ │ ├── file_statistics.h │ │ ├── file_system.h │ │ ├── file_system_helper.h │ │ ├── file_system_test.cc │ │ ├── fingerprint.h │ │ ├── float8.h │ │ ├── gif.h │ │ ├── hash.h │ │ ├── host_info.h │ │ ├── human_readable_json.h │ │ ├── init_main.h │ │ ├── intrusive_ptr.h │ │ ├── jpeg.h │ │ ├── load_library.h │ │ ├── logger.h │ │ ├── logging.h │ │ ├── macros.h │ │ ├── mem.h │ │ ├── mutex.h │ │ ├── net.h │ │ ├── notification.h │ │ ├── null_file_system.h │ │ ├── numa.h │ │ ├── numbers.h │ │ ├── path.h │ │ ├── platform.h │ │ ├── platform_strings.cc │ │ ├── platform_strings.h │ │ ├── platform_strings_computed.h │ │ ├── platform_strings_test.cc │ │ ├── png.h │ │ ├── prefetch.h │ │ ├── profile_utils │ │ │ ├── BUILD │ │ │ ├── android_armv7a_cpu_utils_helper.h │ │ │ ├── clock_cycle_profiler.h │ │ │ ├── cpu_utils.h │ │ │ └── i_cpu_utils_helper.h │ │ ├── protobuf.h │ │ ├── protobuf_internal.h │ │ ├── ram_file_system.h │ │ ├── ram_file_system_test.py │ │ ├── random.h │ │ ├── raw_coding.h │ │ ├── refcount.h │ │ ├── regexp.h │ │ ├── resource.h │ │ ├── resource_loader.h │ │ ├── resource_loader_test.cc │ │ ├── retrying_file_system.h │ │ ├── retrying_utils.h │ │ ├── rocm.h │ │ ├── rocm_rocdl_path.h │ │ ├── rocm_rocdl_path_test.cc │ │ ├── rules_cc.bzl │ │ ├── scanner.h │ │ ├── setround.h │ │ ├── snappy.h │ │ ├── stack_frame.h │ │ ├── stacktrace.h │ │ ├── stacktrace_handler.h │ │ ├── status.h │ │ ├── status_matchers.h │ │ ├── statusor.h │ │ ├── str_util.h │ │ ├── strcat.h │ │ ├── stream_executor.h │ │ ├── stream_executor_no_cuda.h │ │ ├── stringpiece.h │ │ ├── stringprintf.h │ │ ├── strong_hash.h │ │ ├── subprocess.h │ │ ├── tensor_coding.cc │ │ ├── tensor_coding.h │ │ ├── tensor_float_32_utils.h │ │ ├── test.h │ │ ├── test_benchmark.h │ │ ├── testdata │ │ │ ├── BUILD │ │ │ ├── test_echo.cc │ │ │ ├── test_echo_argv_1.cc │ │ │ ├── test_noop.cc │ │ │ └── test_stderr.cc │ │ ├── thread_annotations.h │ │ ├── threadpool.h │ │ ├── threadpool_interface.h │ │ ├── threadpool_options.h │ │ ├── tracing.h │ │ ├── tstring.h │ │ ├── types.h │ │ ├── unbounded_work_queue.h │ │ ├── vmodule_benchmark_test.cc │ │ └── vmodule_test.cc │ ├── profiler │ │ ├── BUILD │ │ ├── backends │ │ │ ├── README.md │ │ │ ├── cpu │ │ │ │ ├── BUILD │ │ │ │ ├── annotation_stack.h │ │ │ │ ├── host_tracer.h │ │ │ │ ├── host_tracer_test.cc │ │ │ │ ├── host_tracer_utils.h │ │ │ │ ├── metadata_utils.h │ │ │ │ ├── python_tracer.h │ │ │ │ └── traceme_recorder.h │ │ │ └── gpu │ │ │ │ ├── BUILD │ │ │ │ ├── cuda_test.h │ │ │ │ ├── cupti_collector.h │ │ │ │ ├── cupti_error_manager.h │ │ │ │ ├── cupti_interface.h │ │ │ │ ├── cupti_tracer.h │ │ │ │ ├── cupti_wrapper.h │ │ │ │ ├── device_tracer_test.cc │ │ │ │ ├── mock_cupti.h │ │ │ │ ├── nvtx_utils.h │ │ │ │ └── rocm_tracer.h │ │ ├── builds │ │ │ ├── BUILD │ │ │ └── build_config.bzl │ │ ├── convert │ │ │ ├── BUILD │ │ │ ├── dcn_analysis.cc │ │ │ ├── dcn_analysis.h │ │ │ ├── dcn_analysis_test.cc │ │ │ ├── dcn_slack_analysis_combiner.cc │ │ │ ├── dcn_slack_analysis_combiner.h │ │ │ ├── dcn_utils.cc │ │ │ ├── dcn_utils.h │ │ │ ├── dcn_utils_test.cc │ │ │ ├── hlo_proto_to_graph_view.cc │ │ │ ├── hlo_proto_to_graph_view.h │ │ │ ├── hlo_proto_to_graph_view_test.cc │ │ │ ├── hlo_proto_to_memory_visualization_utils.cc │ │ │ ├── hlo_proto_to_memory_visualization_utils.h │ │ │ ├── hlo_proto_to_memory_visualization_utils_test.cc │ │ │ ├── hlo_to_tools_data.cc │ │ │ ├── hlo_to_tools_data.h │ │ │ ├── multi_xplanes_to_op_stats.cc │ │ │ ├── multi_xplanes_to_op_stats.h │ │ │ ├── op_metrics_db_combiner.cc │ │ │ ├── op_metrics_db_combiner.h │ │ │ ├── op_metrics_to_record.cc │ │ │ ├── op_metrics_to_record.h │ │ │ ├── op_profile_builder.cc │ │ │ ├── op_profile_builder.h │ │ │ ├── op_stack.h │ │ │ ├── op_stats_combiner.cc │ │ │ ├── op_stats_combiner.h │ │ │ ├── op_stats_combiner_test.cc │ │ │ ├── op_stats_to_input_pipeline_analysis.cc │ │ │ ├── op_stats_to_input_pipeline_analysis.h │ │ │ ├── op_stats_to_op_profile.cc │ │ │ ├── op_stats_to_op_profile.h │ │ │ ├── op_stats_to_overview_page.cc │ │ │ ├── op_stats_to_overview_page.h │ │ │ ├── op_stats_to_pod_stats.cc │ │ │ ├── op_stats_to_pod_stats.h │ │ │ ├── op_stats_to_pod_stats_test.cc │ │ │ ├── op_stats_to_pod_viewer.cc │ │ │ ├── op_stats_to_pod_viewer.h │ │ │ ├── op_stats_to_pod_viewer_test.cc │ │ │ ├── op_stats_to_tf_stats.cc │ │ │ ├── op_stats_to_tf_stats.h │ │ │ ├── op_stats_to_tf_stats_test.cc │ │ │ ├── preprocess_single_host_xplane.cc │ │ │ ├── preprocess_single_host_xplane.h │ │ │ ├── process_megascale_dcn.cc │ │ │ ├── process_megascale_dcn.h │ │ │ ├── repository.cc │ │ │ ├── repository.h │ │ │ ├── repository_test.cc │ │ │ ├── step_events_to_steps_db.cc │ │ │ ├── step_events_to_steps_db.h │ │ │ ├── tool_options.h │ │ │ ├── trace_viewer │ │ │ │ ├── BUILD │ │ │ │ ├── trace_event_arguments_builder.h │ │ │ │ ├── trace_events.cc │ │ │ │ ├── trace_events.h │ │ │ │ ├── trace_events_filter_interface.h │ │ │ │ ├── trace_events_to_json.cc │ │ │ │ ├── trace_events_to_json.h │ │ │ │ ├── trace_events_util.cc │ │ │ │ ├── trace_events_util.h │ │ │ │ ├── trace_viewer_color.cc │ │ │ │ ├── trace_viewer_color.h │ │ │ │ ├── trace_viewer_visibility.cc │ │ │ │ ├── trace_viewer_visibility.h │ │ │ │ └── trace_viewer_visibility_test.cc │ │ │ ├── xplane_to_dcn_collective_stats.cc │ │ │ ├── xplane_to_dcn_collective_stats.h │ │ │ ├── xplane_to_dcn_collective_stats_test.cc │ │ │ ├── xplane_to_hlo.cc │ │ │ ├── xplane_to_hlo.h │ │ │ ├── xplane_to_kernel_stats_db.cc │ │ │ ├── xplane_to_kernel_stats_db.h │ │ │ ├── xplane_to_kernel_stats_db_test.cc │ │ │ ├── xplane_to_memory_profile.cc │ │ │ ├── xplane_to_memory_profile.h │ │ │ ├── xplane_to_memory_profile_test.cc │ │ │ ├── xplane_to_op_metrics_db.cc │ │ │ ├── xplane_to_op_metrics_db.h │ │ │ ├── xplane_to_op_metrics_db_test.cc │ │ │ ├── xplane_to_op_stats.cc │ │ │ ├── xplane_to_op_stats.h │ │ │ ├── xplane_to_op_stats_test.cc │ │ │ ├── xplane_to_step_events.cc │ │ │ ├── xplane_to_step_events.h │ │ │ ├── xplane_to_step_events_test.cc │ │ │ ├── xplane_to_step_stats.cc │ │ │ ├── xplane_to_step_stats.h │ │ │ ├── xplane_to_tf_data_stats.cc │ │ │ ├── xplane_to_tf_data_stats.h │ │ │ ├── xplane_to_tf_data_stats_test.cc │ │ │ ├── xplane_to_tf_functions.cc │ │ │ ├── xplane_to_tf_functions.h │ │ │ ├── xplane_to_tf_functions_test.cc │ │ │ ├── xplane_to_tool_names.cc │ │ │ ├── xplane_to_tool_names.h │ │ │ ├── xplane_to_tool_names_test.cc │ │ │ ├── xplane_to_tools_data.cc │ │ │ ├── xplane_to_tools_data.h │ │ │ ├── xplane_to_trace_container.cc │ │ │ ├── xplane_to_trace_container.h │ │ │ ├── xspace_to_dcn_slack_analysis.cc │ │ │ └── xspace_to_dcn_slack_analysis.h │ │ ├── g3doc │ │ │ ├── advise.md │ │ │ ├── code_timeline.png │ │ │ ├── command_line.md │ │ │ ├── graph_timeline.png │ │ │ ├── options.md │ │ │ ├── pprof.jpg │ │ │ ├── profile_memory.md │ │ │ ├── profile_model_architecture.md │ │ │ ├── profile_time.md │ │ │ ├── profiler_ui.jpg │ │ │ ├── python_api.md │ │ │ └── scope_timeline.png │ │ ├── internal │ │ │ ├── BUILD │ │ │ ├── advisor │ │ │ │ ├── BUILD │ │ │ │ ├── accelerator_utilization_checker.h │ │ │ │ ├── checker.h │ │ │ │ ├── expensive_operation_checker.h │ │ │ │ ├── internal_checker_runner.h │ │ │ │ ├── internal_checker_runner_dummy.cc │ │ │ │ ├── operation_checker.h │ │ │ │ ├── tfprof_advisor.h │ │ │ │ └── tfprof_advisor_test.cc │ │ │ ├── print_model_analysis.cc │ │ │ ├── print_model_analysis.h │ │ │ ├── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── ckpt.data-00000-of-00001 │ │ │ │ ├── ckpt.index │ │ │ │ ├── ckpt.meta │ │ │ │ ├── graph.pbtxt │ │ │ │ ├── run_meta │ │ │ │ └── tfprof_log │ │ │ ├── tfprof_code.cc │ │ │ ├── tfprof_code.h │ │ │ ├── tfprof_constants.h │ │ │ ├── tfprof_graph.cc │ │ │ ├── tfprof_graph.h │ │ │ ├── tfprof_node.cc │ │ │ ├── tfprof_node.h │ │ │ ├── tfprof_node_show.cc │ │ │ ├── tfprof_node_show.h │ │ │ ├── tfprof_op.cc │ │ │ ├── tfprof_op.h │ │ │ ├── tfprof_scope.cc │ │ │ ├── tfprof_scope.h │ │ │ ├── tfprof_show.cc │ │ │ ├── tfprof_show.h │ │ │ ├── tfprof_show_multi.cc │ │ │ ├── tfprof_show_multi.h │ │ │ ├── tfprof_show_test.cc │ │ │ ├── tfprof_stats.cc │ │ │ ├── tfprof_stats.h │ │ │ ├── tfprof_stats_test.cc │ │ │ ├── tfprof_tensor.cc │ │ │ ├── tfprof_tensor.h │ │ │ ├── tfprof_tensor_test.cc │ │ │ ├── tfprof_timeline.cc │ │ │ ├── tfprof_timeline.h │ │ │ ├── tfprof_timeline_test.cc │ │ │ ├── tfprof_utils.cc │ │ │ └── tfprof_utils.h │ │ ├── lib │ │ │ ├── BUILD │ │ │ ├── annotated_traceme.h │ │ │ ├── connected_traceme.h │ │ │ ├── context_types.h │ │ │ ├── device_profiler_session.h │ │ │ ├── profiler_controller.h │ │ │ ├── profiler_disabled_test.cc │ │ │ ├── profiler_factory.h │ │ │ ├── profiler_interface.h │ │ │ ├── profiler_lock.h │ │ │ ├── profiler_session.h │ │ │ ├── scoped_annotation.h │ │ │ ├── scoped_annotation_stack.h │ │ │ ├── scoped_memory_debug_annotation.h │ │ │ ├── traceme.h │ │ │ └── traceme_encode.h │ │ ├── profile.proto │ │ ├── profiler.cc │ │ ├── profiler_analysis.proto │ │ ├── profiler_options.proto │ │ ├── profiler_service.proto │ │ ├── profiler_service_monitor_result.proto │ │ ├── protobuf │ │ │ ├── BUILD │ │ │ ├── dcn_collective_info.proto │ │ │ ├── dcn_slack_analysis.proto │ │ │ ├── diagnostics.proto │ │ │ ├── hardware_types.proto │ │ │ ├── input_pipeline.proto │ │ │ ├── kernel_stats.proto │ │ │ ├── memory_profile.proto │ │ │ ├── memory_viewer_preprocess.proto │ │ │ ├── op_metrics.proto │ │ │ ├── op_profile.proto │ │ │ ├── op_stats.proto │ │ │ ├── overview_page.proto │ │ │ ├── pod_stats.proto │ │ │ ├── pod_viewer.proto │ │ │ ├── power_metrics.proto │ │ │ ├── steps_db.proto │ │ │ ├── task.proto │ │ │ ├── tf_data_stats.proto │ │ │ ├── tf_function.proto │ │ │ ├── tf_stats.proto │ │ │ ├── tfstreamz.proto │ │ │ ├── topology.proto │ │ │ ├── trace_events.proto │ │ │ ├── trace_events_raw.proto │ │ │ └── xplane.proto │ │ ├── rpc │ │ │ ├── BUILD │ │ │ ├── client │ │ │ │ ├── BUILD │ │ │ │ ├── profiler_client.h │ │ │ │ ├── remote_profiler_session_manager.h │ │ │ │ └── save_profile.h │ │ │ ├── grpc.h │ │ │ ├── oss │ │ │ │ ├── BUILD │ │ │ │ └── grpc.cc │ │ │ ├── profiler_server.h │ │ │ └── profiler_service_impl.h │ │ ├── tfprof_log.proto │ │ ├── tfprof_options.cc │ │ ├── tfprof_options.h │ │ ├── tfprof_options.proto │ │ ├── tfprof_output.proto │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── cost_utils.cc │ │ │ ├── cost_utils.h │ │ │ ├── derived_timeline.cc │ │ │ ├── derived_timeline.h │ │ │ ├── derived_timeline_test.cc │ │ │ ├── device_caps_utils.cc │ │ │ ├── device_caps_utils.h │ │ │ ├── diagnostics.cc │ │ │ ├── diagnostics.h │ │ │ ├── event_span.cc │ │ │ ├── event_span.h │ │ │ ├── gpu_event_stats.cc │ │ │ ├── gpu_event_stats.h │ │ │ ├── hardware_type_utils.cc │ │ │ ├── hardware_type_utils.h │ │ │ ├── hlo_module_map.cc │ │ │ ├── hlo_module_map.h │ │ │ ├── hlo_module_utils.h │ │ │ ├── hlo_proto_map.cc │ │ │ ├── hlo_proto_map.h │ │ │ ├── hlo_proto_to_module.cc │ │ │ ├── hlo_proto_to_module.h │ │ │ ├── html_utils.h │ │ │ ├── kernel_stats_utils.cc │ │ │ ├── kernel_stats_utils.h │ │ │ ├── kernel_stats_utils_test.cc │ │ │ ├── math_utils.h │ │ │ ├── op_metrics_db_utils.cc │ │ │ ├── op_metrics_db_utils.h │ │ │ ├── op_metrics_db_utils_test.cc │ │ │ ├── op_utils.cc │ │ │ ├── op_utils.h │ │ │ ├── step_intersection.cc │ │ │ ├── step_intersection.h │ │ │ ├── step_intersection_test.cc │ │ │ ├── tfstreamz_utils.cc │ │ │ ├── tfstreamz_utils.h │ │ │ ├── trace_utils.h │ │ │ ├── xplane_builder.h │ │ │ ├── xplane_schema.h │ │ │ ├── xplane_test_utils.h │ │ │ ├── xplane_utils.h │ │ │ └── xplane_visitor.h │ ├── protobuf │ │ ├── BUILD │ │ ├── autotuning.proto │ │ ├── bfc_memory_map.proto │ │ ├── cluster.proto │ │ ├── composite_tensor_variant.proto │ │ ├── config.proto │ │ ├── control_flow.proto │ │ ├── conv_autotuning.proto │ │ ├── core_platform_payloads.proto │ │ ├── critical_section.proto │ │ ├── data_service.proto │ │ ├── debug.proto │ │ ├── debug_event.proto │ │ ├── device_filters.proto │ │ ├── device_properties.proto │ │ ├── eager_service.proto │ │ ├── error_codes.proto │ │ ├── fingerprint.proto │ │ ├── master.proto │ │ ├── master_service.proto │ │ ├── meta_graph.proto │ │ ├── named_tensor.proto │ │ ├── queue_runner.proto │ │ ├── remote_tensor_handle.proto │ │ ├── replay_log.proto │ │ ├── rewriter_config.proto │ │ ├── rpc_options.proto │ │ ├── saved_model.proto │ │ ├── saved_object_graph.proto │ │ ├── saver.proto │ │ ├── service_config.proto │ │ ├── snapshot.proto │ │ ├── status.proto │ │ ├── struct.proto │ │ ├── tensor_bundle.proto │ │ ├── tensorflow_server.proto │ │ ├── tpu │ │ │ ├── BUILD │ │ │ ├── compilation_result.proto │ │ │ ├── compile_metadata.proto │ │ │ ├── dynamic_padding.proto │ │ │ ├── optimization_parameters.proto │ │ │ ├── topology.proto │ │ │ └── tpu_embedding_configuration.proto │ │ ├── trackable_object_graph.proto │ │ ├── transport_options.proto │ │ ├── verifier_config.proto │ │ ├── worker.proto │ │ └── worker_service.proto │ ├── public │ │ ├── BUILD │ │ ├── README.md │ │ ├── session.h │ │ ├── session_options.h │ │ └── version.h │ ├── runtime_fallback │ │ ├── BUILD │ │ ├── bef_executor_flags.cc │ │ ├── bef_executor_flags.h │ │ ├── conversion │ │ │ ├── BUILD │ │ │ ├── conversion.cc │ │ │ ├── conversion.h │ │ │ └── static_registration.cc │ │ ├── kernel │ │ │ ├── BUILD │ │ │ ├── attr_util.cc │ │ │ ├── attr_util.h │ │ │ ├── attr_util_test.cc │ │ │ ├── conversion │ │ │ │ ├── conversion.cc │ │ │ │ ├── conversion.h │ │ │ │ └── static_registration.cc │ │ │ ├── kernel_fallback_compat_request_state.cc │ │ │ ├── kernel_fallback_compat_request_state.h │ │ │ ├── kernel_fallback_execute.cc │ │ │ ├── kernel_fallback_execute.h │ │ │ ├── kernel_fallback_execute_compat.cc │ │ │ ├── kernel_fallback_execute_compat.h │ │ │ ├── kernel_fallback_execute_compat_eager.cc │ │ │ ├── kernel_fallback_execute_compat_eager.h │ │ │ ├── kernel_fallback_kernels.cc │ │ │ ├── kernel_fallback_op_handler.cc │ │ │ ├── kernel_fallback_op_handler.h │ │ │ ├── kernel_fallback_static_registration.cc │ │ │ ├── kernel_fallback_tensor.cc │ │ │ ├── kernel_fallback_tensor.h │ │ │ ├── kernel_fallback_utils.cc │ │ │ ├── kernel_fallback_utils.h │ │ │ ├── kernels │ │ │ │ └── aggregate_ops.cc │ │ │ ├── tensor_util.cc │ │ │ ├── tensor_util.h │ │ │ ├── tfrt_op_kernel.cc │ │ │ ├── tfrt_op_kernel.h │ │ │ └── tfrt_op_kernel_test.cc │ │ ├── runtime │ │ │ ├── BUILD │ │ │ ├── conversion_function.cc │ │ │ ├── conversion_function.h │ │ │ ├── fallback_batch_kernel.cc │ │ │ ├── fallback_batch_kernel.h │ │ │ ├── kernel_utils.cc │ │ │ ├── kernel_utils.h │ │ │ ├── op_logger.h │ │ │ ├── runtime_fallback_batch_tf_opkernels.cc │ │ │ ├── runtime_fallback_kernels.cc │ │ │ ├── runtime_fallback_kernels.h │ │ │ ├── runtime_fallback_op_handler.cc │ │ │ ├── runtime_fallback_op_handler.h │ │ │ ├── runtime_fallback_tensor.cc │ │ │ ├── runtime_fallback_tensor.h │ │ │ └── static_registration.cc │ │ ├── test │ │ │ ├── BUILD │ │ │ ├── batch_function_fallback_benchmark_test.cc │ │ │ ├── c_api_tfrt_test.cc │ │ │ ├── coreruntime_driver.cc │ │ │ ├── coreruntime_driver.h │ │ │ ├── coreruntime_test.cc │ │ │ ├── fallback_coreruntime_test.cc │ │ │ ├── forwarding_test_kernels.cc │ │ │ ├── gpu │ │ │ │ └── BUILD │ │ │ ├── kernel_fallback_compat_request_state_test.cc │ │ │ ├── kernel_fallback_compat_test.cc │ │ │ ├── kernel_fallback_coreruntime_test.cc │ │ │ ├── runtime_fallback_kernels_test.cc │ │ │ ├── saved_model │ │ │ │ ├── BUILD │ │ │ │ ├── gen_saved_model_v1.py │ │ │ │ └── saved_model_test.cc │ │ │ ├── static_registration.cc │ │ │ ├── test_kernels.cc │ │ │ ├── test_opkernels.cc │ │ │ ├── testdata │ │ │ │ ├── batch_function_fallback.mlir │ │ │ │ ├── create_op.mlir │ │ │ │ └── custom_thread_pool.mlir │ │ │ ├── tfrt_forwarding_kernels.cc │ │ │ └── tfrt_forwarding_static_registration.cc │ │ ├── tf_bef_executor_main.cc │ │ └── util │ │ │ ├── BUILD │ │ │ ├── attr_type.def │ │ │ ├── attr_util.cc │ │ │ ├── attr_util.h │ │ │ ├── attr_util_test.cc │ │ │ ├── dtype.def │ │ │ ├── fallback_test_util.cc │ │ │ ├── fallback_test_util.h │ │ │ ├── gpu │ │ │ ├── gpu_utils.cc │ │ │ └── gpu_utils.h │ │ │ ├── tensor_metadata.h │ │ │ ├── tensor_util.cc │ │ │ ├── tensor_util.h │ │ │ ├── tensor_util_test.cc │ │ │ ├── type_util.h │ │ │ └── type_util_test.cc │ ├── summary │ │ ├── BUILD │ │ ├── loader.cc │ │ ├── schema.cc │ │ ├── schema.h │ │ ├── schema_test.cc │ │ ├── summary_converter.cc │ │ ├── summary_converter.h │ │ ├── summary_db_writer.cc │ │ ├── summary_db_writer.h │ │ ├── summary_db_writer_test.cc │ │ ├── summary_file_writer.cc │ │ ├── summary_file_writer.h │ │ ├── summary_file_writer_test.cc │ │ └── vacuum.cc │ ├── tfrt │ │ ├── BUILD │ │ ├── common │ │ │ ├── BUILD │ │ │ ├── async_value_tensor.cc │ │ │ ├── async_value_tensor.h │ │ │ ├── create_pjrt_client_util.cc │ │ │ ├── create_pjrt_client_util.h │ │ │ ├── create_pjrt_client_util_test.cc │ │ │ ├── global_state.cc │ │ │ ├── global_state.h │ │ │ ├── pjrt_client_factory_options.h │ │ │ ├── pjrt_client_factory_registry.cc │ │ │ ├── pjrt_client_factory_registry.h │ │ │ ├── pjrt_cpu_client_registration.cc │ │ │ ├── pjrt_cpu_client_registration_test.cc │ │ │ ├── pjrt_gpu_client_registration.cc │ │ │ ├── pjrt_gpu_client_registration_test.cc │ │ │ ├── pjrt_state.cc │ │ │ ├── pjrt_state.h │ │ │ ├── pjrt_state_test.cc │ │ │ ├── pjrt_util.cc │ │ │ ├── pjrt_util.h │ │ │ └── pjrt_util_test.cc │ │ ├── fallback │ │ │ ├── BUILD │ │ │ ├── cost_recorder.cc │ │ │ ├── cost_recorder.h │ │ │ ├── cost_recorder_test.cc │ │ │ ├── device_with_custom_allocator.h │ │ │ ├── fallback_state.cc │ │ │ ├── fallback_state.h │ │ │ ├── fallback_state_test.cc │ │ │ ├── op_cost_map.proto │ │ │ ├── op_kernel_runner.cc │ │ │ ├── op_kernel_runner.h │ │ │ ├── op_kernel_runner_cache.cc │ │ │ ├── op_kernel_runner_cache.h │ │ │ └── op_kernel_runner_test.cc │ │ ├── gpu │ │ │ └── kernel │ │ │ │ ├── BUILD │ │ │ │ ├── gpu_runner.cc │ │ │ │ ├── gpu_runner.h │ │ │ │ ├── gpu_runner_test.cc │ │ │ │ ├── gpurt_kernels.cc │ │ │ │ ├── tfrt_gpu_init.cc │ │ │ │ └── tfrt_gpu_init.h │ │ ├── graph_executor │ │ │ ├── BUILD │ │ │ ├── config.cc │ │ │ ├── config.h │ │ │ ├── config.proto │ │ │ ├── config_test.cc │ │ │ ├── executable_context.h │ │ │ ├── export_mlir.h │ │ │ ├── graph_execution_options.cc │ │ │ ├── graph_execution_options.h │ │ │ ├── graph_executor.cc │ │ │ ├── graph_executor.h │ │ │ ├── graph_executor_test.cc │ │ │ ├── python │ │ │ │ ├── BUILD │ │ │ │ ├── _pywrap_graph_execution_options.pyi │ │ │ │ └── graph_execution_options.cc │ │ │ ├── sync_resource_state.h │ │ │ └── test_config.proto │ │ ├── ifrt │ │ │ ├── BUILD │ │ │ ├── ifrt_executable_registry.cc │ │ │ ├── ifrt_executable_registry.h │ │ │ ├── ifrt_model_context.h │ │ │ ├── ifrt_serving_executable.cc │ │ │ ├── ifrt_serving_executable.h │ │ │ ├── ifrt_serving_executable_test.cc │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ └── executable.mlir │ │ ├── kernels │ │ │ ├── BUILD │ │ │ ├── ifrt_program_ops.cc │ │ │ ├── ifrt_program_ops.h │ │ │ ├── stream_ops.cc │ │ │ ├── stream_ops.h │ │ │ ├── stream_ops_util.cc │ │ │ ├── stream_ops_util.h │ │ │ ├── stream_ops_util_constants.h │ │ │ └── stream_ops_util_test.cc │ │ ├── mla │ │ │ ├── BUILD │ │ │ ├── mla_test_utils.h │ │ │ └── mla_utils.h │ │ ├── mlrt │ │ │ ├── attribute │ │ │ │ ├── BUILD │ │ │ │ ├── attribute.cc │ │ │ │ ├── attribute.h │ │ │ │ └── attribute_test.cc │ │ │ ├── bytecode │ │ │ │ ├── BUILD │ │ │ │ ├── bytecode.h │ │ │ │ ├── bytecode_test.cc │ │ │ │ ├── executable.h │ │ │ │ ├── executable_test.cc │ │ │ │ ├── function.h │ │ │ │ ├── function_test.cc │ │ │ │ ├── kernel.h │ │ │ │ ├── kernel_test.cc │ │ │ │ ├── span.h │ │ │ │ └── span_test.cc │ │ │ ├── interpreter │ │ │ │ ├── BUILD │ │ │ │ ├── async_handle.cc │ │ │ │ ├── async_handle.h │ │ │ │ ├── attribute_span.h │ │ │ │ ├── builtin_kernels.cc │ │ │ │ ├── builtin_kernels.h │ │ │ │ ├── context.cc │ │ │ │ ├── context.h │ │ │ │ ├── context_test.cc │ │ │ │ ├── execute.cc │ │ │ │ ├── execute.h │ │ │ │ ├── future.h │ │ │ │ ├── future_test.cc │ │ │ │ ├── interpreter_test.cc │ │ │ │ ├── interpreter_testutil.h │ │ │ │ ├── iterator.h │ │ │ │ ├── register_span.h │ │ │ │ ├── register_span_test.cc │ │ │ │ ├── value.h │ │ │ │ └── value_test.cc │ │ │ └── kernel │ │ │ │ ├── BUILD │ │ │ │ ├── batch_kernel.cc │ │ │ │ ├── batch_kernel.h │ │ │ │ ├── context.h │ │ │ │ ├── kernel.cc │ │ │ │ ├── kernel.h │ │ │ │ ├── kernel_runner_utils.cc │ │ │ │ ├── kernel_runner_utils.h │ │ │ │ └── kernel_test.cc │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── ifrt_program_ops.cc │ │ │ └── stream_ops.cc │ │ ├── run_handler_thread_pool │ │ │ ├── BUILD │ │ │ ├── run_handler.cc │ │ │ ├── run_handler.h │ │ │ ├── run_handler_concurrent_work_queue.cc │ │ │ ├── run_handler_concurrent_work_queue.h │ │ │ ├── run_handler_concurrent_work_queue_test.cc │ │ │ ├── run_handler_test.cc │ │ │ ├── run_handler_util.cc │ │ │ ├── run_handler_util.h │ │ │ └── run_handler_util_test.cc │ │ ├── runtime │ │ │ ├── BUILD │ │ │ ├── runtime.cc │ │ │ ├── runtime.h │ │ │ ├── runtime_test.cc │ │ │ ├── step_id.cc │ │ │ ├── step_id.h │ │ │ ├── stream.cc │ │ │ ├── stream.h │ │ │ ├── stream_test.cc │ │ │ ├── tf_threadpool_concurrent_work_queue.cc │ │ │ ├── tf_threadpool_concurrent_work_queue.h │ │ │ ├── tf_threadpool_concurrent_work_queue_test.cc │ │ │ ├── work_queue_interface.cc │ │ │ ├── work_queue_interface.h │ │ │ └── work_queue_interface_test.cc │ │ ├── saved_model │ │ │ ├── BUILD │ │ │ ├── python │ │ │ │ ├── BUILD │ │ │ │ ├── _pywrap_saved_model.pyi │ │ │ │ ├── saved_model_load_and_run.cc │ │ │ │ ├── saved_model_load_and_run.h │ │ │ │ ├── saved_model_load_and_run_test.py │ │ │ │ └── saved_model_load_and_run_wrapper.cc │ │ │ ├── saved_model.cc │ │ │ ├── saved_model.h │ │ │ ├── saved_model_aot_compile.cc │ │ │ ├── saved_model_aot_compile.h │ │ │ ├── saved_model_import_input.cc │ │ │ ├── saved_model_import_input.h │ │ │ ├── saved_model_testutil.cc │ │ │ ├── saved_model_testutil.h │ │ │ ├── saved_model_util.cc │ │ │ ├── saved_model_util.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── gen_control_flow_v1.py │ │ │ │ ├── gen_data.py │ │ │ │ ├── gen_dtype_coverage_v1.py │ │ │ │ ├── gen_error_v1.py │ │ │ │ ├── gen_hash_table_asset_v1.py │ │ │ │ ├── gen_if_v1.py │ │ │ │ ├── gen_matmul_gpu.py │ │ │ │ ├── gen_pow.py │ │ │ │ ├── gen_pow_v2.py │ │ │ │ ├── gen_ref_type_tensor_input.py │ │ │ │ ├── gen_resource_gather_v1.py │ │ │ │ ├── gen_saved_model.bzl │ │ │ │ ├── gen_saved_model_v1.py │ │ │ │ ├── gen_saved_model_v2.py │ │ │ │ ├── gen_sparse_tensor_input.py │ │ │ │ ├── gen_variable_on_tpu.py │ │ │ │ ├── gen_while_v1.py │ │ │ │ ├── saved_model_aot_test.cc │ │ │ │ ├── saved_model_gpu_test.cc │ │ │ │ ├── saved_model_ifrt_test.cc │ │ │ │ └── saved_model_test.cc │ │ │ └── utils │ │ │ │ ├── BUILD │ │ │ │ ├── serialize_utils.cc │ │ │ │ ├── serialize_utils.h │ │ │ │ └── serialize_utils_test.cc │ │ ├── stubs │ │ │ ├── BUILD │ │ │ ├── model_config_stub.cc │ │ │ ├── model_config_stub.h │ │ │ ├── tfrt_native_lowering_stub.cc │ │ │ └── tfrt_native_lowering_stub.h │ │ ├── tfrt_session │ │ │ ├── BUILD │ │ │ ├── tfrt_session.cc │ │ │ ├── tfrt_session.h │ │ │ ├── tfrt_session_init.h │ │ │ ├── tfrt_session_python_test.py │ │ │ └── tfrt_session_test.cc │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── any_ptr.h │ │ │ ├── bridge_graph_analysis.h │ │ │ ├── debug │ │ │ ├── BUILD │ │ │ ├── node_io_dump_rewriter.cc │ │ │ ├── node_io_dump_rewriter.h │ │ │ └── node_io_dump_rewriter_test.cc │ │ │ ├── device_variables_table.h │ │ │ ├── dtype.def │ │ │ ├── error_type.def │ │ │ ├── error_util.cc │ │ │ ├── error_util.h │ │ │ ├── error_util_test.cc │ │ │ ├── fallback_tensor.cc │ │ │ ├── fallback_tensor.h │ │ │ ├── fallback_tensor_test.cc │ │ │ ├── gpu_variables_table.h │ │ │ ├── graph_partition.cc │ │ │ ├── graph_partition.h │ │ │ ├── graph_partition_test.cc │ │ │ ├── tensor_util.cc │ │ │ ├── tensor_util.h │ │ │ ├── tensor_util_test.cc │ │ │ ├── tfrt_graph_execution_state.cc │ │ │ ├── tfrt_graph_execution_state.h │ │ │ ├── tfrt_graph_execution_state_test.cc │ │ │ ├── thread_pool.h │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ ├── tpu │ │ ├── BUILD │ │ ├── build_defs.bzl │ │ ├── graph_rewrite │ │ │ ├── BUILD │ │ │ ├── combine_tpu_embedding_load_retrieve_pass.cc │ │ │ ├── combine_tpu_embedding_load_retrieve_pass.h │ │ │ ├── cond_builder.cc │ │ │ ├── cond_builder.h │ │ │ ├── configure_tpu_embedding_rewrite_pass.cc │ │ │ ├── configure_tpu_embedding_rewrite_pass.h │ │ │ ├── configure_tpu_embedding_rewrite_registration.cc │ │ │ ├── distributed_tpu_configuration_rewrite_pass.cc │ │ │ ├── distributed_tpu_configuration_rewrite_pass.h │ │ │ ├── distributed_tpu_rewrite_helpers.cc │ │ │ ├── distributed_tpu_rewrite_helpers.h │ │ │ ├── distributed_tpu_rewrite_pass.cc │ │ │ ├── distributed_tpu_rewrite_pass.h │ │ │ ├── distributed_tpu_rewrite_pass_internal.cc │ │ │ ├── distributed_tpu_rewrite_pass_internal.h │ │ │ ├── encapsulate_tpu_computations_pass.cc │ │ │ ├── encapsulate_tpu_computations_pass.h │ │ │ ├── host_training_loop_optimization_util.cc │ │ │ ├── host_training_loop_optimization_util.h │ │ │ ├── incomplete_nodedef_builder.cc │ │ │ ├── incomplete_nodedef_builder.h │ │ │ ├── tpu_embedding_rewrite_pass_utils.cc │ │ │ ├── tpu_embedding_rewrite_pass_utils.h │ │ │ ├── tpu_embedding_software_deduplication_rewrite_pass.cc │ │ │ ├── tpu_embedding_software_deduplication_rewrite_pass.h │ │ │ ├── tpu_rewrite_pass_registration.cc │ │ │ ├── update_tpu_embedding_ops_passes.cc │ │ │ ├── update_tpu_embedding_ops_passes.h │ │ │ ├── variable_merger_pass.cc │ │ │ └── variable_merger_pass.h │ │ ├── kernels │ │ │ ├── BUILD │ │ │ ├── compiled_subgraph.h │ │ │ ├── cross_replica_ops.cc │ │ │ ├── global_iter_id.cc │ │ │ ├── host_compute_ops.cc │ │ │ ├── image_resize_ops.cc │ │ │ ├── infeed_ops.cc │ │ │ ├── infeed_ops.h │ │ │ ├── outfeed_ops.cc │ │ │ ├── outfeed_ops.h │ │ │ ├── replication_ops.cc │ │ │ ├── sharding_util_ops.cc │ │ │ ├── sharding_util_ops_test.cc │ │ │ ├── sparse_core_layout.cc │ │ │ ├── sparse_core_layout.h │ │ │ ├── sparse_core_layout.proto │ │ │ ├── sparse_core_layout_test.cc │ │ │ ├── sparse_core_ops_stats_handler.h │ │ │ ├── sparse_core_ops_utils.cc │ │ │ ├── sparse_core_ops_utils.h │ │ │ ├── sparse_core_ops_utils_test.cc │ │ │ ├── sparse_core_preprocess_ops.cc │ │ │ ├── sparse_core_preprocess_ops.h │ │ │ ├── sparse_core_xla_flags_defaults.h │ │ │ ├── sparse_core_xla_ops.cc │ │ │ ├── sparse_core_xla_ops.h │ │ │ ├── topk_ops.cc │ │ │ ├── tpu_compilation_cache.proto │ │ │ ├── tpu_compilation_cache_common.proto │ │ │ ├── tpu_compilation_cache_entry.h │ │ │ ├── tpu_compilation_cache_entry_unloader.h │ │ │ ├── tpu_compilation_cache_external.cc │ │ │ ├── tpu_compilation_cache_external.h │ │ │ ├── tpu_compilation_cache_factory.cc │ │ │ ├── tpu_compilation_cache_factory.h │ │ │ ├── tpu_compilation_cache_grpc.cc │ │ │ ├── tpu_compilation_cache_grpc.h │ │ │ ├── tpu_compilation_cache_interface.cc │ │ │ ├── tpu_compilation_cache_interface.h │ │ │ ├── tpu_compilation_cache_key.h │ │ │ ├── tpu_compilation_cache_local_lookup.cc │ │ │ ├── tpu_compilation_cache_local_lookup.h │ │ │ ├── tpu_compilation_cache_lookup.h │ │ │ ├── tpu_compilation_cache_rpc_lookup.cc │ │ │ ├── tpu_compilation_cache_rpc_lookup.h │ │ │ ├── tpu_compilation_cache_rpc_support.cc │ │ │ ├── tpu_compilation_cache_rpc_support.h │ │ │ ├── tpu_compilation_cache_service.cc │ │ │ ├── tpu_compilation_cache_service.h │ │ │ ├── tpu_compilation_metrics.cc │ │ │ ├── tpu_compilation_metrics.h │ │ │ ├── tpu_compile.proto │ │ │ ├── tpu_compile_op.cc │ │ │ ├── tpu_compile_op.h │ │ │ ├── tpu_compile_op_common.cc │ │ │ ├── tpu_compile_op_common.h │ │ │ ├── tpu_compile_op_impl.cc │ │ │ ├── tpu_compile_op_impl.h │ │ │ ├── tpu_compile_op_options.cc │ │ │ ├── tpu_compile_op_options.h │ │ │ ├── tpu_compile_op_support.cc │ │ │ ├── tpu_compile_op_support.h │ │ │ ├── tpu_configuration_ops.cc │ │ │ ├── tpu_configuration_ops.h │ │ │ ├── tpu_embedding_configuration_ops.cc │ │ │ ├── tpu_embedding_engine_state_interface.h │ │ │ ├── tpu_embedding_enqueue_ops.cc │ │ │ ├── tpu_embedding_enqueue_ops.h │ │ │ ├── tpu_embedding_load_retrieve_ops.cc │ │ │ ├── tpu_embedding_load_retrieve_ops.h │ │ │ ├── tpu_embedding_ops.cc │ │ │ ├── tpu_executable_info.proto │ │ │ ├── tpu_execute_op.cc │ │ │ ├── tpu_execute_op.h │ │ │ ├── tpu_execute_op_options.cc │ │ │ ├── tpu_execute_op_options.h │ │ │ ├── tpu_fingerprint_lookup.cc │ │ │ ├── tpu_fingerprint_lookup.h │ │ │ ├── tpu_functional_ops.cc │ │ │ ├── tpu_functional_ops.h │ │ │ ├── tpu_handle_to_key_op.cc │ │ │ ├── tpu_mesh_state_interface.h │ │ │ ├── tpu_op_consts.cc │ │ │ ├── tpu_op_consts.h │ │ │ ├── tpu_op_util.cc │ │ │ ├── tpu_op_util.h │ │ │ ├── tpu_ordinal_selector.h │ │ │ ├── tpu_ordinal_selector_interface.h │ │ │ ├── tpu_ordinal_selector_op.cc │ │ │ ├── tpu_pod_state.cc │ │ │ ├── tpu_pod_state.h │ │ │ ├── tpu_program_group.cc │ │ │ ├── tpu_program_group.h │ │ │ ├── tpu_program_group_interface.h │ │ │ ├── tpu_reshard_variables_op.cc │ │ │ ├── tpu_reshard_variables_op.h │ │ │ ├── tpu_reshard_variables_op_util.cc │ │ │ ├── tpu_reshard_variables_op_util.h │ │ │ ├── tpu_util.cc │ │ │ ├── tpu_util.h │ │ │ ├── trace_util.h │ │ │ ├── transfer_ops.cc │ │ │ ├── transfer_ops.h │ │ │ └── xla │ │ │ │ ├── BUILD │ │ │ │ ├── get_item_op.cc │ │ │ │ ├── host_compute_ops.cc │ │ │ │ ├── index_ops.cc │ │ │ │ ├── infeed_op.cc │ │ │ │ ├── inplace_ops.cc │ │ │ │ └── outfeed_ops.cc │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── host_compute_ops.cc │ │ │ ├── sparse_core_ops.cc │ │ │ ├── sparse_core_preprocess_ops.cc │ │ │ ├── topk_ops.cc │ │ │ ├── tpu_compile_op.cc │ │ │ ├── tpu_copy_with_dynamic_shape_op.cc │ │ │ ├── tpu_embedding_ops.cc │ │ │ ├── tpu_embedding_ops.h │ │ │ ├── tpu_embedding_shape_util.cc │ │ │ ├── tpu_embedding_shape_util.h │ │ │ ├── tpu_execute_op.cc │ │ │ ├── tpu_handle_to_key_op.cc │ │ │ ├── tpu_partitioned_input_op.cc │ │ │ ├── tpu_partitioned_output_op.cc │ │ │ ├── tpu_reshard_variables_op.cc │ │ │ └── tpu_round_robin_op.cc │ │ ├── tpu_api_dlsym_initializer.cc │ │ ├── tpu_api_dlsym_initializer_windows.cc │ │ ├── tpu_compile.cc │ │ ├── tpu_compile.h │ │ ├── tpu_compile_interface.cc │ │ ├── tpu_compile_interface.h │ │ ├── tpu_configuration.cc │ │ ├── tpu_configuration.h │ │ ├── tpu_defs.cc │ │ ├── tpu_defs.h │ │ ├── tpu_embedding_configuration_proto_rewrite.cc │ │ ├── tpu_embedding_configuration_proto_rewrite.h │ │ ├── tpu_embedding_configuration_proto_rewrite_test.cc │ │ ├── tpu_embedding_configuration_utils.cc │ │ ├── tpu_embedding_configuration_utils.h │ │ ├── tpu_embedding_errors.cc │ │ ├── tpu_embedding_errors.h │ │ ├── tpu_embedding_errors_test.cc │ │ ├── tpu_embedding_optimization_parameters_utils.cc │ │ ├── tpu_embedding_optimization_parameters_utils.h │ │ ├── tpu_embedding_output_layout_utils.cc │ │ ├── tpu_embedding_output_layout_utils.h │ │ ├── tpu_embedding_spmd_sharding_utils.cc │ │ ├── tpu_embedding_spmd_sharding_utils.h │ │ ├── tpu_execute.cc │ │ ├── tpu_execute.h │ │ ├── tpu_fingerprint_utils.cc │ │ ├── tpu_fingerprint_utils.h │ │ ├── tpu_global_init.cc │ │ ├── tpu_global_init.h │ │ ├── tpu_init_mode.cc │ │ ├── tpu_init_mode.h │ │ ├── tpu_model_server_initializer.cc │ │ ├── tpu_model_server_initializer.h │ │ ├── tpu_node_device_util.cc │ │ ├── tpu_node_device_util.h │ │ ├── virtual_device.cc │ │ └── virtual_device.h │ ├── transforms │ │ ├── BUILD │ │ ├── README.md │ │ ├── cf_sink │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── sink_basic.mlir │ │ │ │ ├── sink_region_op.mlir │ │ │ │ └── sink_rename.mlir │ │ ├── consolidate_attrs │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── cf_op_dropped_attrs.mlir │ │ │ │ ├── cf_op_output_shapes.mlir │ │ │ │ ├── function_arg_handle_data.mlir │ │ │ │ ├── function_arg_output_shapes.mlir │ │ │ │ ├── function_dropped_arg_attrs.mlir │ │ │ │ ├── function_dropped_res_attrs.mlir │ │ │ │ ├── function_input_shapes.mlir │ │ │ │ ├── function_res_handle_data.mlir │ │ │ │ ├── function_res_output_shapes.mlir │ │ │ │ ├── function_with_empty_arg_or_results.mlir │ │ │ │ ├── legalize_cf_op_attrs.mlir │ │ │ │ ├── materialize_cf_output_shapes.mlir │ │ │ │ ├── materialize_function_attrs.mlir │ │ │ │ ├── materialize_output_shapes.mlir │ │ │ │ ├── negative_shape.mlir │ │ │ │ ├── op_output_shapes.mlir │ │ │ │ ├── op_output_shapes_roundtrip.mlir │ │ │ │ └── skip_generic_functions.mlir │ │ ├── const_dedupe_hoist │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── duplicate_constants.mlir │ │ │ │ └── with_legacy_call.mlir │ │ ├── constant_folding │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── add_subtract_tree.mlir │ │ │ │ ├── add_tree.mlir │ │ │ │ ├── constant_pushdown_0.mlir │ │ │ │ ├── constant_pushdown_15.mlir │ │ │ │ ├── constant_pushdown_23.mlir │ │ │ │ ├── constant_pushdown_31.mlir │ │ │ │ ├── constant_pushdown_7.mlir │ │ │ │ ├── constant_pushdown_biasadd.mlir │ │ │ │ ├── enter.mlir │ │ │ │ ├── materialize_broadcast_gradient_args.mlir │ │ │ │ ├── materialize_constant_value_node.mlir │ │ │ │ ├── materialize_reduction_indices.mlir │ │ │ │ ├── materialize_reduction_indices_with_reshape.mlir │ │ │ │ ├── merge_concat.mlir │ │ │ │ ├── merge_concat_partial_folding.mlir │ │ │ │ ├── merge_nodes.mlir │ │ │ │ ├── mul_conv_pushdown_conv2d_scalar_const.mlir │ │ │ │ ├── neutral_element.mlir │ │ │ │ ├── no_op_reshape.mlir │ │ │ │ ├── packing.mlir │ │ │ │ ├── padding_with_zero_size.mlir │ │ │ │ ├── partial_folding_associative_and_commutative.mlir │ │ │ │ ├── partial_folding_identity_n.mlir │ │ │ │ ├── pattern_category.mlir │ │ │ │ ├── preserve_ops.mlir │ │ │ │ ├── random_shuffle_on_scalar_removal.mlir │ │ │ │ ├── reverse_on_size_1dims_removal.mlir │ │ │ │ ├── shape_materialization.mlir │ │ │ │ ├── shape_materialization_shapen.mlir │ │ │ │ ├── simplify_case.mlir │ │ │ │ ├── simplify_select.mlir │ │ │ │ ├── slice_with_same_dimension_removal.mlir │ │ │ │ ├── split_removal.mlir │ │ │ │ ├── splitv_removal.mlir │ │ │ │ ├── squeeze_with_all_dimensions_greater_than_one.mlir │ │ │ │ ├── strength_reduce_reciprocal.mlir │ │ │ │ ├── strided_slice_with_same_dimension_removal.mlir │ │ │ │ ├── switch_identical_inputs.mlir │ │ │ │ ├── tensor_array_size.mlir │ │ │ │ ├── tile_with_multiplies_being_one.mlir │ │ │ │ ├── transpose_on_size_1dims_removal.mlir │ │ │ │ └── trivial_pack.mlir │ │ ├── cse │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ └── cse.mlir │ │ ├── drop_unregistered_attribute │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ └── drop_output_shapes.mlir │ │ ├── eliminate_passthrough_iter_args │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ └── eliminate_passthrough_iter_args.mlir │ │ ├── func_to_graph │ │ │ ├── BUILD │ │ │ ├── func_to_graph.cc │ │ │ ├── func_to_graph.h │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── round_trip.mlir │ │ │ │ └── simple.mlir │ │ ├── functional_to_region │ │ │ ├── BUILD │ │ │ ├── impl.cc │ │ │ ├── impl.h │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── case.mlir │ │ │ │ ├── for.mlir │ │ │ │ ├── function_multi_use.mlir │ │ │ │ ├── function_with_no_args_or_returns.mlir │ │ │ │ ├── if.mlir │ │ │ │ ├── if_in_func.mlir │ │ │ │ ├── if_needs_cast.mlir │ │ │ │ ├── if_nested.mlir │ │ │ │ ├── missing_function.mlir │ │ │ │ ├── name_uniquing.mlir │ │ │ │ └── while.mlir │ │ ├── graph_compactor │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── add_default_attrs.mlir │ │ │ │ ├── rename.mlir │ │ │ │ ├── rename_lots.mlir │ │ │ │ └── strip_default_attrs.mlir │ │ ├── graph_to_func │ │ │ ├── BUILD │ │ │ ├── graph_to_func.cc │ │ │ ├── graph_to_func.h │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── same_feeds_and_fetches.mlir │ │ │ │ └── simple.mlir │ │ ├── graph_transform_wrapper.cc │ │ ├── graph_transform_wrapper.h │ │ ├── graph_transform_wrapper_test.cc │ │ ├── legacy_call │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── legacy_call_not_pruned.mlir │ │ │ │ └── lift_legacy_call.mlir │ │ ├── pass_registration.h │ │ ├── passes.td │ │ ├── region_to_functional │ │ │ ├── BUILD │ │ │ ├── impl.cc │ │ │ ├── impl.h │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── case.mlir │ │ │ │ ├── duplicate_loop_name.mlir │ │ │ │ ├── for.mlir │ │ │ │ ├── func_nested_outline_first.mlir │ │ │ │ ├── idempotence_arg_count.mlir │ │ │ │ ├── idempotence_arg_reorder.mlir │ │ │ │ ├── idempotence_double_specialize.mlir │ │ │ │ ├── if.mlir │ │ │ │ ├── if_barrier.mlir │ │ │ │ ├── if_in_func.mlir │ │ │ │ ├── if_nested.mlir │ │ │ │ ├── nested_name_lookup.mlir │ │ │ │ ├── noop.mlir │ │ │ │ ├── op_count_compare.mlir │ │ │ │ ├── region_with_no_args_or_returns.mlir │ │ │ │ ├── return_type_mismatch.mlir │ │ │ │ ├── roundtrip_attrs.mlir │ │ │ │ ├── roundtrip_multiuse.mlir │ │ │ │ ├── roundtrip_name_uniquing.mlir │ │ │ │ ├── roundtrip_regions_with_no_args_or_results.mlir │ │ │ │ ├── roundtrip_reuse.mlir │ │ │ │ ├── roundtrip_reuse_type_mismatch.mlir │ │ │ │ ├── sink_respecialize.mlir │ │ │ │ └── while.mlir │ │ ├── remapper │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ ├── pdll │ │ │ │ └── mkl_patterns.pdll │ │ │ ├── remapping_helper.h │ │ │ └── tests │ │ │ │ ├── contraction.mlir │ │ │ │ ├── fusedbatchnormEx_test.mlir │ │ │ │ ├── onednn_contraction.mlir │ │ │ │ ├── onednn_fusedbatchnormEx_test.mlir │ │ │ │ ├── onednn_mish.mlir │ │ │ │ ├── onednn_remapper.mlir │ │ │ │ └── tensor_to_hashbucket.mlir │ │ ├── shape_inference │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ └── shape_inference_test.mlir │ │ ├── tfg-transforms-opt.cc │ │ ├── toposort │ │ │ ├── BUILD │ │ │ ├── pass.cc │ │ │ ├── pass.h │ │ │ └── tests │ │ │ │ ├── toposort.mlir │ │ │ │ └── toposort_regions.mlir │ │ └── utils │ │ │ ├── eval_utils.cc │ │ │ ├── eval_utils.h │ │ │ ├── eval_utils_test.cc │ │ │ ├── op_cat_helper.cc │ │ │ ├── op_cat_helper.h │ │ │ ├── pdll │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils.pdll │ │ │ ├── utils.cc │ │ │ └── utils.h │ ├── user_ops │ │ ├── BUILD │ │ └── fact.cc │ └── util │ │ ├── BUILD │ │ ├── activation_mode.cc │ │ ├── activation_mode.h │ │ ├── autotune_maps │ │ ├── BUILD │ │ ├── autotune_map.proto │ │ ├── autotune_serialize.cc │ │ ├── autotune_serialize.h │ │ ├── autotune_serialize_test.cc │ │ ├── conv_autotune_maps.h │ │ ├── conv_parameters.cc │ │ ├── conv_parameters.h │ │ └── conv_parameters.proto │ │ ├── batch_util.cc │ │ ├── batch_util.h │ │ ├── bcast.cc │ │ ├── bcast.h │ │ ├── bcast_test.cc │ │ ├── command_line_flags.h │ │ ├── command_line_flags_test.cc │ │ ├── ctc │ │ ├── BUILD │ │ ├── ctc_beam_entry.h │ │ ├── ctc_beam_scorer.h │ │ ├── ctc_beam_search.h │ │ ├── ctc_beam_search_test.cc │ │ ├── ctc_decoder.h │ │ ├── ctc_loss_calculator.cc │ │ ├── ctc_loss_calculator.h │ │ └── ctc_loss_util.h │ │ ├── cuda_solvers.cc │ │ ├── cuda_sparse.cc │ │ ├── cuda_sparse.h │ │ ├── debug_data_dumper.cc │ │ ├── debug_data_dumper.h │ │ ├── debug_data_dumper_test.cc │ │ ├── debug_events_writer.cc │ │ ├── debug_events_writer.h │ │ ├── debug_events_writer_test.cc │ │ ├── determinism.h │ │ ├── device_name_utils.h │ │ ├── dump_graph.cc │ │ ├── dump_graph.h │ │ ├── dump_graph_test.cc │ │ ├── einsum_op_util.cc │ │ ├── einsum_op_util.h │ │ ├── env_var.h │ │ ├── equal_graph_def.cc │ │ ├── equal_graph_def.h │ │ ├── equal_graph_def_test.cc │ │ ├── event.proto │ │ ├── events_writer.cc │ │ ├── events_writer.h │ │ ├── events_writer_test.cc │ │ ├── example_proto_fast_parsing.cc │ │ ├── example_proto_fast_parsing.h │ │ ├── example_proto_fast_parsing_test.cc │ │ ├── example_proto_fast_parsing_test.proto │ │ ├── example_proto_helper.cc │ │ ├── example_proto_helper.h │ │ ├── example_proto_helper_test.cc │ │ ├── exec_on_stall.h │ │ ├── exec_on_stall_test.cc │ │ ├── fake_clock_env.cc │ │ ├── fake_clock_env.h │ │ ├── fake_clock_env_test.cc │ │ ├── gpu_cuda_alias.h │ │ ├── gpu_device_functions.h │ │ ├── gpu_kernel_helper.h │ │ ├── gpu_kernel_helper_test.cu.cc │ │ ├── gpu_launch_config.h │ │ ├── gpu_solvers.h │ │ ├── guarded_philox_random.cc │ │ ├── guarded_philox_random.h │ │ ├── image_resizer_state.h │ │ ├── incremental_barrier.cc │ │ ├── incremental_barrier.h │ │ ├── incremental_barrier_test.cc │ │ ├── managed_stack_trace.h │ │ ├── matmul_autotune.cc │ │ ├── matmul_autotune.h │ │ ├── matmul_bcast.h │ │ ├── matmul_bcast_test.cc │ │ ├── memmapped_file_system.cc │ │ ├── memmapped_file_system.h │ │ ├── memmapped_file_system.proto │ │ ├── memmapped_file_system_test.cc │ │ ├── memmapped_file_system_writer.cc │ │ ├── memmapped_file_system_writer.h │ │ ├── mirror_pad_mode.cc │ │ ├── mirror_pad_mode.h │ │ ├── mkl_heuristics.h │ │ ├── mkl_heuristics_test.cc │ │ ├── mkl_util.h │ │ ├── mkl_util_test.cc │ │ ├── onednn_env_vars.cc │ │ ├── onednn_env_vars.h │ │ ├── overflow.h │ │ ├── overflow_test.cc │ │ ├── padding.cc │ │ ├── padding.h │ │ ├── permutation_input_iterator.h │ │ ├── permutation_output_iterator.h │ │ ├── port.cc │ │ ├── port.h │ │ ├── presized_cuckoo_map.h │ │ ├── presized_cuckoo_map_test.cc │ │ ├── proto │ │ ├── BUILD │ │ ├── decode.h │ │ ├── descriptor_pool_registry.cc │ │ ├── descriptor_pool_registry.h │ │ ├── descriptor_pool_registry_test.cc │ │ ├── descriptors.cc │ │ ├── descriptors.h │ │ ├── local_descriptor_pool_registration.cc │ │ ├── proto_utils.cc │ │ ├── proto_utils.h │ │ └── proto_utils_test.cc │ │ ├── quantization │ │ ├── BUILD │ │ ├── uniform_quant_ops_attr.proto │ │ ├── uniform_quant_ops_params.cc │ │ ├── uniform_quant_ops_params.h │ │ └── uniform_quant_ops_params_test.cc │ │ ├── ragged_to_dense_util.cc │ │ ├── ragged_to_dense_util.h │ │ ├── ragged_to_dense_util_common.h │ │ ├── ragged_to_dense_util_test.cc │ │ ├── reffed_status_callback.h │ │ ├── reffed_status_callback_test.cc │ │ ├── reporter.h │ │ ├── reporter_test.cc │ │ ├── rocm_solvers.cc │ │ ├── rocm_sparse.cc │ │ ├── saved_tensor_slice.proto │ │ ├── saved_tensor_slice_util.cc │ │ ├── saved_tensor_slice_util.h │ │ ├── saved_tensor_slice_util_test.cc │ │ ├── semver_test.cc │ │ ├── sparse │ │ ├── BUILD │ │ ├── README.md │ │ ├── dim_comparator.h │ │ ├── group_iterator.cc │ │ ├── group_iterator.h │ │ ├── sparse_tensor.cc │ │ ├── sparse_tensor.h │ │ └── sparse_tensor_test.cc │ │ ├── stat_summarizer.cc │ │ ├── stat_summarizer.h │ │ ├── stat_summarizer_options.h │ │ ├── stat_summarizer_test.cc │ │ ├── stats_calculator.h │ │ ├── stream_executor_util.h │ │ ├── strided_slice_op.cc │ │ ├── strided_slice_op.h │ │ ├── strided_slice_op_test.cc │ │ ├── tensor_bundle │ │ ├── BUILD │ │ ├── byte_swap_array.h │ │ ├── byte_swap_tensor.cc │ │ ├── byte_swap_tensor.h │ │ ├── naming.cc │ │ ├── naming.h │ │ ├── tensor_bundle.cc │ │ ├── tensor_bundle.h │ │ ├── tensor_bundle_test.cc │ │ └── testdata │ │ │ ├── BUILD │ │ │ └── old_string_tensors │ │ │ ├── BUILD │ │ │ ├── README │ │ │ ├── foo.data-00000-of-00001 │ │ │ └── foo.index │ │ ├── tensor_format.cc │ │ ├── tensor_format.h │ │ ├── tensor_format_test.cc │ │ ├── tensor_ops_util.h │ │ ├── tensor_slice_reader.cc │ │ ├── tensor_slice_reader.h │ │ ├── tensor_slice_reader_cache.cc │ │ ├── tensor_slice_reader_cache.h │ │ ├── tensor_slice_reader_test.cc │ │ ├── tensor_slice_set.cc │ │ ├── tensor_slice_set.h │ │ ├── tensor_slice_set_test.cc │ │ ├── tensor_slice_util.h │ │ ├── tensor_slice_util_test.cc │ │ ├── tensor_slice_writer.cc │ │ ├── tensor_slice_writer.h │ │ ├── tensor_slice_writer_test.cc │ │ ├── test_log.proto │ │ ├── transform_output_iterator.h │ │ ├── use_cudnn.h │ │ ├── util.cc │ │ ├── util.h │ │ ├── work_sharder.cc │ │ ├── work_sharder.h │ │ ├── work_sharder_test.cc │ │ ├── xla_config_registry.cc │ │ ├── xla_config_registry.h │ │ └── zen_util.h ├── distribute │ └── experimental │ │ └── rpc │ │ ├── README.md │ │ ├── kernels │ │ ├── BUILD │ │ ├── grpc_credentials.h │ │ ├── grpc_rpc_service.h │ │ ├── metadata_for_rpc_ops.cc │ │ ├── oss │ │ │ ├── BUILD │ │ │ ├── defs.bzl │ │ │ └── grpc_credentials.cc │ │ └── rpc_ops.cc │ │ └── proto │ │ ├── BUILD │ │ └── tf_rpc_service.proto ├── docs_src │ └── README.md ├── dtensor │ ├── BUILD │ ├── build_defs.bzl │ ├── cc │ │ ├── BUILD │ │ ├── constants.h │ │ ├── default_parallel_executor.cc │ │ ├── dstatus.h │ │ ├── dtensor_device.cc │ │ ├── dtensor_device.h │ │ ├── dtensor_device_util.cc │ │ ├── dtensor_device_util.h │ │ ├── dtensor_graph_to_mlir_pass.cc │ │ ├── dtensor_graph_to_mlir_pass.h │ │ ├── dtensor_meta_ops.cc │ │ ├── dtensor_operation.h │ │ ├── dtensor_ops.cc │ │ ├── dtensor_tpu_kernels.cc │ │ ├── dtensor_tpu_ops.cc │ │ ├── dtensor_utils.cc │ │ ├── dtensor_utils.h │ │ ├── mesh_type.h │ │ ├── parallel_executor.h │ │ ├── save_restore_util.cc │ │ ├── save_restore_util.h │ │ ├── slice_util.cc │ │ ├── slice_util.h │ │ ├── small_constant_optimization.cc │ │ ├── small_constant_optimization.h │ │ ├── tensor_layout.cc │ │ ├── tensor_layout.h │ │ ├── tensor_with_layout.cc │ │ ├── tensor_with_layout.h │ │ ├── tpu_system_interface.cc │ │ ├── tpu_system_interface.h │ │ └── xla_spmd │ │ │ ├── layout_to_xla_sharding.cc │ │ │ └── layout_to_xla_sharding.h │ ├── mlir │ │ ├── BUILD │ │ ├── Passes.td │ │ ├── annotate_global_shape.cc │ │ ├── cluster_function_conversion.cc │ │ ├── collectives.cc │ │ ├── collectives.h │ │ ├── collectives_common.cc │ │ ├── collectives_common.h │ │ ├── constant_folding.cc │ │ ├── create_dtensor_mlir_passes.h │ │ ├── dce.cc │ │ ├── designate_resource_handle_mesh.cc │ │ ├── device_mesh_cluster_coarsening.cc │ │ ├── device_utils.cc │ │ ├── device_utils.h │ │ ├── dtensor_allreduce_combine_optimization.cc │ │ ├── dtensor_allreduce_scatter_optimization.cc │ │ ├── dtensor_allreduce_sum_optimization.cc │ │ ├── dtensor_collective_type_lowering.cc │ │ ├── dtensor_dialect │ │ │ ├── BUILD │ │ │ └── ir │ │ │ │ ├── dialect.h │ │ │ │ ├── dtensor_attributes.cc │ │ │ │ ├── dtensor_attributes.h │ │ │ │ ├── dtensor_dialect.td │ │ │ │ ├── dtensor_ops.td │ │ │ │ ├── ops.cc │ │ │ │ └── ops.h │ │ ├── dtensor_layout_to_xla_sharding_op.cc │ │ ├── dtensor_location.cc │ │ ├── dtensor_location.h │ │ ├── dtensor_location_test.cc │ │ ├── dtensor_mixed_precision_reduce.cc │ │ ├── dtensor_mlir_passes.cc │ │ ├── dtensor_mlir_passes.h │ │ ├── dtensor_multi_device_expansion.cc │ │ ├── dtensor_remove_dtensorlayout.cc │ │ ├── dtensor_replace_auxiliary_layout_op.cc │ │ ├── dtensor_replace_relayout_with_identity.cc │ │ ├── dtensor_send_recv.cc │ │ ├── dtensor_send_recv.h │ │ ├── dtensor_set_hlo_sharding.cc │ │ ├── expansions │ │ │ ├── argmax_spmd_expander.cc │ │ │ ├── argmax_spmd_expander.h │ │ │ ├── bias_add_spmd_expander.cc │ │ │ ├── bias_add_spmd_expander.h │ │ │ ├── broadcast_to_spmd_expander.cc │ │ │ ├── broadcast_to_spmd_expander.h │ │ │ ├── concat_spmd_expander.cc │ │ │ ├── concat_spmd_expander.h │ │ │ ├── control_flow_spmd_expander.cc │ │ │ ├── control_flow_spmd_expander.h │ │ │ ├── conv_spmd_expander.cc │ │ │ ├── conv_spmd_expander.h │ │ │ ├── cumsum_spmd_expander.cc │ │ │ ├── cumsum_spmd_expander.h │ │ │ ├── dataparallel_spmd_expander.cc │ │ │ ├── dataparallel_spmd_expander.h │ │ │ ├── disable_copy_on_read_spmd_expander.cc │ │ │ ├── disable_copy_on_read_spmd_expander.h │ │ │ ├── dtensor_op_spmd_expander.cc │ │ │ ├── dtensor_op_spmd_expander.h │ │ │ ├── einsum_spmd_expander.cc │ │ │ ├── einsum_spmd_expander.h │ │ │ ├── elementwise_spmd_expander.cc │ │ │ ├── elementwise_spmd_expander.h │ │ │ ├── expanddims_spmd_expander.cc │ │ │ ├── expanddims_spmd_expander.h │ │ │ ├── fft_spmd_expander.cc │ │ │ ├── fft_spmd_expander.h │ │ │ ├── fill_spmd_expander.cc │ │ │ ├── fill_spmd_expander.h │ │ │ ├── gather_spmd_expander.cc │ │ │ ├── gather_spmd_expander.h │ │ │ ├── identity_n_spmd_expander.cc │ │ │ ├── identity_n_spmd_expander.h │ │ │ ├── in_top_k_spmd_expander.cc │ │ │ ├── in_top_k_spmd_expander.h │ │ │ ├── io_op_spmd_expander.cc │ │ │ ├── io_op_spmd_expander.h │ │ │ ├── iterator_spmd_expander.cc │ │ │ ├── iterator_spmd_expander.h │ │ │ ├── matmul_spmd_expander.cc │ │ │ ├── matmul_spmd_expander.h │ │ │ ├── meta_spmd_expander.cc │ │ │ ├── meta_spmd_expander.h │ │ │ ├── nullary_spmd_expander.cc │ │ │ ├── nullary_spmd_expander.h │ │ │ ├── optional_spmd_expander.cc │ │ │ ├── optional_spmd_expander.h │ │ │ ├── qr_spmd_expander.cc │ │ │ ├── qr_spmd_expander.h │ │ │ ├── random_op_spmd_expander.cc │ │ │ ├── random_op_spmd_expander.h │ │ │ ├── range_spmd_expander.cc │ │ │ ├── range_spmd_expander.h │ │ │ ├── reduce_spmd_expander.cc │ │ │ ├── reduce_spmd_expander.h │ │ │ ├── replicated_spmd_expander.cc │ │ │ ├── replicated_spmd_expander.h │ │ │ ├── resource_spmd_expander.cc │ │ │ ├── resource_spmd_expander.h │ │ │ ├── save_restore_spmd_expander.cc │ │ │ ├── save_restore_spmd_expander.h │ │ │ ├── scatter_spmd_expander.cc │ │ │ ├── scatter_spmd_expander.h │ │ │ ├── segmentation_spmd_expander.cc │ │ │ ├── segmentation_spmd_expander.h │ │ │ ├── slice_spmd_expander.cc │ │ │ ├── slice_spmd_expander.h │ │ │ ├── softmax_spmd_expander.cc │ │ │ ├── softmax_spmd_expander.h │ │ │ ├── sparse_to_dense_spmd_expander.cc │ │ │ ├── sparse_to_dense_spmd_expander.h │ │ │ ├── split_spmd_expander.cc │ │ │ ├── split_spmd_expander.h │ │ │ ├── squeeze_spmd_expander.cc │ │ │ ├── squeeze_spmd_expander.h │ │ │ ├── strided_slice_spmd_expander.cc │ │ │ ├── strided_slice_spmd_expander.h │ │ │ ├── tensorlist_getitem_spmd_expander.cc │ │ │ ├── tensorlist_getitem_spmd_expander.h │ │ │ ├── tensorlist_reserve_spmd_expander.cc │ │ │ ├── tensorlist_reserve_spmd_expander.h │ │ │ ├── tensorlist_setitem_spmd_expander.cc │ │ │ ├── tensorlist_setitem_spmd_expander.h │ │ │ ├── top_k_spmd_expander.cc │ │ │ ├── top_k_spmd_expander.h │ │ │ ├── trivial_spmd_expander.cc │ │ │ ├── trivial_spmd_expander.h │ │ │ ├── unsupported_op_spmd_expander.cc │ │ │ ├── unsupported_op_spmd_expander.h │ │ │ ├── where_spmd_expander.cc │ │ │ └── where_spmd_expander.h │ │ ├── function_renaming.cc │ │ ├── handle_cross_cluster_dependencies.cc │ │ ├── handle_sparsetensors.cc │ │ ├── ir │ │ │ ├── tf_dtensor.cc │ │ │ ├── tf_dtensor.h │ │ │ └── tf_dtensor.td │ │ ├── layout_parsing.cc │ │ ├── layout_parsing.h │ │ ├── layout_propagation_v2.cc │ │ ├── lower_send_recv.cc │ │ ├── merge_clusters.cc │ │ ├── mesh_propagation.cc │ │ ├── move_compilation_to_host.cc │ │ ├── op_to_device_cluster.cc │ │ ├── op_utils.cc │ │ ├── op_utils.h │ │ ├── propagate_default_layout.cc │ │ ├── propagate_device_id_to_function_args.cc │ │ ├── restore_shape_inference.cc │ │ ├── set_default_sharding.cc │ │ ├── shape_utils.cc │ │ ├── shape_utils.h │ │ ├── sparse_expander.cc │ │ ├── sparse_expander.h │ │ ├── sparse_expander_common.cc │ │ ├── sparse_expander_common.h │ │ ├── sparse_expanders.cc │ │ ├── sparse_expansion.cc │ │ ├── sparse_expansions │ │ │ ├── dynamic_enqueue_sparse_expander.cc │ │ │ ├── dynamic_enqueue_sparse_expander.h │ │ │ ├── matmul_sparse_expander.cc │ │ │ └── matmul_sparse_expander.h │ │ ├── spmd_expander.cc │ │ ├── spmd_expander.h │ │ ├── spmd_expander_common.cc │ │ ├── spmd_expander_common.h │ │ ├── spmd_expanders.cc │ │ ├── spmd_expansion.cc │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── annotate_global_shape.mlir │ │ │ ├── cluster_function_conversion.mlir │ │ │ ├── constant_folding.mlir │ │ │ ├── cpu_layout.pbtxt │ │ │ ├── decompose_controlflow.mlir │ │ │ ├── designate_resource_handle_mesh.mlir │ │ │ ├── device_mesh_cluster_coarsening.mlir │ │ │ ├── dtensor_all_gather.mlir │ │ │ ├── dtensor_all_scatter.mlir │ │ │ ├── dtensor_allreduce_combine_optimization.mlir │ │ │ ├── dtensor_allreduce_lowering.mlir │ │ │ ├── dtensor_allreduce_scatter_optimization.mlir │ │ │ ├── dtensor_allreduce_sum_optimization.mlir │ │ │ ├── dtensor_alltoall_lowering.mlir │ │ │ ├── dtensor_collective_type_lowering.mlir │ │ │ ├── dtensor_layout_must_execute.mlir │ │ │ ├── dtensor_layout_to_xla_sharding_op.mlir │ │ │ ├── dtensor_mixed_precision_reduce.mlir │ │ │ ├── dtensor_mlir_opt_main.cc │ │ │ ├── dtensor_reduce_scatter_lowering.mlir │ │ │ ├── dtensor_remove_dtensorlayout.mlir │ │ │ ├── dtensor_replace_auxiliary_layout_op.mlir │ │ │ ├── dtensor_replace_relayout_with_identity.mlir │ │ │ ├── dtensor_set_hlo_sharding.mlir │ │ │ ├── dtensor_set_hlo_sharding_default.mlir │ │ │ ├── dtensor_xla_spmd_integration.mlir │ │ │ ├── elide_identity_before_copy_to_mesh.mlir │ │ │ ├── function_renaming.mlir │ │ │ ├── handle_cross_cluster_dependencies.mlir │ │ │ ├── handle_sparsetensors.mlir │ │ │ ├── layout_propagation.mlir │ │ │ ├── layout_propagation_v2.mlir │ │ │ ├── lower_send_recv.mlir │ │ │ ├── merge_clusters.mlir │ │ │ ├── mesh_propagation.mlir │ │ │ ├── move_compilation_to_host.mlir │ │ │ ├── multi_device_expansion.mlir │ │ │ ├── op_to_device_cluster.mlir │ │ │ ├── propagate_default_layout.mlir │ │ │ ├── propagate_device_id_to_function.mlir │ │ │ ├── restore_and_assign.mlir │ │ │ ├── restore_shape_inference.mlir │ │ │ ├── set_default_sharding.mlir │ │ │ ├── sparse_expansion.mlir │ │ │ ├── spmd_batchparallel.mlir │ │ │ ├── spmd_concat.mlir │ │ │ ├── spmd_conv.mlir │ │ │ ├── spmd_dtensor_ops.mlir │ │ │ ├── spmd_einsum.mlir │ │ │ ├── spmd_expansion.mlir │ │ │ ├── spmd_fft.mlir │ │ │ ├── spmd_fill.mlir │ │ │ ├── spmd_io_ops.mlir │ │ │ ├── spmd_iterator.mlir │ │ │ ├── spmd_matmul.mlir │ │ │ ├── spmd_metadata.mlir │ │ │ ├── spmd_random.mlir │ │ │ ├── spmd_reduction.mlir │ │ │ ├── spmd_save_restore.mlir │ │ │ ├── spmd_segment_sum.mlir │ │ │ ├── spmd_slice.mlir │ │ │ ├── spmd_softmax_loss.mlir │ │ │ ├── spmd_squeeze.mlir │ │ │ ├── spmd_tile.mlir │ │ │ ├── spmd_var_handle.mlir │ │ │ ├── tf_dtensor_ops.mlir │ │ │ ├── tpu_add_resource_device_attribute.mlir │ │ │ ├── tpu_integration.mlir │ │ │ ├── undo_merge_const_across_mesh.mlir │ │ │ └── update_tpu_metadata.mlir │ │ ├── topological_iterator.cc │ │ ├── topological_iterator.h │ │ ├── tpu_add_resource_device_attribute.cc │ │ ├── tpu_integration.cc │ │ ├── undo_merge_const_across_mesh.cc │ │ ├── utils │ │ │ ├── BUILD │ │ │ ├── collective_lowering.cc │ │ │ ├── dtensor_mlir_passes_internal.cc │ │ │ ├── dtensor_mlir_passes_internal.h │ │ │ └── update_tpu_metadata.cc │ │ ├── value_utils.cc │ │ └── value_utils.h │ ├── proto │ │ ├── BUILD │ │ └── layout.proto │ ├── python │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── accelerator_util.py │ │ ├── api.py │ │ ├── config.py │ │ ├── d_checkpoint.py │ │ ├── d_random.py │ │ ├── d_variable.py │ │ ├── dtensor_device.py │ │ ├── heartbeat.py │ │ ├── input_util.py │ │ ├── layout.py │ │ ├── mesh_util.py │ │ ├── numpy_util.py │ │ ├── save_restore.py │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── api_test.py │ │ │ ├── array_ops_test.py │ │ │ ├── batchparallel_spmd_test.py │ │ │ ├── cache_test.py │ │ │ ├── collective_test.py │ │ │ ├── config_test.py │ │ │ ├── conv_test.py │ │ │ ├── device_test.py │ │ │ ├── input_util_test.py │ │ │ ├── layout_propagation_test.py │ │ │ ├── layout_test.py │ │ │ ├── mesh_util_test.py │ │ │ ├── mnist_test.py │ │ │ ├── multi_client_input_util_test.py │ │ │ ├── multi_client_test.py │ │ │ ├── multi_client_test_util.py │ │ │ ├── multi_mesh_test.py │ │ │ ├── numerics_test.py │ │ │ ├── numpy_util_test.py │ │ │ ├── rng_test.py │ │ │ ├── save_restore_v2_test.py │ │ │ ├── sparse_test.py │ │ │ ├── spmd_test.py │ │ │ ├── test_backend_name.py │ │ │ ├── test_backend_util.py │ │ │ ├── test_util.py │ │ │ ├── test_util_ops.py │ │ │ ├── tpu_device_assignment_test.py │ │ │ └── variable_test.py │ │ └── tpu_util.py │ └── tests │ │ ├── BUILD │ │ ├── dtensor_operation_test.cc │ │ ├── executable_manager_test.cc │ │ ├── layout_to_xla_sharding_test.cc │ │ ├── slice_util_test.cc │ │ ├── spmd_expander_test.cc │ │ └── tensor_layout_test.cc ├── examples │ ├── README.md │ ├── __init__.py │ ├── adding_an_op │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── attr_examples.cc │ │ ├── cuda_op.py │ │ ├── cuda_op_kernel.cc │ │ ├── cuda_op_kernel.cu.cc │ │ ├── cuda_op_test.py │ │ ├── fact_test.py │ │ ├── zero_out_1_test.py │ │ ├── zero_out_2_test.py │ │ ├── zero_out_3_test.py │ │ ├── zero_out_grad_2.py │ │ ├── zero_out_op_1.py │ │ ├── zero_out_op_2.py │ │ ├── zero_out_op_3.py │ │ ├── zero_out_op_kernel_1.cc │ │ ├── zero_out_op_kernel_2.cc │ │ └── zero_out_op_kernel_3.cc │ ├── android │ │ └── .gitignore │ ├── custom_ops_doc │ │ ├── README.md │ │ ├── multiplex_1 │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── multiplex_1_kernel.cc │ │ │ ├── multiplex_1_op.cc │ │ │ ├── multiplex_1_op.py │ │ │ └── multiplex_1_test.py │ │ ├── multiplex_2 │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── multiplex_2_kernel.cc │ │ │ ├── multiplex_2_kernel.cu.cc │ │ │ ├── multiplex_2_kernel.h │ │ │ ├── multiplex_2_op.cc │ │ │ ├── multiplex_2_op.py │ │ │ └── multiplex_2_test.py │ │ ├── multiplex_3 │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── multiplex_3_kernel.cc │ │ │ ├── multiplex_3_op.cc │ │ │ ├── multiplex_3_op.py │ │ │ └── multiplex_3_test.py │ │ ├── multiplex_4 │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── model_using_multiplex.py │ │ │ ├── multiplex_2_save.py │ │ │ ├── multiplex_4_kernel.cc │ │ │ ├── multiplex_4_load_use.py │ │ │ ├── multiplex_4_op.cc │ │ │ ├── multiplex_4_op.py │ │ │ └── multiplex_4_test.py │ │ ├── simple_hash_table │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── simple_hash_table.py │ │ │ ├── simple_hash_table_kernel.cc │ │ │ ├── simple_hash_table_op.cc │ │ │ ├── simple_hash_table_op.py │ │ │ └── simple_hash_table_test.py │ │ └── sleep │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── sleep_bin.py │ │ │ ├── sleep_kernel.cc │ │ │ ├── sleep_op.cc │ │ │ ├── sleep_op.py │ │ │ └── sleep_test.py │ ├── image_retraining │ │ └── README.md │ ├── label_image │ │ ├── BUILD │ │ ├── README.md │ │ ├── data │ │ │ └── grace_hopper.jpg │ │ ├── label_image.py │ │ └── main.cc │ ├── multibox_detector │ │ ├── BUILD │ │ ├── README.md │ │ ├── data │ │ │ └── surfers.jpg │ │ └── main.cc │ ├── speech_commands │ │ ├── BUILD │ │ ├── README.md │ │ ├── accuracy_utils.cc │ │ ├── accuracy_utils.h │ │ ├── accuracy_utils.py │ │ ├── accuracy_utils_test.cc │ │ ├── freeze.py │ │ ├── freeze_test.py │ │ ├── generate_streaming_test_wav.py │ │ ├── generate_streaming_test_wav_test.py │ │ ├── input_data.py │ │ ├── input_data_test.py │ │ ├── label_wav.cc │ │ ├── label_wav.py │ │ ├── label_wav_dir.py │ │ ├── label_wav_test.py │ │ ├── models.py │ │ ├── models_test.py │ │ ├── recognize_commands.cc │ │ ├── recognize_commands.h │ │ ├── recognize_commands.py │ │ ├── recognize_commands_test.cc │ │ ├── test_streaming_accuracy.cc │ │ ├── test_streaming_accuracy.py │ │ ├── train.py │ │ ├── train_test.py │ │ ├── wav_to_features.py │ │ └── wav_to_features_test.py │ ├── udacity │ │ ├── .gitignore │ │ └── README.md │ └── wav_to_spectrogram │ │ ├── BUILD │ │ ├── README.md │ │ ├── main.cc │ │ ├── wav_to_spectrogram.cc │ │ ├── wav_to_spectrogram.h │ │ └── wav_to_spectrogram_test.cc ├── go │ ├── BUILD │ ├── README.md │ ├── android.go │ ├── attrs.go │ ├── attrs_test.go │ ├── context.go │ ├── context_test.go │ ├── core │ │ ├── framework │ │ │ └── BUILD │ │ └── protobuf │ │ │ └── BUILD │ ├── doc.go │ ├── example_inception_inference_test.go │ ├── genop │ │ ├── .gitignore │ │ ├── BUILD │ │ ├── generate.go │ │ ├── generate.sh │ │ ├── internal │ │ │ ├── BUILD │ │ │ ├── api_def_map.go │ │ │ ├── genop.go │ │ │ ├── genop_test.go │ │ │ └── lib.go │ │ └── main.go │ ├── graph.go │ ├── graph_test.go │ ├── lib.go │ ├── op │ │ ├── BUILD │ │ ├── generate.go │ │ ├── gradients.go │ │ ├── gradients_test.go │ │ ├── op.go │ │ ├── op_test.go │ │ ├── scope.go │ │ ├── scope_test.go │ │ └── wrappers.go │ ├── operation.go │ ├── operation_test.go │ ├── saved_model.go │ ├── saved_model_test.go │ ├── session.go │ ├── session_test.go │ ├── shape.go │ ├── shape_test.go │ ├── signature.go │ ├── signature_test.go │ ├── status.go │ ├── stream_executor │ │ └── BUILD │ ├── tensor.go │ ├── tensor_handle.go │ ├── tensor_handle_test.go │ ├── tensor_test.go │ ├── test.sh │ ├── testdata │ │ ├── label_image │ │ │ └── grace_hopper.jpg │ │ └── saved_model │ │ │ └── half_plus_two │ │ │ └── 00000123 │ │ │ ├── assets │ │ │ └── foo.txt │ │ │ ├── saved_model.pb │ │ │ └── variables │ │ │ ├── variables.data-00000-of-00001 │ │ │ └── variables.index │ ├── tsl │ │ ├── profiler │ │ │ └── protobuf │ │ │ │ └── BUILD │ │ └── protobuf │ │ │ └── BUILD │ ├── util_test.go │ └── version.go ├── java │ ├── BUILD │ ├── LEGACY.md │ ├── README.md │ ├── build_defs.bzl │ ├── config │ │ ├── exported_symbols.lds │ │ └── version_script.lds │ ├── generate_pom.cc │ └── src │ │ ├── gen │ │ ├── cc │ │ │ ├── java_defs.h │ │ │ ├── op_gen_main.cc │ │ │ ├── op_generator.cc │ │ │ ├── op_generator.h │ │ │ ├── op_specs.cc │ │ │ ├── op_specs.h │ │ │ ├── source_writer.cc │ │ │ ├── source_writer.h │ │ │ └── source_writer_test.cc │ │ ├── gen_ops.bzl │ │ ├── java │ │ │ └── org │ │ │ │ └── tensorflow │ │ │ │ └── processor │ │ │ │ └── OperatorProcessor.java │ │ ├── perl │ │ │ ├── tftypes-runall.pl │ │ │ └── tftypes.pl │ │ └── resources │ │ │ ├── META-INF │ │ │ └── services │ │ │ │ └── javax.annotation.processing.Processor │ │ │ ├── Tensors.java.tmpl │ │ │ ├── test.java.snippet │ │ │ └── tftypes.csv │ │ ├── main │ │ ├── java │ │ │ └── org │ │ │ │ └── tensorflow │ │ │ │ ├── AbstractOperation.java │ │ │ │ ├── DataType.java │ │ │ │ ├── EagerOperation.java │ │ │ │ ├── EagerOperationBuilder.java │ │ │ │ ├── EagerSession.java │ │ │ │ ├── ExecutionEnvironment.java │ │ │ │ ├── Graph.java │ │ │ │ ├── GraphOperation.java │ │ │ │ ├── GraphOperationBuilder.java │ │ │ │ ├── NativeLibrary.java │ │ │ │ ├── Operand.java │ │ │ │ ├── Operation.java │ │ │ │ ├── OperationBuilder.java │ │ │ │ ├── Output.java │ │ │ │ ├── SavedModelBundle.java │ │ │ │ ├── Server.java │ │ │ │ ├── Session.java │ │ │ │ ├── Shape.java │ │ │ │ ├── Tensor.java │ │ │ │ ├── TensorFlow.java │ │ │ │ ├── TensorFlowException.java │ │ │ │ ├── Tensors.java │ │ │ │ ├── examples │ │ │ │ ├── .gitignore │ │ │ │ ├── BUILD │ │ │ │ └── LabelImage.java │ │ │ │ ├── op │ │ │ │ ├── NameScope.java │ │ │ │ ├── Op.java │ │ │ │ ├── Operands.java │ │ │ │ ├── PrimitiveOp.java │ │ │ │ ├── Scope.java │ │ │ │ ├── annotation │ │ │ │ │ └── Operator.java │ │ │ │ └── core │ │ │ │ │ ├── Constant.java │ │ │ │ │ ├── Gradients.java │ │ │ │ │ └── Zeros.java │ │ │ │ ├── package-info.java │ │ │ │ └── types │ │ │ │ ├── UInt8.java │ │ │ │ └── package-info.java │ │ └── native │ │ │ ├── BUILD │ │ │ ├── eager_operation_builder_jni.cc │ │ │ ├── eager_operation_builder_jni.h │ │ │ ├── eager_operation_jni.cc │ │ │ ├── eager_operation_jni.h │ │ │ ├── eager_session_jni.cc │ │ │ ├── eager_session_jni.h │ │ │ ├── exception_jni.cc │ │ │ ├── exception_jni.h │ │ │ ├── graph_jni.cc │ │ │ ├── graph_jni.h │ │ │ ├── graph_operation_builder_jni.cc │ │ │ ├── graph_operation_builder_jni.h │ │ │ ├── graph_operation_jni.cc │ │ │ ├── graph_operation_jni.h │ │ │ ├── saved_model_bundle_jni.cc │ │ │ ├── saved_model_bundle_jni.h │ │ │ ├── server_jni.cc │ │ │ ├── server_jni.h │ │ │ ├── session_jni.cc │ │ │ ├── session_jni.h │ │ │ ├── tensor_jni.cc │ │ │ ├── tensor_jni.h │ │ │ ├── tensorflow_jni.cc │ │ │ ├── tensorflow_jni.h │ │ │ ├── utils_jni.cc │ │ │ └── utils_jni.h │ │ └── test │ │ ├── java │ │ └── org │ │ │ └── tensorflow │ │ │ ├── EagerOperationBuilderTest.java │ │ │ ├── EagerOperationTest.java │ │ │ ├── EagerSessionTest.java │ │ │ ├── GraphOperationBuilderTest.java │ │ │ ├── GraphOperationTest.java │ │ │ ├── GraphTest.java │ │ │ ├── SavedModelBundleTest.java │ │ │ ├── SessionTest.java │ │ │ ├── ShapeTest.java │ │ │ ├── TensorFlowTest.java │ │ │ ├── TensorTest.java │ │ │ ├── TestUtil.java │ │ │ ├── op │ │ │ ├── OperandsTest.java │ │ │ ├── PrimitiveOpTest.java │ │ │ ├── ScopeTest.java │ │ │ └── core │ │ │ │ ├── ConstantTest.java │ │ │ │ ├── GeneratedOperationsTest.java │ │ │ │ ├── GradientsTest.java │ │ │ │ └── ZerosTest.java │ │ │ └── processor │ │ │ └── OperatorProcessorTest.java │ │ ├── native │ │ └── my_test_op.cc │ │ └── resources │ │ └── org │ │ └── tensorflow │ │ └── processor │ │ └── operator │ │ ├── bad │ │ └── BasicBad.java │ │ └── good │ │ └── BasicGood.java ├── js │ ├── BUILD │ └── ops │ │ ├── ts_op_gen.cc │ │ ├── ts_op_gen.h │ │ └── ts_op_gen_test.cc ├── lite │ ├── BUILD │ ├── CMakeLists.txt │ ├── README.md │ ├── acceleration │ │ └── configuration │ │ │ ├── BUILD │ │ │ ├── build_defs.bzl │ │ │ ├── c │ │ │ ├── BUILD │ │ │ ├── delegate_plugin.h │ │ │ ├── gpu_plugin.h │ │ │ ├── nnapi_plugin.h │ │ │ ├── stable_delegate.h │ │ │ └── xnnpack_plugin.h │ │ │ ├── configuration.proto │ │ │ ├── configuration_generated.h │ │ │ ├── coreml_plugin.cc │ │ │ ├── delegate_plugin_converter.cc │ │ │ ├── delegate_plugin_converter.h │ │ │ ├── delegate_registry.h │ │ │ ├── flatbuffer_to_proto.cc │ │ │ ├── flatbuffer_to_proto.h │ │ │ ├── flatbuffer_to_proto_test.cc │ │ │ ├── gpu_plugin.cc │ │ │ ├── gpu_plugin.h │ │ │ ├── gpu_plugin_test.cc │ │ │ ├── hexagon_plugin.cc │ │ │ ├── prev_is_different_than_current_test.sh │ │ │ ├── proto_to_flatbuffer.cc │ │ │ ├── proto_to_flatbuffer.h │ │ │ ├── proto_to_flatbuffer_test.cc │ │ │ ├── stable_delegate_plugin.cc │ │ │ ├── stable_delegate_plugin.h │ │ │ ├── stable_delegate_plugin_test.cc │ │ │ ├── testdata │ │ │ ├── configuration.old.fbs │ │ │ └── configuration.proto_prev │ │ │ └── xnnpack_plugin_test.cc │ ├── allocation.cc │ ├── allocation.h │ ├── allocation_test.cc │ ├── arena_planner.cc │ ├── arena_planner.h │ ├── arena_planner_subgraph_test.cc │ ├── arena_planner_test.cc │ ├── array.cc │ ├── array.h │ ├── array_test.cc │ ├── async │ │ ├── BUILD │ │ ├── backend_async_kernel_interface.cc │ │ ├── backend_async_kernel_interface.h │ │ ├── backend_async_kernel_interface_test.cc │ │ ├── c │ │ │ ├── BUILD │ │ │ ├── async_kernel.h │ │ │ ├── async_signature_runner.h │ │ │ ├── task.h │ │ │ └── types.h │ │ ├── interop │ │ │ └── c │ │ │ │ ├── BUILD │ │ │ │ ├── attribute_map.h │ │ │ │ ├── constants.h │ │ │ │ └── types.h │ │ └── testing │ │ │ ├── BUILD │ │ │ └── mock_async_kernel.h │ ├── build_def.bzl │ ├── builtin_op_data.h │ ├── builtin_ops.h │ ├── c │ │ ├── BUILD │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── builtin_op_data.h │ │ ├── c_api.h │ │ ├── c_api_experimental.h │ │ ├── c_api_for_testing.cc │ │ ├── c_api_for_testing.h │ │ ├── c_api_internal.h │ │ ├── c_api_opaque.h │ │ ├── c_api_opaque_internal.cc │ │ ├── c_api_opaque_internal.h │ │ ├── c_api_opaque_internal_test.cc │ │ ├── c_api_signature_runner_test.cc │ │ ├── c_api_types.h │ │ ├── c_test.c │ │ ├── common.h │ │ ├── common_internal.cc │ │ ├── common_internal.h │ │ ├── exported_symbols.lds │ │ ├── jni │ │ │ ├── BUILD │ │ │ ├── jni_utils.cc │ │ │ └── jni_utils.h │ │ ├── test_util.cc │ │ ├── test_util.h │ │ └── version_script.lds │ ├── cmake │ │ └── DownloadPThreadPool.cmake │ ├── context.h │ ├── context_util.h │ ├── core │ │ ├── BUILD │ │ ├── README.md │ │ ├── acceleration │ │ │ └── configuration │ │ │ │ ├── BUILD │ │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ ├── delegate_plugin.h │ │ │ │ ├── gpu_plugin.cc │ │ │ │ ├── gpu_plugin.h │ │ │ │ ├── gpu_plugin_test.cc │ │ │ │ ├── nnapi_plugin.cc │ │ │ │ ├── nnapi_plugin.h │ │ │ │ ├── nnapi_plugin_test.cc │ │ │ │ ├── special_rules.bzl │ │ │ │ ├── stable_delegate.h │ │ │ │ ├── xnnpack_plugin.cc │ │ │ │ ├── xnnpack_plugin.h │ │ │ │ ├── xnnpack_plugin_c_test.cc │ │ │ │ ├── xnnpack_plugin_c_test_lib.c │ │ │ │ ├── xnnpack_plugin_c_test_lib.h │ │ │ │ └── xnnpack_plugin_test.cc │ │ │ │ ├── delegate_registry.cc │ │ │ │ ├── delegate_registry.h │ │ │ │ ├── nnapi_plugin.cc │ │ │ │ ├── nnapi_plugin.h │ │ │ │ ├── nnapi_plugin_test.cc │ │ │ │ ├── stable_delegate_registry.cc │ │ │ │ ├── stable_delegate_registry.h │ │ │ │ ├── stable_delegate_registry_test.cc │ │ │ │ ├── xnnpack_plugin.cc │ │ │ │ └── xnnpack_plugin_test.cc │ │ ├── api │ │ │ ├── BUILD │ │ │ ├── error_reporter.cc │ │ │ ├── error_reporter.h │ │ │ ├── error_reporter_test.cc │ │ │ ├── flatbuffer_conversions.cc │ │ │ ├── flatbuffer_conversions.h │ │ │ ├── flatbuffer_conversions_test.cc │ │ │ ├── op_resolver.cc │ │ │ ├── op_resolver.h │ │ │ ├── op_resolver_internal.h │ │ │ ├── op_resolver_internal_test.cc │ │ │ ├── op_resolver_test.cc │ │ │ ├── profiler.h │ │ │ ├── tensor_utils.cc │ │ │ ├── tensor_utils.h │ │ │ └── verifier.h │ │ ├── async │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── async_kernel_internal.h │ │ │ ├── async_signature_runner.cc │ │ │ ├── async_signature_runner.h │ │ │ ├── async_signature_runner_test.cc │ │ │ ├── async_subgraph.cc │ │ │ ├── async_subgraph.h │ │ │ ├── async_subgraph_test.cc │ │ │ ├── backend_async_kernel_interface.h │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ ├── async_kernel.cc │ │ │ │ ├── async_kernel.h │ │ │ │ ├── async_signature_runner.cc │ │ │ │ ├── async_signature_runner.h │ │ │ │ ├── async_signature_runner_test.cc │ │ │ │ ├── internal.h │ │ │ │ ├── task.cc │ │ │ │ ├── task.h │ │ │ │ ├── task_test.cc │ │ │ │ └── types.h │ │ │ ├── interop │ │ │ │ ├── BUILD │ │ │ │ ├── attribute_map_internal.cc │ │ │ │ ├── attribute_map_internal.h │ │ │ │ ├── attribute_map_internal_test.cc │ │ │ │ ├── c │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── attribute_map.cc │ │ │ │ │ ├── attribute_map.h │ │ │ │ │ ├── attribute_map_test.cc │ │ │ │ │ ├── constants.cc │ │ │ │ │ ├── constants.h │ │ │ │ │ ├── types.cc │ │ │ │ │ ├── types.h │ │ │ │ │ └── types_test.cc │ │ │ │ ├── reconcile_fns.cc │ │ │ │ ├── reconcile_fns.h │ │ │ │ ├── reconcile_fns_test.cc │ │ │ │ ├── variant.cc │ │ │ │ ├── variant.h │ │ │ │ └── variant_test.cc │ │ │ ├── task_internal.cc │ │ │ ├── task_internal.h │ │ │ ├── task_internal_test.cc │ │ │ └── testing │ │ │ │ ├── BUILD │ │ │ │ ├── mock_async_kernel.h │ │ │ │ ├── test_backend.cc │ │ │ │ └── test_backend.h │ │ ├── c │ │ │ ├── BUILD │ │ │ ├── builtin_op_data.h │ │ │ ├── builtin_op_data_test.cc │ │ │ ├── c_api.cc │ │ │ ├── c_api.h │ │ │ ├── c_api_experimental.cc │ │ │ ├── c_api_experimental.h │ │ │ ├── c_api_experimental_test.cc │ │ │ ├── c_api_opaque.cc │ │ │ ├── c_api_opaque.h │ │ │ ├── c_api_opaque_test.cc │ │ │ ├── c_api_test.cc │ │ │ ├── c_api_types.h │ │ │ ├── common.cc │ │ │ ├── common.h │ │ │ ├── common_test.cc │ │ │ ├── registration_external.cc │ │ │ ├── registration_external.h │ │ │ └── special_rules.bzl │ │ ├── create_op_resolver.h │ │ ├── create_op_resolver_with_builtin_ops.cc │ │ ├── experimental │ │ │ └── acceleration │ │ │ │ ├── configuration │ │ │ │ ├── BUILD │ │ │ │ ├── c │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── delegate_plugin.h │ │ │ │ │ ├── gpu_plugin.h │ │ │ │ │ ├── nnapi_plugin.h │ │ │ │ │ ├── special_rules.bzl │ │ │ │ │ ├── stable_delegate.h │ │ │ │ │ └── xnnpack_plugin.h │ │ │ │ ├── delegate_registry.h │ │ │ │ └── stable_delegate_registry.h │ │ │ │ └── mini_benchmark │ │ │ │ └── c │ │ │ │ ├── BUILD │ │ │ │ ├── c_api.cc │ │ │ │ ├── c_api.h │ │ │ │ ├── c_api_test.cc │ │ │ │ └── c_api_types.h │ │ ├── interpreter.cc │ │ ├── interpreter.h │ │ ├── interpreter_builder.cc │ │ ├── interpreter_builder.h │ │ ├── interpreter_experimental.cc │ │ ├── kernels │ │ │ ├── BUILD │ │ │ ├── builtin_op_kernels.h │ │ │ ├── register.cc │ │ │ ├── register.h │ │ │ └── register_test.cc │ │ ├── macros.h │ │ ├── model.h │ │ ├── model_builder.cc │ │ ├── model_builder.h │ │ ├── model_test.cc │ │ ├── shims │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── build_defs.bzl │ │ │ ├── cc_library_with_tflite.bzl │ │ │ └── cc_library_with_tflite_test.bzl │ │ ├── signature_runner.cc │ │ ├── signature_runner.h │ │ ├── signature_runner_test.cc │ │ ├── special_rules.bzl │ │ ├── subgraph.cc │ │ ├── subgraph.h │ │ ├── subgraph_test.cc │ │ └── tools │ │ │ ├── BUILD │ │ │ ├── verifier.cc │ │ │ ├── verifier.h │ │ │ ├── verifier_internal.cc │ │ │ ├── verifier_internal.h │ │ │ ├── verifier_internal_test.cc │ │ │ └── verifier_test.cc │ ├── create_op_resolver.h │ ├── create_op_resolver_with_selected_ops.cc │ ├── delegates │ │ ├── BUILD │ │ ├── coreml │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── builders │ │ │ │ ├── BUILD │ │ │ │ ├── activation_layer_builder.cc │ │ │ │ ├── activation_layer_builder.h │ │ │ │ ├── add_op_builder.cc │ │ │ │ ├── add_op_builder.h │ │ │ │ ├── concatenation_op_builder.cc │ │ │ │ ├── concatenation_op_builder.h │ │ │ │ ├── convolution_op_builder.cc │ │ │ │ ├── convolution_op_builder.h │ │ │ │ ├── dummy_op_builder.cc │ │ │ │ ├── dummy_op_builder.h │ │ │ │ ├── fully_connected_op_builder.cc │ │ │ │ ├── fully_connected_op_builder.h │ │ │ │ ├── hardswish_op_builder.cc │ │ │ │ ├── hardswish_op_builder.h │ │ │ │ ├── mul_op_builder.cc │ │ │ │ ├── mul_op_builder.h │ │ │ │ ├── op_builder.cc │ │ │ │ ├── op_builder.h │ │ │ │ ├── op_factory.h │ │ │ │ ├── op_validator.h │ │ │ │ ├── pad_op_builder.cc │ │ │ │ ├── pad_op_builder.h │ │ │ │ ├── pooling_layer_builder.cc │ │ │ │ ├── pooling_layer_builder.h │ │ │ │ ├── reshape_op_builder.cc │ │ │ │ ├── reshape_op_builder.h │ │ │ │ ├── resize_bilinear_op_builder.cc │ │ │ │ ├── resize_bilinear_op_builder.h │ │ │ │ ├── softmax_op_builder.cc │ │ │ │ ├── softmax_op_builder.h │ │ │ │ ├── test_util.h │ │ │ │ ├── test_util.mm │ │ │ │ ├── threshold_layer_builder.cc │ │ │ │ ├── threshold_layer_builder.h │ │ │ │ ├── util.cc │ │ │ │ ├── util.h │ │ │ │ └── util_test.cc │ │ │ ├── coreml_delegate.h │ │ │ ├── coreml_delegate.mm │ │ │ ├── coreml_delegate_kernel.h │ │ │ ├── coreml_delegate_kernel.mm │ │ │ ├── coreml_executor.h │ │ │ └── coreml_executor.mm │ │ ├── delegate_test.cc │ │ ├── delegate_test_util.cc │ │ ├── delegate_test_util.h │ │ ├── external │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── external_delegate.cc │ │ │ ├── external_delegate.h │ │ │ └── external_delegate_interface.h │ │ ├── flex │ │ │ ├── BUILD │ │ │ ├── allowlisted_flex_ops.cc │ │ │ ├── allowlisted_flex_ops.h │ │ │ ├── allowlisted_flex_ops_internal.h │ │ │ ├── allowlisted_flex_ops_test.cc │ │ │ ├── buffer_map.cc │ │ │ ├── buffer_map.h │ │ │ ├── buffer_map_test.cc │ │ │ ├── buffer_map_util.cc │ │ │ ├── buffer_map_util.h │ │ │ ├── build_def.bzl │ │ │ ├── delegate.cc │ │ │ ├── delegate.h │ │ │ ├── delegate_data.cc │ │ │ ├── delegate_data.h │ │ │ ├── delegate_data_test.cc │ │ │ ├── delegate_symbol.cc │ │ │ ├── delegate_test.cc │ │ │ ├── exported_symbols.lds │ │ │ ├── java │ │ │ │ └── src │ │ │ │ │ └── main │ │ │ │ │ ├── java │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ └── flex │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ └── FlexDelegate.java │ │ │ │ │ └── native │ │ │ │ │ ├── BUILD │ │ │ │ │ └── flex_delegate_jni.cc │ │ │ ├── kernel.cc │ │ │ ├── kernel.h │ │ │ ├── kernel_test.cc │ │ │ ├── subgraph_resource.h │ │ │ ├── test │ │ │ │ └── BUILD │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ ├── tflite_subgraph_execute.cc │ │ │ ├── util.cc │ │ │ ├── util.h │ │ │ ├── util_test.cc │ │ │ └── version_script.lds │ │ ├── gpu │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── android_hardware_buffer.cc │ │ │ ├── android_hardware_buffer.h │ │ │ ├── android_hardware_buffer_test.cc │ │ │ ├── android_version.cc │ │ │ ├── api.cc │ │ │ ├── api.h │ │ │ ├── async_buffers.cc │ │ │ ├── async_buffers.h │ │ │ ├── async_buffers_test.cc │ │ │ ├── build_defs.bzl │ │ │ ├── cl │ │ │ │ ├── BUILD │ │ │ │ ├── api.cc │ │ │ │ ├── api.h │ │ │ │ ├── buffer.cc │ │ │ │ ├── buffer.h │ │ │ │ ├── buffer_test.cc │ │ │ │ ├── cl_arguments.cc │ │ │ │ ├── cl_arguments.h │ │ │ │ ├── cl_arguments_test.cc │ │ │ │ ├── cl_command_queue.cc │ │ │ │ ├── cl_command_queue.h │ │ │ │ ├── cl_context.cc │ │ │ │ ├── cl_context.h │ │ │ │ ├── cl_device.cc │ │ │ │ ├── cl_device.h │ │ │ │ ├── cl_device_test.cc │ │ │ │ ├── cl_errors.h │ │ │ │ ├── cl_event.cc │ │ │ │ ├── cl_event.h │ │ │ │ ├── cl_image_format.cc │ │ │ │ ├── cl_image_format.h │ │ │ │ ├── cl_kernel.cc │ │ │ │ ├── cl_kernel.h │ │ │ │ ├── cl_memory.cc │ │ │ │ ├── cl_memory.h │ │ │ │ ├── cl_operation.cc │ │ │ │ ├── cl_operation.h │ │ │ │ ├── cl_program.cc │ │ │ │ ├── cl_program.h │ │ │ │ ├── cl_test.h │ │ │ │ ├── compiled_program_cache.fbs │ │ │ │ ├── compiled_program_cache_generated.h │ │ │ │ ├── default │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── qcom_wrapper.h │ │ │ │ │ ├── recordable_queue.cc │ │ │ │ │ └── util.cc │ │ │ │ ├── egl_sync.cc │ │ │ │ ├── egl_sync.h │ │ │ │ ├── environment.cc │ │ │ │ ├── environment.h │ │ │ │ ├── gl_interop.cc │ │ │ │ ├── gl_interop.h │ │ │ │ ├── gpu_api_delegate.cc │ │ │ │ ├── gpu_api_delegate.h │ │ │ │ ├── gpu_object.h │ │ │ │ ├── inference_context.cc │ │ │ │ ├── inference_context.h │ │ │ │ ├── kernels │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add_test.cc │ │ │ │ │ ├── cast_test.cc │ │ │ │ │ ├── cl_test.cc │ │ │ │ │ ├── cl_test.h │ │ │ │ │ ├── concat_test.cc │ │ │ │ │ ├── conv_constants_test.cc │ │ │ │ │ ├── conv_generic_test.cc │ │ │ │ │ ├── conv_weights_converter_test.cc │ │ │ │ │ ├── converter.cc │ │ │ │ │ ├── converter.h │ │ │ │ │ ├── convolution_transposed_3x3_test.cc │ │ │ │ │ ├── convolution_transposed_3x3_thin_test.cc │ │ │ │ │ ├── convolution_transposed_4x4_test.cc │ │ │ │ │ ├── convolution_transposed_test.cc │ │ │ │ │ ├── convolution_transposed_thin_test.cc │ │ │ │ │ ├── cumsum_test.cc │ │ │ │ │ ├── depthwise_conv_3x3_test.cc │ │ │ │ │ ├── depthwise_conv_test.cc │ │ │ │ │ ├── elementwise_test.cc │ │ │ │ │ ├── fully_connected_test.cc │ │ │ │ │ ├── gather_test.cc │ │ │ │ │ ├── lstm_full_test.cc │ │ │ │ │ ├── lstm_test.cc │ │ │ │ │ ├── max_unpooling_test.cc │ │ │ │ │ ├── mean_stddev_normalization_test.cc │ │ │ │ │ ├── one_hot_test.cc │ │ │ │ │ ├── padding_test.cc │ │ │ │ │ ├── pooling_test.cc │ │ │ │ │ ├── prelu_test.cc │ │ │ │ │ ├── quantize_and_dequantize_test.cc │ │ │ │ │ ├── reduce_test.cc │ │ │ │ │ ├── relu_test.cc │ │ │ │ │ ├── resampler_test.cc │ │ │ │ │ ├── reshape_test.cc │ │ │ │ │ ├── reshapex4_test.cc │ │ │ │ │ ├── resize_test.cc │ │ │ │ │ ├── select_v2_test.cc │ │ │ │ │ ├── softmax1x1_test.cc │ │ │ │ │ ├── softmax_test.cc │ │ │ │ │ ├── space_to_depth_test.cc │ │ │ │ │ ├── split_test.cc │ │ │ │ │ ├── strided_slice_test.cc │ │ │ │ │ ├── tile_test.cc │ │ │ │ │ ├── transpose_test.cc │ │ │ │ │ └── winograd_test.cc │ │ │ │ ├── opencl_wrapper.cc │ │ │ │ ├── opencl_wrapper.h │ │ │ │ ├── program_cache.cc │ │ │ │ ├── program_cache.h │ │ │ │ ├── qcom_thin_filter.cc │ │ │ │ ├── qcom_thin_filter.h │ │ │ │ ├── recordable_queue.h │ │ │ │ ├── recordable_queue_builder.h │ │ │ │ ├── run_tests.sh │ │ │ │ ├── serialization.fbs │ │ │ │ ├── serialization_generated.h │ │ │ │ ├── tensor.cc │ │ │ │ ├── tensor.h │ │ │ │ ├── tensor_test.cc │ │ │ │ ├── tensor_type_util.cc │ │ │ │ ├── tensor_type_util.h │ │ │ │ ├── testing │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── delegate_testing.cc │ │ │ │ │ ├── gpu_model_test.cc │ │ │ │ │ ├── internal_api_samples.cc │ │ │ │ │ ├── memory_sharing_sample.cc │ │ │ │ │ ├── performance_profiling.cc │ │ │ │ │ ├── run_delegate_testing.sh │ │ │ │ │ ├── run_internal_api_samples.sh │ │ │ │ │ ├── run_memory_sharing_sample.sh │ │ │ │ │ └── run_performance_profiling.sh │ │ │ │ ├── util.cc │ │ │ │ └── util.h │ │ │ ├── common │ │ │ │ ├── BUILD │ │ │ │ ├── access_type.h │ │ │ │ ├── convert.cc │ │ │ │ ├── convert.h │ │ │ │ ├── custom_parsers.h │ │ │ │ ├── custom_transformations.h │ │ │ │ ├── data_type.cc │ │ │ │ ├── data_type.h │ │ │ │ ├── data_type_test.cc │ │ │ │ ├── default │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── custom_parsers.cc │ │ │ │ │ └── custom_transformations.cc │ │ │ │ ├── flops_util.cc │ │ │ │ ├── flops_util.h │ │ │ │ ├── google │ │ │ │ │ ├── BUILD │ │ │ │ │ └── status_macros.h │ │ │ │ ├── gpu_info.cc │ │ │ │ ├── gpu_info.h │ │ │ │ ├── gpu_model.cc │ │ │ │ ├── gpu_model.fbs │ │ │ │ ├── gpu_model.h │ │ │ │ ├── gpu_model_generated.h │ │ │ │ ├── gpu_model_test_util.cc │ │ │ │ ├── gpu_model_test_util.h │ │ │ │ ├── kernel_info.h │ │ │ │ ├── lstm_parser.cc │ │ │ │ ├── lstm_parser.h │ │ │ │ ├── memory_management.cc │ │ │ │ ├── memory_management.h │ │ │ │ ├── memory_management │ │ │ │ │ ├── equality_assignment.h │ │ │ │ │ ├── greedy_by_breadth_assignment.cc │ │ │ │ │ ├── greedy_by_breadth_assignment.h │ │ │ │ │ ├── greedy_by_size_assignment.cc │ │ │ │ │ ├── greedy_by_size_assignment.h │ │ │ │ │ ├── greedy_in_order_assignment.h │ │ │ │ │ ├── internal.cc │ │ │ │ │ ├── internal.h │ │ │ │ │ ├── internal_test.cc │ │ │ │ │ ├── min_cost_flow_assignment.cc │ │ │ │ │ ├── min_cost_flow_assignment.h │ │ │ │ │ ├── naive_assignment.h │ │ │ │ │ ├── types.cc │ │ │ │ │ ├── types.h │ │ │ │ │ └── types_test.cc │ │ │ │ ├── memory_management_test.cc │ │ │ │ ├── model.cc │ │ │ │ ├── model.h │ │ │ │ ├── model_builder.cc │ │ │ │ ├── model_builder.h │ │ │ │ ├── model_builder_helper.cc │ │ │ │ ├── model_builder_helper.h │ │ │ │ ├── model_builder_helper_test.cc │ │ │ │ ├── model_builder_internal.h │ │ │ │ ├── model_builder_test.cc │ │ │ │ ├── model_hints.h │ │ │ │ ├── model_test.cc │ │ │ │ ├── model_transformer.cc │ │ │ │ ├── model_transformer.h │ │ │ │ ├── object_reader.cc │ │ │ │ ├── object_reader.h │ │ │ │ ├── operation_parser.cc │ │ │ │ ├── operation_parser.h │ │ │ │ ├── operations.cc │ │ │ │ ├── operations.h │ │ │ │ ├── precision.cc │ │ │ │ ├── precision.h │ │ │ │ ├── quantization_util.cc │ │ │ │ ├── quantization_util.h │ │ │ │ ├── quantization_util_test.cc │ │ │ │ ├── selectors │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── convolution_selector.h │ │ │ │ │ ├── convolution_transposed_selector.h │ │ │ │ │ ├── default │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── convolution_selector.cc │ │ │ │ │ │ ├── convolution_transposed_selector.cc │ │ │ │ │ │ ├── default_selector.cc │ │ │ │ │ │ ├── dw_convolution_selector.cc │ │ │ │ │ │ └── fully_connected_selector.cc │ │ │ │ │ ├── default_selector.h │ │ │ │ │ ├── dw_convolution_selector.h │ │ │ │ │ ├── fully_connected_selector.h │ │ │ │ │ ├── operation_selector.cc │ │ │ │ │ ├── operation_selector.h │ │ │ │ │ ├── simple_selectors.cc │ │ │ │ │ ├── simple_selectors.h │ │ │ │ │ ├── special_selector.cc │ │ │ │ │ ├── special_selector.h │ │ │ │ │ ├── subgraph.cc │ │ │ │ │ └── subgraph.h │ │ │ │ ├── shape.cc │ │ │ │ ├── shape.h │ │ │ │ ├── shape_test.cc │ │ │ │ ├── status.h │ │ │ │ ├── task │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── arguments.cc │ │ │ │ │ ├── arguments.h │ │ │ │ │ ├── buffer_desc.cc │ │ │ │ │ ├── buffer_desc.h │ │ │ │ │ ├── compiler_options.h │ │ │ │ │ ├── gpu_object_desc.h │ │ │ │ │ ├── gpu_operation.cc │ │ │ │ │ ├── gpu_operation.h │ │ │ │ │ ├── gpu_tensor.h │ │ │ │ │ ├── profiling_info.cc │ │ │ │ │ ├── profiling_info.h │ │ │ │ │ ├── qcom_thin_filter_desc.cc │ │ │ │ │ ├── qcom_thin_filter_desc.h │ │ │ │ │ ├── serialization_base.cc │ │ │ │ │ ├── serialization_base.fbs │ │ │ │ │ ├── serialization_base.h │ │ │ │ │ ├── serialization_base_generated.h │ │ │ │ │ ├── tensor_desc.cc │ │ │ │ │ ├── tensor_desc.h │ │ │ │ │ ├── testing_util.cc │ │ │ │ │ ├── testing_util.h │ │ │ │ │ ├── tuning_type.h │ │ │ │ │ ├── util.cc │ │ │ │ │ ├── util.h │ │ │ │ │ ├── weights_conversion.cc │ │ │ │ │ ├── weights_conversion.h │ │ │ │ │ ├── weights_layout.cc │ │ │ │ │ ├── weights_layout.h │ │ │ │ │ ├── work_group_picking.cc │ │ │ │ │ └── work_group_picking.h │ │ │ │ ├── tasks │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add.cc │ │ │ │ │ ├── add.h │ │ │ │ │ ├── add_test_util.cc │ │ │ │ │ ├── add_test_util.h │ │ │ │ │ ├── cast.cc │ │ │ │ │ ├── cast.h │ │ │ │ │ ├── cast_test_util.cc │ │ │ │ │ ├── cast_test_util.h │ │ │ │ │ ├── concat_test_util.cc │ │ │ │ │ ├── concat_test_util.h │ │ │ │ │ ├── concat_xy.cc │ │ │ │ │ ├── concat_xy.h │ │ │ │ │ ├── concat_z.cc │ │ │ │ │ ├── concat_z.h │ │ │ │ │ ├── conv_constants.cc │ │ │ │ │ ├── conv_constants.h │ │ │ │ │ ├── conv_constants_test_util.cc │ │ │ │ │ ├── conv_constants_test_util.h │ │ │ │ │ ├── conv_generic.cc │ │ │ │ │ ├── conv_generic.h │ │ │ │ │ ├── conv_generic_test_util.cc │ │ │ │ │ ├── conv_generic_test_util.h │ │ │ │ │ ├── conv_metal_simd.cc │ │ │ │ │ ├── conv_metal_simd.h │ │ │ │ │ ├── conv_weights_converter.cc │ │ │ │ │ ├── conv_weights_converter.h │ │ │ │ │ ├── conv_weights_converter_test_util.cc │ │ │ │ │ ├── conv_weights_converter_test_util.h │ │ │ │ │ ├── conversion.cc │ │ │ │ │ ├── conversion.h │ │ │ │ │ ├── convolution_transposed.cc │ │ │ │ │ ├── convolution_transposed.h │ │ │ │ │ ├── convolution_transposed_3x3.cc │ │ │ │ │ ├── convolution_transposed_3x3.h │ │ │ │ │ ├── convolution_transposed_3x3_test_util.cc │ │ │ │ │ ├── convolution_transposed_3x3_test_util.h │ │ │ │ │ ├── convolution_transposed_3x3_thin.cc │ │ │ │ │ ├── convolution_transposed_3x3_thin.h │ │ │ │ │ ├── convolution_transposed_3x3_thin_test_util.cc │ │ │ │ │ ├── convolution_transposed_3x3_thin_test_util.h │ │ │ │ │ ├── convolution_transposed_4x4.cc │ │ │ │ │ ├── convolution_transposed_4x4.h │ │ │ │ │ ├── convolution_transposed_4x4_test_util.cc │ │ │ │ │ ├── convolution_transposed_4x4_test_util.h │ │ │ │ │ ├── convolution_transposed_test_util.cc │ │ │ │ │ ├── convolution_transposed_test_util.h │ │ │ │ │ ├── convolution_transposed_thin.cc │ │ │ │ │ ├── convolution_transposed_thin.h │ │ │ │ │ ├── convolution_transposed_thin_test_util.cc │ │ │ │ │ ├── convolution_transposed_thin_test_util.h │ │ │ │ │ ├── cumsum.cc │ │ │ │ │ ├── cumsum.h │ │ │ │ │ ├── cumsum_test_util.cc │ │ │ │ │ ├── cumsum_test_util.h │ │ │ │ │ ├── depthwise_conv.cc │ │ │ │ │ ├── depthwise_conv.h │ │ │ │ │ ├── depthwise_conv_3x3.cc │ │ │ │ │ ├── depthwise_conv_3x3.h │ │ │ │ │ ├── depthwise_conv_3x3_stride_h2.cc │ │ │ │ │ ├── depthwise_conv_3x3_stride_h2.h │ │ │ │ │ ├── depthwise_conv_3x3_stride_h2_test_util.cc │ │ │ │ │ ├── depthwise_conv_3x3_stride_h2_test_util.h │ │ │ │ │ ├── depthwise_conv_3x3_test_util.cc │ │ │ │ │ ├── depthwise_conv_3x3_test_util.h │ │ │ │ │ ├── depthwise_conv_test_util.cc │ │ │ │ │ ├── depthwise_conv_test_util.h │ │ │ │ │ ├── elementwise.cc │ │ │ │ │ ├── elementwise.h │ │ │ │ │ ├── elementwise_test_util.cc │ │ │ │ │ ├── elementwise_test_util.h │ │ │ │ │ ├── fully_connected.cc │ │ │ │ │ ├── fully_connected.h │ │ │ │ │ ├── fully_connected_test_util.cc │ │ │ │ │ ├── fully_connected_test_util.h │ │ │ │ │ ├── gather.cc │ │ │ │ │ ├── gather.h │ │ │ │ │ ├── gather_test_util.cc │ │ │ │ │ ├── gather_test_util.h │ │ │ │ │ ├── lstm.cc │ │ │ │ │ ├── lstm.h │ │ │ │ │ ├── lstm_test_util.cc │ │ │ │ │ ├── lstm_test_util.h │ │ │ │ │ ├── max_unpooling.cc │ │ │ │ │ ├── max_unpooling.h │ │ │ │ │ ├── max_unpooling_test_util.cc │ │ │ │ │ ├── max_unpooling_test_util.h │ │ │ │ │ ├── mean_stddev_normalization.cc │ │ │ │ │ ├── mean_stddev_normalization.h │ │ │ │ │ ├── mean_stddev_normalization_test_util.cc │ │ │ │ │ ├── mean_stddev_normalization_test_util.h │ │ │ │ │ ├── one_hot.cc │ │ │ │ │ ├── one_hot.h │ │ │ │ │ ├── one_hot_test_util.cc │ │ │ │ │ ├── one_hot_test_util.h │ │ │ │ │ ├── padding.cc │ │ │ │ │ ├── padding.h │ │ │ │ │ ├── padding_test_util.cc │ │ │ │ │ ├── padding_test_util.h │ │ │ │ │ ├── pooling.cc │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── pooling_test_util.cc │ │ │ │ │ ├── pooling_test_util.h │ │ │ │ │ ├── prelu.cc │ │ │ │ │ ├── prelu.h │ │ │ │ │ ├── prelu_test_util.cc │ │ │ │ │ ├── prelu_test_util.h │ │ │ │ │ ├── quantize_and_dequantize.cc │ │ │ │ │ ├── quantize_and_dequantize.h │ │ │ │ │ ├── quantize_and_dequantize_test_util.cc │ │ │ │ │ ├── quantize_and_dequantize_test_util.h │ │ │ │ │ ├── reduce.cc │ │ │ │ │ ├── reduce.h │ │ │ │ │ ├── reduce_test_util.cc │ │ │ │ │ ├── reduce_test_util.h │ │ │ │ │ ├── relu.cc │ │ │ │ │ ├── relu.h │ │ │ │ │ ├── relu_test_util.cc │ │ │ │ │ ├── relu_test_util.h │ │ │ │ │ ├── resampler.cc │ │ │ │ │ ├── resampler.h │ │ │ │ │ ├── resampler_test_util.cc │ │ │ │ │ ├── resampler_test_util.h │ │ │ │ │ ├── reshape.cc │ │ │ │ │ ├── reshape.h │ │ │ │ │ ├── reshape_test_util.cc │ │ │ │ │ ├── reshape_test_util.h │ │ │ │ │ ├── reshapex4.cc │ │ │ │ │ ├── reshapex4.h │ │ │ │ │ ├── resize.cc │ │ │ │ │ ├── resize.h │ │ │ │ │ ├── resize_test_util.cc │ │ │ │ │ ├── resize_test_util.h │ │ │ │ │ ├── select_v2.cc │ │ │ │ │ ├── select_v2.h │ │ │ │ │ ├── select_v2_test_util.cc │ │ │ │ │ ├── select_v2_test_util.h │ │ │ │ │ ├── softmax.cc │ │ │ │ │ ├── softmax.h │ │ │ │ │ ├── softmax1x1.cc │ │ │ │ │ ├── softmax1x1.h │ │ │ │ │ ├── softmax_test_util.cc │ │ │ │ │ ├── softmax_test_util.h │ │ │ │ │ ├── space_to_depth.cc │ │ │ │ │ ├── space_to_depth.h │ │ │ │ │ ├── space_to_depth_test_util.cc │ │ │ │ │ ├── space_to_depth_test_util.h │ │ │ │ │ ├── special │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── conv_pointwise.cc │ │ │ │ │ │ ├── conv_pointwise.h │ │ │ │ │ │ ├── conv_pointwise_test.cc │ │ │ │ │ │ ├── dw7x7_conv2to6_concat_conv8to8.cc │ │ │ │ │ │ ├── dw7x7_conv2to6_concat_conv8to8.h │ │ │ │ │ │ ├── fc_fc_add.cc │ │ │ │ │ │ ├── fc_fc_add.h │ │ │ │ │ │ ├── thin_pointwise_fuser.cc │ │ │ │ │ │ └── thin_pointwise_fuser.h │ │ │ │ │ ├── split.cc │ │ │ │ │ ├── split.h │ │ │ │ │ ├── split_test_util.cc │ │ │ │ │ ├── split_test_util.h │ │ │ │ │ ├── strided_slice.cc │ │ │ │ │ ├── strided_slice.h │ │ │ │ │ ├── strided_slice_test_util.cc │ │ │ │ │ ├── strided_slice_test_util.h │ │ │ │ │ ├── tile.cc │ │ │ │ │ ├── tile.h │ │ │ │ │ ├── tile_test_util.cc │ │ │ │ │ ├── tile_test_util.h │ │ │ │ │ ├── transpose.cc │ │ │ │ │ ├── transpose.h │ │ │ │ │ ├── transpose_test_util.cc │ │ │ │ │ ├── transpose_test_util.h │ │ │ │ │ ├── winograd.cc │ │ │ │ │ ├── winograd.h │ │ │ │ │ ├── winograd_test_util.cc │ │ │ │ │ └── winograd_test_util.h │ │ │ │ ├── tensor.h │ │ │ │ ├── testing │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── feature_parity │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── feature_parity.h │ │ │ │ │ │ ├── generators │ │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ │ ├── add.cc │ │ │ │ │ │ │ └── add.h │ │ │ │ │ │ ├── opencl_test.cc │ │ │ │ │ │ ├── opengl_test.cc │ │ │ │ │ │ ├── utils.cc │ │ │ │ │ │ ├── utils.h │ │ │ │ │ │ └── xnnpack_test.cc │ │ │ │ │ ├── interpreter_utils.cc │ │ │ │ │ └── interpreter_utils.h │ │ │ │ ├── transformations │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add_bias.cc │ │ │ │ │ ├── add_bias.h │ │ │ │ │ ├── add_quant_adjustments.cc │ │ │ │ │ ├── add_quant_adjustments.h │ │ │ │ │ ├── add_quant_adjustments_test.cc │ │ │ │ │ ├── fuse_add_to_conv.cc │ │ │ │ │ ├── fuse_add_to_conv.h │ │ │ │ │ ├── fuse_add_to_conv_test.cc │ │ │ │ │ ├── fuse_mul_to_conv.cc │ │ │ │ │ ├── fuse_mul_to_conv.h │ │ │ │ │ ├── fuse_mul_to_conv_test.cc │ │ │ │ │ ├── global_pooling_to_reduce_op.cc │ │ │ │ │ ├── global_pooling_to_reduce_op.h │ │ │ │ │ ├── global_pooling_to_reduce_op_test.cc │ │ │ │ │ ├── make_fully_connected.cc │ │ │ │ │ ├── make_fully_connected.h │ │ │ │ │ ├── make_fully_connected_test.cc │ │ │ │ │ ├── make_padding.cc │ │ │ │ │ ├── make_padding.h │ │ │ │ │ ├── make_padding_test.cc │ │ │ │ │ ├── matching.h │ │ │ │ │ ├── merge_densify.cc │ │ │ │ │ ├── merge_densify.h │ │ │ │ │ ├── merge_padding_with.cc │ │ │ │ │ ├── merge_padding_with.h │ │ │ │ │ ├── merge_padding_with_test.cc │ │ │ │ │ ├── model_transformations.cc │ │ │ │ │ ├── model_transformations.h │ │ │ │ │ ├── remove_noop.cc │ │ │ │ │ ├── remove_noop.h │ │ │ │ │ └── remove_noop_test.cc │ │ │ │ ├── types.h │ │ │ │ ├── unimplemented_operation_parser.h │ │ │ │ ├── util.h │ │ │ │ ├── util_test.cc │ │ │ │ ├── winograd_util.cc │ │ │ │ ├── winograd_util.h │ │ │ │ ├── winograd_util_test.cc │ │ │ │ ├── workgroup_selection.cc │ │ │ │ └── workgroup_selection.h │ │ │ ├── delegate.cc │ │ │ ├── delegate.h │ │ │ ├── delegate_options.cc │ │ │ ├── delegate_options.h │ │ │ ├── gl │ │ │ │ ├── BUILD │ │ │ │ ├── android_sync.cc │ │ │ │ ├── android_sync.h │ │ │ │ ├── android_sync_test.cc │ │ │ │ ├── api.cc │ │ │ │ ├── api.h │ │ │ │ ├── api2.cc │ │ │ │ ├── api2.h │ │ │ │ ├── command_queue.cc │ │ │ │ ├── command_queue.h │ │ │ │ ├── common.fbs │ │ │ │ ├── compiled_model.fbs │ │ │ │ ├── compiler.cc │ │ │ │ ├── compiler.h │ │ │ │ ├── compiler │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── compiled_node.cc │ │ │ │ │ ├── compiled_node.h │ │ │ │ │ ├── compiled_node_test.cc │ │ │ │ │ ├── fuse_auto_input.cc │ │ │ │ │ ├── fuse_auto_input.h │ │ │ │ │ ├── fuse_auto_input_test.cc │ │ │ │ │ ├── fuse_inline.cc │ │ │ │ │ ├── fuse_inline.h │ │ │ │ │ ├── fuse_inplace.cc │ │ │ │ │ ├── fuse_inplace.h │ │ │ │ │ ├── object_accessor.cc │ │ │ │ │ ├── object_accessor.h │ │ │ │ │ ├── object_accessor_test.cc │ │ │ │ │ ├── preprocessor.cc │ │ │ │ │ ├── preprocessor.h │ │ │ │ │ ├── preprocessor_test.cc │ │ │ │ │ ├── rename.cc │ │ │ │ │ ├── rename.h │ │ │ │ │ ├── shader_code.h │ │ │ │ │ ├── shader_codegen.cc │ │ │ │ │ ├── shader_codegen.h │ │ │ │ │ ├── variable_accessor.cc │ │ │ │ │ ├── variable_accessor.h │ │ │ │ │ └── variable_accessor_test.cc │ │ │ │ ├── compiler_options.h │ │ │ │ ├── converters │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── bhwc_to_phwc4.cc │ │ │ │ │ ├── bhwc_to_phwc4.h │ │ │ │ │ ├── bhwc_to_phwc4_test.cc │ │ │ │ │ ├── phwc4_to_bhwc.cc │ │ │ │ │ ├── phwc4_to_bhwc.h │ │ │ │ │ ├── phwc4_to_bhwc_test.cc │ │ │ │ │ └── util.h │ │ │ │ ├── egl_context.cc │ │ │ │ ├── egl_context.h │ │ │ │ ├── egl_environment.cc │ │ │ │ ├── egl_environment.h │ │ │ │ ├── egl_surface.cc │ │ │ │ ├── egl_surface.h │ │ │ │ ├── float16_conversions.cc │ │ │ │ ├── float16_conversions.h │ │ │ │ ├── gl_buffer.cc │ │ │ │ ├── gl_buffer.h │ │ │ │ ├── gl_buffer_test.cc │ │ │ │ ├── gl_call.h │ │ │ │ ├── gl_errors.cc │ │ │ │ ├── gl_errors.h │ │ │ │ ├── gl_program.cc │ │ │ │ ├── gl_program.h │ │ │ │ ├── gl_shader.cc │ │ │ │ ├── gl_shader.h │ │ │ │ ├── gl_sync.cc │ │ │ │ ├── gl_sync.h │ │ │ │ ├── gl_texture.cc │ │ │ │ ├── gl_texture.h │ │ │ │ ├── gl_texture_helper.cc │ │ │ │ ├── gl_texture_helper.h │ │ │ │ ├── kernels │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add.cc │ │ │ │ │ ├── add.h │ │ │ │ │ ├── add_test.cc │ │ │ │ │ ├── concat.cc │ │ │ │ │ ├── concat.h │ │ │ │ │ ├── concat_test.cc │ │ │ │ │ ├── conv.cc │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── conv_test.cc │ │ │ │ │ ├── converter.cc │ │ │ │ │ ├── converter.h │ │ │ │ │ ├── converter_test.cc │ │ │ │ │ ├── custom_registry.cc │ │ │ │ │ ├── custom_registry.h │ │ │ │ │ ├── depthwise_conv.cc │ │ │ │ │ ├── depthwise_conv.h │ │ │ │ │ ├── depthwise_conv_test.cc │ │ │ │ │ ├── elementwise.cc │ │ │ │ │ ├── elementwise.h │ │ │ │ │ ├── elementwise_test.cc │ │ │ │ │ ├── fully_connected.cc │ │ │ │ │ ├── fully_connected.h │ │ │ │ │ ├── fully_connected_test.cc │ │ │ │ │ ├── lstm.cc │ │ │ │ │ ├── lstm.h │ │ │ │ │ ├── lstm_test.cc │ │ │ │ │ ├── max_unpooling.cc │ │ │ │ │ ├── max_unpooling.h │ │ │ │ │ ├── max_unpooling_test.cc │ │ │ │ │ ├── mean.cc │ │ │ │ │ ├── mean.h │ │ │ │ │ ├── mean_test.cc │ │ │ │ │ ├── mul.cc │ │ │ │ │ ├── mul.h │ │ │ │ │ ├── mul_test.cc │ │ │ │ │ ├── pad.cc │ │ │ │ │ ├── pad.h │ │ │ │ │ ├── pad_test.cc │ │ │ │ │ ├── pooling.cc │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── pooling_test.cc │ │ │ │ │ ├── prelu.cc │ │ │ │ │ ├── prelu.h │ │ │ │ │ ├── prelu_test.cc │ │ │ │ │ ├── quantize_and_dequantize.cc │ │ │ │ │ ├── quantize_and_dequantize.h │ │ │ │ │ ├── quantize_and_dequantize_test.cc │ │ │ │ │ ├── registry.cc │ │ │ │ │ ├── registry.h │ │ │ │ │ ├── relu.cc │ │ │ │ │ ├── relu.h │ │ │ │ │ ├── relu_test.cc │ │ │ │ │ ├── resampler.cc │ │ │ │ │ ├── resampler.h │ │ │ │ │ ├── resampler_test.cc │ │ │ │ │ ├── reshape.cc │ │ │ │ │ ├── reshape.h │ │ │ │ │ ├── reshape_test.cc │ │ │ │ │ ├── resize.cc │ │ │ │ │ ├── resize.h │ │ │ │ │ ├── resize_test.cc │ │ │ │ │ ├── slice.cc │ │ │ │ │ ├── slice.h │ │ │ │ │ ├── slice_test.cc │ │ │ │ │ ├── softmax.cc │ │ │ │ │ ├── softmax.h │ │ │ │ │ ├── softmax_test.cc │ │ │ │ │ ├── space_to_depth.cc │ │ │ │ │ ├── space_to_depth.h │ │ │ │ │ ├── space_to_depth_test.cc │ │ │ │ │ ├── test_util.cc │ │ │ │ │ ├── test_util.h │ │ │ │ │ ├── tile.cc │ │ │ │ │ ├── tile.h │ │ │ │ │ ├── tile_test.cc │ │ │ │ │ ├── transpose_conv.cc │ │ │ │ │ ├── transpose_conv.h │ │ │ │ │ └── transpose_conv_test.cc │ │ │ │ ├── metadata.fbs │ │ │ │ ├── node_shader.h │ │ │ │ ├── object.h │ │ │ │ ├── object_manager.cc │ │ │ │ ├── object_manager.h │ │ │ │ ├── portable_egl.h │ │ │ │ ├── portable_gl31.h │ │ │ │ ├── request_gpu_info.cc │ │ │ │ ├── request_gpu_info.h │ │ │ │ ├── runtime.cc │ │ │ │ ├── runtime.h │ │ │ │ ├── runtime │ │ │ │ │ ├── BUILD │ │ │ │ │ └── shared_buffer.h │ │ │ │ ├── runtime_options.h │ │ │ │ ├── serialization.cc │ │ │ │ ├── serialization.h │ │ │ │ ├── serialization_test.cc │ │ │ │ ├── stats.h │ │ │ │ ├── variable.h │ │ │ │ ├── workgroups.fbs │ │ │ │ └── workgroups │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── best_effort_calculator.cc │ │ │ │ │ ├── best_effort_calculator.h │ │ │ │ │ ├── calculator.cc │ │ │ │ │ ├── calculator.h │ │ │ │ │ ├── calculator_from_metadata.cc │ │ │ │ │ ├── calculator_from_metadata.h │ │ │ │ │ ├── default_calculator.cc │ │ │ │ │ ├── default_calculator.h │ │ │ │ │ ├── ideal_workgroup_picker.cc │ │ │ │ │ └── ideal_workgroup_picker.h │ │ │ ├── gl_delegate.cc │ │ │ ├── gl_delegate.h │ │ │ ├── java │ │ │ │ └── src │ │ │ │ │ └── main │ │ │ │ │ ├── java │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ └── gpu │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── CompatibilityList.java │ │ │ │ │ │ ├── GpuDelegate.java │ │ │ │ │ │ ├── GpuDelegateFactory.java │ │ │ │ │ │ └── GpuDelegateNative.java │ │ │ │ │ └── native │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── compatibility_list_jni.cc │ │ │ │ │ ├── gpu_delegate_jni.cc │ │ │ │ │ └── gpu_delegate_native_jni.cc │ │ │ ├── metal │ │ │ │ ├── BUILD │ │ │ │ ├── ComponentsTests.tulsiproj │ │ │ │ │ ├── Configs │ │ │ │ │ │ └── ComponentsTests.tulsigen │ │ │ │ │ └── project.tulsiconf │ │ │ │ ├── Info.plist │ │ │ │ ├── benchmarking │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── Benchmark.tulsiproj │ │ │ │ │ │ ├── Configs │ │ │ │ │ │ │ └── Benchmark.tulsigen │ │ │ │ │ │ └── project.tulsiconf │ │ │ │ │ ├── Info.plist │ │ │ │ │ ├── main.mm │ │ │ │ │ └── models │ │ │ │ │ │ └── README.md │ │ │ │ ├── buffer.cc │ │ │ │ ├── buffer.h │ │ │ │ ├── buffer_convert.h │ │ │ │ ├── buffer_convert.mm │ │ │ │ ├── buffer_test.mm │ │ │ │ ├── common.h │ │ │ │ ├── common.mm │ │ │ │ ├── common_test.mm │ │ │ │ ├── compute_task.cc │ │ │ │ ├── compute_task.h │ │ │ │ ├── gpu_object.h │ │ │ │ ├── host.m │ │ │ │ ├── inference_context.cc │ │ │ │ ├── inference_context.fbs │ │ │ │ ├── inference_context.h │ │ │ │ ├── kernels │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add_test.mm │ │ │ │ │ ├── cast_test.mm │ │ │ │ │ ├── concat_test.mm │ │ │ │ │ ├── conv_test.mm │ │ │ │ │ ├── conv_weights_converter_test.mm │ │ │ │ │ ├── cumsum_test.mm │ │ │ │ │ ├── depthwise_conv_test.mm │ │ │ │ │ ├── elementwise_test.mm │ │ │ │ │ ├── fully_connected_test.mm │ │ │ │ │ ├── gather_test.mm │ │ │ │ │ ├── lstm_test.mm │ │ │ │ │ ├── max_unpooling_test.mm │ │ │ │ │ ├── mean_stddev_normalization_test.mm │ │ │ │ │ ├── one_hot_test.mm │ │ │ │ │ ├── padding_test.mm │ │ │ │ │ ├── pooling_test.mm │ │ │ │ │ ├── prelu_test.mm │ │ │ │ │ ├── quantize_and_dequantize_test.mm │ │ │ │ │ ├── reduce_test.mm │ │ │ │ │ ├── relu_test.mm │ │ │ │ │ ├── resampler_test.mm │ │ │ │ │ ├── reshape_test.mm │ │ │ │ │ ├── resize_test.mm │ │ │ │ │ ├── select_v2_test.mm │ │ │ │ │ ├── slice_test.mm │ │ │ │ │ ├── softmax_test.mm │ │ │ │ │ ├── space_to_depth_test.mm │ │ │ │ │ ├── split_test.mm │ │ │ │ │ ├── test_util.cc │ │ │ │ │ ├── test_util.h │ │ │ │ │ ├── tile_test.mm │ │ │ │ │ ├── transpose_conv_test.mm │ │ │ │ │ ├── transpose_test.mm │ │ │ │ │ └── winograd_test.mm │ │ │ │ ├── metal_arguments.cc │ │ │ │ ├── metal_arguments.h │ │ │ │ ├── metal_device.cc │ │ │ │ ├── metal_device.h │ │ │ │ ├── metal_spatial_tensor.cc │ │ │ │ ├── metal_spatial_tensor.h │ │ │ │ └── metal_spatial_tensor_test.mm │ │ │ ├── metal_delegate.h │ │ │ ├── metal_delegate.mm │ │ │ ├── metal_delegate_internal.h │ │ │ ├── spi.h │ │ │ ├── tflite_profile.cc │ │ │ └── tflite_profile.h │ │ ├── hexagon │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── builders │ │ │ │ ├── BUILD │ │ │ │ ├── activation_builder.cc │ │ │ │ ├── activation_builder.h │ │ │ │ ├── arg_min_max_builder.cc │ │ │ │ ├── arg_min_max_builder.h │ │ │ │ ├── arithmetic_builder.cc │ │ │ │ ├── arithmetic_builder.h │ │ │ │ ├── batch_seq_builder.cc │ │ │ │ ├── batch_seq_builder.h │ │ │ │ ├── cast_builder.cc │ │ │ │ ├── cast_builder.h │ │ │ │ ├── concat_builder.cc │ │ │ │ ├── concat_builder.h │ │ │ │ ├── conv_2d_builder.cc │ │ │ │ ├── conv_2d_builder.h │ │ │ │ ├── conv_2d_helpers.cc │ │ │ │ ├── hardswish_builder.cc │ │ │ │ ├── hardswish_builder.h │ │ │ │ ├── l2_normalization_builder.cc │ │ │ │ ├── l2_normalization_builder.h │ │ │ │ ├── matmul_builder.cc │ │ │ │ ├── matmul_builder.h │ │ │ │ ├── min_max_builder.cc │ │ │ │ ├── min_max_builder.h │ │ │ │ ├── mirror_pad_builder.cc │ │ │ │ ├── mirror_pad_builder.h │ │ │ │ ├── neg_op_builder.cc │ │ │ │ ├── neg_op_builder.h │ │ │ │ ├── op_builder.cc │ │ │ │ ├── op_builder.h │ │ │ │ ├── op_factory.h │ │ │ │ ├── pack_builder.cc │ │ │ │ ├── pack_builder.h │ │ │ │ ├── pad_builder.cc │ │ │ │ ├── pad_builder.h │ │ │ │ ├── pool_2d_builder.cc │ │ │ │ ├── pool_2d_builder.h │ │ │ │ ├── quantize_builder.cc │ │ │ │ ├── quantize_builder.h │ │ │ │ ├── reduce_builder.cc │ │ │ │ ├── reduce_builder.h │ │ │ │ ├── reshape_builder.cc │ │ │ │ ├── reshape_builder.h │ │ │ │ ├── resize_bilinear_builder.cc │ │ │ │ ├── resize_bilinear_builder.h │ │ │ │ ├── resize_nearest_neighbor_builder.cc │ │ │ │ ├── resize_nearest_neighbor_builder.h │ │ │ │ ├── rsqrt_builder.cc │ │ │ │ ├── slice_builder.cc │ │ │ │ ├── slice_builder.h │ │ │ │ ├── softmax_builder.cc │ │ │ │ ├── softmax_builder.h │ │ │ │ ├── space_to_depth_builder.cc │ │ │ │ ├── space_to_depth_builder.h │ │ │ │ ├── split_builder.cc │ │ │ │ ├── split_builder.h │ │ │ │ ├── squared_difference.cc │ │ │ │ ├── strided_slice_builder.cc │ │ │ │ ├── strided_slice_builder.h │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── activations_test.cc │ │ │ │ │ ├── arg_min_max_test.cc │ │ │ │ │ ├── arithmetic_test.cc │ │ │ │ │ ├── batch_seq_config_test.cc │ │ │ │ │ ├── concat_test.cc │ │ │ │ │ ├── conv_test.cc │ │ │ │ │ ├── hexagon_delegate_op_model.h │ │ │ │ │ ├── l2_norm_test.cc │ │ │ │ │ ├── matmul_test.cc │ │ │ │ │ ├── min_max_builder_test.cc │ │ │ │ │ ├── mirror_pad_test.cc │ │ │ │ │ ├── mul_test.cc │ │ │ │ │ ├── neg_test.cc │ │ │ │ │ ├── pack_test.cc │ │ │ │ │ ├── pad_test.cc │ │ │ │ │ ├── pool_test.cc │ │ │ │ │ ├── quantize_test.cc │ │ │ │ │ ├── reduce_test.cc │ │ │ │ │ ├── reshape_test.cc │ │ │ │ │ ├── resize_test.cc │ │ │ │ │ ├── rsqrt_test.cc │ │ │ │ │ ├── run_tests.sh │ │ │ │ │ ├── slice_test.cc │ │ │ │ │ ├── softmax_test.cc │ │ │ │ │ ├── space_to_depth_test.cc │ │ │ │ │ ├── split_test.cc │ │ │ │ │ ├── squared_difference_test.cc │ │ │ │ │ ├── strided_slice_test.cc │ │ │ │ │ ├── tests.bzl │ │ │ │ │ ├── transpose_conv_test.cc │ │ │ │ │ └── transpose_test.cc │ │ │ │ ├── transpose_builder.cc │ │ │ │ ├── transpose_builder.h │ │ │ │ ├── transpose_conv_2d_builder.cc │ │ │ │ └── transpose_conv_2d_builder.h │ │ │ ├── hexagon_delegate.cc │ │ │ ├── hexagon_delegate.h │ │ │ ├── hexagon_delegate_kernel.cc │ │ │ ├── hexagon_delegate_kernel.h │ │ │ ├── hexagon_implementation.cc │ │ │ ├── hexagon_implementation.h │ │ │ ├── hexagon_nn │ │ │ │ ├── BUILD │ │ │ │ ├── adsprpc_interface.cc │ │ │ │ ├── hexagon_nn.h │ │ │ │ ├── hexagon_nn_init.cc │ │ │ │ ├── hexagon_nn_init.h │ │ │ │ ├── soc_model.cc │ │ │ │ ├── soc_model.h │ │ │ │ └── version_scripts.lds │ │ │ ├── hexagon_nn_interface.h │ │ │ ├── java │ │ │ │ ├── AndroidManifest.xml │ │ │ │ ├── BUILD │ │ │ │ ├── proguard.flags │ │ │ │ └── src │ │ │ │ │ └── main │ │ │ │ │ ├── java │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ └── HexagonDelegate.java │ │ │ │ │ └── native │ │ │ │ │ ├── BUILD │ │ │ │ │ └── hexagon_delegate_jni.cc │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ ├── utils_test.cc │ │ │ └── version_script.lds │ │ ├── interpreter_utils.cc │ │ ├── interpreter_utils.h │ │ ├── interpreter_utils_test.cc │ │ ├── nnapi │ │ │ ├── BUILD │ │ │ ├── acceleration_test_list.cc │ │ │ ├── acceleration_test_util.cc │ │ │ ├── acceleration_test_util.h │ │ │ ├── java │ │ │ │ └── src │ │ │ │ │ └── main │ │ │ │ │ ├── java │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ └── nnapi │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── NnApiDelegate.java │ │ │ │ │ │ └── NnApiDelegateImpl.java │ │ │ │ │ └── native │ │ │ │ │ ├── BUILD │ │ │ │ │ └── nnapi_delegate_impl_jni.cc │ │ │ ├── nnapi_delegate.cc │ │ │ ├── nnapi_delegate.h │ │ │ ├── nnapi_delegate_c_api.cc │ │ │ ├── nnapi_delegate_c_api.h │ │ │ ├── nnapi_delegate_c_api_test.cc │ │ │ ├── nnapi_delegate_device_selection_test.cc │ │ │ ├── nnapi_delegate_disabled.cc │ │ │ ├── nnapi_delegate_errno_test.cc │ │ │ ├── nnapi_delegate_kernel.h │ │ │ ├── nnapi_delegate_mock_test.h │ │ │ ├── nnapi_delegate_nnapi_failure_handling_test.cc │ │ │ ├── nnapi_delegate_plugin.h │ │ │ ├── nnapi_delegate_signed_quantization_test.cc │ │ │ ├── nnapi_delegate_test.cc │ │ │ ├── quant_lstm_sup.cc │ │ │ ├── quant_lstm_sup.h │ │ │ └── quant_lstm_sup_test.cc │ │ ├── opaque_delegate_test.cc │ │ ├── serialization.cc │ │ ├── serialization.h │ │ ├── serialization_test.cc │ │ ├── telemetry.cc │ │ ├── telemetry.h │ │ ├── telemetry_test.cc │ │ ├── utils.cc │ │ ├── utils.h │ │ ├── utils │ │ │ ├── BUILD │ │ │ ├── async_type_helpers.cc │ │ │ ├── async_type_helpers.h │ │ │ ├── dummy_delegate │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── dummy_delegate.cc │ │ │ │ ├── dummy_delegate.h │ │ │ │ ├── dummy_delegate_provider.cc │ │ │ │ ├── external_delegate_adaptor.cc │ │ │ │ └── external_delegate_test.sh │ │ │ ├── experimental │ │ │ │ ├── sample_stable_delegate │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── sample_app_using_stable_delegate.cc │ │ │ │ │ ├── sample_stable_delegate.cc │ │ │ │ │ ├── sample_stable_delegate.h │ │ │ │ │ ├── sample_stable_delegate_external.cc │ │ │ │ │ ├── sample_stable_delegate_external_test.cc │ │ │ │ │ ├── sample_stable_delegate_test.cc │ │ │ │ │ ├── sample_stable_delegate_with_control_flow.cc │ │ │ │ │ ├── sample_stable_delegate_with_control_flow.h │ │ │ │ │ └── sample_stable_delegate_with_control_flow_test.cc │ │ │ │ └── stable_delegate │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── delegate_loader.cc │ │ │ │ │ ├── delegate_loader.h │ │ │ │ │ ├── delegate_loader_test.cc │ │ │ │ │ ├── kernel_test_main.cc │ │ │ │ │ ├── stable_delegate_interface.h │ │ │ │ │ ├── stable_xnnpack_delegate.cc │ │ │ │ │ ├── test_xnnpack_settings.json │ │ │ │ │ ├── tflite_settings_json_parser.cc │ │ │ │ │ ├── tflite_settings_json_parser.h │ │ │ │ │ ├── tflite_settings_json_parser_test.cc │ │ │ │ │ └── version_script.lds │ │ │ ├── ret_macros.h │ │ │ ├── simple_delegate.cc │ │ │ ├── simple_delegate.h │ │ │ ├── simple_delegate_test.cc │ │ │ ├── simple_opaque_delegate.cc │ │ │ ├── simple_opaque_delegate.h │ │ │ ├── simple_opaque_delegate_test.cc │ │ │ ├── sync_fence.cc │ │ │ ├── sync_fence.h │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── utils_test.cc │ │ └── xnnpack │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── abs_test.cc │ │ │ ├── add_test.cc │ │ │ ├── average_pool_2d_test.cc │ │ │ ├── batch_matrix_multiply_test.cc │ │ │ ├── batch_matrix_multiply_tester.cc │ │ │ ├── batch_matrix_multiply_tester.h │ │ │ ├── binary_elementwise_tester.cc │ │ │ ├── binary_elementwise_tester.h │ │ │ ├── ceil_test.cc │ │ │ ├── channelwise_quantized_conv_2d_test.cc │ │ │ ├── channelwise_quantized_depthwise_conv_2d_test.cc │ │ │ ├── concatenation_test.cc │ │ │ ├── concatenation_tester.cc │ │ │ ├── concatenation_tester.h │ │ │ ├── conv_2d_test.cc │ │ │ ├── conv_2d_tester.cc │ │ │ ├── conv_2d_tester.h │ │ │ ├── delegate_test.cc │ │ │ ├── depth_to_space_test.cc │ │ │ ├── depth_to_space_tester.cc │ │ │ ├── depth_to_space_tester.h │ │ │ ├── depthwise_conv_2d_test.cc │ │ │ ├── depthwise_conv_2d_tester.cc │ │ │ ├── depthwise_conv_2d_tester.h │ │ │ ├── dequantize_tester.cc │ │ │ ├── dequantize_tester.h │ │ │ ├── div_test.cc │ │ │ ├── dynamically_quantized_conv_2d_test.cc │ │ │ ├── dynamically_quantized_conv_2d_tester.cc │ │ │ ├── dynamically_quantized_conv_2d_tester.h │ │ │ ├── dynamically_quantized_fully_connected_test.cc │ │ │ ├── dynamically_quantized_fully_connected_tester.cc │ │ │ ├── dynamically_quantized_fully_connected_tester.h │ │ │ ├── elu_test.cc │ │ │ ├── floor_test.cc │ │ │ ├── fully_connected_test.cc │ │ │ ├── fully_connected_tester.cc │ │ │ ├── fully_connected_tester.h │ │ │ ├── hard_swish_test.cc │ │ │ ├── leaky_relu_test.cc │ │ │ ├── leaky_relu_tester.cc │ │ │ ├── leaky_relu_tester.h │ │ │ ├── logistic_test.cc │ │ │ ├── max_pool_2d_test.cc │ │ │ ├── maximum_test.cc │ │ │ ├── mean_test.cc │ │ │ ├── minimum_test.cc │ │ │ ├── mul_test.cc │ │ │ ├── neg_test.cc │ │ │ ├── pad_test.cc │ │ │ ├── pad_tester.cc │ │ │ ├── pad_tester.h │ │ │ ├── pool_2d_tester.cc │ │ │ ├── pool_2d_tester.h │ │ │ ├── prelu_test.cc │ │ │ ├── prelu_tester.cc │ │ │ ├── prelu_tester.h │ │ │ ├── quantization_util.cc │ │ │ ├── quantization_util.h │ │ │ ├── quantization_util_test.cc │ │ │ ├── quantize_float32_to_int8_test.cc │ │ │ ├── quantize_float32_to_uint8_test.cc │ │ │ ├── quantize_int8_to_int8_test.cc │ │ │ ├── quantize_tester.cc │ │ │ ├── quantize_tester.h │ │ │ ├── quantize_uint8_to_uint8_test.cc │ │ │ ├── quantized_binary_elementwise_tester.cc │ │ │ ├── quantized_binary_elementwise_tester.h │ │ │ ├── quantized_conv_2d_tester.cc │ │ │ ├── quantized_conv_2d_tester.h │ │ │ ├── quantized_depthwise_conv_2d_tester.cc │ │ │ ├── quantized_depthwise_conv_2d_tester.h │ │ │ ├── quantized_fully_connected_tester.cc │ │ │ ├── quantized_fully_connected_tester.h │ │ │ ├── quantized_leaky_relu_tester.cc │ │ │ ├── quantized_leaky_relu_tester.h │ │ │ ├── quantized_pad_tester.cc │ │ │ ├── quantized_pad_tester.h │ │ │ ├── quantized_pool_2d_tester.cc │ │ │ ├── quantized_pool_2d_tester.h │ │ │ ├── quantized_reduce_tester.cc │ │ │ ├── quantized_reduce_tester.h │ │ │ ├── quantized_resize_bilinear_tester.cc │ │ │ ├── quantized_resize_bilinear_tester.h │ │ │ ├── quantized_transpose_conv_tester.cc │ │ │ ├── quantized_transpose_conv_tester.h │ │ │ ├── quantized_unary_elementwise_tester.cc │ │ │ ├── quantized_unary_elementwise_tester.h │ │ │ ├── quantized_variable_ops_tester.cc │ │ │ ├── quantized_variable_ops_tester.h │ │ │ ├── reduce_tester.cc │ │ │ ├── reduce_tester.h │ │ │ ├── relu6_test.cc │ │ │ ├── relu_n1_to_1_test.cc │ │ │ ├── relu_test.cc │ │ │ ├── reshape_test.cc │ │ │ ├── reshape_tester.cc │ │ │ ├── reshape_tester.h │ │ │ ├── resize_bilinear_test.cc │ │ │ ├── resize_bilinear_tester.cc │ │ │ ├── resize_bilinear_tester.h │ │ │ ├── round_test.cc │ │ │ ├── signed_dequantize_test.cc │ │ │ ├── signed_quantized_add_test.cc │ │ │ ├── signed_quantized_concatenation_test.cc │ │ │ ├── signed_quantized_conv_2d_test.cc │ │ │ ├── signed_quantized_depth_to_space_test.cc │ │ │ ├── signed_quantized_depthwise_conv_2d_test.cc │ │ │ ├── signed_quantized_elu_test.cc │ │ │ ├── signed_quantized_fully_connected_test.cc │ │ │ ├── signed_quantized_leaky_relu_test.cc │ │ │ ├── signed_quantized_logistic_test.cc │ │ │ ├── signed_quantized_max_pool_2d_test.cc │ │ │ ├── signed_quantized_mean_test.cc │ │ │ ├── signed_quantized_mul_test.cc │ │ │ ├── signed_quantized_pad_test.cc │ │ │ ├── signed_quantized_reshape_test.cc │ │ │ ├── signed_quantized_resize_bilinear_test.cc │ │ │ ├── signed_quantized_slice_test.cc │ │ │ ├── signed_quantized_space_to_depth_test.cc │ │ │ ├── signed_quantized_split_test.cc │ │ │ ├── signed_quantized_strided_slice_test.cc │ │ │ ├── signed_quantized_sub_test.cc │ │ │ ├── signed_quantized_tanh_test.cc │ │ │ ├── signed_quantized_transpose_conv_test.cc │ │ │ ├── signed_quantized_transpose_test.cc │ │ │ ├── signed_quantized_variable_ops_multiple_subgraph_test.cc │ │ │ ├── signed_quantized_variable_ops_test.cc │ │ │ ├── slice_test.cc │ │ │ ├── slice_tester.cc │ │ │ ├── slice_tester.h │ │ │ ├── softmax_test.cc │ │ │ ├── softmax_tester.cc │ │ │ ├── softmax_tester.h │ │ │ ├── space_to_depth_test.cc │ │ │ ├── space_to_depth_tester.cc │ │ │ ├── space_to_depth_tester.h │ │ │ ├── split_test.cc │ │ │ ├── split_tester.cc │ │ │ ├── split_tester.h │ │ │ ├── sqrt_test.cc │ │ │ ├── square_test.cc │ │ │ ├── squared_difference_test.cc │ │ │ ├── strided_slice_test.cc │ │ │ ├── strided_slice_tester.cc │ │ │ ├── strided_slice_tester.h │ │ │ ├── sub_test.cc │ │ │ ├── sum_test.cc │ │ │ ├── tanh_test.cc │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ ├── transpose_conv_test.cc │ │ │ ├── transpose_conv_tester.cc │ │ │ ├── transpose_conv_tester.h │ │ │ ├── transpose_test.cc │ │ │ ├── transpose_tester.cc │ │ │ ├── transpose_tester.h │ │ │ ├── unary_elementwise_tester.cc │ │ │ ├── unary_elementwise_tester.h │ │ │ ├── unsigned_dequantize_test.cc │ │ │ ├── unsigned_quantized_add_test.cc │ │ │ ├── unsigned_quantized_concatenation_test.cc │ │ │ ├── unsigned_quantized_conv_2d_test.cc │ │ │ ├── unsigned_quantized_depth_to_space_test.cc │ │ │ ├── unsigned_quantized_depthwise_conv_2d_test.cc │ │ │ ├── unsigned_quantized_fully_connected_test.cc │ │ │ ├── unsigned_quantized_leaky_relu_test.cc │ │ │ ├── unsigned_quantized_logistic_test.cc │ │ │ ├── unsigned_quantized_max_pool_2d_test.cc │ │ │ ├── unsigned_quantized_mean_test.cc │ │ │ ├── unsigned_quantized_mul_test.cc │ │ │ ├── unsigned_quantized_pad_test.cc │ │ │ ├── unsigned_quantized_reshape_test.cc │ │ │ ├── unsigned_quantized_resize_bilinear_test.cc │ │ │ ├── unsigned_quantized_slice_test.cc │ │ │ ├── unsigned_quantized_space_to_depth_test.cc │ │ │ ├── unsigned_quantized_split_test.cc │ │ │ ├── unsigned_quantized_strided_slice_test.cc │ │ │ ├── unsigned_quantized_sub_test.cc │ │ │ ├── unsigned_quantized_tanh_test.cc │ │ │ ├── unsigned_quantized_transpose_conv_test.cc │ │ │ ├── unsigned_quantized_transpose_test.cc │ │ │ ├── unsigned_quantized_variable_ops_multiple_subgraph_test.cc │ │ │ ├── unsigned_quantized_variable_ops_test.cc │ │ │ ├── variable_ops_multiple_subgraph_test.cc │ │ │ ├── variable_ops_test.cc │ │ │ ├── variable_ops_tester.cc │ │ │ ├── variable_ops_tester.h │ │ │ ├── weights_cache_test.cc │ │ │ ├── xnnpack_delegate.cc │ │ │ ├── xnnpack_delegate.h │ │ │ └── xnnpack_delegate_test.h │ ├── error_reporter.h │ ├── examples │ │ ├── android │ │ │ └── app │ │ │ │ └── README.md │ │ ├── experimental_new_converter │ │ │ ├── BUILD │ │ │ ├── Keras_LSTM_fusion_Codelab.ipynb │ │ │ ├── keras_lstm.ipynb │ │ │ ├── keras_lstm.png │ │ │ └── stack_trace_example.py │ │ ├── ios │ │ │ ├── camera │ │ │ │ ├── .gitignore │ │ │ │ ├── CameraExampleAppDelegate.h │ │ │ │ ├── CameraExampleAppDelegate.m │ │ │ │ ├── CameraExampleViewController.h │ │ │ │ ├── CameraExampleViewController.mm │ │ │ │ ├── Info.plist │ │ │ │ ├── MainStoryboard_iPhone.storyboard │ │ │ │ ├── Podfile │ │ │ │ ├── README.md │ │ │ │ ├── data │ │ │ │ │ ├── .gitignore │ │ │ │ │ └── labels.txt │ │ │ │ ├── main.mm │ │ │ │ └── tflite_camera_example.xcodeproj │ │ │ │ │ └── project.pbxproj │ │ │ ├── download_models.sh │ │ │ └── simple │ │ │ │ ├── AppDelegate.h │ │ │ │ ├── AppDelegate.mm │ │ │ │ ├── Podfile │ │ │ │ ├── RunModel-Info.plist │ │ │ │ ├── RunModelViewController.h │ │ │ │ ├── RunModelViewController.mm │ │ │ │ ├── RunModelViewController.xib │ │ │ │ ├── data │ │ │ │ ├── grace_hopper.jpg │ │ │ │ └── labels.txt │ │ │ │ ├── ios_image_load.h │ │ │ │ ├── ios_image_load.mm │ │ │ │ ├── main.mm │ │ │ │ └── simple.xcodeproj │ │ │ │ └── project.pbxproj │ │ ├── label_image │ │ │ ├── BUILD │ │ │ ├── CMakeLists.txt │ │ │ ├── README.md │ │ │ ├── bitmap_helpers.cc │ │ │ ├── bitmap_helpers.h │ │ │ ├── bitmap_helpers_impl.h │ │ │ ├── get_top_n.h │ │ │ ├── get_top_n_impl.h │ │ │ ├── label_image.cc │ │ │ ├── label_image.h │ │ │ ├── label_image_test.cc │ │ │ ├── log.h │ │ │ └── testdata │ │ │ │ └── grace_hopper.bmp │ │ ├── minimal │ │ │ ├── BUILD │ │ │ ├── CMakeLists.txt │ │ │ ├── README.md │ │ │ └── minimal.cc │ │ └── python │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ └── label_image.py │ ├── experimental │ │ ├── acceleration │ │ │ ├── README.md │ │ │ ├── compatibility │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── android_info.cc │ │ │ │ ├── android_info.h │ │ │ │ ├── canonicalize_value.cc │ │ │ │ ├── canonicalize_value.h │ │ │ │ ├── canonicalize_value_test.cc │ │ │ │ ├── convert_binary_to_cc_source.py │ │ │ │ ├── database.fbs │ │ │ │ ├── devicedb-sample.json │ │ │ │ ├── devicedb.cc │ │ │ │ ├── devicedb.h │ │ │ │ ├── devicedb_test.cc │ │ │ │ ├── gpu_compatibility.bin │ │ │ │ ├── gpu_compatibility.cc │ │ │ │ ├── gpu_compatibility.h │ │ │ │ ├── gpu_compatibility_test.cc │ │ │ │ ├── json_to_fb.cc │ │ │ │ └── variables.h │ │ │ ├── configuration │ │ │ │ ├── BUILD │ │ │ │ ├── build_defs.bzl │ │ │ │ ├── c │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── delegate_plugin.h │ │ │ │ │ ├── gpu_plugin.h │ │ │ │ │ ├── nnapi_plugin.h │ │ │ │ │ ├── stable_delegate.h │ │ │ │ │ └── xnnpack_plugin.h │ │ │ │ ├── configuration.proto │ │ │ │ ├── configuration_generated.h │ │ │ │ ├── coreml_plugin.cc │ │ │ │ ├── delegate_plugin_converter.h │ │ │ │ ├── delegate_registry.h │ │ │ │ ├── flatbuffer_to_proto.h │ │ │ │ ├── gpu_plugin.h │ │ │ │ ├── hexagon_plugin.cc │ │ │ │ ├── nnapi_plugin.h │ │ │ │ ├── proto_to_flatbuffer.h │ │ │ │ └── stable_delegate_plugin.h │ │ │ └── mini_benchmark │ │ │ │ ├── BUILD │ │ │ │ ├── benchmark_result_evaluator.cc │ │ │ │ ├── benchmark_result_evaluator.h │ │ │ │ ├── big_little_affinity.cc │ │ │ │ ├── big_little_affinity.h │ │ │ │ ├── big_little_affinity_test.cc │ │ │ │ ├── blocking_validator_runner.cc │ │ │ │ ├── blocking_validator_runner.h │ │ │ │ ├── blocking_validator_runner_test.cc │ │ │ │ ├── build_defs.bzl │ │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ └── c_api.h │ │ │ │ ├── call.cc │ │ │ │ ├── call_register.h │ │ │ │ ├── call_test.cc │ │ │ │ ├── chessboard.jpg │ │ │ │ ├── constants.h │ │ │ │ ├── copy_associated_files.py │ │ │ │ ├── data │ │ │ │ ├── chair_224_224_3.jpeg │ │ │ │ ├── city_224_224_3.jpeg │ │ │ │ ├── early_color_photo_face_128_224_3.jpeg │ │ │ │ ├── early_color_photo_faces_320_320_3.jpeg │ │ │ │ ├── face_in_painting_128_224_3.jpeg │ │ │ │ ├── face_in_snow_128_224_3.jpeg │ │ │ │ ├── faces_in_painting_320_320_3.jpeg │ │ │ │ ├── faces_in_snow_320_320_3.jpeg │ │ │ │ ├── flowers_192_192_3.jpeg │ │ │ │ ├── food_224_224_3.jpeg │ │ │ │ ├── person_192_192_3.jpeg │ │ │ │ ├── plant_224_224_3.jpeg │ │ │ │ ├── rural_192_192_3.jpeg │ │ │ │ ├── shoe_224_224_3.jpeg │ │ │ │ ├── single_face_128_224_3.jpeg │ │ │ │ ├── single_face_320_320_3.jpeg │ │ │ │ ├── snowman_128_224_3.jpeg │ │ │ │ ├── snowman_320_320_3.jpeg │ │ │ │ ├── street_192_192_3.jpeg │ │ │ │ └── toys_192_192_3.jpeg │ │ │ │ ├── decode_jpeg.cc │ │ │ │ ├── decode_jpeg_register.h │ │ │ │ ├── decode_jpeg_status.h │ │ │ │ ├── decode_jpeg_test.cc │ │ │ │ ├── fb_storage.cc │ │ │ │ ├── fb_storage.h │ │ │ │ ├── fb_storage_test.cc │ │ │ │ ├── file_lock.cc │ │ │ │ ├── file_lock.h │ │ │ │ ├── file_lock_test.cc │ │ │ │ ├── gpu_module_plugin.cc │ │ │ │ ├── gpu_module_plugin.h │ │ │ │ ├── jpeg_common.h │ │ │ │ ├── jpeg_decompress_buffered_struct.h │ │ │ │ ├── jpeg_decompress_buffered_struct_test.cc │ │ │ │ ├── jpeg_header_parser.cc │ │ │ │ ├── jpeg_header_parser.h │ │ │ │ ├── jpeg_header_parser_test.cc │ │ │ │ ├── libc_handle.cc │ │ │ │ ├── libc_handle.h │ │ │ │ ├── libc_handle_test.cc │ │ │ │ ├── libjpeg.h │ │ │ │ ├── libjpeg_decoder.cc │ │ │ │ ├── libjpeg_decoder.h │ │ │ │ ├── libjpeg_decoder_test.cc │ │ │ │ ├── libjpeg_decoder_test_helper.h │ │ │ │ ├── libjpeg_handle.h │ │ │ │ ├── libjpeg_handle_dynamic_link.cc │ │ │ │ ├── libjpeg_handle_static_link.cc │ │ │ │ ├── libjpeg_handle_test.cc │ │ │ │ ├── metrics │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── blazeface_metrics.py │ │ │ │ ├── kl_divergence.py │ │ │ │ └── mobilenet.py │ │ │ │ ├── mini_benchmark.cc │ │ │ │ ├── mini_benchmark.h │ │ │ │ ├── mini_benchmark_implementation.cc │ │ │ │ ├── mini_benchmark_test.cc │ │ │ │ ├── mini_benchmark_test_helper.cc │ │ │ │ ├── mini_benchmark_test_helper.h │ │ │ │ ├── model_modifier │ │ │ │ ├── BUILD │ │ │ │ ├── custom_validation_embedder.cc │ │ │ │ ├── custom_validation_embedder.h │ │ │ │ ├── custom_validation_embedder_test.cc │ │ │ │ ├── embedder.cc │ │ │ │ ├── embedder.h │ │ │ │ ├── embedder_main.cc │ │ │ │ ├── grafter.cc │ │ │ │ ├── grafter.h │ │ │ │ ├── validation_graph_builder.cc │ │ │ │ └── validation_graph_builder.h │ │ │ │ ├── model_validation_test.cc │ │ │ │ ├── models │ │ │ │ ├── BUILD │ │ │ │ └── blazeface_mlkit_v1.tfl │ │ │ │ ├── nnapi_sl_fake_impl.cc │ │ │ │ ├── nnapi_sl_fake_impl.h │ │ │ │ ├── runner.cc │ │ │ │ ├── runner.h │ │ │ │ ├── runner_main.c │ │ │ │ ├── runner_test.cc │ │ │ │ ├── runner_test_entry_points.cc │ │ │ │ ├── set_big_core_affinity.cc │ │ │ │ ├── set_big_core_affinity.h │ │ │ │ ├── snow_4032_3024_3.jpg │ │ │ │ ├── special_rules.bzl │ │ │ │ ├── status_codes.h │ │ │ │ ├── test_card.jpg │ │ │ │ ├── validator.cc │ │ │ │ ├── validator.h │ │ │ │ ├── validator_runner.cc │ │ │ │ ├── validator_runner.h │ │ │ │ ├── validator_runner_entrypoint.cc │ │ │ │ ├── validator_runner_entrypoint.h │ │ │ │ ├── validator_runner_entrypoint_test.cc │ │ │ │ ├── validator_runner_impl.cc │ │ │ │ ├── validator_runner_impl.h │ │ │ │ ├── validator_runner_impl_opaque_delegate_test.cc │ │ │ │ ├── validator_runner_impl_test.cc │ │ │ │ ├── validator_runner_options.cc │ │ │ │ ├── validator_runner_options.h │ │ │ │ ├── validator_runner_test.cc │ │ │ │ └── validator_test.cc │ │ ├── examples │ │ │ └── unity │ │ │ │ └── TensorFlowLitePlugin │ │ │ │ ├── .gitignore │ │ │ │ ├── Assets │ │ │ │ ├── TensorFlowLite.meta │ │ │ │ └── TensorFlowLite │ │ │ │ │ ├── Examples.meta │ │ │ │ │ ├── Examples │ │ │ │ │ ├── HelloTFLite.meta │ │ │ │ │ └── HelloTFLite │ │ │ │ │ │ ├── Scenes.meta │ │ │ │ │ │ ├── Scenes │ │ │ │ │ │ ├── HelloTFLite.unity │ │ │ │ │ │ ├── HelloTFLite.unity.meta │ │ │ │ │ │ ├── add.bytes │ │ │ │ │ │ └── add.bytes.meta │ │ │ │ │ │ ├── Scripts.meta │ │ │ │ │ │ └── Scripts │ │ │ │ │ │ ├── HelloTFLite.cs │ │ │ │ │ │ └── HelloTFLite.cs.meta │ │ │ │ │ ├── SDK.meta │ │ │ │ │ └── SDK │ │ │ │ │ ├── Scripts.meta │ │ │ │ │ └── Scripts │ │ │ │ │ ├── Interpreter.cs │ │ │ │ │ └── Interpreter.cs.meta │ │ │ │ ├── ProjectSettings │ │ │ │ ├── AudioManager.asset │ │ │ │ ├── ClusterInputManager.asset │ │ │ │ ├── DynamicsManager.asset │ │ │ │ ├── EditorBuildSettings.asset │ │ │ │ ├── EditorSettings.asset │ │ │ │ ├── GraphicsSettings.asset │ │ │ │ ├── InputManager.asset │ │ │ │ ├── NavMeshAreas.asset │ │ │ │ ├── NetworkManager.asset │ │ │ │ ├── Physics2DSettings.asset │ │ │ │ ├── ProjectSettings.asset │ │ │ │ ├── ProjectVersion.txt │ │ │ │ ├── QualitySettings.asset │ │ │ │ ├── TagManager.asset │ │ │ │ ├── TimeManager.asset │ │ │ │ └── UnityConnectSettings.asset │ │ │ │ ├── README.md │ │ │ │ └── UnityPackageManager │ │ │ │ └── manifest.json │ │ ├── microfrontend │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── audio_microfrontend.cc │ │ │ ├── audio_microfrontend.h │ │ │ ├── audio_microfrontend_test.cc │ │ │ ├── lib │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── bits.h │ │ │ │ ├── fft.cc │ │ │ │ ├── fft.h │ │ │ │ ├── fft_io.c │ │ │ │ ├── fft_io.h │ │ │ │ ├── fft_util.cc │ │ │ │ ├── fft_util.h │ │ │ │ ├── filterbank.c │ │ │ │ ├── filterbank.h │ │ │ │ ├── filterbank_io.c │ │ │ │ ├── filterbank_io.h │ │ │ │ ├── filterbank_util.c │ │ │ │ ├── filterbank_util.h │ │ │ │ ├── frontend.c │ │ │ │ ├── frontend.h │ │ │ │ ├── frontend_io.c │ │ │ │ ├── frontend_io.h │ │ │ │ ├── frontend_main.c │ │ │ │ ├── frontend_memmap_generator.c │ │ │ │ ├── frontend_memmap_main.c │ │ │ │ ├── frontend_util.c │ │ │ │ ├── frontend_util.h │ │ │ │ ├── log_lut.c │ │ │ │ ├── log_lut.h │ │ │ │ ├── log_scale.c │ │ │ │ ├── log_scale.h │ │ │ │ ├── log_scale_io.c │ │ │ │ ├── log_scale_io.h │ │ │ │ ├── log_scale_util.c │ │ │ │ ├── log_scale_util.h │ │ │ │ ├── noise_reduction.c │ │ │ │ ├── noise_reduction.h │ │ │ │ ├── noise_reduction_io.c │ │ │ │ ├── noise_reduction_io.h │ │ │ │ ├── noise_reduction_util.c │ │ │ │ ├── noise_reduction_util.h │ │ │ │ ├── pcan_gain_control.c │ │ │ │ ├── pcan_gain_control.h │ │ │ │ ├── pcan_gain_control_util.c │ │ │ │ ├── pcan_gain_control_util.h │ │ │ │ ├── window.c │ │ │ │ ├── window.h │ │ │ │ ├── window_io.c │ │ │ │ ├── window_io.h │ │ │ │ ├── window_util.c │ │ │ │ └── window_util.h │ │ │ ├── ops │ │ │ │ └── audio_microfrontend_op.cc │ │ │ └── python │ │ │ │ ├── kernel_tests │ │ │ │ └── audio_microfrontend_op_test.py │ │ │ │ └── ops │ │ │ │ └── audio_microfrontend_op.py │ │ ├── ml_adjacent │ │ │ ├── BUILD │ │ │ ├── algo │ │ │ │ ├── BUILD │ │ │ │ ├── crop.cc │ │ │ │ ├── crop.h │ │ │ │ ├── crop_test.cc │ │ │ │ ├── flip_left_right.cc │ │ │ │ ├── flip_left_right.h │ │ │ │ ├── flip_left_right_test.cc │ │ │ │ ├── flip_up_down.cc │ │ │ │ ├── flip_up_down.h │ │ │ │ ├── flip_up_down_test.cc │ │ │ │ ├── image_utils.h │ │ │ │ ├── per_image_standardization.cc │ │ │ │ ├── per_image_standardization.h │ │ │ │ ├── per_image_standardization_test.cc │ │ │ │ ├── resize.cc │ │ │ │ ├── resize.h │ │ │ │ ├── resize_test.cc │ │ │ │ ├── rgb_to_grayscale.cc │ │ │ │ ├── rgb_to_grayscale.h │ │ │ │ ├── rgb_to_grayscale_test.cc │ │ │ │ ├── rgb_to_yuv.cc │ │ │ │ ├── rgb_to_yuv.h │ │ │ │ ├── rgb_to_yuv_test.cc │ │ │ │ ├── rotate.cc │ │ │ │ ├── rotate.h │ │ │ │ ├── rotate_test.cc │ │ │ │ ├── yuv_to_rgb.cc │ │ │ │ ├── yuv_to_rgb.h │ │ │ │ └── yuv_to_rgb_test.cc │ │ │ ├── data │ │ │ │ ├── BUILD │ │ │ │ ├── owning_vector_ref.cc │ │ │ │ ├── owning_vector_ref.h │ │ │ │ └── owning_vector_ref_test.cc │ │ │ ├── lib.h │ │ │ └── tflite │ │ │ │ ├── BUILD │ │ │ │ ├── extern_call.cc │ │ │ │ ├── extern_call.h │ │ │ │ ├── extern_call_test.cc │ │ │ │ ├── tfl_tensor_ref.cc │ │ │ │ ├── tfl_tensor_ref.h │ │ │ │ └── tfl_tensor_ref_test.cc │ │ ├── remat │ │ │ ├── BUILD │ │ │ ├── metadata_util.cc │ │ │ ├── metadata_util.h │ │ │ └── metadata_util_test.cc │ │ └── resource │ │ │ ├── BUILD │ │ │ ├── initialization_status.cc │ │ │ ├── initialization_status.h │ │ │ ├── lookup_interfaces.h │ │ │ ├── lookup_util.h │ │ │ ├── resource_base.h │ │ │ ├── resource_variable.cc │ │ │ ├── resource_variable.h │ │ │ ├── resource_variable_test.cc │ │ │ ├── static_hashtable.cc │ │ │ └── static_hashtable.h │ ├── external_cpu_backend_context.cc │ ├── external_cpu_backend_context.h │ ├── g3doc │ │ ├── _book.yaml │ │ ├── android │ │ │ ├── acceleration_service.md │ │ │ ├── delegates │ │ │ │ ├── gpu.md │ │ │ │ ├── gpu_native.md │ │ │ │ ├── gpu_task.md │ │ │ │ ├── hexagon.md │ │ │ │ └── nnapi.md │ │ │ ├── development.md │ │ │ ├── index.md │ │ │ ├── lite_build.md │ │ │ ├── play_services.md │ │ │ ├── quickstart.md │ │ │ └── tutorials │ │ │ │ ├── audio_classification.md │ │ │ │ ├── object_detection.md │ │ │ │ ├── question_answer.md │ │ │ │ └── text_classification.md │ │ ├── api_docs │ │ │ └── index.md │ │ ├── examples │ │ │ ├── audio_classification │ │ │ │ ├── images │ │ │ │ │ └── android_audio_classification.png │ │ │ │ └── overview.md │ │ │ ├── auto_complete │ │ │ │ ├── images │ │ │ │ │ ├── context_window.png │ │ │ │ │ └── tflite_workflow.png │ │ │ │ └── overview.md │ │ │ ├── bert_qa │ │ │ │ ├── images │ │ │ │ │ └── screenshot.gif │ │ │ │ └── overview.md │ │ │ ├── image_classification │ │ │ │ ├── images │ │ │ │ │ ├── android_banana.png │ │ │ │ │ ├── build_and_execute.png │ │ │ │ │ ├── bundle_identifier.png │ │ │ │ │ ├── device_selection.png │ │ │ │ │ └── dog.png │ │ │ │ └── overview.md │ │ │ ├── images │ │ │ │ ├── audio.png │ │ │ │ ├── blank.png │ │ │ │ ├── camera.png │ │ │ │ ├── detection.png │ │ │ │ ├── image.png │ │ │ │ ├── object.png │ │ │ │ ├── output_stride.png │ │ │ │ ├── pose.png │ │ │ │ ├── segmentation.png │ │ │ │ ├── sentiment.png │ │ │ │ ├── smart_reply.png │ │ │ │ ├── tabular.png │ │ │ │ ├── text.png │ │ │ │ ├── tflite_models.png │ │ │ │ └── video.png │ │ │ ├── jax_conversion │ │ │ │ └── overview.ipynb │ │ │ ├── object_detection │ │ │ │ ├── images │ │ │ │ │ ├── android_apple_banana.png │ │ │ │ │ └── false_positive.png │ │ │ │ └── overview.md │ │ │ ├── on_device_training │ │ │ │ └── overview.ipynb │ │ │ ├── optical_character_recognition │ │ │ │ ├── images │ │ │ │ │ └── screenshot.gif │ │ │ │ └── overview.md │ │ │ ├── pose_estimation │ │ │ │ └── overview.md │ │ │ ├── recommendation │ │ │ │ ├── images │ │ │ │ │ └── screenshot.gif │ │ │ │ └── overview.md │ │ │ ├── reinforcement_learning │ │ │ │ ├── images │ │ │ │ │ └── screenshot.gif │ │ │ │ └── overview.md │ │ │ ├── segmentation │ │ │ │ ├── images │ │ │ │ │ └── segmentation.gif │ │ │ │ └── overview.md │ │ │ ├── smart_reply │ │ │ │ ├── images │ │ │ │ │ └── smart_reply.gif │ │ │ │ └── overview.md │ │ │ ├── style_transfer │ │ │ │ └── overview.ipynb │ │ │ ├── super_resolution │ │ │ │ └── overview.ipynb │ │ │ ├── text_classification │ │ │ │ ├── images │ │ │ │ │ └── screenshot.gif │ │ │ │ └── overview.md │ │ │ └── video_classification │ │ │ │ └── overview.md │ │ ├── guide │ │ │ ├── authoring.ipynb │ │ │ ├── build_arm.md │ │ │ ├── build_cmake.md │ │ │ ├── build_cmake_arm.md │ │ │ ├── build_cmake_pip.md │ │ │ ├── build_ios.md │ │ │ ├── faq.md │ │ │ ├── index.md │ │ │ ├── inference.md │ │ │ ├── ios.md │ │ │ ├── model_analyzer.ipynb │ │ │ ├── op_select_allowlist.md │ │ │ ├── ops_compatibility.md │ │ │ ├── ops_custom.md │ │ │ ├── ops_select.md │ │ │ ├── ops_version.md │ │ │ ├── python.md │ │ │ ├── reduce_binary_size.md │ │ │ ├── roadmap.md │ │ │ └── signatures.ipynb │ │ ├── images │ │ │ ├── acceleration │ │ │ │ ├── acceleration_service.png │ │ │ │ └── acceleration_service_steps.png │ │ │ ├── build │ │ │ │ └── build_workflow_diag.png │ │ │ ├── convert │ │ │ │ ├── convert.png │ │ │ │ ├── convert_workflow_diag.png │ │ │ │ ├── model_with_metadata.png │ │ │ │ ├── op_fusion.png │ │ │ │ ├── op_fusion_banner.jpg │ │ │ │ ├── tf_operators_relationships.png │ │ │ │ └── workflow.svg │ │ │ ├── ios │ │ │ │ ├── build_and_execute.png │ │ │ │ ├── bundle_identifier.png │ │ │ │ └── device_selection.png │ │ │ ├── landing-page │ │ │ │ ├── ai_in_motion.png │ │ │ │ ├── assistant_logo.png │ │ │ │ ├── detect_crop_disease_in_africa.png │ │ │ │ ├── facial_contour_detection.png │ │ │ │ ├── fishbrain_logo.png │ │ │ │ ├── fishbrain_logo_big.png │ │ │ │ ├── gboard_logo.png │ │ │ │ ├── gmail_logo.png │ │ │ │ ├── loseit_logo.png │ │ │ │ ├── loseit_logo_big.png │ │ │ │ ├── nest_logo.png │ │ │ │ ├── photos_logo.png │ │ │ │ ├── shazam_logo.png │ │ │ │ └── vsco_logo.png │ │ │ └── performance │ │ │ │ ├── accuracy_vs_latency.png │ │ │ │ ├── model_size_vs_accuracy.png │ │ │ │ ├── tflite_delegate_graph_1.png │ │ │ │ └── tflite_delegate_graph_2.png │ │ ├── inference_with_metadata │ │ │ ├── codegen.md │ │ │ ├── lite_support.md │ │ │ ├── overview.md │ │ │ └── task_library │ │ │ │ ├── audio_classifier.md │ │ │ │ ├── bert_nl_classifier.md │ │ │ │ ├── bert_question_answerer.md │ │ │ │ ├── customized_task_api.md │ │ │ │ ├── image_classifier.md │ │ │ │ ├── image_embedder.md │ │ │ │ ├── image_searcher.md │ │ │ │ ├── image_segmenter.md │ │ │ │ ├── images │ │ │ │ ├── android_task_api.svg │ │ │ │ ├── detection-output.png │ │ │ │ ├── dogs.jpg │ │ │ │ ├── ios_task_api.svg │ │ │ │ ├── native_task_api.svg │ │ │ │ ├── plane.jpg │ │ │ │ ├── prebuilt_task_apis.svg │ │ │ │ ├── segmentation-output.png │ │ │ │ └── sparrow.jpg │ │ │ │ ├── nl_classifier.md │ │ │ │ ├── object_detector.md │ │ │ │ ├── overview.md │ │ │ │ ├── text_embedder.md │ │ │ │ └── text_searcher.md │ │ ├── ios │ │ │ └── delegates │ │ │ │ └── gpu.md │ │ ├── microcontrollers │ │ │ ├── build_convert.md │ │ │ ├── get_started_low_level.md │ │ │ ├── index.md │ │ │ └── library.md │ │ ├── models │ │ │ ├── _index.yaml │ │ │ ├── build │ │ │ │ └── index.md │ │ │ ├── convert │ │ │ │ ├── api_updates.md │ │ │ │ ├── convert_models.md │ │ │ │ ├── index.md │ │ │ │ ├── metadata.md │ │ │ │ ├── metadata_writer_tutorial.ipynb │ │ │ │ ├── operation_fusion.md │ │ │ │ └── rnn.md │ │ │ ├── modify │ │ │ │ └── model_maker │ │ │ │ │ ├── audio_classification.ipynb │ │ │ │ │ ├── image_classification.ipynb │ │ │ │ │ ├── index.md │ │ │ │ │ ├── object_detection.ipynb │ │ │ │ │ ├── question_answer.ipynb │ │ │ │ │ ├── speech_recognition.ipynb │ │ │ │ │ ├── text_classification.ipynb │ │ │ │ │ └── text_searcher.ipynb │ │ │ └── trained │ │ │ │ └── index.md │ │ ├── performance │ │ │ ├── best_practices.md │ │ │ ├── coreml_delegate.md │ │ │ ├── delegates.md │ │ │ ├── gpu.md │ │ │ ├── images │ │ │ │ ├── android_gpu_demo.gif │ │ │ │ ├── as_select_profiling_mode.png │ │ │ │ ├── as_traces.png │ │ │ │ ├── delegate_runtime.png │ │ │ │ ├── optimization.jpg │ │ │ │ ├── perfetto_traces.png │ │ │ │ ├── quantization_decision_tree.png │ │ │ │ ├── tflite_delegate.png │ │ │ │ ├── xcode_profile_environment.png │ │ │ │ └── xcode_traces.png │ │ │ ├── implementing_delegate.md │ │ │ ├── measurement.md │ │ │ ├── model_optimization.md │ │ │ ├── post_training_float16_quant.ipynb │ │ │ ├── post_training_integer_quant.ipynb │ │ │ ├── post_training_integer_quant_16x8.ipynb │ │ │ ├── post_training_quant.ipynb │ │ │ ├── post_training_quantization.md │ │ │ ├── quantization_debugger.ipynb │ │ │ └── quantization_spec.md │ │ ├── r1 │ │ │ ├── convert │ │ │ │ ├── cmdline_examples.md │ │ │ │ ├── cmdline_reference.md │ │ │ │ ├── index.md │ │ │ │ └── python_api.md │ │ │ └── images │ │ │ │ └── convert │ │ │ │ ├── sample_after.png │ │ │ │ ├── sample_before.png │ │ │ │ └── workflow.svg │ │ ├── tf_docs_reviews.gwsq │ │ ├── tools │ │ │ ├── BUILD │ │ │ ├── build_java_api_docs.py │ │ │ └── build_py_api_docs.py │ │ └── tutorials │ │ │ └── pose_classification.ipynb │ ├── generate-pc.sh │ ├── graph_info.cc │ ├── graph_info.h │ ├── graph_info_test.cc │ ├── internal │ │ ├── BUILD │ │ └── signature_def.h │ ├── interpreter.h │ ├── interpreter_builder.h │ ├── interpreter_options.h │ ├── interpreter_test.cc │ ├── interpreter_test_util.h │ ├── ios │ │ ├── BUILD.apple │ │ ├── TensorFlowLiteC.h │ │ ├── TensorFlowLiteC.md │ │ ├── TensorFlowLiteC.podspec │ │ ├── TensorFlowLiteC.podspec.template │ │ ├── TensorFlowLiteSelectTfOps.md │ │ ├── TensorFlowLiteSelectTfOps.podspec │ │ ├── TensorFlowLiteSelectTfOps.podspec.template │ │ ├── allowlist_TensorFlowLiteC.txt │ │ ├── allowlist_TensorFlowLiteCCoreML.txt │ │ ├── allowlist_TensorFlowLiteCMetal.txt │ │ ├── build_frameworks.sh │ │ ├── extract_object_files.py │ │ ├── extract_object_files_main.py │ │ ├── extract_object_files_test.py │ │ ├── hide_symbols_with_allowlist.sh │ │ ├── hide_xcframework_symbols_with_allowlist.sh │ │ ├── ios.bzl │ │ └── testdata │ │ │ ├── duplicate_names │ │ │ ├── foo.o │ │ │ ├── foo_1.o │ │ │ ├── foo_2.o │ │ │ └── input.a │ │ │ ├── extended_filename │ │ │ ├── input.a │ │ │ ├── long_file_name_with_extended_format.o │ │ │ └── short.o │ │ │ ├── odd_bytes │ │ │ ├── even.o │ │ │ ├── input.a │ │ │ └── odd.o │ │ │ ├── simple │ │ │ ├── bar.o │ │ │ ├── foo.o │ │ │ └── input.a │ │ │ └── skip_same_file │ │ │ ├── foo.o │ │ │ └── input.a │ ├── java │ │ ├── AndroidManifest.xml │ │ ├── AndroidManifestApi.xml │ │ ├── AndroidManifestGpu.xml │ │ ├── AndroidManifestGpuApi.xml │ │ ├── BUILD │ │ ├── aar_with_jni.bzl │ │ ├── build_aar_for_release.sh │ │ ├── demo │ │ │ ├── .gitignore │ │ │ ├── README.md │ │ │ ├── app │ │ │ │ ├── build.gradle │ │ │ │ └── src │ │ │ │ │ └── main │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── assets │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── labels_imagenet_slim.txt │ │ │ │ │ └── labels_mobilenet_quant_v1_224.txt │ │ │ │ │ ├── java │ │ │ │ │ └── com │ │ │ │ │ │ └── example │ │ │ │ │ │ └── android │ │ │ │ │ │ └── tflitecamerademo │ │ │ │ │ │ ├── AutoFitTextureView.java │ │ │ │ │ │ ├── Camera2BasicFragment.java │ │ │ │ │ │ ├── CameraActivity.java │ │ │ │ │ │ ├── ImageClassifier.java │ │ │ │ │ │ ├── ImageClassifierFloatInception.java │ │ │ │ │ │ ├── ImageClassifierFloatMobileNet.java │ │ │ │ │ │ └── ImageClassifierQuantizedMobileNet.java │ │ │ │ │ └── res │ │ │ │ │ ├── drawable-hdpi │ │ │ │ │ ├── ic_action_info.png │ │ │ │ │ ├── ic_launcher.png │ │ │ │ │ └── tile.9.png │ │ │ │ │ ├── drawable-mdpi │ │ │ │ │ ├── ic_action_info.png │ │ │ │ │ └── ic_launcher.png │ │ │ │ │ ├── drawable-xhdpi │ │ │ │ │ ├── ic_action_info.png │ │ │ │ │ └── ic_launcher.png │ │ │ │ │ ├── drawable-xxhdpi │ │ │ │ │ ├── ic_action_info.png │ │ │ │ │ ├── ic_launcher.png │ │ │ │ │ └── logo.png │ │ │ │ │ ├── drawable │ │ │ │ │ └── item_selector.xml │ │ │ │ │ ├── layout-land │ │ │ │ │ └── fragment_camera2_basic.xml │ │ │ │ │ ├── layout-v26 │ │ │ │ │ └── fragment_camera2_basic.xml │ │ │ │ │ ├── layout │ │ │ │ │ ├── activity_camera.xml │ │ │ │ │ ├── fragment_camera2_basic.xml │ │ │ │ │ └── listview_row.xml │ │ │ │ │ ├── values-sw600dp │ │ │ │ │ ├── template-dimens.xml │ │ │ │ │ └── template-styles.xml │ │ │ │ │ ├── values-v11 │ │ │ │ │ └── template-styles.xml │ │ │ │ │ ├── values-v21 │ │ │ │ │ ├── base-colors.xml │ │ │ │ │ └── base-template-styles.xml │ │ │ │ │ └── values │ │ │ │ │ ├── base-strings.xml │ │ │ │ │ ├── colors.xml │ │ │ │ │ ├── strings.xml │ │ │ │ │ ├── styles.xml │ │ │ │ │ ├── template-dimens.xml │ │ │ │ │ └── template-styles.xml │ │ │ ├── build.gradle │ │ │ ├── gradle.properties │ │ │ ├── gradle │ │ │ │ └── wrapper │ │ │ │ │ ├── gradle-wrapper.jar │ │ │ │ │ └── gradle-wrapper.properties │ │ │ ├── gradlew │ │ │ ├── gradlew.bat │ │ │ └── settings.gradle │ │ ├── gpu_version_script.lds │ │ ├── jni │ │ │ └── BUILD │ │ ├── ovic │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── Winner_OSS_Template.md │ │ │ ├── demo │ │ │ │ ├── app │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── OvicBenchmarkerActivity.java │ │ │ │ │ ├── build.gradle │ │ │ │ │ └── res │ │ │ │ │ │ ├── drawable-mdpi │ │ │ │ │ │ └── ic_launcher.png │ │ │ │ │ │ ├── drawable-xhdpi │ │ │ │ │ │ └── ic_launcher.png │ │ │ │ │ │ ├── drawable │ │ │ │ │ │ └── start_button_color.xml │ │ │ │ │ │ ├── layout │ │ │ │ │ │ └── activity_main.xml │ │ │ │ │ │ └── values │ │ │ │ │ │ ├── dimens.xml │ │ │ │ │ │ └── strings.xml │ │ │ │ ├── build.gradle │ │ │ │ ├── gradle.properties │ │ │ │ ├── gradle │ │ │ │ │ └── wrapper │ │ │ │ │ │ ├── gradle-wrapper.jar │ │ │ │ │ │ └── gradle-wrapper.properties │ │ │ │ ├── gradlew │ │ │ │ ├── gradlew.bat │ │ │ │ └── settings.gradle │ │ │ └── src │ │ │ │ ├── main │ │ │ │ └── java │ │ │ │ │ └── org │ │ │ │ │ └── tensorflow │ │ │ │ │ └── ovic │ │ │ │ │ ├── BoundingBox.java │ │ │ │ │ ├── OvicBenchmarker.java │ │ │ │ │ ├── OvicClassificationResult.java │ │ │ │ │ ├── OvicClassifier.java │ │ │ │ │ ├── OvicClassifierBenchmarker.java │ │ │ │ │ ├── OvicDetectionResult.java │ │ │ │ │ ├── OvicDetector.java │ │ │ │ │ ├── OvicDetectorBenchmarker.java │ │ │ │ │ └── OvicValidator.java │ │ │ │ ├── test │ │ │ │ └── java │ │ │ │ │ └── org │ │ │ │ │ └── tensorflow │ │ │ │ │ └── ovic │ │ │ │ │ ├── OvicClassifierTest.java │ │ │ │ │ └── OvicDetectorTest.java │ │ │ │ └── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── coco_labels.txt │ │ │ │ └── labels.txt │ │ ├── proguard.flags │ │ ├── src │ │ │ ├── main │ │ │ │ ├── java │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ ├── DataType.java │ │ │ │ │ │ ├── DataTypeUtils.java │ │ │ │ │ │ ├── Delegate.java │ │ │ │ │ │ ├── DelegateFactory.java │ │ │ │ │ │ ├── Interpreter.java │ │ │ │ │ │ ├── InterpreterApi.java │ │ │ │ │ │ ├── InterpreterFactory.java │ │ │ │ │ │ ├── InterpreterFactoryApi.java │ │ │ │ │ │ ├── InterpreterFactoryImpl.java │ │ │ │ │ │ ├── InterpreterImpl.java │ │ │ │ │ │ ├── NativeInterpreterWrapper.java │ │ │ │ │ │ ├── NativeInterpreterWrapperExperimental.java │ │ │ │ │ │ ├── NativeSignatureRunnerWrapper.java │ │ │ │ │ │ ├── RuntimeFlavor.java │ │ │ │ │ │ ├── Tensor.java │ │ │ │ │ │ ├── TensorFlowLite.java │ │ │ │ │ │ ├── TensorImpl.java │ │ │ │ │ │ ├── acceleration │ │ │ │ │ │ └── ValidatedAccelerationConfig.java │ │ │ │ │ │ ├── annotations │ │ │ │ │ │ └── UsedByReflection.java │ │ │ │ │ │ └── package-info.java │ │ │ │ └── native │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── exported_symbols.lds │ │ │ │ │ ├── interpreter_factory_impl_jni.cc │ │ │ │ │ ├── jni_utils.cc │ │ │ │ │ ├── jni_utils.h │ │ │ │ │ ├── nativeinterpreterwrapper_jni.cc │ │ │ │ │ ├── nativeinterpreterwrapperexperimental_jni.cc │ │ │ │ │ ├── nativesignaturerunner_jni.cc │ │ │ │ │ ├── op_resolver_lazy_delegate_proxy.cc │ │ │ │ │ ├── op_resolver_lazy_delegate_proxy.h │ │ │ │ │ ├── tensor_jni.cc │ │ │ │ │ ├── tensorflow_lite_jni.cc │ │ │ │ │ └── version_script.lds │ │ │ ├── test │ │ │ │ ├── java │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ ├── DataTypeTest.java │ │ │ │ │ │ ├── InterpreterApiNoRuntimeTest.java │ │ │ │ │ │ ├── InterpreterApiTest.java │ │ │ │ │ │ ├── InterpreterCustomizedAndroidBuildTest.java │ │ │ │ │ │ ├── InterpreterFlexTest.java │ │ │ │ │ │ ├── InterpreterFlexWithCustomOpsTest.java │ │ │ │ │ │ ├── InterpreterMobileNetTest.java │ │ │ │ │ │ ├── InterpreterTest.java │ │ │ │ │ │ ├── InterpreterTestHelper.java │ │ │ │ │ │ ├── NativeInterpreterWrapperTest.java │ │ │ │ │ │ ├── NnApiDelegateNativeTest.java │ │ │ │ │ │ ├── SupportedFeatures.java │ │ │ │ │ │ ├── TensorFlowLiteInvalidNativeLibTest.java │ │ │ │ │ │ ├── TensorFlowLiteNoNativeLibTest.java │ │ │ │ │ │ ├── TensorFlowLiteTest.java │ │ │ │ │ │ ├── TensorTest.java │ │ │ │ │ │ ├── TestInit.java │ │ │ │ │ │ ├── TestUtils.java │ │ │ │ │ │ ├── gpu │ │ │ │ │ │ ├── CompatibilityListTest.java │ │ │ │ │ │ └── GpuDelegateTest.java │ │ │ │ │ │ └── nnapi │ │ │ │ │ │ └── NnApiDelegateTest.java │ │ │ │ └── native │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── interpreter_test_jni.cc │ │ │ │ │ ├── nnapi_delegate_test_jni.cc │ │ │ │ │ ├── supported_features_jni.cc │ │ │ │ │ └── test_init_jni.cc │ │ │ ├── testdata │ │ │ │ ├── add.bin │ │ │ │ ├── add_unknown_dimensions.bin │ │ │ │ ├── float32.bin │ │ │ │ ├── grace_hopper_224.jpg │ │ │ │ ├── int32.bin │ │ │ │ ├── int64.bin │ │ │ │ ├── invalid_model.bin │ │ │ │ ├── mul_add_signature_def.bin │ │ │ │ ├── multi_signature_def.bin │ │ │ │ ├── quantized.bin │ │ │ │ ├── string.bin │ │ │ │ ├── string_scalar.bin │ │ │ │ ├── tile_with_bool_input.bin │ │ │ │ ├── uint8.bin │ │ │ │ └── with_custom_op.lite │ │ │ └── testhelper │ │ │ │ └── java │ │ │ │ └── org │ │ │ │ └── tensorflow │ │ │ │ └── lite │ │ │ │ ├── BUILD │ │ │ │ └── TestHelper.java │ │ └── tflite_version_script.lds │ ├── kernels │ │ ├── BUILD │ │ ├── CMakeLists.txt │ │ ├── acceleration_test_util.cc │ │ ├── acceleration_test_util.h │ │ ├── acceleration_test_util_internal.cc │ │ ├── acceleration_test_util_internal.h │ │ ├── acceleration_test_util_internal_test.cc │ │ ├── activations.cc │ │ ├── activations_test.cc │ │ ├── add.cc │ │ ├── add_n.cc │ │ ├── add_n_test.cc │ │ ├── add_n_test_common.h │ │ ├── add_test.cc │ │ ├── arg_min_max.cc │ │ ├── arg_min_max_test.cc │ │ ├── assign_variable.cc │ │ ├── atan2.cc │ │ ├── atan2_custom.cc │ │ ├── atan2_custom_test.cc │ │ ├── atan2_test.cc │ │ ├── audio_spectrogram.cc │ │ ├── audio_spectrogram_test.cc │ │ ├── basic_rnn.cc │ │ ├── basic_rnn_test.cc │ │ ├── batch_matmul.cc │ │ ├── batch_matmul_test.cc │ │ ├── batch_to_space_nd.cc │ │ ├── batch_to_space_nd_test.cc │ │ ├── bidirectional_sequence_lstm.cc │ │ ├── bidirectional_sequence_lstm_test.cc │ │ ├── bidirectional_sequence_rnn.cc │ │ ├── bidirectional_sequence_rnn_test.cc │ │ ├── bitcast.cc │ │ ├── bitcast_test.cc │ │ ├── bitwise_xor.cc │ │ ├── bitwise_xor_test.cc │ │ ├── broadcast_args.cc │ │ ├── broadcast_args_test.cc │ │ ├── broadcast_to.cc │ │ ├── broadcast_to_test.cc │ │ ├── bucketize.cc │ │ ├── bucketize_test.cc │ │ ├── builtin_op_kernels.h │ │ ├── builtin_ops_list.inc │ │ ├── call_once.cc │ │ ├── call_once_test.cc │ │ ├── cast.cc │ │ ├── cast_test.cc │ │ ├── cast_test_common.h │ │ ├── ceil.cc │ │ ├── ceil_test.cc │ │ ├── comparisons.cc │ │ ├── comparisons_test.cc │ │ ├── complex_support.cc │ │ ├── complex_support_test.cc │ │ ├── concatenation.cc │ │ ├── concatenation_test.cc │ │ ├── control_flow_common.h │ │ ├── conv.cc │ │ ├── conv3d.cc │ │ ├── conv3d_test.cc │ │ ├── conv3d_transpose.cc │ │ ├── conv3d_transpose_test.cc │ │ ├── conv_mem_test.cc │ │ ├── conv_test.cc │ │ ├── cpu_backend_context.cc │ │ ├── cpu_backend_context.h │ │ ├── cpu_backend_gemm.h │ │ ├── cpu_backend_gemm_custom_gemv.h │ │ ├── cpu_backend_gemm_eigen.cc │ │ ├── cpu_backend_gemm_eigen.h │ │ ├── cpu_backend_gemm_gemmlowp.h │ │ ├── cpu_backend_gemm_params.h │ │ ├── cpu_backend_gemm_ruy.h │ │ ├── cpu_backend_gemm_test.cc │ │ ├── cpu_backend_gemm_x86.h │ │ ├── cpu_backend_threadpool.h │ │ ├── cpu_backend_threadpool_test.cc │ │ ├── ctc │ │ │ ├── BUILD │ │ │ ├── ctc_beam_entry.h │ │ │ ├── ctc_beam_scorer.h │ │ │ ├── ctc_beam_search.h │ │ │ ├── ctc_beam_search_decoder.cc │ │ │ ├── ctc_beam_search_decoder_test.cc │ │ │ ├── ctc_decoder.h │ │ │ ├── ctc_loss_util.h │ │ │ └── top_n.h │ │ ├── cumsum.cc │ │ ├── cumsum_test.cc │ │ ├── custom_ops_register.h │ │ ├── densify.cc │ │ ├── densify_test.cc │ │ ├── deprecated_backends.cc │ │ ├── depth_to_space.cc │ │ ├── depth_to_space_test.cc │ │ ├── depthwise_conv.cc │ │ ├── depthwise_conv_hybrid_test.cc │ │ ├── depthwise_conv_test.cc │ │ ├── dequantize.cc │ │ ├── dequantize.h │ │ ├── dequantize_test.cc │ │ ├── detection_postprocess.cc │ │ ├── detection_postprocess_test.cc │ │ ├── dilate.cc │ │ ├── dilate_test.cc │ │ ├── div.cc │ │ ├── div_test.cc │ │ ├── dynamic_update_slice.cc │ │ ├── dynamic_update_slice_test.cc │ │ ├── eigen_support.cc │ │ ├── eigen_support.h │ │ ├── eigen_support_test.cc │ │ ├── elementwise.cc │ │ ├── elementwise_test.cc │ │ ├── embedding_lookup.cc │ │ ├── embedding_lookup_sparse.cc │ │ ├── embedding_lookup_sparse_test.cc │ │ ├── embedding_lookup_test.cc │ │ ├── exp.cc │ │ ├── exp_test.cc │ │ ├── expand_dims.cc │ │ ├── expand_dims_test.cc │ │ ├── fake_quant.cc │ │ ├── fake_quant_test.cc │ │ ├── fill.cc │ │ ├── fill_test.cc │ │ ├── floor.cc │ │ ├── floor_div.cc │ │ ├── floor_div_test.cc │ │ ├── floor_mod.cc │ │ ├── floor_mod_test.cc │ │ ├── floor_mod_test_common.h │ │ ├── floor_test.cc │ │ ├── fully_connected.cc │ │ ├── fully_connected.h │ │ ├── fully_connected_4bit_test.cc │ │ ├── fully_connected_test.cc │ │ ├── gather.cc │ │ ├── gather_nd.cc │ │ ├── gather_nd_test.cc │ │ ├── gather_test.cc │ │ ├── gradient │ │ │ ├── BUILD │ │ │ ├── bcast_grad_args.cc │ │ │ ├── bcast_grad_args.h │ │ │ ├── bcast_grad_args_test.cc │ │ │ ├── gradient_ops.cc │ │ │ ├── gradient_ops.h │ │ │ ├── gradient_ops_wrapper.cc │ │ │ └── pywrap_gradient_ops.pyi │ │ ├── gru_cell.cc │ │ ├── gru_cell.h │ │ ├── hashtable.cc │ │ ├── hashtable.md │ │ ├── hashtable_find.cc │ │ ├── hashtable_import.cc │ │ ├── hashtable_lookup.cc │ │ ├── hashtable_lookup_test.cc │ │ ├── hashtable_ops_test.cc │ │ ├── hashtable_size.cc │ │ ├── if.cc │ │ ├── if_test.cc │ │ ├── internal │ │ │ ├── BUILD │ │ │ ├── averagepool_quantized_test.cc │ │ │ ├── batch_to_space_nd_test.cc │ │ │ ├── common.cc │ │ │ ├── common.h │ │ │ ├── compatibility.h │ │ │ ├── constants.h │ │ │ ├── conv_per_channel_quantized_16x8_test.cc │ │ │ ├── cppmath.h │ │ │ ├── depthwiseconv_float_test.cc │ │ │ ├── depthwiseconv_per_channel_quantized_16x8_test.cc │ │ │ ├── depthwiseconv_per_channel_quantized_test.cc │ │ │ ├── depthwiseconv_quantized_test.cc │ │ │ ├── kernel_utils.cc │ │ │ ├── kernel_utils.h │ │ │ ├── legacy_types.h │ │ │ ├── log_quantized_test.cc │ │ │ ├── logsoftmax_quantized_test.cc │ │ │ ├── max.h │ │ │ ├── maxpool_quantized_test.cc │ │ │ ├── mfcc.cc │ │ │ ├── mfcc.h │ │ │ ├── mfcc_dct.cc │ │ │ ├── mfcc_dct.h │ │ │ ├── mfcc_mel_filterbank.cc │ │ │ ├── mfcc_mel_filterbank.h │ │ │ ├── min.h │ │ │ ├── non_max_suppression_test.cc │ │ │ ├── opaque_tensor_ctypes.cc │ │ │ ├── opaque_tensor_ctypes.h │ │ │ ├── optimized │ │ │ │ ├── 4bit │ │ │ │ │ ├── fully_connected_common.h │ │ │ │ │ ├── fully_connected_reference.cc │ │ │ │ │ ├── fully_connected_reference.h │ │ │ │ │ ├── fully_connected_reference_impl.h │ │ │ │ │ ├── neon_fully_connected.cc │ │ │ │ │ ├── neon_fully_connected.h │ │ │ │ │ ├── neon_fully_connected_aarch64_nosdot.cc │ │ │ │ │ ├── neon_fully_connected_aarch64_sdot.cc │ │ │ │ │ ├── neon_fully_connected_arm32.cc │ │ │ │ │ ├── neon_fully_connected_impl.h │ │ │ │ │ ├── sse_fully_connected.cc │ │ │ │ │ ├── sse_fully_connected.h │ │ │ │ │ └── sse_fully_connected_impl.h │ │ │ │ ├── avx2_quantization_utils.h │ │ │ │ ├── avx2_quantization_utils_test.cc │ │ │ │ ├── batch_matmul.h │ │ │ │ ├── cpu_check.cc │ │ │ │ ├── cpu_check.h │ │ │ │ ├── depthwiseconv_3x3_filter_common.h │ │ │ │ ├── depthwiseconv_float.h │ │ │ │ ├── depthwiseconv_multithread.h │ │ │ │ ├── depthwiseconv_uint8.h │ │ │ │ ├── depthwiseconv_uint8_3x3_filter.h │ │ │ │ ├── depthwiseconv_uint8_transitional.h │ │ │ │ ├── eigen_spatial_convolutions.h │ │ │ │ ├── fully_connected_4bit.h │ │ │ │ ├── im2col_utils.h │ │ │ │ ├── integer_ops │ │ │ │ │ ├── README.md │ │ │ │ │ ├── add.h │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── depthwise_conv.h │ │ │ │ │ ├── depthwise_conv_3x3_filter.h │ │ │ │ │ ├── depthwise_conv_hybrid.h │ │ │ │ │ ├── depthwise_conv_hybrid_3x3_filter.h │ │ │ │ │ ├── fully_connected.h │ │ │ │ │ ├── leaky_relu.h │ │ │ │ │ ├── lut.h │ │ │ │ │ ├── mean.h │ │ │ │ │ ├── mul.h │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── sub.h │ │ │ │ │ └── transpose_conv.h │ │ │ │ ├── legacy_optimized_ops.h │ │ │ │ ├── multithreaded_conv.h │ │ │ │ ├── neon_check.h │ │ │ │ ├── neon_tensor_utils.cc │ │ │ │ ├── neon_tensor_utils.h │ │ │ │ ├── neon_tensor_utils_impl.h │ │ │ │ ├── optimized_4bit_test.cc │ │ │ │ ├── optimized_ops.h │ │ │ │ ├── optimized_ops_utils.h │ │ │ │ ├── reduce.h │ │ │ │ ├── reduce_utils.h │ │ │ │ ├── reduce_utils_test.cc │ │ │ │ ├── resize_bilinear.h │ │ │ │ ├── sparse_ops │ │ │ │ │ └── fully_connected.h │ │ │ │ ├── sse_check.h │ │ │ │ ├── sse_tensor_utils.cc │ │ │ │ ├── sse_tensor_utils.h │ │ │ │ └── sse_tensor_utils_impl.h │ │ │ ├── per_channel_dequantize_test.cc │ │ │ ├── per_channel_quantize_test.cc │ │ │ ├── portable_tensor.h │ │ │ ├── portable_tensor_utils.cc │ │ │ ├── portable_tensor_utils.h │ │ │ ├── quantization_util.cc │ │ │ ├── quantization_util.h │ │ │ ├── quantization_util_test.cc │ │ │ ├── reduce_common.h │ │ │ ├── reference │ │ │ │ ├── add.h │ │ │ │ ├── add_n.h │ │ │ │ ├── arg_min_max.h │ │ │ │ ├── batch_matmul.h │ │ │ │ ├── batch_to_space_nd.h │ │ │ │ ├── binary_function.h │ │ │ │ ├── broadcast_args.h │ │ │ │ ├── broadcast_to.h │ │ │ │ ├── cast.h │ │ │ │ ├── ceil.h │ │ │ │ ├── comparisons.cc │ │ │ │ ├── comparisons.h │ │ │ │ ├── concatenation.h │ │ │ │ ├── conv.h │ │ │ │ ├── conv3d.h │ │ │ │ ├── conv3d_transpose.h │ │ │ │ ├── cumsum.h │ │ │ │ ├── densify.h │ │ │ │ ├── depth_to_space.h │ │ │ │ ├── depthwiseconv_float.h │ │ │ │ ├── depthwiseconv_uint8.h │ │ │ │ ├── dequantize.h │ │ │ │ ├── div.h │ │ │ │ ├── elu.h │ │ │ │ ├── exp.h │ │ │ │ ├── fill.h │ │ │ │ ├── floor.h │ │ │ │ ├── floor_div.h │ │ │ │ ├── floor_mod.h │ │ │ │ ├── fully_connected.h │ │ │ │ ├── gather.h │ │ │ │ ├── gelu.h │ │ │ │ ├── hard_swish.h │ │ │ │ ├── integer_ops │ │ │ │ │ ├── README.md │ │ │ │ │ ├── add.h │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── depthwise_conv.h │ │ │ │ │ ├── dequantize.h │ │ │ │ │ ├── fully_connected.h │ │ │ │ │ ├── l2normalization.h │ │ │ │ │ ├── log_softmax.h │ │ │ │ │ ├── logistic.h │ │ │ │ │ ├── lut.h │ │ │ │ │ ├── mean.h │ │ │ │ │ ├── mul.h │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── tanh.h │ │ │ │ │ └── transpose_conv.h │ │ │ │ ├── l2normalization.h │ │ │ │ ├── leaky_relu.h │ │ │ │ ├── legacy_reference_ops.h │ │ │ │ ├── log_softmax.h │ │ │ │ ├── logistic.h │ │ │ │ ├── lstm_cell.h │ │ │ │ ├── maximum_minimum.h │ │ │ │ ├── mul.h │ │ │ │ ├── neg.h │ │ │ │ ├── non_max_suppression.h │ │ │ │ ├── pad.h │ │ │ │ ├── pooling.h │ │ │ │ ├── portable_tensor_utils.cc │ │ │ │ ├── portable_tensor_utils.h │ │ │ │ ├── portable_tensor_utils_impl.h │ │ │ │ ├── prelu.h │ │ │ │ ├── process_broadcast_shapes.h │ │ │ │ ├── quantize.h │ │ │ │ ├── reduce.h │ │ │ │ ├── reference_ops.h │ │ │ │ ├── requantize.h │ │ │ │ ├── resize_bilinear.h │ │ │ │ ├── resize_nearest_neighbor.h │ │ │ │ ├── round.h │ │ │ │ ├── select.h │ │ │ │ ├── slice.h │ │ │ │ ├── softmax.h │ │ │ │ ├── space_to_batch_nd.h │ │ │ │ ├── space_to_depth.h │ │ │ │ ├── sparse_ops │ │ │ │ │ └── fully_connected.h │ │ │ │ ├── strided_slice.h │ │ │ │ ├── string_comparisons.h │ │ │ │ ├── sub.h │ │ │ │ ├── svdf.h │ │ │ │ ├── tanh.h │ │ │ │ ├── transpose.h │ │ │ │ └── transpose_conv.h │ │ │ ├── resize_bilinear_test.cc │ │ │ ├── resize_nearest_neighbor_test.cc │ │ │ ├── runtime_shape.cc │ │ │ ├── runtime_shape.h │ │ │ ├── runtime_shape_test.cc │ │ │ ├── softmax_quantized_test.cc │ │ │ ├── spectrogram.cc │ │ │ ├── spectrogram.h │ │ │ ├── strided_slice_logic.h │ │ │ ├── strided_slice_logic_test.cc │ │ │ ├── tensor.h │ │ │ ├── tensor_ctypes.cc │ │ │ ├── tensor_ctypes.h │ │ │ ├── tensor_test.cc │ │ │ ├── tensor_utils.cc │ │ │ ├── tensor_utils.h │ │ │ ├── tensor_utils_test.cc │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ ├── transpose_utils.cc │ │ │ ├── transpose_utils.h │ │ │ ├── transpose_utils_test.cc │ │ │ ├── types.h │ │ │ └── utils │ │ │ │ ├── BUILD │ │ │ │ ├── sparsity_format_converter.cc │ │ │ │ ├── sparsity_format_converter.h │ │ │ │ └── sparsity_format_converter_test.cc │ │ ├── irfft2d.cc │ │ ├── irfft2d_test.cc │ │ ├── kernel_util.cc │ │ ├── kernel_util.h │ │ ├── kernel_util_test.cc │ │ ├── l2norm.cc │ │ ├── l2norm_test.cc │ │ ├── local_response_norm.cc │ │ ├── local_response_norm_test.cc │ │ ├── log_softmax_test.cc │ │ ├── logical.cc │ │ ├── logical_test.cc │ │ ├── lsh_projection.cc │ │ ├── lsh_projection_test.cc │ │ ├── lstm.cc │ │ ├── lstm_eval.cc │ │ ├── lstm_eval.h │ │ ├── lstm_eval_test.cc │ │ ├── lstm_shared.h │ │ ├── lstm_test.cc │ │ ├── matrix_diag.cc │ │ ├── matrix_diag_test.cc │ │ ├── matrix_set_diag.cc │ │ ├── matrix_set_diag_test.cc │ │ ├── maximum_minimum.cc │ │ ├── maximum_minimum_test.cc │ │ ├── mfcc.cc │ │ ├── mfcc_test.cc │ │ ├── mirror_pad.cc │ │ ├── mirror_pad_test.cc │ │ ├── mul.cc │ │ ├── mul_test.cc │ │ ├── multinomial.cc │ │ ├── multinomial_test.cc │ │ ├── neg.cc │ │ ├── neg_test.cc │ │ ├── non_max_suppression.cc │ │ ├── non_max_suppression_test.cc │ │ ├── numeric_verify.cc │ │ ├── numeric_verify_test.cc │ │ ├── one_hot.cc │ │ ├── one_hot_test.cc │ │ ├── op_macros.h │ │ ├── optional_tensor_test.cc │ │ ├── pack.cc │ │ ├── pack_test.cc │ │ ├── pad.cc │ │ ├── pad_test.cc │ │ ├── padding.h │ │ ├── parse_example │ │ │ ├── BUILD │ │ │ ├── example_proto_fast_parsing.cc │ │ │ ├── example_proto_fast_parsing.h │ │ │ ├── parse_example.cc │ │ │ ├── parse_example.h │ │ │ └── parse_example_test.cc │ │ ├── perception │ │ │ ├── BUILD │ │ │ ├── dense_image_warp.cc │ │ │ ├── dense_image_warp_test.cc │ │ │ ├── max_pool_with_argmax.cc │ │ │ ├── max_pool_with_argmax_test.cc │ │ │ ├── max_unpooling_2d.cc │ │ │ ├── max_unpooling_2d_test.cc │ │ │ ├── perception_ops.cc │ │ │ ├── perception_ops.h │ │ │ ├── perception_ops_wrapper.cc │ │ │ └── pywrap_perception_ops.pyi │ │ ├── pooling.cc │ │ ├── pooling3d.cc │ │ ├── pooling3d_test.cc │ │ ├── pooling_test.cc │ │ ├── pow.cc │ │ ├── pow_test.cc │ │ ├── pywrap_variable_ops.pyi │ │ ├── quant_basic_lstm_test.cc │ │ ├── quantize.cc │ │ ├── quantize_test.cc │ │ ├── random_ops.cc │ │ ├── random_ops_test.cc │ │ ├── random_standard_normal_custom.cc │ │ ├── random_standard_normal_custom_test.cc │ │ ├── random_uniform_custom.cc │ │ ├── random_uniform_custom_test.cc │ │ ├── range.cc │ │ ├── range_test.cc │ │ ├── rank.cc │ │ ├── rank_test.cc │ │ ├── read_variable.cc │ │ ├── reduce.cc │ │ ├── reduce_test.cc │ │ ├── reduce_test_common.h │ │ ├── reduce_window_test.cc │ │ ├── register.h │ │ ├── register_ref.cc │ │ ├── register_ref.h │ │ ├── reshape.cc │ │ ├── reshape_test.cc │ │ ├── reshape_test_common.h │ │ ├── resize_bilinear.cc │ │ ├── resize_bilinear_test.cc │ │ ├── resize_nearest_neighbor.cc │ │ ├── resize_nearest_neighbor_test.cc │ │ ├── reverse.cc │ │ ├── reverse_sequence.cc │ │ ├── reverse_sequence_test.cc │ │ ├── reverse_test.cc │ │ ├── rfft2d.cc │ │ ├── rfft2d_test.cc │ │ ├── right_shift.cc │ │ ├── right_shift_test.cc │ │ ├── rng_bit_generator.cc │ │ ├── rng_bit_generator_test.cc │ │ ├── rng_util.cc │ │ ├── rng_util.h │ │ ├── rng_util_test.cc │ │ ├── roll.cc │ │ ├── roll_test.cc │ │ ├── round.cc │ │ ├── round_test.cc │ │ ├── scatter_nd.cc │ │ ├── scatter_nd_test.cc │ │ ├── segment_sum.cc │ │ ├── segment_sum_test.cc │ │ ├── select.cc │ │ ├── select_test.cc │ │ ├── shape.cc │ │ ├── shape_test.cc │ │ ├── shim │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── op_kernel.h │ │ │ ├── shape.cc │ │ │ ├── shape.h │ │ │ ├── shape_test.cc │ │ │ ├── status_macros.h │ │ │ ├── tensor_view.h │ │ │ ├── test_op │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── simple_op.h │ │ │ │ ├── simple_tf_op.cc │ │ │ │ ├── simple_tf_op.h │ │ │ │ ├── simple_tf_op_test.cc │ │ │ │ ├── simple_tflite_op.cc │ │ │ │ ├── simple_tflite_op.h │ │ │ │ ├── simple_tflite_op_test.cc │ │ │ │ ├── tmpl_op.h │ │ │ │ ├── tmpl_tf_op.cc │ │ │ │ ├── tmpl_tf_op.h │ │ │ │ ├── tmpl_tf_op_test.cc │ │ │ │ ├── tmpl_tflite_op.cc │ │ │ │ ├── tmpl_tflite_op.h │ │ │ │ └── tmpl_tflite_op_test.cc │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ ├── test_util_test.cc │ │ │ ├── tf_op_shim.cc │ │ │ ├── tf_op_shim.h │ │ │ ├── tf_tensor_view.cc │ │ │ ├── tf_tensor_view.h │ │ │ ├── tf_tensor_view_test.cc │ │ │ ├── tflite_op_shim.cc │ │ │ ├── tflite_op_shim.h │ │ │ ├── tflite_op_wrapper.h │ │ │ ├── tflite_op_wrapper_test.cc │ │ │ ├── tflite_tensor_view.cc │ │ │ ├── tflite_tensor_view.h │ │ │ └── tflite_tensor_view_test.cc │ │ ├── sign.cc │ │ ├── sign_custom.cc │ │ ├── sign_custom_test.cc │ │ ├── sign_test.cc │ │ ├── skip_gram.cc │ │ ├── skip_gram_test.cc │ │ ├── slice.cc │ │ ├── slice_test.cc │ │ ├── softmax_test.cc │ │ ├── space_to_batch_nd.cc │ │ ├── space_to_batch_nd_test.cc │ │ ├── space_to_depth.cc │ │ ├── space_to_depth_test.cc │ │ ├── sparse_to_dense.cc │ │ ├── sparse_to_dense_test.cc │ │ ├── split.cc │ │ ├── split_test.cc │ │ ├── split_v.cc │ │ ├── split_v_test.cc │ │ ├── squared_difference.cc │ │ ├── squared_difference_test.cc │ │ ├── squeeze.cc │ │ ├── squeeze_test.cc │ │ ├── stablehlo_add.cc │ │ ├── stablehlo_add_test.cc │ │ ├── stablehlo_elementwise.cc │ │ ├── stablehlo_elementwise.h │ │ ├── stablehlo_gather.cc │ │ ├── stablehlo_gather_test.cc │ │ ├── stablehlo_min_max.cc │ │ ├── stablehlo_min_max_test.cc │ │ ├── stablehlo_multiply.cc │ │ ├── stablehlo_multiply_test.cc │ │ ├── stablehlo_pad.cc │ │ ├── stablehlo_pad_test.cc │ │ ├── stablehlo_reduce_window.cc │ │ ├── stablehlo_reduce_window_test.cc │ │ ├── stablehlo_reduce_window_test_util.h │ │ ├── stablehlo_reduce_window_test_util_test.cc │ │ ├── stablehlo_scatter.cc │ │ ├── stablehlo_scatter_test.cc │ │ ├── strided_slice.cc │ │ ├── strided_slice_test.cc │ │ ├── sub.cc │ │ ├── sub_test.cc │ │ ├── subgraph_test_util.cc │ │ ├── subgraph_test_util.h │ │ ├── subgraph_test_util_test.cc │ │ ├── svdf.cc │ │ ├── svdf_test.cc │ │ ├── table.cc │ │ ├── table_test.cc │ │ ├── tensor_slice_util.cc │ │ ├── tensor_slice_util.h │ │ ├── tensor_slice_util_test.cc │ │ ├── test_delegate_providers.cc │ │ ├── test_delegate_providers.h │ │ ├── test_delegate_providers_test.cc │ │ ├── test_main.cc │ │ ├── test_util.cc │ │ ├── test_util.h │ │ ├── test_util_test.cc │ │ ├── tile.cc │ │ ├── tile_test.cc │ │ ├── topk_v2.cc │ │ ├── topk_v2_test.cc │ │ ├── transpose.cc │ │ ├── transpose_conv.cc │ │ ├── transpose_conv_test.cc │ │ ├── transpose_test.cc │ │ ├── transpose_test_utils.h │ │ ├── transpose_test_utils_test.cc │ │ ├── unidirectional_sequence_gru.cc │ │ ├── unidirectional_sequence_gru_test.cc │ │ ├── unidirectional_sequence_lstm.cc │ │ ├── unidirectional_sequence_lstm_test.cc │ │ ├── unidirectional_sequence_lstm_test_util.h │ │ ├── unidirectional_sequence_rnn.cc │ │ ├── unidirectional_sequence_rnn_test.cc │ │ ├── unique.cc │ │ ├── unique_test.cc │ │ ├── unpack.cc │ │ ├── unpack_test.cc │ │ ├── unsorted_segment.cc │ │ ├── unsorted_segment_max_test.cc │ │ ├── unsorted_segment_min_test.cc │ │ ├── unsorted_segment_prod_test.cc │ │ ├── unsorted_segment_sum_test.cc │ │ ├── unsorted_segment_test.cc │ │ ├── unsorted_segment_test.h │ │ ├── var_handle.cc │ │ ├── variable_ops_test.cc │ │ ├── variable_ops_wrapper.cc │ │ ├── variants │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── list_kernels │ │ │ │ ├── list_element_shape.cc │ │ │ │ ├── list_element_shape_test.cc │ │ │ │ ├── list_from_tensor.cc │ │ │ │ ├── list_from_tensor_test.cc │ │ │ │ ├── list_get_item.cc │ │ │ │ ├── list_get_item_test.cc │ │ │ │ ├── list_length.cc │ │ │ │ ├── list_length_test.cc │ │ │ │ ├── list_pop_back_test.cc │ │ │ │ ├── list_push_back_test.cc │ │ │ │ ├── list_reserve.cc │ │ │ │ ├── list_reserve_test.cc │ │ │ │ ├── list_set_item.cc │ │ │ │ ├── list_set_item_test.cc │ │ │ │ ├── list_stack.cc │ │ │ │ ├── list_stack_test.cc │ │ │ │ ├── test_util.cc │ │ │ │ ├── test_util.h │ │ │ │ ├── variant_add_n.cc │ │ │ │ ├── variant_add_n_test.cc │ │ │ │ └── variant_zeros_like_test.cc │ │ │ ├── list_ops_lib.h │ │ │ ├── list_ops_subgraph_test.cc │ │ │ ├── list_ops_subgraph_test_util.cc │ │ │ ├── list_ops_subgraph_test_util.h │ │ │ ├── list_ops_util.cc │ │ │ ├── list_ops_util.h │ │ │ ├── list_ops_util_test.cc │ │ │ ├── py │ │ │ │ ├── BUILD │ │ │ │ ├── end_to_end_test.py │ │ │ │ ├── register_list_ops_py.cc │ │ │ │ └── register_list_ops_py.pyi │ │ │ ├── register_list_ops.cc │ │ │ ├── register_list_ops.h │ │ │ ├── tensor_array.cc │ │ │ ├── tensor_array.h │ │ │ └── tensor_array_test.cc │ │ ├── where.cc │ │ ├── where_test.cc │ │ ├── while.cc │ │ ├── while_test.cc │ │ ├── zeros_like.cc │ │ └── zeros_like_test.cc │ ├── lib_package │ │ ├── BUILD │ │ └── concat_licenses.sh │ ├── logger.cc │ ├── logger.h │ ├── memory_planner.h │ ├── micro │ │ └── README.md │ ├── minimal_logging.cc │ ├── minimal_logging.h │ ├── minimal_logging_android.cc │ ├── minimal_logging_default.cc │ ├── minimal_logging_ios.cc │ ├── minimal_logging_test.cc │ ├── mmap_allocation.cc │ ├── mmap_allocation_disabled.cc │ ├── model.h │ ├── model_builder.h │ ├── model_flex_test.cc │ ├── model_xnnpack_test.cc │ ├── mutable_op_resolver.cc │ ├── mutable_op_resolver.h │ ├── mutable_op_resolver_test.cc │ ├── namespace.h │ ├── nnapi │ │ ├── BUILD │ │ ├── NeuralNetworksShim.h │ │ ├── NeuralNetworksTypes.h │ │ ├── README.md │ │ ├── nnapi_handler.cc │ │ ├── nnapi_handler.h │ │ ├── nnapi_handler_test.cc │ │ ├── nnapi_implementation.cc │ │ ├── nnapi_implementation.h │ │ ├── nnapi_implementation_disabled.cc │ │ ├── nnapi_implementation_test.cc │ │ ├── nnapi_util.cc │ │ ├── nnapi_util.h │ │ └── sl │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── SupportLibrary.cc │ │ │ ├── include │ │ │ ├── SupportLibrary.h │ │ │ └── SupportLibrarySymbols.h │ │ │ └── public │ │ │ └── NeuralNetworksSupportLibraryImpl.h │ ├── objc │ │ ├── BUILD.apple │ │ ├── README.md │ │ ├── TensorFlowLite.tulsiproj │ │ │ ├── Configs │ │ │ │ └── TensorFlowLite.tulsigen │ │ │ └── project.tulsiconf │ │ ├── TensorFlowLiteObjC.podspec │ │ ├── TensorFlowLiteObjC.podspec.template │ │ ├── apis │ │ │ ├── TFLCoreMLDelegate.h │ │ │ ├── TFLDelegate.h │ │ │ ├── TFLInterpreter.h │ │ │ ├── TFLInterpreterOptions.h │ │ │ ├── TFLMetalDelegate.h │ │ │ ├── TFLQuantizationParameters.h │ │ │ ├── TFLSignatureRunner.h │ │ │ ├── TFLTensor.h │ │ │ └── TFLTensorFlowLite.h │ │ ├── apps │ │ │ └── TestApp │ │ │ │ ├── Podfile │ │ │ │ ├── TestApp.xcodeproj │ │ │ │ └── project.pbxproj │ │ │ │ └── TestApp │ │ │ │ ├── AppDelegate.h │ │ │ │ ├── AppDelegate.m │ │ │ │ ├── Assets.xcassets │ │ │ │ ├── AppIcon.appiconset │ │ │ │ │ └── Contents.json │ │ │ │ └── Contents.json │ │ │ │ ├── Base.lproj │ │ │ │ ├── LaunchScreen.storyboard │ │ │ │ └── Main.storyboard │ │ │ │ ├── Info.plist │ │ │ │ ├── ViewController.h │ │ │ │ ├── ViewController.m │ │ │ │ └── main.m │ │ ├── sources │ │ │ ├── TFLCommonUtil.h │ │ │ ├── TFLCommonUtil.mm │ │ │ ├── TFLCoreMLDelegate.m │ │ │ ├── TFLDelegate.m │ │ │ ├── TFLErrorUtil.h │ │ │ ├── TFLErrorUtil.m │ │ │ ├── TFLInterpreter+Internal.h │ │ │ ├── TFLInterpreter.mm │ │ │ ├── TFLInterpreterOptions.m │ │ │ ├── TFLMetalDelegate.m │ │ │ ├── TFLQuantizationParameters+Internal.h │ │ │ ├── TFLQuantizationParameters.m │ │ │ ├── TFLSignatureRunner+Internal.h │ │ │ ├── TFLSignatureRunner.mm │ │ │ ├── TFLTensor+Internal.h │ │ │ ├── TFLTensor.m │ │ │ └── TFLTensorDataAccessor.h │ │ └── tests │ │ │ ├── TFLCoreMLDelegateTests.m │ │ │ ├── TFLInterpreterOptionsTests.m │ │ │ ├── TFLInterpreterTests.m │ │ │ ├── TFLMetalDelegateTests.m │ │ │ ├── TFLQuantizationParametersTests.m │ │ │ └── TFLSignatureRunnerTest.m │ ├── op_resolver.h │ ├── optional_debug_tools.cc │ ├── optional_debug_tools.h │ ├── optional_debug_tools_test.cc │ ├── portable_type_to_tflitetype.h │ ├── profiling │ │ ├── BUILD │ │ ├── atrace_profiler.cc │ │ ├── atrace_profiler.h │ │ ├── atrace_profiler_test.cc │ │ ├── buffered_profiler.h │ │ ├── memory_info.cc │ │ ├── memory_info.h │ │ ├── memory_info_test.cc │ │ ├── memory_usage_monitor.cc │ │ ├── memory_usage_monitor.h │ │ ├── memory_usage_monitor_test.cc │ │ ├── noop_profiler.h │ │ ├── platform_profiler.cc │ │ ├── platform_profiler.h │ │ ├── profile_buffer.cc │ │ ├── profile_buffer.h │ │ ├── profile_buffer_test.cc │ │ ├── profile_summarizer.cc │ │ ├── profile_summarizer.h │ │ ├── profile_summarizer_test.cc │ │ ├── profile_summary_formatter.cc │ │ ├── profile_summary_formatter.h │ │ ├── profile_summary_formatter_test.cc │ │ ├── profiler.h │ │ ├── profiler_test.cc │ │ ├── root_profiler.cc │ │ ├── root_profiler.h │ │ ├── root_profiler_test.cc │ │ ├── signpost_profiler.h │ │ ├── signpost_profiler.mm │ │ ├── subgraph_tensor_profiler.cc │ │ ├── subgraph_tensor_profiler.h │ │ ├── subgraph_tensor_profiler_test.cc │ │ ├── telemetry │ │ │ ├── BUILD │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ ├── profiler.h │ │ │ │ ├── telemetry_setting.h │ │ │ │ ├── telemetry_setting_internal.cc │ │ │ │ └── telemetry_setting_internal.h │ │ │ ├── profiler.cc │ │ │ ├── profiler.h │ │ │ ├── profiler_test.cc │ │ │ ├── telemetry.cc │ │ │ ├── telemetry.h │ │ │ ├── telemetry_status.h │ │ │ └── telemetry_test.cc │ │ ├── time.cc │ │ ├── time.h │ │ └── time_test.cc │ ├── python │ │ ├── BUILD │ │ ├── analyzer.py │ │ ├── analyzer_test.py │ │ ├── analyzer_wrapper │ │ │ ├── BUILD │ │ │ ├── _pywrap_analyzer_wrapper.pyi │ │ │ ├── analyzer_wrapper.cc │ │ │ ├── model_analyzer.cc │ │ │ └── model_analyzer.h │ │ ├── authoring │ │ │ ├── BUILD │ │ │ ├── authoring.py │ │ │ └── authoring_test.py │ │ ├── convert.py │ │ ├── convert_file_to_c_source.py │ │ ├── convert_file_to_c_source_test.sh │ │ ├── convert_phase.py │ │ ├── convert_saved_model.py │ │ ├── convert_saved_model_test.py │ │ ├── convert_test.py │ │ ├── interpreter.py │ │ ├── interpreter_test.py │ │ ├── interpreter_wrapper │ │ │ ├── BUILD │ │ │ ├── _pywrap_tensorflow_interpreter_wrapper.pyi │ │ │ ├── interpreter_wrapper.cc │ │ │ ├── interpreter_wrapper.h │ │ │ ├── interpreter_wrapper_pybind11.cc │ │ │ ├── numpy.cc │ │ │ ├── numpy.h │ │ │ ├── python_error_reporter.cc │ │ │ ├── python_error_reporter.h │ │ │ ├── python_utils.cc │ │ │ └── python_utils.h │ │ ├── lite.py │ │ ├── lite_constants.py │ │ ├── lite_flex_test.py │ │ ├── lite_test.py │ │ ├── lite_v2_test.py │ │ ├── lite_v2_test_util.py │ │ ├── metrics │ │ │ ├── BUILD │ │ │ ├── _pywrap_tensorflow_lite_metrics_wrapper.pyi │ │ │ ├── converter_error_data.proto │ │ │ ├── metrics_interface.py │ │ │ ├── metrics_nonportable.py │ │ │ ├── metrics_nonportable_test.py │ │ │ ├── metrics_portable.py │ │ │ ├── metrics_portable_test.py │ │ │ └── wrapper │ │ │ │ ├── metrics_wrapper.h │ │ │ │ ├── metrics_wrapper.py │ │ │ │ ├── metrics_wrapper_nonportable.cc │ │ │ │ ├── metrics_wrapper_portable.cc │ │ │ │ ├── metrics_wrapper_pybind11.cc │ │ │ │ └── metrics_wrapper_test.py │ │ ├── op_hint.py │ │ ├── optimize │ │ │ ├── BUILD │ │ │ ├── _pywrap_tensorflow_lite_calibration_wrapper.pyi │ │ │ ├── calibration_wrapper.cc │ │ │ ├── calibration_wrapper.h │ │ │ ├── calibration_wrapper_pybind11.cc │ │ │ ├── calibrator.py │ │ │ ├── calibrator_test.py │ │ │ └── test_data │ │ │ │ ├── mobilenet_like_model.bin │ │ │ │ └── string_input_flex_model.bin │ │ ├── schema_util.py │ │ ├── test_util.py │ │ ├── test_util_test.py │ │ ├── testdata │ │ │ ├── BUILD │ │ │ ├── _pywrap_test_registerer.pyi │ │ │ ├── control_flow_v1.pbtxt │ │ │ ├── control_flow_v1_saved_model │ │ │ │ ├── BUILD │ │ │ │ └── saved_model.pb │ │ │ ├── double_op.cc │ │ │ ├── double_op.py │ │ │ ├── gather.pbtxt │ │ │ ├── gather_0d.pbtxt │ │ │ ├── pc_conv.bin │ │ │ ├── permute.pbtxt │ │ │ ├── test_delegate.cc │ │ │ ├── test_registerer.cc │ │ │ ├── test_registerer.h │ │ │ └── test_registerer_wrapper.cc │ │ ├── tflite_convert.py │ │ ├── tflite_convert_test.py │ │ ├── tflite_keras_util.py │ │ ├── util.py │ │ ├── util_test.py │ │ └── wrap_toco.py │ ├── schema │ │ ├── BUILD │ │ ├── builtin_ops_header │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── consistency_test.cc │ │ │ ├── generate.cc │ │ │ ├── generator.cc │ │ │ ├── generator.h │ │ │ └── generator_test.cc │ │ ├── builtin_ops_list │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── consistency_test.cc │ │ │ ├── generate.cc │ │ │ ├── generator.cc │ │ │ ├── generator.h │ │ │ └── generator_test.cc │ │ ├── conversion_metadata.fbs │ │ ├── conversion_metadata_generated.h │ │ ├── flatbuffer_compatibility_test.cc │ │ ├── schema.fbs │ │ ├── schema_conversion_utils.cc │ │ ├── schema_conversion_utils.h │ │ ├── schema_generated.h │ │ ├── schema_utils.cc │ │ ├── schema_utils.h │ │ ├── schema_v0.fbs │ │ ├── schema_v1.fbs │ │ ├── schema_v2.fbs │ │ ├── schema_v3.fbs │ │ ├── schema_v3a.fbs │ │ ├── schema_v3b.fbs │ │ ├── schema_v3c.fbs │ │ ├── upgrade_schema.py │ │ └── upgrade_schema_test.py │ ├── shared_library.h │ ├── signature_runner.h │ ├── simple_memory_arena.cc │ ├── simple_memory_arena.h │ ├── simple_memory_arena_debug_dump.cc │ ├── simple_memory_arena_test.cc │ ├── simple_planner.cc │ ├── simple_planner.h │ ├── simple_planner_test.cc │ ├── special_rules.bzl │ ├── stablehlo │ │ └── schema │ │ │ ├── BUILD │ │ │ └── schema.fbs │ ├── stateful_error_reporter.h │ ├── stderr_reporter.cc │ ├── stderr_reporter.h │ ├── stderr_reporter_test.cc │ ├── string_type.h │ ├── string_util.cc │ ├── string_util.h │ ├── string_util_test.cc │ ├── swift │ │ ├── BUILD.apple │ │ ├── README.md │ │ ├── Sources │ │ │ ├── CoreMLDelegate.swift │ │ │ ├── Delegate.swift │ │ │ ├── Interpreter.swift │ │ │ ├── InterpreterError.swift │ │ │ ├── MetalDelegate.swift │ │ │ ├── Model.swift │ │ │ ├── QuantizationParameters.swift │ │ │ ├── SignatureRunner.swift │ │ │ ├── SignatureRunnerError.swift │ │ │ ├── Tensor.swift │ │ │ └── TensorFlowLite.swift │ │ ├── TensorFlowLite.tulsiproj │ │ │ ├── Configs │ │ │ │ └── TensorFlowLite.tulsigen │ │ │ └── project.tulsiconf │ │ ├── TensorFlowLiteSwift.podspec │ │ ├── TensorFlowLiteSwift.podspec.template │ │ ├── TestApp │ │ │ └── README.md │ │ ├── Tests │ │ │ ├── InterpreterTests.swift │ │ │ ├── MetalDelegateTests.swift │ │ │ ├── ModelTests.swift │ │ │ ├── QuantizationParametersTests.swift │ │ │ ├── SignatureRunnerTest.swift │ │ │ ├── TensorFlowLiteTests.swift │ │ │ └── TensorTests.swift │ │ └── docsgen │ │ │ ├── README.md │ │ │ └── TensorFlowLiteSwift │ │ │ ├── TensorFlowLiteSwift.xcodeproj │ │ │ ├── project.pbxproj │ │ │ └── project.xcworkspace │ │ │ │ ├── contents.xcworkspacedata │ │ │ │ └── xcshareddata │ │ │ │ └── IDEWorkspaceChecks.plist │ │ │ ├── TensorFlowLiteSwift │ │ │ ├── Info.plist │ │ │ └── TensorFlowLiteSwift.h │ │ │ └── scripts │ │ │ └── download_frameworks.sh │ ├── tensorflow_profiler_logger.h │ ├── tensorflow_profiler_logger_shim.cc │ ├── test_util.h │ ├── test_util_test.cc │ ├── testdata │ │ ├── 0_subgraphs.bin │ │ ├── 2_subgraphs.bin │ │ ├── 2_subgraphs_dont_delegate_name.bin │ │ ├── add.bin │ │ ├── add.json │ │ ├── add_quantized.bin │ │ ├── add_quantized.json │ │ ├── add_quantized_int8.bin │ │ ├── add_quantized_int8.json │ │ ├── add_shared_tensors.bin │ │ ├── call_once_mul.bin │ │ ├── conv3d_huge_im2col.bin │ │ ├── conv_huge_im2col.bin │ │ ├── custom_lstm.bin │ │ ├── custom_sinh.bin │ │ ├── double_flex.bin │ │ ├── dynamic_shapes.bin │ │ ├── empty_model.bin │ │ ├── lstm.bin │ │ ├── multi_add.bin │ │ ├── multi_add.json │ │ ├── multi_add.pb │ │ ├── multi_add_flex.bin │ │ ├── multi_signatures.bin │ │ ├── multi_subgraphs_while.bin │ │ ├── nested_while.bin │ │ ├── no_subgraphs.bin │ │ ├── segment_sum_invalid_buffer.bin │ │ ├── softplus_flex.bin │ │ ├── sparse_tensor.bin │ │ ├── sparse_tensor.json │ │ ├── src │ │ │ ├── BUILD │ │ │ └── intermediate_tensor_output.py │ │ ├── string_input_model.bin │ │ ├── sub.bin │ │ ├── tensor_string_data_type.bin │ │ ├── test_custom_node_with_init_data.bin │ │ ├── test_min_runtime.bin │ │ ├── test_model.bin │ │ ├── test_model_broken.bin │ │ ├── test_model_broken.json │ │ ├── test_model_redux_precision.bin │ │ ├── test_model_versioned_ops.bin │ │ ├── two_subgraphs.bin │ │ ├── unidirectional_sequence_lstm.bin │ │ ├── while_op_with_forwarding_input.bin │ │ └── zero_size_constant.bin │ ├── testing │ │ ├── BUILD │ │ ├── _pywrap_string_util.pyi │ │ ├── build_def.bzl │ │ ├── generate_examples.py │ │ ├── generate_examples_lib.py │ │ ├── generate_examples_report.py │ │ ├── generate_testspec.cc │ │ ├── generate_testspec.h │ │ ├── generate_testspec_test.cc │ │ ├── generated_examples_zip_test.cc │ │ ├── init_tensorflow.cc │ │ ├── init_tensorflow.h │ │ ├── join.h │ │ ├── join_test.cc │ │ ├── kernel_test │ │ │ ├── BUILD │ │ │ ├── diff_analyzer.cc │ │ │ ├── diff_analyzer.h │ │ │ ├── diff_analyzer_test.cc │ │ │ ├── generate_diff_report.cc │ │ │ ├── input_generator.cc │ │ │ ├── input_generator.h │ │ │ ├── input_generator_test.cc │ │ │ ├── testdata │ │ │ │ └── test_input.csv │ │ │ ├── tflite_kernel_runner.cc │ │ │ ├── util.h │ │ │ └── util_test.cc │ │ ├── message.cc │ │ ├── message.h │ │ ├── message_test.cc │ │ ├── mlir_convert.py │ │ ├── nnapi_example.cc │ │ ├── op_tests │ │ │ ├── abs.py │ │ │ ├── add_n.py │ │ │ ├── arg_min_max.py │ │ │ ├── atan2.py │ │ │ ├── batch_to_space_nd.py │ │ │ ├── batchmatmul.py │ │ │ ├── binary_op.py │ │ │ ├── bitcast.py │ │ │ ├── bitwise_xor.py │ │ │ ├── broadcast_args.py │ │ │ ├── broadcast_gradient_args.py │ │ │ ├── broadcast_to.py │ │ │ ├── cast.py │ │ │ ├── ceil.py │ │ │ ├── complex_abs.py │ │ │ ├── concat.py │ │ │ ├── cond.py │ │ │ ├── constant.py │ │ │ ├── control_dep.py │ │ │ ├── conv.py │ │ │ ├── conv2d_transpose.py │ │ │ ├── conv3d.py │ │ │ ├── conv3d_transpose.py │ │ │ ├── conv_activation.py │ │ │ ├── conv_bias_activation.py │ │ │ ├── conv_to_depthwiseconv_with_shared_weights.py │ │ │ ├── conv_with_shared_weights.py │ │ │ ├── cos.py │ │ │ ├── cumsum.py │ │ │ ├── depth_to_space.py │ │ │ ├── depthwiseconv.py │ │ │ ├── dynamic_rnn.py │ │ │ ├── dynamic_update_slice.py │ │ │ ├── einsum.py │ │ │ ├── elementwise.py │ │ │ ├── elu.py │ │ │ ├── embedding_lookup.py │ │ │ ├── equal.py │ │ │ ├── exp.py │ │ │ ├── expand_dims.py │ │ │ ├── expm1.py │ │ │ ├── eye.py │ │ │ ├── fill.py │ │ │ ├── floor.py │ │ │ ├── fully_connected.py │ │ │ ├── fully_connected_4bit_hybrid.py │ │ │ ├── fused_batch_norm.py │ │ │ ├── gather.py │ │ │ ├── gather_nd.py │ │ │ ├── gather_with_constant.py │ │ │ ├── gelu.py │ │ │ ├── global_batch_norm.py │ │ │ ├── greater.py │ │ │ ├── greater_equal.py │ │ │ ├── hardswish.py │ │ │ ├── identify_dilated_conv.py │ │ │ ├── identify_dilated_conv1d.py │ │ │ ├── identity.py │ │ │ ├── imag.py │ │ │ ├── irfft2d.py │ │ │ ├── is_finite.py │ │ │ ├── l2norm.py │ │ │ ├── l2norm_shared_epsilon.py │ │ │ ├── leaky_relu.py │ │ │ ├── less.py │ │ │ ├── less_equal.py │ │ │ ├── local_response_norm.py │ │ │ ├── log_softmax.py │ │ │ ├── logic.py │ │ │ ├── lstm.py │ │ │ ├── matrix_band_part.py │ │ │ ├── matrix_diag.py │ │ │ ├── matrix_set_diag.py │ │ │ ├── max_pool_with_argmax.py │ │ │ ├── maximum.py │ │ │ ├── minimum.py │ │ │ ├── mirror_pad.py │ │ │ ├── multinomial.py │ │ │ ├── nearest_upsample.py │ │ │ ├── neg.py │ │ │ ├── not_equal.py │ │ │ ├── one_hot.py │ │ │ ├── pack.py │ │ │ ├── pad.py │ │ │ ├── padv2.py │ │ │ ├── parse_example.py │ │ │ ├── placeholder_with_default.py │ │ │ ├── pool.py │ │ │ ├── pool3d.py │ │ │ ├── prelu.py │ │ │ ├── random_standard_normal.py │ │ │ ├── random_uniform.py │ │ │ ├── range.py │ │ │ ├── rank.py │ │ │ ├── real.py │ │ │ ├── reciprocal.py │ │ │ ├── reduce.py │ │ │ ├── relu.py │ │ │ ├── relu1.py │ │ │ ├── relu6.py │ │ │ ├── reshape.py │ │ │ ├── resize_bilinear.py │ │ │ ├── resize_nearest_neighbor.py │ │ │ ├── resolve_constant_strided_slice.py │ │ │ ├── reverse_sequence.py │ │ │ ├── reverse_v2.py │ │ │ ├── rfft.py │ │ │ ├── rfft2d.py │ │ │ ├── right_shift.py │ │ │ ├── roll.py │ │ │ ├── round.py │ │ │ ├── scatter_nd.py │ │ │ ├── segment_sum.py │ │ │ ├── shape.py │ │ │ ├── shape_to_strided_slice.py │ │ │ ├── sigmoid.py │ │ │ ├── sigmoid_grad.py │ │ │ ├── sign.py │ │ │ ├── slice.py │ │ │ ├── softmax.py │ │ │ ├── softplus.py │ │ │ ├── space_to_batch_nd.py │ │ │ ├── space_to_depth.py │ │ │ ├── sparse_to_dense.py │ │ │ ├── split.py │ │ │ ├── splitv.py │ │ │ ├── squeeze.py │ │ │ ├── squeeze_transpose.py │ │ │ ├── static_hashtable.py │ │ │ ├── static_rnn_with_control_flow_v2.py │ │ │ ├── stft.py │ │ │ ├── strided_slice.py │ │ │ ├── strided_slice_np_style.py │ │ │ ├── tanh.py │ │ │ ├── tensor_list_concat.py │ │ │ ├── tensor_list_dynamic_shape.py │ │ │ ├── tensor_list_get_item.py │ │ │ ├── tensor_list_length.py │ │ │ ├── tensor_list_resize.py │ │ │ ├── tensor_list_set_item.py │ │ │ ├── tensor_scatter_add.py │ │ │ ├── tensor_scatter_update.py │ │ │ ├── tile.py │ │ │ ├── topk.py │ │ │ ├── transpose.py │ │ │ ├── transpose_conv.py │ │ │ ├── unfused_gru.py │ │ │ ├── unique.py │ │ │ ├── unpack.py │ │ │ ├── unroll_batch_matmul.py │ │ │ ├── unsorted_segment.py │ │ │ ├── where.py │ │ │ ├── where_v2.py │ │ │ ├── while_loop.py │ │ │ └── zeros_like.py │ │ ├── parse_testdata.cc │ │ ├── parse_testdata.h │ │ ├── result_expectations.cc │ │ ├── result_expectations.h │ │ ├── selective_build_test.cc │ │ ├── split.cc │ │ ├── split.h │ │ ├── split_test.cc │ │ ├── string_util.cc │ │ ├── string_util.h │ │ ├── string_util_wrapper.cc │ │ ├── test_runner.h │ │ ├── test_runner_test.cc │ │ ├── tf_driver.cc │ │ ├── tf_driver.h │ │ ├── tf_driver_test.cc │ │ ├── tflite_diff_example_test.cc │ │ ├── tflite_diff_flags.h │ │ ├── tflite_diff_util.cc │ │ ├── tflite_diff_util.h │ │ ├── tflite_driver.cc │ │ ├── tflite_driver.h │ │ ├── tflite_driver_test.cc │ │ ├── tflite_model_test.bzl │ │ ├── tokenize.cc │ │ ├── tokenize.h │ │ ├── tokenize_test.cc │ │ ├── util.h │ │ └── zip_test_utils.py │ ├── tflite_exported_symbols.lds │ ├── tflite_version_script.lds │ ├── tflite_with_xnnpack.cc │ ├── tflite_with_xnnpack_optional.cc │ ├── tflite_with_xnnpack_optional.h │ ├── toco │ │ ├── BUILD │ │ ├── README.md │ │ ├── allocate_transient_arrays.cc │ │ ├── allocate_transient_arrays.h │ │ ├── args.cc │ │ ├── args.h │ │ ├── dump_graphviz.cc │ │ ├── dump_graphviz.h │ │ ├── export_tensorflow.cc │ │ ├── export_tensorflow.h │ │ ├── format_port.h │ │ ├── g3doc │ │ │ └── README.md │ │ ├── graph_transformations │ │ │ ├── convert_expanddims_to_reshape.cc │ │ │ ├── convert_matrix_diag_v2_or_v3_to_v1.cc │ │ │ ├── convert_matrix_set_diag_v2_or_v3_to_v1.cc │ │ │ ├── convert_pure_conv_to_depthwise.cc │ │ │ ├── convert_reorder_axes.cc │ │ │ ├── convert_squeeze_to_reshape.cc │ │ │ ├── convert_trivial_addn_to_add.cc │ │ │ ├── convert_trivial_pack_to_reshape.cc │ │ │ ├── convert_trivial_tile_to_concat.cc │ │ │ ├── convert_trivial_transpose_to_reshape.cc │ │ │ ├── create_im2col_arrays.cc │ │ │ ├── dequantize.cc │ │ │ ├── drop_fake_quant.cc │ │ │ ├── drop_im2col_arrays.cc │ │ │ ├── ensure_bias_vectors.cc │ │ │ ├── ensure_uint8_weights_safe_for_fast_int8_kernels.cc │ │ │ ├── fuse_activation_functions.cc │ │ │ ├── fuse_binary_into_following_affine.cc │ │ │ ├── fuse_binary_into_preceding_affine.cc │ │ │ ├── fuse_broadcast_into_following_binary.cc │ │ │ ├── graph_transformations.cc │ │ │ ├── graph_transformations.h │ │ │ ├── group_bidirectional_sequence_ops.cc │ │ │ ├── hardcode_min_max.cc │ │ │ ├── identify_dilated_conv.cc │ │ │ ├── identify_hardswish.cc │ │ │ ├── identify_l2_normalization.cc │ │ │ ├── identify_l2_pool.cc │ │ │ ├── identify_lstm.cc │ │ │ ├── identify_lstm_merge_inputs.cc │ │ │ ├── identify_lstm_split_inputs.cc │ │ │ ├── identify_nearest_upsample.cc │ │ │ ├── identify_prelu.cc │ │ │ ├── identify_relu1.cc │ │ │ ├── identify_util.cc │ │ │ ├── identify_util.h │ │ │ ├── lstm_utils.cc │ │ │ ├── lstm_utils.h │ │ │ ├── make_initial_dequantize_operator.cc │ │ │ ├── merge_reshape_into_preceding_transpose.cc │ │ │ ├── move_binary_operator_before_reshape.cc │ │ │ ├── propagate_activation_function_into_constants.cc │ │ │ ├── propagate_array_data_types.cc │ │ │ ├── propagate_default_min_max.cc │ │ │ ├── propagate_fake_quant_num_bits.cc │ │ │ ├── propagate_fixed_sizes.cc │ │ │ ├── quantization_util.cc │ │ │ ├── quantization_util.h │ │ │ ├── quantize.cc │ │ │ ├── read_array_minmax_and_narrow_range_from_fake_quant.cc │ │ │ ├── remove_final_dequantize_op.cc │ │ │ ├── remove_successive_transpose.cc │ │ │ ├── remove_tensorflow_assert.cc │ │ │ ├── remove_tensorflow_identity.cc │ │ │ ├── remove_trivial_binary.cc │ │ │ ├── remove_trivial_concatenation.cc │ │ │ ├── remove_trivial_concatenation_input.cc │ │ │ ├── remove_trivial_fake_quant.cc │ │ │ ├── remove_trivial_passthrough.cc │ │ │ ├── remove_trivial_passthrough.h │ │ │ ├── remove_trivial_quantized_activation_func.cc │ │ │ ├── remove_trivial_quantized_min_max.cc │ │ │ ├── remove_trivial_reshape.cc │ │ │ ├── remove_trivial_slice.cc │ │ │ ├── remove_unused_op.cc │ │ │ ├── reorder_elementwise_unary.cc │ │ │ ├── reorder_reshape_transpose.cc │ │ │ ├── resolve_batch_normalization.cc │ │ │ ├── resolve_batch_to_space_nd_attributes.cc │ │ │ ├── resolve_constant_binary.cc │ │ │ ├── resolve_constant_concatenation.cc │ │ │ ├── resolve_constant_fake_quant.cc │ │ │ ├── resolve_constant_fill.cc │ │ │ ├── resolve_constant_gather.cc │ │ │ ├── resolve_constant_pack.cc │ │ │ ├── resolve_constant_random_uniform.cc │ │ │ ├── resolve_constant_range.cc │ │ │ ├── resolve_constant_reshape.cc │ │ │ ├── resolve_constant_select.cc │ │ │ ├── resolve_constant_shape_or_rank.cc │ │ │ ├── resolve_constant_slice.cc │ │ │ ├── resolve_constant_strided_slice.cc │ │ │ ├── resolve_constant_tile.cc │ │ │ ├── resolve_constant_transpose.cc │ │ │ ├── resolve_constant_unary.cc │ │ │ ├── resolve_fake_quant_args_from_vars.cc │ │ │ ├── resolve_gather_attributes.cc │ │ │ ├── resolve_multiply_by_zero.cc │ │ │ ├── resolve_pad_attributes.cc │ │ │ ├── resolve_padv2_attributes.cc │ │ │ ├── resolve_reduce_attributes.cc │ │ │ ├── resolve_reorder_axes.cc │ │ │ ├── resolve_reshape_attributes.cc │ │ │ ├── resolve_slice_attributes.cc │ │ │ ├── resolve_space_to_batch_nd_attributes.cc │ │ │ ├── resolve_squeeze_attributes.cc │ │ │ ├── resolve_strided_slice_attributes.cc │ │ │ ├── resolve_tensorflow_concat.cc │ │ │ ├── resolve_tensorflow_matmul.cc │ │ │ ├── resolve_tensorflow_merge.cc │ │ │ ├── resolve_tensorflow_switch.cc │ │ │ ├── resolve_transpose_attributes.cc │ │ │ ├── shuffle_fc_weights.cc │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── fuse_binary_into_following_affine_test.cc │ │ │ │ ├── fuse_binary_into_preceding_affine_test.cc │ │ │ │ ├── identify_l2_normalization_test.cc │ │ │ │ ├── identify_l2_pool_test.cc │ │ │ │ ├── lstm_utils_test.cc │ │ │ │ ├── remove_successive_transpose_test.cc │ │ │ │ ├── resolve_constant_concatenation_test.cc │ │ │ │ ├── resolve_constant_unary_test.cc │ │ │ │ └── unpack_quantize_test.cc │ │ │ ├── unfuse_activation_functions.cc │ │ │ ├── unpartition_embedding_lookup.cc │ │ │ └── unroll_batch_matmul.cc │ │ ├── import_tensorflow.cc │ │ ├── import_tensorflow.h │ │ ├── import_tensorflow_test.cc │ │ ├── logging │ │ │ ├── BUILD │ │ │ ├── conversion_log_util.cc │ │ │ ├── conversion_log_util.h │ │ │ ├── conversion_log_util_test.cc │ │ │ ├── gen_html.py │ │ │ ├── gen_html_test.py │ │ │ ├── template.html │ │ │ ├── testdata │ │ │ │ ├── BUILD │ │ │ │ ├── generated.html │ │ │ │ ├── toco_log_after.pb │ │ │ │ ├── toco_log_before.pb │ │ │ │ ├── toco_tf_graph.dot │ │ │ │ └── toco_tflite_graph.dot │ │ │ └── toco_conversion_log.proto │ │ ├── model.h │ │ ├── model_cmdline_flags.cc │ │ ├── model_cmdline_flags.h │ │ ├── model_cmdline_flags_test.cc │ │ ├── model_flags.proto │ │ ├── python │ │ │ ├── BUILD │ │ │ ├── tensorflow_wrap_toco.py │ │ │ ├── toco_from_protos.py │ │ │ ├── toco_from_protos_test.py │ │ │ ├── toco_python_api.cc │ │ │ └── toco_python_api.h │ │ ├── runtime │ │ │ ├── common.h │ │ │ └── types.h │ │ ├── tensorflow_graph_matching │ │ │ ├── BUILD │ │ │ ├── cluster.cc │ │ │ ├── cluster.h │ │ │ ├── cluster_utils.cc │ │ │ ├── cluster_utils.h │ │ │ ├── resolve_cluster.cc │ │ │ ├── resolve_cluster.h │ │ │ ├── resolve_svdf.cc │ │ │ ├── resolve_svdf.h │ │ │ └── resolve_svdf_test.cc │ │ ├── tensorflow_util.cc │ │ ├── tensorflow_util.h │ │ ├── tflite │ │ │ ├── BUILD │ │ │ ├── builtin_operator.h │ │ │ ├── custom_operator.h │ │ │ ├── export.cc │ │ │ ├── export.h │ │ │ ├── export_test.cc │ │ │ ├── import.cc │ │ │ ├── import.h │ │ │ ├── import_test.cc │ │ │ ├── operator.cc │ │ │ ├── operator.h │ │ │ ├── operator_test.cc │ │ │ ├── simple_operator.h │ │ │ ├── types.cc │ │ │ ├── types.h │ │ │ └── types_test.cc │ │ ├── toco.cc │ │ ├── toco_cmdline_flags.cc │ │ ├── toco_cmdline_flags.h │ │ ├── toco_cmdline_flags_test.cc │ │ ├── toco_convert.cc │ │ ├── toco_convert.h │ │ ├── toco_convert_test.cc │ │ ├── toco_flags.proto │ │ ├── toco_graphviz_dump_options.cc │ │ ├── toco_graphviz_dump_options.h │ │ ├── toco_port.cc │ │ ├── toco_port.h │ │ ├── toco_port_test.cc │ │ ├── toco_tooling.cc │ │ ├── toco_tooling.h │ │ ├── toco_types.h │ │ ├── tooling_util.cc │ │ ├── tooling_util.h │ │ ├── tooling_util_test.cc │ │ └── types.proto │ ├── tools │ │ ├── BUILD │ │ ├── benchmark │ │ │ ├── BUILD │ │ │ ├── CMakeLists.txt │ │ │ ├── README.md │ │ │ ├── android │ │ │ │ ├── AndroidManifest.xml │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── jni │ │ │ │ │ └── benchmark_model_jni.cc │ │ │ │ └── src │ │ │ │ │ └── org │ │ │ │ │ └── tensorflow │ │ │ │ │ └── lite │ │ │ │ │ └── benchmark │ │ │ │ │ ├── BenchmarkModel.java │ │ │ │ │ └── BenchmarkModelActivity.java │ │ │ ├── benchmark_main.cc │ │ │ ├── benchmark_model.cc │ │ │ ├── benchmark_model.h │ │ │ ├── benchmark_multirun_stats_recorder.h │ │ │ ├── benchmark_params.h │ │ │ ├── benchmark_performance_options.cc │ │ │ ├── benchmark_performance_options.h │ │ │ ├── benchmark_plus_flex_main.cc │ │ │ ├── benchmark_test.cc │ │ │ ├── benchmark_tflite_model.cc │ │ │ ├── benchmark_tflite_model.h │ │ │ ├── benchmark_tflite_model_test.cc │ │ │ ├── benchmark_tflite_performance_options_main.cc │ │ │ ├── benchmark_utils.cc │ │ │ ├── benchmark_utils.h │ │ │ ├── benchmark_utils_test.cc │ │ │ ├── experimental │ │ │ │ ├── c │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── benchmark_c_api.cc │ │ │ │ │ └── benchmark_c_api.h │ │ │ │ ├── delegate_performance │ │ │ │ │ └── android │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── build_defs.bzl │ │ │ │ │ │ ├── models │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── mobilenet_v1_1.0_224.textproto │ │ │ │ │ │ └── mobilenet_v1_1.0_224_quant.textproto │ │ │ │ │ │ ├── proto.bzl │ │ │ │ │ │ ├── proto │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── default_latency_criteria.textproto │ │ │ │ │ │ └── delegate_performance.proto │ │ │ │ │ │ └── src │ │ │ │ │ │ ├── main │ │ │ │ │ │ ├── java │ │ │ │ │ │ │ └── org │ │ │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ │ │ └── lite │ │ │ │ │ │ │ │ └── benchmark │ │ │ │ │ │ │ │ └── delegateperformance │ │ │ │ │ │ │ │ ├── AccuracyBenchmarkReport.java │ │ │ │ │ │ │ │ ├── BenchmarkAccuracy.java │ │ │ │ │ │ │ │ ├── BenchmarkAccuracyActivity.java │ │ │ │ │ │ │ │ ├── BenchmarkAccuracyImpl.java │ │ │ │ │ │ │ │ ├── BenchmarkLatencyActivity.java │ │ │ │ │ │ │ │ ├── BenchmarkLatencyImpl.java │ │ │ │ │ │ │ │ ├── BenchmarkReport.java │ │ │ │ │ │ │ │ ├── BenchmarkResultType.java │ │ │ │ │ │ │ │ ├── CsvWriter.java │ │ │ │ │ │ │ │ ├── DelegateMetricsEntry.java │ │ │ │ │ │ │ │ ├── DelegatePerformanceBenchmark.java │ │ │ │ │ │ │ │ ├── HtmlWriter.java │ │ │ │ │ │ │ │ ├── JsonWriter.java │ │ │ │ │ │ │ │ ├── LatencyBenchmarkReport.java │ │ │ │ │ │ │ │ ├── MetricsEntry.java │ │ │ │ │ │ │ │ ├── ModelBenchmarkReport.java │ │ │ │ │ │ │ │ ├── ModelBenchmarkReportInterface.java │ │ │ │ │ │ │ │ ├── Preconditions.java │ │ │ │ │ │ │ │ ├── RawDelegateMetricsEntry.java │ │ │ │ │ │ │ │ ├── ReportWriter.java │ │ │ │ │ │ │ │ └── TfLiteSettingsListEntry.java │ │ │ │ │ │ └── native │ │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ │ ├── accuracy_benchmark.cc │ │ │ │ │ │ │ ├── accuracy_benchmark.h │ │ │ │ │ │ │ ├── delegate_performance_benchmark_jni.cc │ │ │ │ │ │ │ ├── latency_benchmark.cc │ │ │ │ │ │ │ ├── latency_benchmark.h │ │ │ │ │ │ │ └── status_codes.h │ │ │ │ │ │ └── test │ │ │ │ │ │ └── native │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── accuracy_benchmark_test.cc │ │ │ │ │ │ └── latency_benchmark_test.cc │ │ │ │ ├── firebase │ │ │ │ │ └── android │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── jni │ │ │ │ │ │ └── benchmark_model_jni.cc │ │ │ │ │ │ └── src │ │ │ │ │ │ └── org │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── lite │ │ │ │ │ │ └── benchmark │ │ │ │ │ │ └── firebase │ │ │ │ │ │ ├── BenchmarkModel.java │ │ │ │ │ │ └── BenchmarkModelActivity.java │ │ │ │ └── ios │ │ │ │ │ └── BUILD.apple │ │ │ ├── ios │ │ │ │ ├── README.md │ │ │ │ ├── TFLiteBenchmark │ │ │ │ │ ├── TFLiteBenchmark.xcodeproj │ │ │ │ │ │ └── project.pbxproj │ │ │ │ │ └── TFLiteBenchmark │ │ │ │ │ │ ├── AppDelegate.h │ │ │ │ │ │ ├── AppDelegate.m │ │ │ │ │ │ ├── Assets.xcassets │ │ │ │ │ │ ├── AppIcon.appiconset │ │ │ │ │ │ │ └── Contents.json │ │ │ │ │ │ └── Contents.json │ │ │ │ │ │ ├── Base.lproj │ │ │ │ │ │ ├── LaunchScreen.storyboard │ │ │ │ │ │ └── Main.storyboard │ │ │ │ │ │ ├── BenchmarkViewController.h │ │ │ │ │ │ ├── BenchmarkViewController.mm │ │ │ │ │ │ ├── Frameworks │ │ │ │ │ │ └── .gitignore │ │ │ │ │ │ ├── Info.plist │ │ │ │ │ │ ├── benchmark_data │ │ │ │ │ │ └── benchmark_params.json │ │ │ │ │ │ └── main.m │ │ │ │ └── build_benchmark_framework.sh │ │ │ ├── profiling_listener.cc │ │ │ ├── profiling_listener.h │ │ │ ├── register_custom_op.cc │ │ │ └── register_custom_op.h │ │ ├── build_aar.sh │ │ ├── build_aar_with_docker.sh │ │ ├── cmake │ │ │ ├── README.md │ │ │ ├── download_toolchains.sh │ │ │ ├── modules │ │ │ │ ├── FindEigen3.cmake │ │ │ │ ├── FindFlatBuffers.cmake │ │ │ │ ├── FindNEON_2_SSE.cmake │ │ │ │ ├── FindXNNPACK.cmake │ │ │ │ ├── Findabsl.cmake │ │ │ │ ├── Findcpuinfo.cmake │ │ │ │ ├── Findegl_headers.cmake │ │ │ │ ├── Findfarmhash.cmake │ │ │ │ ├── Findfft2d.cmake │ │ │ │ ├── Findfp16_headers.cmake │ │ │ │ ├── Findgemmlowp.cmake │ │ │ │ ├── Findgoogle_benchmark.cmake │ │ │ │ ├── Findgoogletest.cmake │ │ │ │ ├── Findml_dtypes.cmake │ │ │ │ ├── Findnsync.cmake │ │ │ │ ├── Findopencl_headers.cmake │ │ │ │ ├── Findopengl_headers.cmake │ │ │ │ ├── Findre2.cmake │ │ │ │ ├── Findruy.cmake │ │ │ │ ├── Findvulkan_headers.cmake │ │ │ │ ├── OverridableFetchContent.cmake │ │ │ │ ├── abseil-cpp.cmake │ │ │ │ ├── cpuinfo.cmake │ │ │ │ ├── egl_headers.cmake │ │ │ │ ├── eigen.cmake │ │ │ │ ├── farmhash.cmake │ │ │ │ ├── farmhash │ │ │ │ │ └── CMakeLists.txt │ │ │ │ ├── fft2d.cmake │ │ │ │ ├── fft2d │ │ │ │ │ └── CMakeLists.txt │ │ │ │ ├── flatbuffers.cmake │ │ │ │ ├── fp16_headers.cmake │ │ │ │ ├── gemmlowp.cmake │ │ │ │ ├── ml_dtypes.cmake │ │ │ │ ├── ml_dtypes │ │ │ │ │ └── CMakeLists.txt │ │ │ │ ├── neon2sse.cmake │ │ │ │ ├── opencl_headers.cmake │ │ │ │ ├── opengl_headers.cmake │ │ │ │ ├── ruy.cmake │ │ │ │ ├── vulkan_headers.cmake │ │ │ │ ├── xnnpack.cmake │ │ │ │ └── xnnpack │ │ │ │ │ └── CMakeLists.txt │ │ │ ├── native_tools │ │ │ │ └── flatbuffers │ │ │ │ │ └── CMakeLists.txt │ │ │ ├── tensorflow-liteConfig.cmake.in │ │ │ └── test_utils │ │ │ │ └── run-tests.cmake │ │ ├── command_line_flags.cc │ │ ├── command_line_flags.h │ │ ├── command_line_flags_test.cc │ │ ├── convert_image_to_csv.py │ │ ├── convert_image_to_csv_test.py │ │ ├── delegates │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── compatibility │ │ │ │ ├── README.md │ │ │ │ ├── common │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── delegate_compatibility_checker_base.cc │ │ │ │ │ ├── delegate_compatibility_checker_base.h │ │ │ │ │ ├── delegate_compatibility_checker_util.h │ │ │ │ │ ├── online_helper_delegate.cc │ │ │ │ │ └── online_helper_delegate.h │ │ │ │ ├── gpu │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── gpu_delegate_compatibility_checker.cc │ │ │ │ │ ├── gpu_delegate_compatibility_checker.h │ │ │ │ │ └── gpu_delegate_compatibility_checker_test.cc │ │ │ │ ├── nnapi │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── nnapi_compatibility_lib.cc │ │ │ │ │ ├── nnapi_compatibility_lib.h │ │ │ │ │ ├── nnapi_compatibility_lib_test.cc │ │ │ │ │ ├── nnapi_delegate_compatibility_checker.cc │ │ │ │ │ ├── nnapi_delegate_compatibility_checker.h │ │ │ │ │ └── nnapi_delegate_compatibility_checker_test.cc │ │ │ │ └── protos │ │ │ │ │ ├── BUILD │ │ │ │ │ └── compatibility_result.proto │ │ │ ├── coreml_delegate_provider.cc │ │ │ ├── default_execution_provider.cc │ │ │ ├── delegate_provider.cc │ │ │ ├── delegate_provider.h │ │ │ ├── delegate_provider_test.cc │ │ │ ├── experimental │ │ │ │ └── stable_delegate │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── stable_delegate_provider.cc │ │ │ │ │ ├── stable_delegate_provider_test.cc │ │ │ │ │ ├── test_invalid_settings.json │ │ │ │ │ ├── test_missing_delegate_path_settings.json │ │ │ │ │ ├── test_missing_stable_delegate_settings.json │ │ │ │ │ ├── test_sample_stable_delegate_settings.json │ │ │ │ │ └── test_stable_xnnpack_settings.json │ │ │ ├── external_delegate_provider.cc │ │ │ ├── gpu_delegate_provider.cc │ │ │ ├── hexagon_delegate_provider.cc │ │ │ ├── nnapi_delegate_provider.cc │ │ │ └── xnnpack_delegate_provider.cc │ │ ├── evaluation │ │ │ ├── BUILD │ │ │ ├── evaluation_delegate_provider.cc │ │ │ ├── evaluation_delegate_provider.h │ │ │ ├── evaluation_delegate_provider_test.cc │ │ │ ├── evaluation_stage.h │ │ │ ├── proto │ │ │ │ ├── BUILD │ │ │ │ ├── evaluation_config.proto │ │ │ │ ├── evaluation_stages.proto │ │ │ │ └── preprocessing_steps.proto │ │ │ ├── stages │ │ │ │ ├── BUILD │ │ │ │ ├── image_classification_stage.cc │ │ │ │ ├── image_classification_stage.h │ │ │ │ ├── image_preprocessing_stage.cc │ │ │ │ ├── image_preprocessing_stage.h │ │ │ │ ├── image_preprocessing_stage_test.cc │ │ │ │ ├── inference_profiler_stage.cc │ │ │ │ ├── inference_profiler_stage.h │ │ │ │ ├── inference_profiler_stage_test.cc │ │ │ │ ├── object_detection_average_precision_stage.cc │ │ │ │ ├── object_detection_average_precision_stage.h │ │ │ │ ├── object_detection_average_precision_stage_test.cc │ │ │ │ ├── object_detection_stage.cc │ │ │ │ ├── object_detection_stage.h │ │ │ │ ├── testdata │ │ │ │ │ └── grace_hopper.jpg │ │ │ │ ├── tflite_inference_stage.cc │ │ │ │ ├── tflite_inference_stage.h │ │ │ │ ├── tflite_inference_stage_test.cc │ │ │ │ ├── topk_accuracy_eval_stage.cc │ │ │ │ ├── topk_accuracy_eval_stage.h │ │ │ │ ├── topk_accuracy_eval_stage_test.cc │ │ │ │ └── utils │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── image_metrics.cc │ │ │ │ │ ├── image_metrics.h │ │ │ │ │ └── image_metrics_test.cc │ │ │ ├── tasks │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── build_def.bzl │ │ │ │ ├── coco_object_detection │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── preprocess_coco_minival.py │ │ │ │ │ └── run_eval.cc │ │ │ │ ├── imagenet_image_classification │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── generate_validation_labels.py │ │ │ │ │ └── run_eval.cc │ │ │ │ ├── inference_diff │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ └── run_eval.cc │ │ │ │ ├── ios │ │ │ │ │ ├── BUILD.apple │ │ │ │ │ ├── README.md │ │ │ │ │ ├── TFLiteEvaluation │ │ │ │ │ │ ├── TFLiteEvaluation.xcodeproj │ │ │ │ │ │ │ └── project.pbxproj │ │ │ │ │ │ └── TFLiteEvaluation │ │ │ │ │ │ │ ├── AppDelegate.h │ │ │ │ │ │ │ ├── AppDelegate.m │ │ │ │ │ │ │ ├── Assets.xcassets │ │ │ │ │ │ │ ├── AccentColor.colorset │ │ │ │ │ │ │ │ └── Contents.json │ │ │ │ │ │ │ ├── AppIcon.appiconset │ │ │ │ │ │ │ │ └── Contents.json │ │ │ │ │ │ │ └── Contents.json │ │ │ │ │ │ │ ├── Base.lproj │ │ │ │ │ │ │ └── Main.storyboard │ │ │ │ │ │ │ ├── EvaluationViewController.h │ │ │ │ │ │ │ ├── EvaluationViewController.mm │ │ │ │ │ │ │ ├── Info.plist │ │ │ │ │ │ │ ├── evaluation_data │ │ │ │ │ │ │ └── evaluation_params.json │ │ │ │ │ │ │ └── main.m │ │ │ │ │ └── build_evaluation_framework.sh │ │ │ │ ├── task_executor.cc │ │ │ │ ├── task_executor.h │ │ │ │ ├── task_executor_c_api.cc │ │ │ │ ├── task_executor_c_api.h │ │ │ │ └── task_executor_main.cc │ │ │ ├── testdata │ │ │ │ ├── empty.txt │ │ │ │ └── labels.txt │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ │ ├── flatbuffer_utils.py │ │ ├── flatbuffer_utils_test.py │ │ ├── gen_op_registration.cc │ │ ├── gen_op_registration.h │ │ ├── gen_op_registration_main.cc │ │ ├── gen_op_registration_test.cc │ │ ├── list_flex_ops.cc │ │ ├── list_flex_ops.h │ │ ├── list_flex_ops_main.cc │ │ ├── list_flex_ops_no_kernel.cc │ │ ├── list_flex_ops_test.cc │ │ ├── logging.h │ │ ├── make │ │ │ └── README.md │ │ ├── model_loader.cc │ │ ├── model_loader.h │ │ ├── model_loader_test.cc │ │ ├── optimize │ │ │ ├── BUILD │ │ │ ├── calibration │ │ │ │ ├── BUILD │ │ │ │ ├── builtin_logging_ops │ │ │ │ │ ├── lstm.cc │ │ │ │ │ └── lstm.h │ │ │ │ ├── calibration_common.h │ │ │ │ ├── calibration_logger.cc │ │ │ │ ├── calibration_logger.h │ │ │ │ ├── calibration_reader.cc │ │ │ │ ├── calibration_reader.h │ │ │ │ ├── calibrator.cc │ │ │ │ ├── calibrator.h │ │ │ │ ├── calibrator_test.cc │ │ │ │ ├── custom_logging_ops │ │ │ │ │ ├── lstm.cc │ │ │ │ │ └── lstm.h │ │ │ │ ├── logging_op.h │ │ │ │ ├── logging_op_resolver.cc │ │ │ │ ├── logging_op_resolver.h │ │ │ │ └── logging_op_resolver_test.cc │ │ │ ├── debugging │ │ │ │ ├── README.md │ │ │ │ └── python │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── debugger.py │ │ │ │ │ └── debugger_test.py │ │ │ ├── g3doc │ │ │ │ └── quantize_weights.md │ │ │ ├── model_utils.cc │ │ │ ├── model_utils.h │ │ │ ├── model_utils_test.cc │ │ │ ├── modify_model_interface.cc │ │ │ ├── modify_model_interface.h │ │ │ ├── modify_model_interface_main.cc │ │ │ ├── modify_model_interface_test.cc │ │ │ ├── operator_property.cc │ │ │ ├── operator_property.h │ │ │ ├── python │ │ │ │ ├── BUILD │ │ │ │ ├── _pywrap_modify_model_interface.pyi │ │ │ │ ├── modify_model_interface.cc │ │ │ │ ├── modify_model_interface.py │ │ │ │ ├── modify_model_interface_constants.py │ │ │ │ ├── modify_model_interface_lib.py │ │ │ │ └── modify_model_interface_lib_test.py │ │ │ ├── quantization_utils.cc │ │ │ ├── quantization_utils.h │ │ │ ├── quantization_utils_test.cc │ │ │ ├── quantization_wrapper.cc │ │ │ ├── quantization_wrapper.h │ │ │ ├── quantization_wrapper_utils.cc │ │ │ ├── quantization_wrapper_utils.h │ │ │ ├── quantization_wrapper_utils_custom_test.cc │ │ │ ├── quantization_wrapper_utils_test.cc │ │ │ ├── quantize_model.cc │ │ │ ├── quantize_model.h │ │ │ ├── quantize_model_test.cc │ │ │ ├── quantize_weights.cc │ │ │ ├── quantize_weights.h │ │ │ ├── quantize_weights_portable.cc │ │ │ ├── quantize_weights_test.cc │ │ │ ├── reduced_precision_support.h │ │ │ ├── reduced_precision_support_test.cc │ │ │ ├── sparsity │ │ │ │ ├── BUILD │ │ │ │ ├── format_converter_wrapper_pybind11.cc │ │ │ │ ├── format_converter_wrapper_pybind11.pyi │ │ │ │ └── format_converter_wrapper_pybind11_test.py │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ └── testdata │ │ │ │ ├── README.md │ │ │ │ ├── add_with_const_input.bin │ │ │ │ ├── argmax.bin │ │ │ │ ├── broadcast_to.bin │ │ │ │ ├── concat.bin │ │ │ │ ├── custom_op.bin │ │ │ │ ├── fc.bin │ │ │ │ ├── fc_qat.bin │ │ │ │ ├── gather_nd.bin │ │ │ │ ├── lstm_calibrated.bin │ │ │ │ ├── lstm_calibrated2.bin │ │ │ │ ├── lstm_quantized.bin │ │ │ │ ├── lstm_quantized2.bin │ │ │ │ ├── maximum.bin │ │ │ │ ├── minimum.bin │ │ │ │ ├── mixed.bin │ │ │ │ ├── mixed16x8.bin │ │ │ │ ├── multi_input_add_reshape.bin │ │ │ │ ├── pack.bin │ │ │ │ ├── quantized_with_gather.bin │ │ │ │ ├── resource_vars_calibrated.bin │ │ │ │ ├── single_avg_pool_min_minus_5_max_plus_5.bin │ │ │ │ ├── single_conv_no_bias.bin │ │ │ │ ├── single_conv_weights_min_0_max_plus_10.bin │ │ │ │ ├── single_conv_weights_min_minus_127_max_plus_127.bin │ │ │ │ ├── single_softmax_min_minus_5_max_plus_5.bin │ │ │ │ ├── split.bin │ │ │ │ ├── svdf_calibrated.bin │ │ │ │ ├── svdf_quantized.bin │ │ │ │ ├── transpose.bin │ │ │ │ ├── unidirectional_sequence_lstm_calibrated.bin │ │ │ │ ├── unidirectional_sequence_lstm_quantized.bin │ │ │ │ ├── unpack.bin │ │ │ │ ├── weight_shared_between_convs.bin │ │ │ │ └── where.bin │ │ ├── pip_package │ │ │ ├── Dockerfile.py3 │ │ │ ├── MANIFEST.in │ │ │ ├── Makefile │ │ │ ├── README.md │ │ │ ├── build_pip_package_with_bazel.sh │ │ │ ├── build_pip_package_with_cmake.sh │ │ │ ├── debian │ │ │ │ ├── changelog │ │ │ │ ├── compat │ │ │ │ ├── control │ │ │ │ ├── copyright │ │ │ │ └── rules │ │ │ ├── setup_with_binary.py │ │ │ ├── update_sources.sh │ │ │ └── with_the_same_user │ │ ├── randomize_weights.py │ │ ├── reverse_xxd_dump_from_cc.py │ │ ├── serialization │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── enum_mapping.h │ │ │ ├── option_writer_generator.cc │ │ │ ├── writer.cc │ │ │ ├── writer_lib.cc │ │ │ ├── writer_lib.h │ │ │ ├── writer_lib_test.cc │ │ │ └── writer_test.cc │ │ ├── signature │ │ │ ├── BUILD │ │ │ ├── _pywrap_signature_def_util_wrapper.pyi │ │ │ ├── signature_def_util.cc │ │ │ ├── signature_def_util.h │ │ │ ├── signature_def_util_test.cc │ │ │ ├── signature_def_util_wrapper_pybind11.cc │ │ │ ├── signature_def_utils.py │ │ │ └── signature_def_utils_test.py │ │ ├── strip_buffers │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── reconstitute_buffers_into_fb.cc │ │ │ ├── strip_buffers_from_fb.cc │ │ │ ├── stripping_lib.cc │ │ │ └── stripping_lib.h │ │ ├── strip_strings.py │ │ ├── test_utils.py │ │ ├── tflite-android.Dockerfile │ │ ├── tool_params.cc │ │ ├── tool_params.h │ │ ├── tool_params_test.cc │ │ ├── utils.cc │ │ ├── utils.h │ │ ├── verifier.h │ │ ├── verifier_internal.h │ │ ├── versioning │ │ │ ├── BUILD │ │ │ ├── gpu_compatibility.cc │ │ │ ├── gpu_compatibility.h │ │ │ ├── gpu_compatibility_test.cc │ │ │ ├── op_signature.cc │ │ │ ├── op_signature.h │ │ │ ├── op_signature_test.cc │ │ │ ├── op_version.cc │ │ │ ├── op_version.h │ │ │ ├── op_version_test.cc │ │ │ ├── runtime_version.cc │ │ │ ├── runtime_version.h │ │ │ └── runtime_version_test.cc │ │ ├── visualize.py │ │ ├── visualize_test.py │ │ └── zip_files.py │ ├── tutorials │ │ ├── BUILD │ │ ├── dataset.py │ │ └── mnist_tflite.py │ ├── type_to_tflitetype.h │ ├── type_to_tflitetype_test.cc │ ├── util.cc │ ├── util.h │ ├── util_test.cc │ └── version.h ├── opensource_only.files ├── py.default.bzl ├── python │ ├── BUILD │ ├── __init__.py │ ├── _pywrap_dtensor_device.pyi │ ├── _pywrap_mlir.pyi │ ├── _pywrap_parallel_device.pyi │ ├── _pywrap_py_exception_registry.pyi │ ├── _pywrap_quantize_training.pyi │ ├── _pywrap_sanitizers.pyi │ ├── _pywrap_tfcompile.pyi │ ├── _pywrap_tfe.pyi │ ├── _pywrap_tfe_monitoring_reader.pyi │ ├── _pywrap_toco_api.pyi │ ├── autograph │ │ ├── BUILD │ │ ├── CONTRIBUTING.md │ │ ├── STYLE_GUIDE.md │ │ ├── __init__.py │ │ ├── converters │ │ │ ├── BUILD │ │ │ ├── asserts.py │ │ │ ├── asserts_test.py │ │ │ ├── break_statements.py │ │ │ ├── break_statements_test.py │ │ │ ├── call_trees.py │ │ │ ├── call_trees_test.py │ │ │ ├── conditional_expressions.py │ │ │ ├── conditional_expressions_test.py │ │ │ ├── continue_statements.py │ │ │ ├── continue_statements_test.py │ │ │ ├── control_flow.py │ │ │ ├── control_flow_test.py │ │ │ ├── directives.py │ │ │ ├── directives_test.py │ │ │ ├── functions.py │ │ │ ├── functions_test.py │ │ │ ├── lists.py │ │ │ ├── lists_test.py │ │ │ ├── logical_expressions.py │ │ │ ├── logical_expressions_test.py │ │ │ ├── return_statements.py │ │ │ ├── return_statements_test.py │ │ │ ├── slices.py │ │ │ ├── slices_test.py │ │ │ ├── variables.py │ │ │ └── variables_test.py │ │ ├── core │ │ │ ├── BUILD │ │ │ ├── ag_ctx.py │ │ │ ├── config.py │ │ │ ├── config_lib.py │ │ │ ├── converter.py │ │ │ ├── converter_test.py │ │ │ ├── converter_testing.py │ │ │ ├── function_wrappers.py │ │ │ ├── function_wrappers_test.py │ │ │ └── unsupported_features_checker.py │ │ ├── g3doc │ │ │ ├── pyct_tutorial.ipynb │ │ │ └── reference │ │ │ │ ├── _control_flow_tutorial.ipynb │ │ │ │ ├── common_errors.md │ │ │ │ ├── control_flow.md │ │ │ │ ├── debugging.md │ │ │ │ ├── error_handling.md │ │ │ │ ├── functions.md │ │ │ │ ├── generated_code.md │ │ │ │ ├── index.md │ │ │ │ ├── intro.md │ │ │ │ ├── limitations.md │ │ │ │ └── operators.md │ │ ├── impl │ │ │ ├── BUILD │ │ │ ├── api.py │ │ │ ├── api_test.py │ │ │ ├── conversion.py │ │ │ ├── conversion_test.py │ │ │ └── testing │ │ │ │ ├── BUILD │ │ │ │ ├── pybind_for_testing.cc │ │ │ │ └── pybind_for_testing.pyi │ │ ├── lang │ │ │ ├── BUILD │ │ │ ├── directives.py │ │ │ ├── special_functions.py │ │ │ └── special_functions_test.py │ │ ├── operators │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── conditional_expressions.py │ │ │ ├── conditional_expressions_test.py │ │ │ ├── control_flow.py │ │ │ ├── control_flow_test.py │ │ │ ├── data_structures.py │ │ │ ├── data_structures_test.py │ │ │ ├── dispatch_context.py │ │ │ ├── exceptions.py │ │ │ ├── exceptions_test.py │ │ │ ├── logical.py │ │ │ ├── logical_test.py │ │ │ ├── py_builtins.py │ │ │ ├── py_builtins_test.py │ │ │ ├── slices.py │ │ │ ├── slices_test.py │ │ │ ├── variables.py │ │ │ └── variables_test.py │ │ ├── pyct │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── anno.py │ │ │ ├── anno_test.py │ │ │ ├── ast_util.py │ │ │ ├── ast_util_test.py │ │ │ ├── cache.py │ │ │ ├── cache_test.py │ │ │ ├── cfg.py │ │ │ ├── cfg_test.py │ │ │ ├── common_transformers │ │ │ │ ├── BUILD │ │ │ │ ├── __init__.py │ │ │ │ ├── anf.py │ │ │ │ └── anf_test.py │ │ │ ├── error_utils.py │ │ │ ├── error_utils_test.py │ │ │ ├── errors.py │ │ │ ├── gast_util.py │ │ │ ├── inspect_utils.py │ │ │ ├── inspect_utils_test.py │ │ │ ├── inspect_utils_test.sh │ │ │ ├── loader.py │ │ │ ├── loader_test.py │ │ │ ├── naming.py │ │ │ ├── naming_test.py │ │ │ ├── origin_info.py │ │ │ ├── origin_info_test.py │ │ │ ├── parser.py │ │ │ ├── parser_test.py │ │ │ ├── pretty_printer.py │ │ │ ├── pretty_printer_test.py │ │ │ ├── qual_names.py │ │ │ ├── qual_names_test.py │ │ │ ├── static_analysis │ │ │ │ ├── BUILD │ │ │ │ ├── __init__.py │ │ │ │ ├── activity.py │ │ │ │ ├── activity_test.py │ │ │ │ ├── annos.py │ │ │ │ ├── liveness.py │ │ │ │ ├── liveness_py3_test.py │ │ │ │ ├── liveness_test.py │ │ │ │ ├── reaching_definitions.py │ │ │ │ ├── reaching_definitions_py3_test.py │ │ │ │ ├── reaching_definitions_test.py │ │ │ │ ├── reaching_fndefs.py │ │ │ │ ├── reaching_fndefs_test.py │ │ │ │ ├── type_inference.py │ │ │ │ └── type_inference_test.py │ │ │ ├── templates.py │ │ │ ├── templates_test.py │ │ │ ├── testing │ │ │ │ ├── BUILD │ │ │ │ ├── basic_definitions.py │ │ │ │ ├── codegen.py │ │ │ │ ├── codegen_test.py │ │ │ │ └── decorators.py │ │ │ ├── transformer.py │ │ │ ├── transformer_test.py │ │ │ ├── transpiler.py │ │ │ └── transpiler_test.py │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── assertion_test.py │ │ │ ├── basic_ifexp_test.py │ │ │ ├── basic_list_test.py │ │ │ ├── call_to_builtin_function_test.py │ │ │ ├── call_to_lambda_function_test.py │ │ │ ├── call_to_named_tuple_test.py │ │ │ ├── call_to_numpy_function_test.py │ │ │ ├── call_to_print_function_test.py │ │ │ ├── call_to_tf_api_test.py │ │ │ ├── call_to_user_function_test.py │ │ │ ├── composite_names_in_control_flow_test.py │ │ │ ├── cond_basic_test.py │ │ │ ├── datasets_test.py │ │ │ ├── early_return_test.py │ │ │ ├── ext_slice_test.py │ │ │ ├── generator_test.py │ │ │ ├── logical_expression_test.py │ │ │ ├── loop_basic_test.py │ │ │ ├── loop_control_flow_illegal_cases_test.py │ │ │ ├── loop_control_flow_test.py │ │ │ ├── loop_created_variables_test.py │ │ │ ├── loop_distributed_test.py │ │ │ ├── loop_scoping_test.py │ │ │ ├── loop_with_function_call_test.py │ │ │ ├── loop_with_variable_type_illegal_cases_test.py │ │ │ ├── loop_with_variable_type_test.py │ │ │ ├── nested_control_flow_test.py │ │ │ ├── reference_test.bzl │ │ │ ├── reference_test_base.py │ │ │ └── type_annotations_test.py │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── ag_logging.py │ │ │ ├── context_managers.py │ │ │ ├── context_managers_test.py │ │ │ ├── misc.py │ │ │ ├── misc_test.py │ │ │ ├── tensor_list.py │ │ │ ├── tensor_list_test.py │ │ │ ├── tensors.py │ │ │ ├── tensors_test.py │ │ │ ├── testing.py │ │ │ └── type_registry.py │ ├── build_defs.bzl │ ├── checkpoint │ │ ├── BUILD │ │ ├── async_checkpoint_helper.py │ │ ├── benchmarks_test.py │ │ ├── checkpoint.py │ │ ├── checkpoint_context.py │ │ ├── checkpoint_management.py │ │ ├── checkpoint_management_test.py │ │ ├── checkpoint_metrics_test.py │ │ ├── checkpoint_options.py │ │ ├── checkpoint_test.py │ │ ├── checkpoint_view.py │ │ ├── checkpoint_view_test.py │ │ ├── checkpoint_with_v1_optimizers_test.py │ │ ├── functional_saver.py │ │ ├── functional_saver_test.py │ │ ├── graph_view.py │ │ ├── restore.py │ │ ├── restore_test.py │ │ ├── save_util.py │ │ ├── save_util_v1.py │ │ ├── save_util_v1_test.py │ │ ├── saveable_compat.py │ │ ├── saveable_compat_test.py │ │ ├── sharding │ │ │ ├── BUILD │ │ │ ├── sharding_policies.py │ │ │ ├── sharding_policies_test.py │ │ │ ├── sharding_util.py │ │ │ └── sharding_util_test.py │ │ ├── tensor_callable.py │ │ ├── tensor_callable_test.py │ │ ├── testdata │ │ │ ├── generate_checkpoint.py │ │ │ ├── table_legacy_saveable_object.data-00000-of-00001 │ │ │ └── table_legacy_saveable_object.index │ │ ├── trackable_view.py │ │ ├── trackable_view_test.py │ │ └── util.py │ ├── client │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── _pywrap_debug_events_writer.pyi │ │ ├── _pywrap_device_lib.pyi │ │ ├── _pywrap_events_writer.pyi │ │ ├── _pywrap_tf_session.pyi │ │ ├── client_lib.py │ │ ├── debug_events_writer_wrapper.cc │ │ ├── device_lib.py │ │ ├── device_lib_test.py │ │ ├── device_lib_wrapper.cc │ │ ├── events_writer_test.py │ │ ├── events_writer_wrapper.cc │ │ ├── notebook.py │ │ ├── pywrap_tf_session.py │ │ ├── session.py │ │ ├── session_benchmark.py │ │ ├── session_clusterspec_prop_test.py │ │ ├── session_list_devices_test.py │ │ ├── session_partial_run_test.py │ │ ├── session_ref.cc │ │ ├── session_ref.h │ │ ├── session_test.py │ │ ├── test_construction_fails_op.cc │ │ ├── tf_session_helper.cc │ │ ├── tf_session_helper.h │ │ ├── tf_session_wrapper.cc │ │ ├── timeline.py │ │ ├── timeline_test.py │ │ └── virtual_gpu_test.py │ ├── compat │ │ ├── BUILD │ │ ├── compat.py │ │ ├── compat_test.py │ │ ├── disable_v2_behavior_test.py │ │ └── v2_compat.py │ ├── compiler │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── mlir │ │ │ ├── BUILD │ │ │ ├── mlir.py │ │ │ ├── mlir_test.py │ │ │ └── multi_add.tflite │ │ ├── tensorrt │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── model_tests │ │ │ │ ├── BUILD │ │ │ │ ├── model_handler.py │ │ │ │ ├── result_analyzer.py │ │ │ │ ├── run_models.py │ │ │ │ └── sample_model │ │ │ │ │ └── saved_model.pb │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── annotate_max_batch_sizes_test.py │ │ │ │ ├── base_test.py │ │ │ │ ├── batch_matmul_test.py │ │ │ │ ├── biasadd_matmul_test.py │ │ │ │ ├── binary_tensor_weight_broadcast_test.py │ │ │ │ ├── bool_test.py │ │ │ │ ├── cast_test.py │ │ │ │ ├── combined_nms_test.py │ │ │ │ ├── concatenation_test.py │ │ │ │ ├── const_broadcast_test.py │ │ │ │ ├── conv2d_test.py │ │ │ │ ├── data_dependent_shape_test.py │ │ │ │ ├── dynamic_input_shapes_test.py │ │ │ │ ├── identity_output_test.py │ │ │ │ ├── int32_test.py │ │ │ │ ├── lru_cache_test.py │ │ │ │ ├── memory_alignment_test.py │ │ │ │ ├── multi_connection_neighbor_engine_test.py │ │ │ │ ├── neighboring_engine_test.py │ │ │ │ ├── quantization_mnist_test.py │ │ │ │ ├── quantization_test.py │ │ │ │ ├── rank_two_test.py │ │ │ │ ├── reshape_transpose_test.py │ │ │ │ ├── shape_output_test.py │ │ │ │ ├── test_utils.py │ │ │ │ ├── testdata │ │ │ │ │ ├── gen_tf_readvariableop_model.py │ │ │ │ │ ├── gen_tf_variablev2_model.py │ │ │ │ │ ├── gen_tftrt_model.py │ │ │ │ │ ├── mnist │ │ │ │ │ │ ├── checkpoint │ │ │ │ │ │ ├── model.ckpt-46900.data-00000-of-00001 │ │ │ │ │ │ └── model.ckpt-46900.index │ │ │ │ │ ├── tf_readvariableop_saved_model │ │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ │ └── variables │ │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ │ └── variables.index │ │ │ │ │ ├── tf_variablev2_saved_model │ │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ │ └── variables │ │ │ │ │ │ │ ├── variables.data-00000-of-00001 │ │ │ │ │ │ │ └── variables.index │ │ │ │ │ └── tftrt_2.0_saved_model │ │ │ │ │ │ ├── saved_model.pb │ │ │ │ │ │ └── variables │ │ │ │ │ │ ├── variables.data-00000-of-00002 │ │ │ │ │ │ ├── variables.data-00001-of-00002 │ │ │ │ │ │ └── variables.index │ │ │ │ ├── tf_function_test.py │ │ │ │ ├── tf_trt_integration_test_base.py │ │ │ │ ├── topk_test.py │ │ │ │ ├── trt_engine_op_shape_test.py │ │ │ │ ├── trt_mode_test.py │ │ │ │ ├── unary_test.py │ │ │ │ ├── vgg_block_nchw_test.py │ │ │ │ └── vgg_block_test.py │ │ │ ├── trt_convert.py │ │ │ ├── trt_convert_test.py │ │ │ └── utils.py │ │ └── xla │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── experimental │ │ │ ├── BUILD │ │ │ ├── xla_sharding.py │ │ │ └── xla_sharding_test.py │ │ │ ├── jit.py │ │ │ ├── pjrt_compile_virtual_device_test.py │ │ │ ├── tests │ │ │ ├── jit_compile_test.py │ │ │ ├── jit_test.py │ │ │ ├── pjrt_compile_test.py │ │ │ └── xla_test.py │ │ │ └── xla.py │ ├── data │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── benchmarks │ │ │ ├── BUILD │ │ │ ├── batch_benchmark.py │ │ │ ├── benchmark_base.py │ │ │ ├── filter_benchmark.py │ │ │ ├── from_tensor_slices_benchmark.py │ │ │ ├── interleave_benchmark.py │ │ │ ├── list_files_benchmark.py │ │ │ ├── map_benchmark.py │ │ │ ├── meta_benchmark.py │ │ │ ├── prefetch_benchmark.py │ │ │ └── range_benchmark.py │ │ ├── experimental │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── benchmarks │ │ │ │ ├── BUILD │ │ │ │ ├── autotune_benchmark.py │ │ │ │ ├── csv_dataset_benchmark.py │ │ │ │ ├── map_and_batch_benchmark.py │ │ │ │ ├── map_defun_benchmark.py │ │ │ │ ├── matching_files_benchmark.py │ │ │ │ ├── optimize_benchmark.py │ │ │ │ ├── parameter_value_benchmark.py │ │ │ │ ├── rejection_resample_benchmark.py │ │ │ │ ├── snapshot_dataset_benchmark.py │ │ │ │ └── unbatch_benchmark.py │ │ │ ├── kernel_tests │ │ │ │ ├── BUILD │ │ │ │ ├── assert_cardinality_test.py │ │ │ │ ├── assert_next_test.py │ │ │ │ ├── assert_prev_test.py │ │ │ │ ├── auto_shard_dataset_test.py │ │ │ │ ├── checkpoint_input_pipeline_hook_test.py │ │ │ │ ├── compression_ops_test.py │ │ │ │ ├── copy_to_device_test.py │ │ │ │ ├── csv_dataset_test.py │ │ │ │ ├── dense_to_sparse_batch_test.py │ │ │ │ ├── from_list_test.py │ │ │ │ ├── group_by_reducer_test.py │ │ │ │ ├── index_shuffle_test.py │ │ │ │ ├── io_test.py │ │ │ │ ├── lookup_ops_test.py │ │ │ │ ├── make_batched_features_dataset_test.py │ │ │ │ ├── make_csv_dataset_test.py │ │ │ │ ├── make_saveable_from_iterator_test.py │ │ │ │ ├── make_tf_record_dataset_test.py │ │ │ │ ├── map_and_batch_test.py │ │ │ │ ├── map_defun_op_test.py │ │ │ │ ├── matching_files_dataset_test.py │ │ │ │ ├── model_dataset_test.py │ │ │ │ ├── non_serializable_test.py │ │ │ │ ├── optimization │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── filter_fusion_test.py │ │ │ │ │ ├── filter_parallelization_test.py │ │ │ │ │ ├── grappler_test.py │ │ │ │ │ ├── make_deterministic_test.py │ │ │ │ │ ├── map_and_batch_fusion_test.py │ │ │ │ │ ├── map_and_filter_fusion_test.py │ │ │ │ │ ├── map_fusion_test.py │ │ │ │ │ ├── map_parallelization_test.py │ │ │ │ │ ├── noop_elimination_test.py │ │ │ │ │ ├── optimization_test.py │ │ │ │ │ └── shuffle_and_repeat_fusion_test.py │ │ │ │ ├── pad_to_cardinality_test.py │ │ │ │ ├── parallel_interleave_test.py │ │ │ │ ├── parse_example_dataset_test.py │ │ │ │ ├── prefetch_to_device_test.py │ │ │ │ ├── prefetch_with_slack_test.py │ │ │ │ ├── rebatch_dataset_test.py │ │ │ │ ├── replicate_test.py │ │ │ │ ├── service │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── auto_shard_test.py │ │ │ │ │ ├── coordinated_read_ft_test.py │ │ │ │ │ ├── coordinated_read_test.py │ │ │ │ │ ├── cross_trainer_cache_ft_test.py │ │ │ │ │ ├── cross_trainer_cache_test.py │ │ │ │ │ ├── data_service_ops_test.py │ │ │ │ │ ├── distributed_save_ft_test.py │ │ │ │ │ ├── distributed_save_test.py │ │ │ │ │ ├── dynamic_sharding_test.py │ │ │ │ │ ├── fault_tolerance_test.py │ │ │ │ │ ├── local_workers_test.py │ │ │ │ │ ├── metadata_test.py │ │ │ │ │ ├── multi_device_test.py │ │ │ │ │ ├── multi_process_cluster.py │ │ │ │ │ ├── multi_process_cluster_test.py │ │ │ │ │ ├── test_base.py │ │ │ │ │ └── worker_tags_test.py │ │ │ │ ├── shuffle_and_repeat_test.py │ │ │ │ ├── sleep_test.py │ │ │ │ ├── sql_dataset_test.py │ │ │ │ ├── tf_record_writer_test.py │ │ │ │ ├── variant_test.py │ │ │ │ └── wrap_unwrap_test.py │ │ │ ├── ops │ │ │ │ ├── BUILD │ │ │ │ ├── batching.py │ │ │ │ ├── cardinality.py │ │ │ │ ├── compression_ops.py │ │ │ │ ├── counter.py │ │ │ │ ├── data_service_ops.py │ │ │ │ ├── distribute.py │ │ │ │ ├── distributed_save_op.py │ │ │ │ ├── enumerate_ops.py │ │ │ │ ├── error_ops.py │ │ │ │ ├── from_list.py │ │ │ │ ├── get_single_element.py │ │ │ │ ├── grouping.py │ │ │ │ ├── interleave_ops.py │ │ │ │ ├── io.py │ │ │ │ ├── iterator_ops.py │ │ │ │ ├── lookup_ops.py │ │ │ │ ├── map_defun.py │ │ │ │ ├── matching_files.py │ │ │ │ ├── pad_to_cardinality.py │ │ │ │ ├── parsing_ops.py │ │ │ │ ├── prefetching_ops.py │ │ │ │ ├── random_access.py │ │ │ │ ├── random_ops.py │ │ │ │ ├── readers.py │ │ │ │ ├── resampling.py │ │ │ │ ├── scan_ops.py │ │ │ │ ├── shuffle_ops.py │ │ │ │ ├── sleep.py │ │ │ │ ├── snapshot.py │ │ │ │ ├── take_while_ops.py │ │ │ │ ├── testing.py │ │ │ │ ├── unique.py │ │ │ │ └── writers.py │ │ │ └── service │ │ │ │ ├── BUILD │ │ │ │ ├── __init__.py │ │ │ │ ├── _pywrap_server_lib.pyi │ │ │ │ ├── _pywrap_snapshot_utils.pyi │ │ │ │ ├── _pywrap_utils.pyi │ │ │ │ ├── server_lib.py │ │ │ │ ├── server_lib_test.py │ │ │ │ ├── server_lib_wrapper.cc │ │ │ │ ├── snapshot_utils_wrapper.cc │ │ │ │ └── utils_wrapper.cc │ │ ├── kernel_tests │ │ │ ├── BUILD │ │ │ ├── as_numpy_iterator_test.py │ │ │ ├── batch_test.py │ │ │ ├── bucket_by_sequence_length_test.py │ │ │ ├── cache_test.py │ │ │ ├── cardinality_test.py │ │ │ ├── checkpoint_test.py │ │ │ ├── checkpoint_test_base.py │ │ │ ├── choose_from_datasets_test.py │ │ │ ├── concatenate_test.py │ │ │ ├── counter_test.py │ │ │ ├── dataset_spec_test.py │ │ │ ├── dataset_test.py │ │ │ ├── enumerate_test.py │ │ │ ├── filter_test.py │ │ │ ├── fixed_length_record_dataset_test.py │ │ │ ├── flat_map_test.py │ │ │ ├── from_generator_test.py │ │ │ ├── from_sparse_tensor_slices_test.py │ │ │ ├── from_tensor_slices_test.py │ │ │ ├── from_tensors_test.py │ │ │ ├── get_single_element_test.py │ │ │ ├── group_by_window_test.py │ │ │ ├── ignore_errors_test.py │ │ │ ├── interleave_test.py │ │ │ ├── io_test.py │ │ │ ├── iterator_cluster_test.py │ │ │ ├── iterator_test.py │ │ │ ├── len_test.py │ │ │ ├── list_files_test.py │ │ │ ├── map_test.py │ │ │ ├── memory_cleanup_test.py │ │ │ ├── multi_device_iterator_test.py │ │ │ ├── optional_test.py │ │ │ ├── options_test.py │ │ │ ├── padded_batch_test.py │ │ │ ├── placement_test.py │ │ │ ├── prefetch_test.py │ │ │ ├── ragged_batch_test.py │ │ │ ├── random_test.py │ │ │ ├── range_test.py │ │ │ ├── rebatch_test.py │ │ │ ├── reduce_test.py │ │ │ ├── rejection_resample_test.py │ │ │ ├── repeat_test.py │ │ │ ├── sample_from_datasets_test.py │ │ │ ├── scan_test.py │ │ │ ├── shard_test.py │ │ │ ├── shuffle_test.py │ │ │ ├── skip_test.py │ │ │ ├── snapshot_test.py │ │ │ ├── sparse_batch_test.py │ │ │ ├── take_test.py │ │ │ ├── take_while_test.py │ │ │ ├── test_base.py │ │ │ ├── text_line_dataset_test.py │ │ │ ├── tf_record_dataset_test.py │ │ │ ├── tf_record_test_base.py │ │ │ ├── unbatch_test.py │ │ │ ├── unique_test.py │ │ │ ├── window_test.py │ │ │ └── zip_test.py │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── batch_op.py │ │ │ ├── cache_op.py │ │ │ ├── choose_from_datasets_op.py │ │ │ ├── concatenate_op.py │ │ │ ├── counter_op.py │ │ │ ├── dataset_autograph.py │ │ │ ├── dataset_ops.py │ │ │ ├── debug_mode.py │ │ │ ├── directed_interleave_op.py │ │ │ ├── filter_op.py │ │ │ ├── flat_map_op.py │ │ │ ├── from_generator_op.py │ │ │ ├── from_sparse_tensor_slices_op.py │ │ │ ├── from_tensor_slices_op.py │ │ │ ├── from_tensors_op.py │ │ │ ├── group_by_window_op.py │ │ │ ├── ignore_errors_op.py │ │ │ ├── interleave_op.py │ │ │ ├── iterator_autograph.py │ │ │ ├── iterator_ops.py │ │ │ ├── load_op.py │ │ │ ├── map_op.py │ │ │ ├── multi_device_iterator_ops.py │ │ │ ├── optional_ops.py │ │ │ ├── options.py │ │ │ ├── padded_batch_op.py │ │ │ ├── prefetch_op.py │ │ │ ├── ragged_batch_op.py │ │ │ ├── random_op.py │ │ │ ├── range_op.py │ │ │ ├── readers.py │ │ │ ├── rebatch_op.py │ │ │ ├── repeat_op.py │ │ │ ├── sample_from_datasets_op.py │ │ │ ├── save_op.py │ │ │ ├── scan_op.py │ │ │ ├── shard_op.py │ │ │ ├── shuffle_op.py │ │ │ ├── skip_op.py │ │ │ ├── snapshot_op.py │ │ │ ├── sparse_batch_op.py │ │ │ ├── structured_function.py │ │ │ ├── take_op.py │ │ │ ├── take_while_op.py │ │ │ ├── test_mode.py │ │ │ ├── unbatch_op.py │ │ │ ├── unique_op.py │ │ │ ├── window_op.py │ │ │ └── zip_op.py │ │ └── util │ │ │ ├── BUILD │ │ │ ├── convert.py │ │ │ ├── convert_test.py │ │ │ ├── nest.py │ │ │ ├── nest_test.py │ │ │ ├── options.py │ │ │ ├── options_test.py │ │ │ ├── random_seed.py │ │ │ ├── random_seed_test.py │ │ │ ├── sparse.py │ │ │ ├── sparse_test.py │ │ │ ├── structure.py │ │ │ ├── structure_test.py │ │ │ ├── traverse.py │ │ │ └── traverse_test.py │ ├── debug │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── cli │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── analyzer_cli.py │ │ │ ├── analyzer_cli_test.py │ │ │ ├── base_ui.py │ │ │ ├── cli_config.py │ │ │ ├── cli_config_test.py │ │ │ ├── cli_shared.py │ │ │ ├── cli_shared_test.py │ │ │ ├── cli_test_utils.py │ │ │ ├── command_parser.py │ │ │ ├── command_parser_test.py │ │ │ ├── debugger_cli_common.py │ │ │ ├── debugger_cli_common_test.py │ │ │ ├── evaluator.py │ │ │ ├── evaluator_test.py │ │ │ ├── offline_analyzer.py │ │ │ ├── profile_analyzer_cli.py │ │ │ ├── profile_analyzer_cli_test.py │ │ │ ├── readline_ui.py │ │ │ ├── readline_ui_test.py │ │ │ ├── tensor_format.py │ │ │ ├── tensor_format_test.py │ │ │ └── ui_factory.py │ │ ├── examples │ │ │ ├── README.md │ │ │ ├── v1 │ │ │ │ ├── BUILD │ │ │ │ ├── debug_errors.py │ │ │ │ ├── debug_fibonacci.py │ │ │ │ ├── debug_keras.py │ │ │ │ ├── debug_mnist_v1.py │ │ │ │ ├── debug_tflearn_iris.py │ │ │ │ ├── examples_v1_debug_errors_test.sh │ │ │ │ ├── examples_v1_debug_fibonacci_test.sh │ │ │ │ ├── examples_v1_debug_keras_test.sh │ │ │ │ ├── examples_v1_debug_mnist_test.sh │ │ │ │ ├── examples_v1_debug_tflearn_iris_test.sh │ │ │ │ ├── examples_v1_offline_analyzer_test.sh │ │ │ │ └── examples_v1_test.sh │ │ │ └── v2 │ │ │ │ ├── BUILD │ │ │ │ ├── debug_fibonacci_v2.py │ │ │ │ ├── debug_mnist_v2.py │ │ │ │ └── examples_v2_test.sh │ │ ├── lib │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── check_numerics_callback.py │ │ │ ├── check_numerics_callback_test.py │ │ │ ├── common.py │ │ │ ├── common_test.py │ │ │ ├── debug_data.py │ │ │ ├── debug_data_test.py │ │ │ ├── debug_events_monitors.py │ │ │ ├── debug_events_monitors_test.py │ │ │ ├── debug_events_reader.py │ │ │ ├── debug_events_writer.py │ │ │ ├── debug_events_writer_test.py │ │ │ ├── debug_gradients.py │ │ │ ├── debug_gradients_test.py │ │ │ ├── debug_graph_reconstruction_test.py │ │ │ ├── debug_graphs.py │ │ │ ├── debug_graphs_test.py │ │ │ ├── debug_grappler_test.py │ │ │ ├── debug_service_pb2_grpc.py │ │ │ ├── debug_utils.py │ │ │ ├── debug_utils_test.py │ │ │ ├── debug_v2_ops_test.py │ │ │ ├── dumping_callback.py │ │ │ ├── dumping_callback_test.py │ │ │ ├── dumping_callback_test_lib.py │ │ │ ├── grpc_debug_server.py │ │ │ ├── grpc_debug_test_server.py │ │ │ ├── grpc_tensorflow_server.py │ │ │ ├── op_callbacks_common.py │ │ │ ├── profiling.py │ │ │ ├── profiling_test.py │ │ │ ├── session_debug_file_test.py │ │ │ ├── session_debug_multi_gpu_test.py │ │ │ ├── session_debug_testlib.py │ │ │ ├── source_remote.py │ │ │ ├── source_remote_test.py │ │ │ ├── source_utils.py │ │ │ └── source_utils_test.py │ │ └── wrappers │ │ │ ├── BUILD │ │ │ ├── disk_usage_test.py │ │ │ ├── dumping_wrapper.py │ │ │ ├── dumping_wrapper_test.py │ │ │ ├── framework.py │ │ │ ├── framework_test.py │ │ │ ├── grpc_wrapper.py │ │ │ ├── hooks.py │ │ │ ├── local_cli_wrapper.py │ │ │ └── local_cli_wrapper_test.py │ ├── distribute │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── central_storage_strategy.py │ │ ├── checkpoint_utils_test.py │ │ ├── checkpointing_test.py │ │ ├── cluster_resolver │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── README_Slurm.md │ │ │ ├── __init__.py │ │ │ ├── cluster_resolver.py │ │ │ ├── cluster_resolver_test.py │ │ │ ├── gce_cluster_resolver.py │ │ │ ├── gce_cluster_resolver_test.py │ │ │ ├── kubernetes_cluster_resolver.py │ │ │ ├── kubernetes_cluster_resolver_test.py │ │ │ ├── sagemaker_cluster_resolver.py │ │ │ ├── sagemaker_cluster_resolver_test.py │ │ │ ├── slurm_cluster_resolver.py │ │ │ ├── slurm_cluster_resolver_test.py │ │ │ ├── tfconfig_cluster_resolver.py │ │ │ ├── tfconfig_cluster_resolver_test.py │ │ │ ├── tpu │ │ │ │ ├── BUILD │ │ │ │ ├── tpu_cluster_resolver.py │ │ │ │ └── tpu_cluster_resolver_test.py │ │ │ └── tpu_cluster_resolver.py │ │ ├── collective_all_reduce_strategy.py │ │ ├── collective_all_reduce_strategy_test.py │ │ ├── collective_util.py │ │ ├── collective_util_test.py │ │ ├── combinations.py │ │ ├── combinations_test.py │ │ ├── coordinator │ │ │ ├── BUILD │ │ │ ├── cluster_coordinator.py │ │ │ ├── cluster_coordinator_test.py │ │ │ ├── coordinator_context.py │ │ │ ├── fault_tolerance_coordination_service_test.py │ │ │ ├── fault_tolerance_test.py │ │ │ ├── fault_tolerance_test_base.py │ │ │ ├── get_task_states_test.py │ │ │ ├── metric_utils.py │ │ │ ├── metric_utils_test.py │ │ │ ├── remote_value.py │ │ │ ├── utils.py │ │ │ ├── values.py │ │ │ ├── watchdog.py │ │ │ └── watchdog_test.py │ │ ├── cross_device_ops.py │ │ ├── cross_device_ops_test.py │ │ ├── cross_device_utils.py │ │ ├── cross_device_utils_test.py │ │ ├── custom_training_loop_gradient_test.py │ │ ├── custom_training_loop_input_test.py │ │ ├── device_util.py │ │ ├── device_util_test.py │ │ ├── distribute_config.py │ │ ├── distribute_coordinator.py │ │ ├── distribute_coordinator_context.py │ │ ├── distribute_coordinator_test.py │ │ ├── distribute_lib.py │ │ ├── distribute_lib_test.py │ │ ├── distribute_utils.py │ │ ├── distribute_utils_test.py │ │ ├── distributed_table_test.py │ │ ├── distributed_variable_test.py │ │ ├── estimator_training.py │ │ ├── experimental │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── dtensor_strategy_extended.py │ │ │ ├── dtensor_util.py │ │ │ ├── dtensor_util_test.py │ │ │ ├── mirrored_strategy.py │ │ │ ├── mirrored_strategy_test.py │ │ │ ├── multi_worker_mirrored_strategy.py │ │ │ ├── multi_worker_mirrored_strategy_test.py │ │ │ └── rpc │ │ │ │ ├── BUILD │ │ │ │ ├── rpc_ops.py │ │ │ │ └── rpc_ops_test.py │ │ ├── failure_handling │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── failure_handler_test.py │ │ │ ├── failure_handling.py │ │ │ ├── failure_handling_util.py │ │ │ ├── gce_failure_handler_test.py │ │ │ └── preemption_watcher.py │ │ ├── input_lib.py │ │ ├── input_lib_test.py │ │ ├── input_lib_type_spec_test.py │ │ ├── input_ops.py │ │ ├── input_ops_test.py │ │ ├── input_util.py │ │ ├── integration_test │ │ │ ├── BUILD │ │ │ ├── mwms_peer_failure_test.py │ │ │ ├── saved_model_test.py │ │ │ └── tpu_memory_test.py │ │ ├── load_context.py │ │ ├── merge_call_interim.py │ │ ├── metrics_v1_test.py │ │ ├── mirrored_run.py │ │ ├── mirrored_strategy.py │ │ ├── mirrored_strategy_test.py │ │ ├── mirrored_values_test.py │ │ ├── mirrored_variable_test.py │ │ ├── moving_averages_test.py │ │ ├── multi_process_lib.py │ │ ├── multi_process_runner.py │ │ ├── multi_process_runner_no_init_test.py │ │ ├── multi_process_runner_test.py │ │ ├── multi_worker_continuous_run_test.py │ │ ├── multi_worker_test_base.py │ │ ├── multi_worker_test_base_test.py │ │ ├── multi_worker_util.py │ │ ├── multi_worker_util_test.py │ │ ├── numpy_dataset.py │ │ ├── numpy_dataset_test.py │ │ ├── one_device_strategy.py │ │ ├── one_device_strategy_test.py │ │ ├── packed_distributed_variable.py │ │ ├── packed_distributed_variable_test.py │ │ ├── parallel_device │ │ │ ├── BUILD │ │ │ ├── parallel_device.py │ │ │ ├── parallel_device_test.py │ │ │ └── pywrap_parallel_device.cc │ │ ├── parameter_server_strategy.py │ │ ├── parameter_server_strategy_test.py │ │ ├── parameter_server_strategy_v2.py │ │ ├── parameter_server_strategy_v2_test.py │ │ ├── per_replica_test.py │ │ ├── ps_values.py │ │ ├── ps_values_test.py │ │ ├── random_generator_test.py │ │ ├── reduce_util.py │ │ ├── remote_mirrored_strategy_eager_test.py │ │ ├── sharded_variable.py │ │ ├── sharded_variable_test.py │ │ ├── shared_variable_creator.py │ │ ├── shared_variable_creator_test.py │ │ ├── single_loss_example.py │ │ ├── step_fn.py │ │ ├── strategy_combinations.py │ │ ├── strategy_combinations_test.py │ │ ├── strategy_common_test.py │ │ ├── strategy_gather_test.py │ │ ├── strategy_test_lib.py │ │ ├── summary_op_util.py │ │ ├── template_mirrored_strategy_test.py │ │ ├── test_util.py │ │ ├── test_util_test.py │ │ ├── tf_function_test.py │ │ ├── tpu_replicated_variable.py │ │ ├── tpu_replicated_variable_test.py │ │ ├── tpu_strategy.py │ │ ├── tpu_strategy_compilation_test.py │ │ ├── tpu_strategy_model_parallelism_test.py │ │ ├── tpu_strategy_test.py │ │ ├── tpu_util.py │ │ ├── tpu_values.py │ │ ├── v1 │ │ │ ├── BUILD │ │ │ ├── all_reduce.py │ │ │ ├── all_reduce_test.py │ │ │ ├── cross_device_ops_test.py │ │ │ └── input_lib.py │ │ ├── values.py │ │ ├── values_test.py │ │ ├── values_util.py │ │ ├── values_v2.py │ │ ├── values_v2_test.py │ │ ├── vars_test.py │ │ ├── warm_starting_util_test.py │ │ └── zero_batch_test.py │ ├── dlpack │ │ ├── BUILD │ │ ├── dlpack.py │ │ └── dlpack_test.py │ ├── eager │ │ ├── BUILD │ │ ├── backprop.py │ │ ├── backprop_test.py │ │ ├── backprop_util.py │ │ ├── benchmarks │ │ │ ├── BUILD │ │ │ ├── kpi_benchmark_test.py │ │ │ └── resnet50 │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── hvp_test.py │ │ │ │ ├── resnet50.py │ │ │ │ ├── resnet50_graph_test.py │ │ │ │ ├── resnet50_test.py │ │ │ │ └── resnet50_test_util.py │ │ ├── benchmarks_test.py │ │ ├── benchmarks_test_base.py │ │ ├── cancellation.py │ │ ├── cancellation_test.py │ │ ├── context.py │ │ ├── context_cross_platform_test.py │ │ ├── context_test.py │ │ ├── core.py │ │ ├── core_test.py │ │ ├── custom_device_test.py │ │ ├── custom_device_testutil.cc │ │ ├── custom_device_testutil.pyi │ │ ├── def_function.py │ │ ├── device_placement_test.py │ │ ├── execute.py │ │ ├── executor.py │ │ ├── forwardprop.py │ │ ├── forwardprop_test.py │ │ ├── forwardprop_util.py │ │ ├── function.py │ │ ├── gen_gradient_input_output_exclusions.py │ │ ├── gradient_input_output_exclusions.py │ │ ├── gradient_input_output_exclusions_test.py │ │ ├── graph_only_ops.py │ │ ├── graph_only_ops_test.py │ │ ├── imperative_grad.py │ │ ├── lift_to_graph.py │ │ ├── lift_to_graph_test.py │ │ ├── memory_tests │ │ │ ├── BUILD │ │ │ ├── memory_test.py │ │ │ ├── memory_test_util.py │ │ │ └── remote_memory_test.py │ │ ├── monitoring.py │ │ ├── monitoring_test.py │ │ ├── ops_test.py │ │ ├── polymorphic_function │ │ │ ├── BUILD │ │ │ ├── argument_naming_test.py │ │ │ ├── atomic_function.py │ │ │ ├── atomic_function_test.py │ │ │ ├── attributes.py │ │ │ ├── autograph_util.py │ │ │ ├── collection_test.py │ │ │ ├── compiler_ir.py │ │ │ ├── compiler_ir_test.py │ │ │ ├── composite_tensor_utils.py │ │ │ ├── concrete_function.py │ │ │ ├── concrete_function_test.py │ │ │ ├── eager_function_run.py │ │ │ ├── function_context.py │ │ │ ├── function_spec_test.py │ │ │ ├── function_type_utils.py │ │ │ ├── gradients_test.py │ │ │ ├── polymorphic_function.py │ │ │ ├── polymorphic_function_test.py │ │ │ ├── polymorphic_function_test_cpu_only.py │ │ │ ├── polymorphic_function_xla_jit_test.py │ │ │ ├── polymorphic_function_xla_test.py │ │ │ ├── saved_model_exported_concrete.py │ │ │ ├── saved_model_utils.py │ │ │ ├── tf_method_target.py │ │ │ ├── tracing_compilation.py │ │ │ ├── tracing_compilation_test.py │ │ │ └── transform.py │ │ ├── profiler.py │ │ ├── profiler_client.py │ │ ├── profiler_client_test.py │ │ ├── profiler_test.py │ │ ├── pywrap_gradient_exclusions.cc │ │ ├── pywrap_gradient_exclusions.h │ │ ├── pywrap_tensor.cc │ │ ├── pywrap_tensor.h │ │ ├── pywrap_tensor_conversion.cc │ │ ├── pywrap_tensor_conversion.h │ │ ├── pywrap_tensor_test.py │ │ ├── pywrap_tensor_test_util.cc │ │ ├── pywrap_tensor_test_util.pyi │ │ ├── pywrap_tfe.h │ │ ├── pywrap_tfe_src.cc │ │ ├── pywrap_tfe_test.py │ │ ├── record.py │ │ ├── record_test.py │ │ ├── remote.py │ │ ├── remote_benchmarks_test.py │ │ ├── remote_cloud_tpu_test.py │ │ ├── remote_cluster_test.py │ │ ├── remote_execution_test.py │ │ ├── remote_test.py │ │ ├── run_eager_op_as_function_test.py │ │ ├── run_eager_op_as_function_xla_test.py │ │ ├── small_constants_optimizer_test.py │ │ ├── summary_optimizer_test.py │ │ ├── tape.py │ │ ├── tensor_test.py │ │ ├── test.py │ │ ├── wrap_function.py │ │ ├── wrap_function_device_test.py │ │ └── wrap_function_test.py │ ├── estimator │ │ ├── BUILD │ │ ├── canned │ │ │ ├── baseline.py │ │ │ ├── dnn.py │ │ │ ├── dnn_linear_combined.py │ │ │ ├── dnn_testing_utils.py │ │ │ ├── head.py │ │ │ ├── linear.py │ │ │ ├── linear_testing_utils.py │ │ │ ├── metric_keys.py │ │ │ ├── optimizers.py │ │ │ ├── parsing_utils.py │ │ │ └── prediction_keys.py │ │ ├── estimator.py │ │ ├── estimator_lib.py │ │ ├── export │ │ │ ├── export.py │ │ │ ├── export_lib.py │ │ │ └── export_output.py │ │ ├── exporter.py │ │ ├── gc.py │ │ ├── inputs │ │ │ ├── inputs.py │ │ │ ├── numpy_io.py │ │ │ ├── pandas_io.py │ │ │ └── queues │ │ │ │ ├── __init__.py │ │ │ │ ├── feeding_functions.py │ │ │ │ └── feeding_queue_runner.py │ │ ├── keras.py │ │ ├── model_fn.py │ │ ├── run_config.py │ │ ├── training.py │ │ └── util.py │ ├── feature_column │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── feature_column.py │ │ ├── feature_column_lib.py │ │ ├── feature_column_test.py │ │ ├── feature_column_v2.py │ │ ├── feature_column_v2_test.py │ │ ├── feature_column_v2_types.py │ │ ├── sequence_feature_column.py │ │ ├── sequence_feature_column_integration_test.py │ │ ├── sequence_feature_column_test.py │ │ ├── serialization.py │ │ ├── serialization_test.py │ │ ├── testdata │ │ │ ├── embedding.ckpt.data-00000-of-00001 │ │ │ ├── embedding.ckpt.index │ │ │ ├── embedding.ckpt.meta │ │ │ ├── unicode_vocabulary │ │ │ ├── unicode_vocabulary.tfrecord.gz │ │ │ ├── warriors_vocabulary.tfrecord.gz │ │ │ ├── warriors_vocabulary.txt │ │ │ ├── wire_vocabulary.tfrecord.gz │ │ │ └── wire_vocabulary.txt │ │ └── utils.py │ ├── flags_pybind.pyi │ ├── framework │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── _dtypes.pyi │ │ ├── _errors_test_helper.pyi │ │ ├── _memory_checker_test_helper.pyi │ │ ├── _op_def_library_pybind.pyi │ │ ├── _op_def_registry.pyi │ │ ├── _op_def_util.pyi │ │ ├── _proto_comparators.pyi │ │ ├── _py_context_manager.pyi │ │ ├── _python_memory_checker_helper.pyi │ │ ├── _pywrap_python_api_dispatcher.pyi │ │ ├── _pywrap_python_api_info.pyi │ │ ├── _pywrap_python_api_parameter_converter.pyi │ │ ├── _pywrap_python_op_gen.pyi │ │ ├── _pywrap_python_tensor_converter.pyi │ │ ├── _test_metrics_util.pyi │ │ ├── auto_control_deps.py │ │ ├── auto_control_deps_test.py │ │ ├── auto_control_deps_utils.py │ │ ├── byte_swap_tensor.py │ │ ├── c_api_util.py │ │ ├── c_api_util_test.py │ │ ├── combinations.py │ │ ├── common_shapes.py │ │ ├── common_shapes_test.py │ │ ├── composite_tensor.py │ │ ├── composite_tensor_gradient.py │ │ ├── composite_tensor_test.py │ │ ├── config.py │ │ ├── config_test.py │ │ ├── config_vgpu_test.py │ │ ├── constant_op.py │ │ ├── constant_op_test.py │ │ ├── convert_to_constants.py │ │ ├── convert_to_constants_test.py │ │ ├── cpp_shape_inference.proto │ │ ├── device.py │ │ ├── device_spec.py │ │ ├── device_spec_test.py │ │ ├── device_test.py │ │ ├── dtypes.cc │ │ ├── dtypes.py │ │ ├── dtypes_test.py │ │ ├── error_interpolation.py │ │ ├── error_interpolation_test.py │ │ ├── errors.py │ │ ├── errors_impl.py │ │ ├── errors_test.py │ │ ├── errors_test_helper.cc │ │ ├── experimental │ │ │ ├── BUILD │ │ │ ├── _math_ops.pyi │ │ │ ├── _nn_ops.pyi │ │ │ ├── _unified_api.pyi │ │ │ ├── context_stack.py │ │ │ ├── def_function.py │ │ │ ├── gradient_registry.py │ │ │ ├── graph_building_test.py │ │ │ ├── math_ops.cc │ │ │ ├── math_ops.py │ │ │ ├── nn_ops.cc │ │ │ ├── nn_ops.py │ │ │ ├── tape.cc │ │ │ ├── tape.py │ │ │ ├── thread_local_stack.py │ │ │ ├── unified_api.cc │ │ │ └── unified_api_test.py │ │ ├── extension_type.py │ │ ├── extension_type_field.py │ │ ├── extension_type_field_test.py │ │ ├── extension_type_test.py │ │ ├── fast_tensor_util.pyx │ │ ├── file_system_test.py │ │ ├── flexible_dtypes.py │ │ ├── flexible_dtypes_test.py │ │ ├── framework_lib.py │ │ ├── func_graph.py │ │ ├── function.py │ │ ├── function_def_to_graph.py │ │ ├── function_def_to_graph_test.py │ │ ├── function_test.py │ │ ├── gpu_util.py │ │ ├── graph_building_benchmark.py │ │ ├── graph_io.py │ │ ├── graph_to_function_def.py │ │ ├── graph_util.py │ │ ├── graph_util_impl.py │ │ ├── graph_util_test.py │ │ ├── immutable_dict.py │ │ ├── immutable_dict_test.py │ │ ├── importer.py │ │ ├── importer_test.py │ │ ├── indexed_slices.py │ │ ├── indexed_slices_test.py │ │ ├── is_mlir_bridge_test_false.py │ │ ├── is_mlir_bridge_test_true.py │ │ ├── is_xla_test_true.py │ │ ├── kernels.py │ │ ├── kernels_test.py │ │ ├── kythe_metadata.proto │ │ ├── load_library.py │ │ ├── memory_checker.py │ │ ├── memory_checker_test.py │ │ ├── memory_checker_test_helper.cc │ │ ├── meta_graph.py │ │ ├── meta_graph_test.py │ │ ├── node_file_writer_test.py │ │ ├── none_tensor.py │ │ ├── offset_counter.cc │ │ ├── offset_counter_helper.cc │ │ ├── offset_counter_helper.h │ │ ├── offset_counter_helper_test.cc │ │ ├── op_allowlist_namespace_test.py │ │ ├── op_callbacks.py │ │ ├── op_callbacks_test.py │ │ ├── op_def_library.py │ │ ├── op_def_library_pybind.cc │ │ ├── op_def_library_pybind.py │ │ ├── op_def_library_test.py │ │ ├── op_def_registry.cc │ │ ├── op_def_registry.py │ │ ├── op_def_util.cc │ │ ├── op_def_util.h │ │ ├── op_def_util_pybind.cc │ │ ├── op_def_util_test.py │ │ ├── op_reg_offset.proto │ │ ├── ops.py │ │ ├── ops_enable_eager_test.py │ │ ├── ops_test.py │ │ ├── proto_comparators.cc │ │ ├── proto_test.py │ │ ├── py_context_manager.cc │ │ ├── py_context_manager.h │ │ ├── py_context_manager_pybind.cc │ │ ├── py_context_manager_test.py │ │ ├── python_api_dispatcher.cc │ │ ├── python_api_dispatcher.h │ │ ├── python_api_dispatcher_test.py │ │ ├── python_api_dispatcher_wrapper.cc │ │ ├── python_api_info.cc │ │ ├── python_api_info.h │ │ ├── python_api_info_test.py │ │ ├── python_api_info_wrapper.cc │ │ ├── python_api_parameter_converter.cc │ │ ├── python_api_parameter_converter.h │ │ ├── python_api_parameter_converter_test.py │ │ ├── python_api_parameter_converter_wrapper.cc │ │ ├── python_memory_checker.py │ │ ├── python_memory_checker_helper.cc │ │ ├── python_op_gen.cc │ │ ├── python_op_gen.h │ │ ├── python_op_gen_annotation_test.py │ │ ├── python_op_gen_annotator.cc │ │ ├── python_op_gen_annotator.h │ │ ├── python_op_gen_annotator_test.cc │ │ ├── python_op_gen_main.cc │ │ ├── python_op_gen_test.cc │ │ ├── python_op_gen_wrapper.cc │ │ ├── python_tensor_converter.cc │ │ ├── python_tensor_converter.h │ │ ├── python_tensor_converter_test.py │ │ ├── python_tensor_converter_wrapper.cc │ │ ├── random_seed.py │ │ ├── random_seed_test.py │ │ ├── registry.py │ │ ├── registry_test.py │ │ ├── smart_cond.py │ │ ├── smart_cond_test.py │ │ ├── sparse_tensor.py │ │ ├── sparse_tensor_test.py │ │ ├── stack.py │ │ ├── strict_mode.py │ │ ├── subscribe.py │ │ ├── subscribe_test.py │ │ ├── summary_test_util.py │ │ ├── tensor.py │ │ ├── tensor_conversion.py │ │ ├── tensor_conversion_registry.py │ │ ├── tensor_shape.py │ │ ├── tensor_shape_casters.h │ │ ├── tensor_shape_test.py │ │ ├── tensor_spec.py │ │ ├── tensor_test.py │ │ ├── tensor_util.py │ │ ├── tensor_util_test.py │ │ ├── test_combinations.py │ │ ├── test_combinations_test.py │ │ ├── test_file_system.cc │ │ ├── test_metrics_util.cc │ │ ├── test_ops.cc │ │ ├── test_ops.cu.cc │ │ ├── test_ops.h │ │ ├── test_util.py │ │ ├── test_util_test.py │ │ ├── testdata │ │ │ └── metrics_export_meta_graph.pb │ │ ├── tf2_test.py │ │ ├── tfrt_utils.py │ │ ├── traceable_stack.py │ │ ├── traceable_stack_test.py │ │ ├── type_spec.py │ │ ├── type_spec_registry.py │ │ ├── type_spec_test.py │ │ ├── type_utils.py │ │ ├── versions.py │ │ ├── versions_test.py │ │ ├── weak_tensor.py │ │ └── weak_tensor_test.py │ ├── grappler │ │ ├── BUILD │ │ ├── _pywrap_cost_analyzer.pyi │ │ ├── _pywrap_graph_analyzer.pyi │ │ ├── _pywrap_model_analyzer.pyi │ │ ├── _pywrap_tf_cluster.pyi │ │ ├── _pywrap_tf_item.pyi │ │ ├── _pywrap_tf_optimizer.pyi │ │ ├── arithmetic_optimizer_test.py │ │ ├── auto_mixed_precision_test.py │ │ ├── cluster.py │ │ ├── cluster_test.py │ │ ├── cluster_wrapper.cc │ │ ├── constant_folding_test.py │ │ ├── cost_analyzer.cc │ │ ├── cost_analyzer.h │ │ ├── cost_analyzer.py │ │ ├── cost_analyzer_test.py │ │ ├── cost_analyzer_tool.py │ │ ├── cost_analyzer_wrapper.cc │ │ ├── datasets_test.py │ │ ├── graph_analyzer.py │ │ ├── graph_analyzer_tool_wrapper.cc │ │ ├── item.py │ │ ├── item_test.py │ │ ├── item_wrapper.cc │ │ ├── layout_optimizer_test.py │ │ ├── memory_optimizer_test.py │ │ ├── model_analyzer.cc │ │ ├── model_analyzer.h │ │ ├── model_analyzer.py │ │ ├── model_analyzer_test.py │ │ ├── model_analyzer_wrapper.cc │ │ ├── remapper_test.py │ │ ├── tf_optimizer.py │ │ ├── tf_optimizer_test.py │ │ └── tf_optimizer_wrapper.cc │ ├── integration_testing │ │ └── BUILD │ ├── keras │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── activations.py │ │ ├── backend.py │ │ ├── backend_config.py │ │ ├── callbacks.py │ │ ├── callbacks_v1.py │ │ ├── combinations.py │ │ ├── constraints.py │ │ ├── distribute │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── distribute_coordinator_utils.py │ │ │ ├── distributed_file_utils.py │ │ │ ├── distributed_training_utils.py │ │ │ ├── distributed_training_utils_v1.py │ │ │ ├── optimizer_combinations.py │ │ │ └── worker_training_state.py │ │ ├── engine │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── base_layer.py │ │ │ ├── base_layer_utils.py │ │ │ ├── base_layer_v1.py │ │ │ ├── base_preprocessing_layer.py │ │ │ ├── compile_utils.py │ │ │ ├── data_adapter.py │ │ │ ├── functional.py │ │ │ ├── input_layer.py │ │ │ ├── input_spec.py │ │ │ ├── keras_tensor.py │ │ │ ├── node.py │ │ │ ├── partial_batch_padding_handler.py │ │ │ ├── saving.py │ │ │ ├── sequential.py │ │ │ ├── training.py │ │ │ ├── training_arrays_v1.py │ │ │ ├── training_distributed_v1.py │ │ │ ├── training_eager_v1.py │ │ │ ├── training_generator_v1.py │ │ │ ├── training_utils.py │ │ │ ├── training_utils_v1.py │ │ │ └── training_v1.py │ │ ├── initializers │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── initializers_v1.py │ │ │ └── initializers_v2.py │ │ ├── keras_parameterized.py │ │ ├── layers │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── advanced_activations.py │ │ │ ├── convolutional.py │ │ │ ├── convolutional_recurrent.py │ │ │ ├── core.py │ │ │ ├── dense_attention.py │ │ │ ├── embeddings.py │ │ │ ├── legacy_rnn │ │ │ │ ├── BUILD │ │ │ │ ├── rnn_cell_impl.py │ │ │ │ └── rnn_cell_wrapper_impl.py │ │ │ ├── merge.py │ │ │ ├── pooling.py │ │ │ ├── recurrent.py │ │ │ ├── rnn_cell_wrapper_v2.py │ │ │ └── serialization.py │ │ ├── legacy_tf_layers │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── convolutional.py │ │ │ ├── core.py │ │ │ ├── pooling.py │ │ │ └── variable_scope_shim.py │ │ ├── losses.py │ │ ├── metrics.py │ │ ├── mixed_precision │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── autocast_variable.py │ │ │ ├── device_compatibility_check.py │ │ │ ├── get_layer_policy.py │ │ │ ├── loss_scale.py │ │ │ ├── loss_scale_optimizer.py │ │ │ ├── policy.py │ │ │ └── test_util.py │ │ ├── models.py │ │ ├── optimizer_v1.py │ │ ├── optimizer_v2 │ │ │ ├── BUILD │ │ │ ├── adadelta.py │ │ │ ├── adagrad.py │ │ │ ├── adam.py │ │ │ ├── adamax.py │ │ │ ├── ftrl.py │ │ │ ├── gradient_descent.py │ │ │ ├── learning_rate_schedule.py │ │ │ ├── legacy_learning_rate_decay.py │ │ │ ├── nadam.py │ │ │ ├── optimizer_v2.py │ │ │ ├── rmsprop.py │ │ │ └── utils.py │ │ ├── optimizers.py │ │ ├── protobuf │ │ │ ├── BUILD │ │ │ ├── projector_config.proto │ │ │ ├── saved_metadata.proto │ │ │ └── versions.proto │ │ ├── regularizers.py │ │ ├── saving │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── hdf5_format.py │ │ │ ├── model_config.py │ │ │ ├── save.py │ │ │ ├── saved_model │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── base_serialization.py │ │ │ │ ├── constants.py │ │ │ │ ├── json_utils.py │ │ │ │ ├── layer_serialization.py │ │ │ │ ├── load.py │ │ │ │ ├── load_context.py │ │ │ │ ├── metric_serialization.py │ │ │ │ ├── model_serialization.py │ │ │ │ ├── network_serialization.py │ │ │ │ ├── save.py │ │ │ │ ├── save_impl.py │ │ │ │ ├── serialized_attributes.py │ │ │ │ └── utils.py │ │ │ ├── saved_model_experimental.py │ │ │ ├── saving_utils.py │ │ │ └── utils_v1 │ │ │ │ ├── BUILD │ │ │ │ ├── __init__.py │ │ │ │ ├── export_output.py │ │ │ │ ├── export_utils.py │ │ │ │ ├── mode_keys.py │ │ │ │ ├── signature_def_utils.py │ │ │ │ └── unexported_constants.py │ │ ├── testing_utils.py │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── all_utils.py │ │ │ ├── control_flow_util.py │ │ │ ├── conv_utils.py │ │ │ ├── data_utils.py │ │ │ ├── dataset_creator.py │ │ │ ├── generic_utils.py │ │ │ ├── io_utils.py │ │ │ ├── kernelized_utils.py │ │ │ ├── layer_utils.py │ │ │ ├── losses_utils.py │ │ │ ├── metrics_utils.py │ │ │ ├── mode_keys.py │ │ │ ├── np_utils.py │ │ │ ├── object_identity.py │ │ │ ├── tf_contextlib.py │ │ │ ├── tf_inspect.py │ │ │ ├── tf_utils.py │ │ │ ├── version_utils.py │ │ │ └── vis_utils.py │ ├── kernel_tests │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── array_ops │ │ │ ├── BUILD │ │ │ ├── array_ops_test.py │ │ │ ├── batch_gather_op_test.py │ │ │ ├── batch_scatter_ops_test.py │ │ │ ├── batchtospace_op_test.py │ │ │ ├── bcast_ops_test.py │ │ │ ├── bitcast_op_test.py │ │ │ ├── broadcast_to_ops_test.py │ │ │ ├── cast_op_test.py │ │ │ ├── concat_op_test.py │ │ │ ├── constant_op_eager_test.py │ │ │ ├── constant_op_test.py │ │ │ ├── denormal_test.py │ │ │ ├── depthtospace_op_test.py │ │ │ ├── diag_op_test.py │ │ │ ├── edit_distance_op_test.py │ │ │ ├── fingerprint_op_test.py │ │ │ ├── gather_nd_op_test.py │ │ │ ├── gather_op_test.py │ │ │ ├── huge_slice_op_test.py │ │ │ ├── identity_n_op_py_test.py │ │ │ ├── identity_op_py_test.py │ │ │ ├── init_ops_test.py │ │ │ ├── inplace_ops_test.py │ │ │ ├── large_concat_op_test.py │ │ │ ├── manip_ops_test.py │ │ │ ├── matrix_band_part_op_test.py │ │ │ ├── one_hot_op_test.py │ │ │ ├── pad_op_test.py │ │ │ ├── reshape_op_test.py │ │ │ ├── reverse_sequence_op_test.py │ │ │ ├── scalar_test.py │ │ │ ├── scatter_nd_ops_test.py │ │ │ ├── scatter_ops_test.py │ │ │ ├── shape_ops_test.py │ │ │ ├── slice_op_test.py │ │ │ ├── spacetobatch_op_test.py │ │ │ ├── spacetodepth_op_test.py │ │ │ ├── split_op_test.py │ │ │ ├── stack_op_test.py │ │ │ ├── unique_op_test.py │ │ │ ├── unstack_op_test.py │ │ │ ├── weights_broadcast_test.py │ │ │ └── where_op_test.py │ │ ├── benchmark_test.py │ │ ├── check_ops_test.py │ │ ├── collective_ops_multi_worker_test.py │ │ ├── collective_ops_test.py │ │ ├── composite_tensor_ops_test.py │ │ ├── control_flow │ │ │ ├── BUILD │ │ │ ├── cond_v2_test.py │ │ │ ├── control_flow_ops_py_test.py │ │ │ ├── control_flow_util_test.py │ │ │ ├── control_flow_util_v2_test.py │ │ │ ├── functional_ops_test.py │ │ │ ├── map_fn_test.py │ │ │ ├── py_func_test.py │ │ │ ├── scan_ops_test.py │ │ │ └── while_v2_test.py │ │ ├── critical_section_test.py │ │ ├── custom_ops │ │ │ ├── BUILD │ │ │ ├── ackermann_op.cc │ │ │ ├── ackermann_test.py │ │ │ ├── duplicate_op.cc │ │ │ ├── duplicate_op_test.py │ │ │ ├── invalid_op.cc │ │ │ └── invalid_op_test.py │ │ ├── data_structures │ │ │ ├── BUILD │ │ │ ├── barrier_ops_test.py │ │ │ ├── conditional_accumulator_test.py │ │ │ ├── dynamic_partition_op_test.py │ │ │ ├── dynamic_stitch_op_test.py │ │ │ ├── fifo_queue_test.py │ │ │ ├── list_ops_test.py │ │ │ ├── listdiff_op_test.py │ │ │ ├── lookup_ops_test.py │ │ │ ├── map_ops_test.py │ │ │ ├── map_stage_op_test.py │ │ │ ├── padding_fifo_queue_test.py │ │ │ ├── priority_queue_test.py │ │ │ ├── stack_ops_test.py │ │ │ ├── stage_op_test.py │ │ │ └── tensor_array_ops_test.py │ │ ├── distributions │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── bernoulli_test.py │ │ │ ├── beta_test.py │ │ │ ├── bijector_test.py │ │ │ ├── categorical_test.py │ │ │ ├── dirichlet_multinomial_test.py │ │ │ ├── dirichlet_test.py │ │ │ ├── exponential_test.py │ │ │ ├── gamma_test.py │ │ │ ├── identity_bijector_test.py │ │ │ ├── kullback_leibler_test.py │ │ │ ├── laplace_test.py │ │ │ ├── multinomial_test.py │ │ │ ├── normal_test.py │ │ │ ├── special_math_test.py │ │ │ ├── student_t_test.py │ │ │ ├── uniform_test.py │ │ │ └── util_test.py │ │ ├── garbage_collection_test.py │ │ ├── gradient_correctness_test.py │ │ ├── histogram_ops_test.py │ │ ├── image_ops │ │ │ ├── BUILD │ │ │ ├── attention_ops_test.py │ │ │ ├── decode_bmp_op_test.py │ │ │ ├── decode_compressed_op_test.py │ │ │ ├── decode_image_op_test.py │ │ │ ├── decode_jpeg_op_test.py │ │ │ ├── decode_png_op_test.py │ │ │ ├── decode_raw_op_test.py │ │ │ ├── draw_bounding_box_op_test.py │ │ │ ├── extract_image_patches_grad_test.py │ │ │ ├── extract_image_patches_op_test.py │ │ │ ├── extract_volume_patches_grad_test.py │ │ │ └── extract_volume_patches_op_test.py │ │ ├── io_ops │ │ │ ├── BUILD │ │ │ ├── checkpoint_ops_test.py │ │ │ ├── decode_csv_op_test.py │ │ │ ├── io_ops_test.py │ │ │ ├── parse_single_example_op_test.py │ │ │ ├── parsing_ops_test.py │ │ │ ├── reader_ops_test.py │ │ │ ├── record_input_test.py │ │ │ └── save_restore_ops_test.py │ │ ├── linalg │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── cholesky_op_test.py │ │ │ ├── determinant_op_test.py │ │ │ ├── eig_op_test.py │ │ │ ├── einsum_op_test.py │ │ │ ├── linalg_grad_test.py │ │ │ ├── linalg_ops_test.py │ │ │ ├── linear_operator_addition_test.py │ │ │ ├── linear_operator_adjoint_test.py │ │ │ ├── linear_operator_block_diag_test.py │ │ │ ├── linear_operator_block_lower_triangular_test.py │ │ │ ├── linear_operator_circulant_test.py │ │ │ ├── linear_operator_composition_test.py │ │ │ ├── linear_operator_diag_test.py │ │ │ ├── linear_operator_full_matrix_test.py │ │ │ ├── linear_operator_householder_test.py │ │ │ ├── linear_operator_identity_test.py │ │ │ ├── linear_operator_inversion_test.py │ │ │ ├── linear_operator_kronecker_test.py │ │ │ ├── linear_operator_low_rank_update_test.py │ │ │ ├── linear_operator_lower_triangular_test.py │ │ │ ├── linear_operator_permutation_test.py │ │ │ ├── linear_operator_test.py │ │ │ ├── linear_operator_toeplitz_test.py │ │ │ ├── linear_operator_tridiag_test.py │ │ │ ├── linear_operator_util_test.py │ │ │ ├── linear_operator_zeros_test.py │ │ │ ├── lu_op_test.py │ │ │ ├── matrix_exponential_op_test.py │ │ │ ├── matrix_inverse_op_test.py │ │ │ ├── matrix_logarithm_op_test.py │ │ │ ├── matrix_solve_ls_op_test.py │ │ │ ├── matrix_solve_op_test.py │ │ │ ├── matrix_square_root_op_test.py │ │ │ ├── matrix_triangular_solve_op_test.py │ │ │ ├── norm_op_test.py │ │ │ ├── normalize_op_test.py │ │ │ ├── qr_op_test.py │ │ │ ├── self_adjoint_eig_op_test.py │ │ │ ├── slicing_test.py │ │ │ ├── sparse │ │ │ │ ├── BUILD │ │ │ │ ├── conjugate_gradient_test.py │ │ │ │ ├── csr_sparse_matrix_dense_mat_mul_grad_test.py │ │ │ │ ├── csr_sparse_matrix_grad_test.py │ │ │ │ ├── csr_sparse_matrix_ops_test.py │ │ │ │ ├── csr_sparse_matrix_sparse_mat_mul_grad_test.py │ │ │ │ └── csr_sparse_matrix_test.py │ │ │ ├── svd_op_test.py │ │ │ ├── testdata │ │ │ │ ├── BUILD │ │ │ │ └── self_adjoint_eig_fail_if_denorms_flushed.txt │ │ │ ├── tridiagonal_matmul_op_test.py │ │ │ └── tridiagonal_solve_op_test.py │ │ ├── logging_ops_logging_level_test.py │ │ ├── logging_ops_test.py │ │ ├── math_ops │ │ │ ├── BUILD │ │ │ ├── aggregate_ops_test.py │ │ │ ├── approx_topk_test.py │ │ │ ├── argmax_op_test.py │ │ │ ├── banded_triangular_solve_op_test.py │ │ │ ├── basic_gpu_test.py │ │ │ ├── batch_matmul_op_test.py │ │ │ ├── bincount_op_test.py │ │ │ ├── bucketize_op_test.py │ │ │ ├── clip_ops_test.py │ │ │ ├── confusion_matrix_test.py │ │ │ ├── cross_grad_test.py │ │ │ ├── cumulative_logsumexp_test.py │ │ │ ├── cwise_ops_binary_test.py │ │ │ ├── cwise_ops_test.py │ │ │ ├── cwise_ops_unary_test.py │ │ │ ├── division_future_test.py │ │ │ ├── division_past_test.py │ │ │ ├── in_topk_op_test.py │ │ │ ├── matmul_op_test.py │ │ │ ├── reduce_benchmark_test.py │ │ │ ├── reduction_ops_test.py │ │ │ ├── reduction_ops_test_big.py │ │ │ ├── segment_reduction_ops_d9m_test.py │ │ │ ├── segment_reduction_ops_test.py │ │ │ ├── sets_test.py │ │ │ ├── tensordot_op_test.py │ │ │ ├── topk_op_test.py │ │ │ ├── transpose_op_test.py │ │ │ └── zero_division_test.py │ │ ├── metrics_test.py │ │ ├── nn_ops │ │ │ ├── BUILD │ │ │ ├── atrous_conv2d_test.py │ │ │ ├── atrous_convolution_test.py │ │ │ ├── betainc_op_test.py │ │ │ ├── bias_op_base.py │ │ │ ├── bias_op_d9m_test.py │ │ │ ├── bias_op_test.py │ │ │ ├── conv1d_test.py │ │ │ ├── conv1d_transpose_test.py │ │ │ ├── conv2d_backprop_filter_grad_test.py │ │ │ ├── conv2d_transpose_test.py │ │ │ ├── conv3d_backprop_filter_v2_grad_test.py │ │ │ ├── conv3d_transpose_test.py │ │ │ ├── conv_ops_3d_test.py │ │ │ ├── conv_ops_test.py │ │ │ ├── ctc_decoder_ops_test.py │ │ │ ├── ctc_loss_op_test.py │ │ │ ├── cudnn_d9m_test.py │ │ │ ├── cudnn_deterministic_base.py │ │ │ ├── cudnn_deterministic_ops_test.py │ │ │ ├── depthwise_conv_op_base.py │ │ │ ├── depthwise_conv_op_d9m_test.py │ │ │ ├── depthwise_conv_op_test.py │ │ │ ├── embedding_ops_test.py │ │ │ ├── fractional_avg_pool_op_test.py │ │ │ ├── fractional_max_pool_op_test.py │ │ │ ├── losses_test.py │ │ │ ├── lrn_op_test.py │ │ │ ├── morphological_ops_test.py │ │ │ ├── nth_element_op_test.py │ │ │ ├── pool_test.py │ │ │ ├── pooling_ops_3d_test.py │ │ │ ├── pooling_ops_test.py │ │ │ ├── relu_op_test.py │ │ │ ├── rnn_cell_test.py │ │ │ ├── rnn_test.py │ │ │ ├── softmax_op_test.py │ │ │ ├── softplus_op_test.py │ │ │ ├── softsign_op_test.py │ │ │ ├── xent_op_d9m_test.py │ │ │ ├── xent_op_test.py │ │ │ └── xent_op_test_base.py │ │ ├── numerics_test.py │ │ ├── proto │ │ │ ├── BUILD │ │ │ ├── decode_proto_op_test.py │ │ │ ├── decode_proto_op_test_base.py │ │ │ ├── descriptor_source_test.py │ │ │ ├── descriptor_source_test_base.py │ │ │ ├── encode_proto_op_test.py │ │ │ ├── encode_proto_op_test_base.py │ │ │ ├── proto_op_test_base.py │ │ │ └── test_example.proto │ │ ├── quantization_ops │ │ │ ├── BUILD │ │ │ └── quantization_ops_test.py │ │ ├── random │ │ │ ├── BUILD │ │ │ ├── candidate_sampler_ops_test.py │ │ │ ├── multinomial_op_big_test.py │ │ │ ├── multinomial_op_test.py │ │ │ ├── parameterized_truncated_normal_op_test.py │ │ │ ├── random_binomial_test.py │ │ │ ├── random_crop_test.py │ │ │ ├── random_gamma_test.py │ │ │ ├── random_grad_test.py │ │ │ ├── random_index_shuffle_test.py │ │ │ ├── random_ops_test.py │ │ │ ├── random_poisson_test.py │ │ │ ├── random_shuffle_queue_test.py │ │ │ ├── stateful_random_ops_test.py │ │ │ ├── stateless_random_ops_test.py │ │ │ └── util.py │ │ ├── signal │ │ │ ├── BUILD │ │ │ ├── dct_ops_test.py │ │ │ ├── fft_ops_test.py │ │ │ ├── mel_ops_test.py │ │ │ ├── mfcc_ops_test.py │ │ │ ├── reconstruction_ops_test.py │ │ │ ├── shape_ops_test.py │ │ │ ├── spectral_ops_test.py │ │ │ ├── test_util.py │ │ │ └── window_ops_test.py │ │ ├── sparse_ops │ │ │ ├── BUILD │ │ │ ├── sparse_add_op_test.py │ │ │ ├── sparse_concat_op_test.py │ │ │ ├── sparse_conditional_accumulator_test.py │ │ │ ├── sparse_cross_op_test.py │ │ │ ├── sparse_matmul_op_test.py │ │ │ ├── sparse_ops_test.py │ │ │ ├── sparse_reorder_op_test.py │ │ │ ├── sparse_reshape_op_test.py │ │ │ ├── sparse_serialization_ops_test.py │ │ │ ├── sparse_slice_op_test.py │ │ │ ├── sparse_split_op_test.py │ │ │ ├── sparse_tensor_dense_matmul_grad_test.py │ │ │ ├── sparse_tensor_dense_matmul_op_d9m_test.py │ │ │ ├── sparse_tensor_dense_matmul_op_test.py │ │ │ ├── sparse_tensors_map_ops_test.py │ │ │ ├── sparse_to_dense_op_py_test.py │ │ │ ├── sparse_xent_op_d9m_test.py │ │ │ ├── sparse_xent_op_test.py │ │ │ ├── sparse_xent_op_test_base.py │ │ │ └── sparsemask_op_test.py │ │ ├── strings_ops │ │ │ ├── BUILD │ │ │ ├── as_string_op_test.py │ │ │ ├── base64_ops_test.py │ │ │ ├── reduce_join_op_test.py │ │ │ ├── regex_full_match_op_test.py │ │ │ ├── regex_replace_op_test.py │ │ │ ├── string_bytes_split_op_test.py │ │ │ ├── string_format_op_test.py │ │ │ ├── string_join_op_test.py │ │ │ ├── string_length_op_test.py │ │ │ ├── string_lower_op_test.py │ │ │ ├── string_split_op_test.py │ │ │ ├── string_strip_op_test.py │ │ │ ├── string_to_hash_bucket_op_test.py │ │ │ ├── string_to_number_op_test.py │ │ │ ├── string_upper_op_test.py │ │ │ ├── substr_op_test.py │ │ │ ├── unicode_decode_op_test.py │ │ │ ├── unicode_encode_op_test.py │ │ │ ├── unicode_script_op_test.py │ │ │ ├── unicode_transcode_op_test.py │ │ │ └── unsorted_segment_join_op_test.py │ │ ├── summary_ops │ │ │ ├── BUILD │ │ │ ├── summary_ops_test.py │ │ │ ├── summary_v1_audio_op_test.py │ │ │ ├── summary_v1_image_op_test.py │ │ │ ├── summary_v1_ops_test.py │ │ │ └── summary_v1_tensor_op_test.py │ │ ├── template_test.py │ │ ├── trace_op_test.py │ │ ├── v1_compat_tests │ │ │ ├── BUILD │ │ │ ├── array_ops_test.py │ │ │ ├── dense_update_ops_test.py │ │ │ ├── identity_op_py_test.py │ │ │ ├── scatter_nd_ops_test.py │ │ │ ├── session_ops_test.py │ │ │ └── stack_op_test.py │ │ └── variables │ │ │ ├── BUILD │ │ │ ├── dense_update_ops_no_tsan_test.py │ │ │ ├── dense_update_ops_test.py │ │ │ ├── partitioned_variables_test.py │ │ │ ├── resource_variable_ops_test.py │ │ │ ├── variable_ops_test.py │ │ │ ├── variable_scope_test.py │ │ │ └── variables_test.py │ ├── layers │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── base.py │ │ ├── convolutional.py │ │ ├── core.py │ │ ├── layers.py │ │ ├── normalization.py │ │ ├── pooling.py │ │ ├── utils.py │ │ └── utils_test.py │ ├── lib │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── core │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── _pywrap_py_func.pyi │ │ │ ├── ndarray_tensor.cc │ │ │ ├── ndarray_tensor.h │ │ │ ├── ndarray_tensor_bridge.cc │ │ │ ├── ndarray_tensor_bridge.h │ │ │ ├── py_exception_registry.cc │ │ │ ├── py_exception_registry.h │ │ │ ├── py_func.cc │ │ │ ├── py_func.h │ │ │ ├── py_func_wrapper.cc │ │ │ ├── py_seq_tensor.cc │ │ │ ├── py_seq_tensor.h │ │ │ ├── py_util.cc │ │ │ ├── py_util.h │ │ │ ├── pybind11_absl.h │ │ │ ├── pybind11_lib.h │ │ │ ├── pybind11_proto.h │ │ │ ├── pybind11_status.h │ │ │ ├── safe_pyobject_ptr.cc │ │ │ └── safe_pyobject_ptr.h │ │ └── io │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── _pywrap_file_io.pyi │ │ │ ├── _pywrap_record_io.pyi │ │ │ ├── file_io.py │ │ │ ├── file_io_test.py │ │ │ ├── file_io_wrapper.cc │ │ │ ├── python_io.py │ │ │ ├── record_io_wrapper.cc │ │ │ ├── tf_record.py │ │ │ └── tf_record_test.py │ ├── lite │ │ └── toco_python_api_wrapper.cc │ ├── mlir_wrapper.cc │ ├── module │ │ ├── BUILD │ │ ├── module.py │ │ └── module_test.py │ ├── modules_with_exports.py │ ├── ops │ │ ├── BUILD │ │ ├── accumulate_n_benchmark.py │ │ ├── array_grad.py │ │ ├── array_grad_test.py │ │ ├── array_ops.py │ │ ├── array_ops_shape_test.py │ │ ├── array_ops_stack.py │ │ ├── array_ops_test.py │ │ ├── autograph_ops.py │ │ ├── autograph_ops_test.py │ │ ├── batch_norm_benchmark.py │ │ ├── batch_ops.py │ │ ├── batch_ops_test.py │ │ ├── bincount_ops.py │ │ ├── bincount_ops_test.py │ │ ├── bitwise_ops.py │ │ ├── bitwise_ops_test.py │ │ ├── boosted_trees_ops.py │ │ ├── candidate_sampling_ops.py │ │ ├── check_ops.py │ │ ├── clip_ops.py │ │ ├── clip_ops_test.py │ │ ├── clustering_ops.py │ │ ├── clustering_ops_test.py │ │ ├── collective_ops.py │ │ ├── collective_ops_benchmark.py │ │ ├── collective_ops_gpu_test.py │ │ ├── collective_ops_test.py │ │ ├── collective_ops_xla_test.py │ │ ├── compiled_collective_ops_gpu_test.py │ │ ├── composite_tensor_ops.py │ │ ├── concat_benchmark.py │ │ ├── cond.py │ │ ├── cond_v2.py │ │ ├── confusion_matrix.py │ │ ├── control_flow_assert.py │ │ ├── control_flow_case.py │ │ ├── control_flow_grad.py │ │ ├── control_flow_ops.py │ │ ├── control_flow_ops_benchmark.py │ │ ├── control_flow_ops_test.py │ │ ├── control_flow_state.py │ │ ├── control_flow_switch_case.py │ │ ├── control_flow_util.py │ │ ├── control_flow_util_v2.py │ │ ├── control_flow_v2_disable_test.py │ │ ├── control_flow_v2_enable_test.py │ │ ├── control_flow_v2_func_graphs.py │ │ ├── control_flow_v2_toggles.py │ │ ├── control_flow_v2_toggles_test.py │ │ ├── conv2d_benchmark.py │ │ ├── critical_section_ops.py │ │ ├── ctc_ops.py │ │ ├── cudnn_rnn_grad.py │ │ ├── custom_gradient.py │ │ ├── data_flow_grad.py │ │ ├── data_flow_ops.py │ │ ├── default_gradient.py │ │ ├── dequantize_op_test.py │ │ ├── distributions │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── bernoulli.py │ │ │ ├── beta.py │ │ │ ├── bijector.py │ │ │ ├── bijector_impl.py │ │ │ ├── bijector_test_util.py │ │ │ ├── categorical.py │ │ │ ├── dirichlet.py │ │ │ ├── dirichlet_multinomial.py │ │ │ ├── distribution.py │ │ │ ├── distributions.py │ │ │ ├── exponential.py │ │ │ ├── gamma.py │ │ │ ├── identity_bijector.py │ │ │ ├── kullback_leibler.py │ │ │ ├── laplace.py │ │ │ ├── multinomial.py │ │ │ ├── normal.py │ │ │ ├── special_math.py │ │ │ ├── student_t.py │ │ │ ├── transformed_distribution.py │ │ │ ├── uniform.py │ │ │ └── util.py │ │ ├── embedding_ops.py │ │ ├── embedding_ops_test.py │ │ ├── factory_ops_test.py │ │ ├── filesystem_ops.py │ │ ├── functional_ops.py │ │ ├── functional_ops_test.py │ │ ├── gradient_checker.py │ │ ├── gradient_checker_v2.py │ │ ├── gradient_checker_v2_test.py │ │ ├── gradients.py │ │ ├── gradients_impl.py │ │ ├── gradients_test.py │ │ ├── gradients_util.py │ │ ├── handle_data_util.py │ │ ├── histogram_ops.py │ │ ├── image_grad.py │ │ ├── image_grad_d9m_test.py │ │ ├── image_grad_test.py │ │ ├── image_grad_test_base.py │ │ ├── image_ops.py │ │ ├── image_ops_impl.py │ │ ├── image_ops_test.py │ │ ├── init_ops.py │ │ ├── init_ops_test.py │ │ ├── init_ops_v2.py │ │ ├── init_ops_v2_test.py │ │ ├── initializers_ns.py │ │ ├── inplace_ops.py │ │ ├── io_ops.py │ │ ├── linalg │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── linalg.py │ │ │ ├── linalg_impl.py │ │ │ ├── linear_operator.py │ │ │ ├── linear_operator_addition.py │ │ │ ├── linear_operator_adjoint.py │ │ │ ├── linear_operator_block_diag.py │ │ │ ├── linear_operator_block_lower_triangular.py │ │ │ ├── linear_operator_circulant.py │ │ │ ├── linear_operator_composition.py │ │ │ ├── linear_operator_diag.py │ │ │ ├── linear_operator_full_matrix.py │ │ │ ├── linear_operator_householder.py │ │ │ ├── linear_operator_identity.py │ │ │ ├── linear_operator_inversion.py │ │ │ ├── linear_operator_kronecker.py │ │ │ ├── linear_operator_low_rank_update.py │ │ │ ├── linear_operator_lower_triangular.py │ │ │ ├── linear_operator_permutation.py │ │ │ ├── linear_operator_test_util.py │ │ │ ├── linear_operator_toeplitz.py │ │ │ ├── linear_operator_tridiag.py │ │ │ ├── linear_operator_util.py │ │ │ ├── linear_operator_zeros.py │ │ │ ├── property_hint_util.py │ │ │ ├── slicing.py │ │ │ └── sparse │ │ │ │ ├── BUILD │ │ │ │ ├── __init__.py │ │ │ │ ├── conjugate_gradient.py │ │ │ │ ├── sparse.py │ │ │ │ ├── sparse_csr_matrix_grad.py │ │ │ │ └── sparse_csr_matrix_ops.py │ │ ├── linalg_grad.py │ │ ├── linalg_ops.py │ │ ├── linalg_ops_impl.py │ │ ├── list_ops.py │ │ ├── logging_ops.py │ │ ├── lookup_grad.py │ │ ├── lookup_ops.py │ │ ├── lookup_ops_async_checkpoint_test.py │ │ ├── losses │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── losses.py │ │ │ ├── losses_impl.py │ │ │ ├── util.py │ │ │ └── util_test.py │ │ ├── manip_grad.py │ │ ├── manip_ops.py │ │ ├── map_fn.py │ │ ├── map_ops.py │ │ ├── math_grad.py │ │ ├── math_grad_test.py │ │ ├── math_ops.py │ │ ├── math_ops_linspace_test.py │ │ ├── math_ops_test.py │ │ ├── matmul_benchmark.py │ │ ├── memory_tests │ │ │ ├── BUILD │ │ │ └── custom_gradient_memory_test.py │ │ ├── metrics.py │ │ ├── metrics_impl.py │ │ ├── nccl_ops.py │ │ ├── nccl_ops_test.py │ │ ├── nn.py │ │ ├── nn_batchnorm_test.py │ │ ├── nn_fused_batch_norm_grad.py │ │ ├── nn_fused_batchnorm_d9m_test.py │ │ ├── nn_fused_batchnorm_test.py │ │ ├── nn_grad.py │ │ ├── nn_grad_test.py │ │ ├── nn_impl.py │ │ ├── nn_impl_distribute.py │ │ ├── nn_loss_scaling_utilities_test.py │ │ ├── nn_ops.py │ │ ├── nn_test.py │ │ ├── nn_xent_test.py │ │ ├── numerics.py │ │ ├── numpy_ops │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── g3doc │ │ │ │ ├── TensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb │ │ │ │ ├── TensorFlow_NumPy_Text_Generation.ipynb │ │ │ │ └── TensorFlow_Numpy_Distributed_Image_Classification.ipynb │ │ │ ├── integration_test │ │ │ │ ├── BUILD │ │ │ │ ├── benchmarks │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── micro_benchmarks.py │ │ │ │ │ ├── numpy_mlp.py │ │ │ │ │ └── tf_numpy_mlp.py │ │ │ │ ├── np_config_test.py │ │ │ │ └── public_symbol_test.py │ │ │ ├── np_array_ops.py │ │ │ ├── np_array_ops_test.py │ │ │ ├── np_arrays.py │ │ │ ├── np_arrays_test.py │ │ │ ├── np_config.py │ │ │ ├── np_dtypes.py │ │ │ ├── np_dtypes_test.py │ │ │ ├── np_interop_test.py │ │ │ ├── np_logic_test.py │ │ │ ├── np_math_ops.py │ │ │ ├── np_math_ops_test.py │ │ │ ├── np_random.py │ │ │ ├── np_random_test.py │ │ │ ├── np_utils.py │ │ │ ├── np_utils_test.py │ │ │ └── tests │ │ │ │ ├── BUILD │ │ │ │ ├── config.py │ │ │ │ ├── extensions.py │ │ │ │ ├── np_einsum_test.py │ │ │ │ ├── np_indexing_test.py │ │ │ │ ├── np_test.py │ │ │ │ ├── np_wrapper.py │ │ │ │ └── test_util.py │ │ ├── op_selector.py │ │ ├── op_selector_test.py │ │ ├── optional_grad.py │ │ ├── parallel_for │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── array_test.py │ │ │ ├── control_flow_ops.py │ │ │ ├── control_flow_ops_test.py │ │ │ ├── gradients.py │ │ │ ├── gradients_test.py │ │ │ ├── index.md │ │ │ ├── math_test.py │ │ │ ├── pfor.py │ │ │ ├── pfor_test.py │ │ │ ├── test_util.py │ │ │ └── xla_control_flow_ops_test.py │ │ ├── parsing_config.py │ │ ├── parsing_grad.py │ │ ├── parsing_ops.py │ │ ├── partitioned_variables.py │ │ ├── proto_ops.py │ │ ├── quantized_conv_ops_test.py │ │ ├── quantized_ops_test.py │ │ ├── ragged │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── convert_to_tensor_or_ragged_tensor_op_test.py │ │ │ ├── dynamic_ragged_shape.py │ │ │ ├── dynamic_ragged_shape_test.py │ │ │ ├── ragged_array_ops.py │ │ │ ├── ragged_autograph.py │ │ │ ├── ragged_batch_gather_op_test.py │ │ │ ├── ragged_batch_gather_ops.py │ │ │ ├── ragged_batch_gather_with_default_op.py │ │ │ ├── ragged_bincount_ops.py │ │ │ ├── ragged_bincount_ops_test.py │ │ │ ├── ragged_bitcast_op_test.py │ │ │ ├── ragged_boolean_mask_op_test.py │ │ │ ├── ragged_check_ops.py │ │ │ ├── ragged_concat_op_test.py │ │ │ ├── ragged_concat_ops.py │ │ │ ├── ragged_config.py │ │ │ ├── ragged_const_op_test.py │ │ │ ├── ragged_constant_value_op_test.py │ │ │ ├── ragged_conversion_ops.py │ │ │ ├── ragged_cross_op_test.py │ │ │ ├── ragged_dispatch.py │ │ │ ├── ragged_dispatch_test.py │ │ │ ├── ragged_dynamic_partition_op_test.py │ │ │ ├── ragged_eager_test.py │ │ │ ├── ragged_embedding_ops.py │ │ │ ├── ragged_expand_dims_op_test.py │ │ │ ├── ragged_factory_ops.py │ │ │ ├── ragged_factory_ops_test.py │ │ │ ├── ragged_fill_empty_rows_op_test.py │ │ │ ├── ragged_from_sparse_op_test.py │ │ │ ├── ragged_from_tensor_op_test.py │ │ │ ├── ragged_functional_ops.py │ │ │ ├── ragged_gather_nd_op_test.py │ │ │ ├── ragged_gather_op_test.py │ │ │ ├── ragged_gather_ops.py │ │ │ ├── ragged_getitem.py │ │ │ ├── ragged_getitem_test.py │ │ │ ├── ragged_image_ops.py │ │ │ ├── ragged_map_flat_values_op_test.py │ │ │ ├── ragged_map_fn_op_test.py │ │ │ ├── ragged_map_ops.py │ │ │ ├── ragged_math_ops.py │ │ │ ├── ragged_math_ops_test.py │ │ │ ├── ragged_matmul_op_test.py │ │ │ ├── ragged_merge_dims_op_test.py │ │ │ ├── ragged_one_hot_op_test.py │ │ │ ├── ragged_operators.py │ │ │ ├── ragged_operators_test.py │ │ │ ├── ragged_ops.py │ │ │ ├── ragged_placeholder_op_test.py │ │ │ ├── ragged_print_op_test.py │ │ │ ├── ragged_range_op_test.py │ │ │ ├── ragged_rank_op_test.py │ │ │ ├── ragged_reduce_op_test.py │ │ │ ├── ragged_resize_image_op_test.py │ │ │ ├── ragged_reverse_op_test.py │ │ │ ├── ragged_row_lengths_op_test.py │ │ │ ├── ragged_row_splits_to_segment_ids_op_test.py │ │ │ ├── ragged_segment_ids_to_row_splits_op_test.py │ │ │ ├── ragged_segment_op_test.py │ │ │ ├── ragged_size_op_test.py │ │ │ ├── ragged_split_op_test.py │ │ │ ├── ragged_squeeze_op.py │ │ │ ├── ragged_squeeze_op_test.py │ │ │ ├── ragged_stack_op_test.py │ │ │ ├── ragged_string_ops.py │ │ │ ├── ragged_tensor.py │ │ │ ├── ragged_tensor_bounding_shape_op_test.py │ │ │ ├── ragged_tensor_shape.py │ │ │ ├── ragged_tensor_shape_test.py │ │ │ ├── ragged_tensor_supported_values_test.py │ │ │ ├── ragged_tensor_test.py │ │ │ ├── ragged_tensor_test_ops.py │ │ │ ├── ragged_tensor_value.py │ │ │ ├── ragged_tile_op_test.py │ │ │ ├── ragged_to_sparse_op_test.py │ │ │ ├── ragged_to_tensor_op_test.py │ │ │ ├── ragged_util.py │ │ │ ├── ragged_util_test.py │ │ │ ├── ragged_where_op.py │ │ │ ├── ragged_where_op_test.py │ │ │ ├── row_partition.py │ │ │ ├── row_partition_test.py │ │ │ ├── segment_id_ops.py │ │ │ ├── string_ngrams_op_test.py │ │ │ └── strings_reduce_join_op_test.py │ │ ├── random_crop_ops.py │ │ ├── random_grad.py │ │ ├── random_ops.py │ │ ├── random_ops_util.py │ │ ├── raw_ops_test.py │ │ ├── ref_variable.py │ │ ├── resource_variable_ops.py │ │ ├── resource_variables_toggle.py │ │ ├── resources.py │ │ ├── rnn.py │ │ ├── rnn_cell.py │ │ ├── rnn_cell_impl.py │ │ ├── rnn_cell_wrapper_impl.py │ │ ├── rnn_grad.py │ │ ├── rnn_grad_test.py │ │ ├── script_ops.py │ │ ├── script_ops_test.py │ │ ├── sdca_ops.py │ │ ├── session_ops.py │ │ ├── sets.py │ │ ├── sets_impl.py │ │ ├── shape_util.py │ │ ├── signal │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── dct_ops.py │ │ │ ├── fft_ops.py │ │ │ ├── mel_ops.py │ │ │ ├── mfcc_ops.py │ │ │ ├── reconstruction_ops.py │ │ │ ├── shape_ops.py │ │ │ ├── signal.py │ │ │ ├── spectral_ops.py │ │ │ ├── util_ops.py │ │ │ └── window_ops.py │ │ ├── sobol_ops_test.py │ │ ├── sort_ops.py │ │ ├── sort_ops_test.py │ │ ├── sparse_bincount_ops_test.py │ │ ├── sparse_grad.py │ │ ├── sparse_ops.py │ │ ├── sparse_ops_test.py │ │ ├── special_math_ops.py │ │ ├── special_math_ops_test.py │ │ ├── split_benchmark.py │ │ ├── standard_ops.py │ │ ├── state_grad.py │ │ ├── state_ops.py │ │ ├── stateful_random_ops.py │ │ ├── stateless_random_ops.py │ │ ├── stochastic_cast_op.py │ │ ├── string_ops.py │ │ ├── structured │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── structured_array_ops.py │ │ │ ├── structured_array_ops_test.py │ │ │ ├── structured_ops.py │ │ │ ├── structured_tensor.py │ │ │ ├── structured_tensor_dynamic.py │ │ │ ├── structured_tensor_slice_test.py │ │ │ ├── structured_tensor_spec_test.py │ │ │ └── structured_tensor_test.py │ │ ├── summary_op_util.py │ │ ├── summary_ops_v2.py │ │ ├── template.py │ │ ├── tensor_array_grad.py │ │ ├── tensor_array_ops.py │ │ ├── tensor_array_ops_test.py │ │ ├── transpose_benchmark.py │ │ ├── unconnected_gradients.py │ │ ├── v1_compat_tests │ │ │ ├── BUILD │ │ │ └── gradient_checker_test.py │ │ ├── variable_scope.py │ │ ├── variable_spec_test.py │ │ ├── variable_v1.py │ │ ├── variables.py │ │ ├── weak_tensor_array_ops_test.py │ │ ├── weak_tensor_constant_op_test.py │ │ ├── weak_tensor_image_ops_test.py │ │ ├── weak_tensor_math_ops_test.py │ │ ├── weak_tensor_nn_test.py │ │ ├── weak_tensor_np_array_ops_test.py │ │ ├── weak_tensor_np_math_ops_test.py │ │ ├── weak_tensor_ops.py │ │ ├── weak_tensor_ops_test.py │ │ ├── weak_tensor_special_math_ops_test.py │ │ ├── weak_tensor_test_util.py │ │ ├── weights_broadcast_ops.py │ │ ├── while_loop.py │ │ ├── while_v2.py │ │ └── while_v2_indexed_slices_rewriter.py │ ├── platform │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── _pywrap_cpu_feature_guard.pyi │ │ ├── _pywrap_stacktrace_handler.pyi │ │ ├── _pywrap_tf2.pyi │ │ ├── analytics.py │ │ ├── app.py │ │ ├── app_test.py │ │ ├── benchmark.py │ │ ├── benchmark_test.py │ │ ├── build_info_test.py │ │ ├── control_imports.py │ │ ├── cpu_feature_guard_wrapper.cc │ │ ├── device_context.py │ │ ├── enable_tf2.cc │ │ ├── flags.py │ │ ├── flags_test.py │ │ ├── gfile.py │ │ ├── googletest.py │ │ ├── logging_test.py │ │ ├── parameterized.py │ │ ├── remote_utils.py │ │ ├── resource_loader.py │ │ ├── resource_loader_test.py │ │ ├── self_check.py │ │ ├── stacktrace_handler_test.py │ │ ├── stacktrace_handler_wrapper.cc │ │ ├── sysconfig.py │ │ ├── sysconfig_test.py │ │ ├── test.py │ │ └── tf_logging.py │ ├── profiler │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── integration_test │ │ │ ├── BUILD │ │ │ ├── mnist_testing_utils.py │ │ │ └── profiler_api_test.py │ │ ├── internal │ │ │ ├── BUILD │ │ │ ├── _pywrap_profiler.pyi │ │ │ ├── _pywrap_traceme.pyi │ │ │ ├── flops_registry.py │ │ │ ├── flops_registry_test.py │ │ │ ├── model_analyzer_testlib.py │ │ │ ├── print_model_analysis_test.py │ │ │ ├── profiler_pywrap_impl.cc │ │ │ ├── profiler_pywrap_impl.h │ │ │ ├── profiler_wrapper.cc │ │ │ ├── python_hooks.h │ │ │ ├── run_metadata_test.py │ │ │ └── traceme_wrapper.cc │ │ ├── model_analyzer.py │ │ ├── model_analyzer_test.py │ │ ├── option_builder.py │ │ ├── pprof_profiler.py │ │ ├── pprof_profiler_test.py │ │ ├── profile_context.py │ │ ├── profile_context_test.py │ │ ├── profiler.py │ │ ├── profiler_client.py │ │ ├── profiler_client_test.py │ │ ├── profiler_test.py │ │ ├── profiler_v2.py │ │ ├── profiler_v2_test.py │ │ ├── profiler_wrapper_test.py │ │ ├── tfprof_logger.py │ │ ├── tfprof_logger_test.py │ │ └── trace.py │ ├── proto_exports.py │ ├── py_exception_registry_wrapper.cc │ ├── pywrap_dlopen_global_flags.py │ ├── pywrap_dtensor_device.cc │ ├── pywrap_mlir.py │ ├── pywrap_sanitizers.py │ ├── pywrap_tensorflow.py │ ├── pywrap_tensorflow_internal.cc │ ├── pywrap_tfe.py │ ├── pywrap_tfe_monitoring_reader.py │ ├── sanitizers_wrapper.cc │ ├── saved_model │ │ ├── BUILD │ │ ├── README.md │ │ ├── builder.py │ │ ├── builder_impl.py │ │ ├── constants.py │ │ ├── fingerprinting.md │ │ ├── fingerprinting.py │ │ ├── fingerprinting_test.py │ │ ├── fingerprinting_utils.py │ │ ├── function_deserialization.py │ │ ├── function_serialization.py │ │ ├── keras_injection_test.py │ │ ├── load.py │ │ ├── load_optimizer_test.py │ │ ├── load_options.py │ │ ├── load_test.py │ │ ├── load_v1_in_v2.py │ │ ├── load_v1_in_v2_test.py │ │ ├── loader.py │ │ ├── loader_impl.py │ │ ├── loader_test.py │ │ ├── main_op.py │ │ ├── main_op_impl.py │ │ ├── method_name_updater.py │ │ ├── method_name_updater_test.py │ │ ├── metrics_test.py │ │ ├── model_utils │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── export_output.py │ │ │ ├── export_output_test.py │ │ │ ├── export_test.py │ │ │ ├── export_utils.py │ │ │ ├── mode_keys.py │ │ │ └── mode_keys_test.py │ │ ├── nested_structure_coder.py │ │ ├── nested_structure_coder_test.py │ │ ├── path_helpers.py │ │ ├── proto_splitter_save_test.py │ │ ├── pywrap_saved_model.cc │ │ ├── pywrap_saved_model │ │ │ ├── __init__.pyi │ │ │ ├── constants.pyi │ │ │ ├── fingerprinting.pyi │ │ │ ├── merger.pyi │ │ │ └── metrics.pyi │ │ ├── pywrap_saved_model_constants.cc │ │ ├── pywrap_saved_model_constants.h │ │ ├── pywrap_saved_model_fingerprinting.cc │ │ ├── pywrap_saved_model_fingerprinting.h │ │ ├── pywrap_saved_model_fingerprinting_test.py │ │ ├── pywrap_saved_model_metrics.cc │ │ ├── pywrap_saved_model_metrics.h │ │ ├── pywrap_saved_model_metrics_test.py │ │ ├── registration │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── registration.py │ │ │ ├── registration_saving_test.py │ │ │ ├── registration_test.py │ │ │ ├── test_util.py │ │ │ ├── tf_checkpoint_saver_allowlist.txt │ │ │ ├── tf_registration_test.py │ │ │ └── tf_serializable_allowlist.txt │ │ ├── revived_types.py │ │ ├── revived_types_test.py │ │ ├── save.py │ │ ├── save_context.py │ │ ├── save_context_test.py │ │ ├── save_options.py │ │ ├── save_test.py │ │ ├── saved_model.py │ │ ├── saved_model_test.py │ │ ├── signature_constants.py │ │ ├── signature_def_utils.py │ │ ├── signature_def_utils_impl.py │ │ ├── signature_def_utils_test.py │ │ ├── signature_serialization.py │ │ ├── simple_save.py │ │ ├── simple_save_test.py │ │ ├── tag_constants.py │ │ ├── tests │ │ │ ├── BUILD │ │ │ └── variable_wrapper_test.py │ │ ├── tracing_utils.py │ │ ├── tracing_utils_test.py │ │ ├── utils.py │ │ ├── utils_impl.py │ │ └── utils_test.py │ ├── summary │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── plugin_asset.py │ │ ├── plugin_asset_test.py │ │ ├── summary.py │ │ ├── summary_iterator.py │ │ ├── summary_iterator_test.py │ │ ├── summary_test.py │ │ ├── summary_v2_test.py │ │ ├── tb_summary.py │ │ └── writer │ │ │ ├── BUILD │ │ │ ├── event_file_writer.py │ │ │ ├── event_file_writer_v2.py │ │ │ ├── fake_summary_writer.py │ │ │ ├── writer.py │ │ │ ├── writer_cache.py │ │ │ └── writer_test.py │ ├── tf2.py │ ├── tfcompile_wrapper.cc │ ├── tfe_wrapper.cc │ ├── tfe_wrapper_monitoring_reader.cc │ ├── tools │ │ ├── BUILD │ │ ├── aot_compiled_test.cc │ │ ├── api │ │ │ ├── generator │ │ │ │ ├── BUILD │ │ │ │ ├── api_gen.bzl │ │ │ │ ├── api_init_files.bzl │ │ │ │ ├── api_init_files_v1.bzl │ │ │ │ ├── create_python_api.py │ │ │ │ ├── create_python_api_test.py │ │ │ │ ├── doc_srcs.py │ │ │ │ ├── doc_srcs_test.py │ │ │ │ └── output_init_files_test.py │ │ │ └── generator2 │ │ │ │ ├── BUILD │ │ │ │ ├── apis.bzl │ │ │ │ ├── docstrings.py │ │ │ │ ├── extractor │ │ │ │ ├── BUILD │ │ │ │ ├── extractor.py │ │ │ │ ├── extractor_test.py │ │ │ │ └── main.py │ │ │ │ ├── generate_api.bzl │ │ │ │ ├── generator │ │ │ │ ├── BUILD │ │ │ │ ├── generator.py │ │ │ │ ├── generator_test.py │ │ │ │ └── main.py │ │ │ │ ├── patterns.bzl │ │ │ │ └── shared │ │ │ │ ├── BUILD │ │ │ │ ├── exported_api.py │ │ │ │ └── exported_api_test.py │ │ ├── freeze_graph.py │ │ ├── freeze_graph_test.py │ │ ├── import_pb_to_tensorboard.py │ │ ├── inspect_checkpoint.py │ │ ├── make_aot_compile_models.py │ │ ├── module_util.py │ │ ├── no_xla_multithread_symbols_test.sh │ │ ├── optimize_for_inference.py │ │ ├── optimize_for_inference_lib.py │ │ ├── optimize_for_inference_test.py │ │ ├── print_selective_registration_header.py │ │ ├── print_selective_registration_header_test.py │ │ ├── saved_model_aot_compile.py │ │ ├── saved_model_cli.py │ │ ├── saved_model_cli_test.py │ │ ├── saved_model_utils.py │ │ ├── saved_model_utils_test.py │ │ ├── selective_registration_header_lib.py │ │ ├── skip_test.sh │ │ ├── strip_unused.py │ │ ├── strip_unused_lib.py │ │ ├── strip_unused_test.py │ │ ├── tf_import_time.py │ │ ├── tools.bzl │ │ └── xla_multithread_symbols_test.sh │ ├── tpu │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── _pywrap_tpu_embedding.pyi │ │ ├── api.py │ │ ├── async_checkpoint.py │ │ ├── async_checkpoint_test.py │ │ ├── bfloat16.py │ │ ├── bfloat16_test.py │ │ ├── client │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── client.py │ │ │ ├── client_test.py │ │ │ ├── pip_package │ │ │ │ ├── BUILD │ │ │ │ ├── README │ │ │ │ ├── build_pip_package.sh │ │ │ │ └── setup.py │ │ │ └── version.py │ │ ├── datasets.py │ │ ├── datasets_test.py │ │ ├── device_assignment.py │ │ ├── experimental │ │ │ ├── BUILD │ │ │ └── __init__.py │ │ ├── feature_column.py │ │ ├── feature_column_test.py │ │ ├── feature_column_v2.py │ │ ├── feature_column_v2_test.py │ │ ├── functional.py │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── tpu_ops.py │ │ │ └── tpu_ordinal_selector_op.py │ │ ├── preempted_hook.py │ │ ├── profiler │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── capture_tpu_profile.py │ │ │ ├── pip_package │ │ │ │ ├── BUILD │ │ │ │ ├── README │ │ │ │ ├── build_pip_package.sh │ │ │ │ └── setup.py │ │ │ ├── profiler_analysis_pb2_grpc.py │ │ │ └── version.py │ │ ├── pywrap_tpu_embedding.cc │ │ ├── session_support.py │ │ ├── tensor_tracer.proto │ │ ├── tensor_tracer.py │ │ ├── tensor_tracer_flags.py │ │ ├── tensor_tracer_report.py │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── tpu_embedding_base_test.py │ │ │ ├── tpu_embedding_v1_checkpoint_test.py │ │ │ ├── tpu_embedding_v1_correctness_test.py │ │ │ ├── tpu_embedding_v2_checkpoint_test.py │ │ │ ├── tpu_embedding_v2_correctness_base_test.py │ │ │ ├── tpu_embedding_v2_correctness_dense_lookup_test.py │ │ │ ├── tpu_embedding_v2_correctness_hd_ragged_forward_test.py │ │ │ ├── tpu_embedding_v2_correctness_hd_ragged_training_test.py │ │ │ ├── tpu_embedding_v2_correctness_hd_sparse_forward_test.py │ │ │ ├── tpu_embedding_v2_correctness_hd_sparse_training_test.py │ │ │ ├── tpu_embedding_v2_correctness_ragged_forward_test.py │ │ │ ├── tpu_embedding_v2_correctness_ragged_training_test.py │ │ │ ├── tpu_embedding_v2_correctness_sequence_feature_test.py │ │ │ ├── tpu_embedding_v2_correctness_sparse_forward_test.py │ │ │ ├── tpu_embedding_v2_correctness_sparse_training_test.py │ │ │ ├── tpu_embedding_v2_enqueue_mode_test.py │ │ │ ├── tpu_embedding_v2_hd_invalid_input_test.py │ │ │ ├── tpu_embedding_v2_hd_valid_input_test.py │ │ │ ├── tpu_embedding_v2_initialization_test.py │ │ │ ├── tpu_embedding_v2_invalid_input_test.py │ │ │ ├── tpu_embedding_v2_mp_strategy_test.py │ │ │ ├── tpu_embedding_v2_optimizer_test.py │ │ │ ├── tpu_embedding_v2_sequence_feature_test.py │ │ │ ├── tpu_embedding_v2_valid_input_test.py │ │ │ └── tpu_initialization_test.py │ │ ├── topology.py │ │ ├── topology_test.py │ │ ├── tpu.bzl │ │ ├── tpu.py │ │ ├── tpu_embedding.py │ │ ├── tpu_embedding_base.py │ │ ├── tpu_embedding_for_serving.py │ │ ├── tpu_embedding_for_serving_test.py │ │ ├── tpu_embedding_gradient.py │ │ ├── tpu_embedding_v1.py │ │ ├── tpu_embedding_v2.py │ │ ├── tpu_embedding_v2_utils.py │ │ ├── tpu_embedding_v2_utils_test.py │ │ ├── tpu_embedding_v3.py │ │ ├── tpu_embedding_v3_utils.py │ │ ├── tpu_embedding_v3_utils_test.py │ │ ├── tpu_feed.py │ │ ├── tpu_function.py │ │ ├── tpu_hardware_feature.py │ │ ├── tpu_infeed_test.py │ │ ├── tpu_name_util.py │ │ ├── tpu_optimizer.py │ │ ├── tpu_outside_compilation_test.py │ │ ├── tpu_replication.py │ │ ├── tpu_sharding.py │ │ ├── tpu_sharding_test.py │ │ ├── tpu_strategy_util.py │ │ ├── tpu_system_metadata.py │ │ ├── tpu_test.py │ │ ├── tpu_test_wrapper.bzl │ │ ├── tpu_test_wrapper.py │ │ ├── tpu_test_wrapper_test.py │ │ └── training_loop.py │ ├── trackable │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── asset.py │ │ ├── autotrackable.py │ │ ├── autotrackable_test.py │ │ ├── base.py │ │ ├── base_delegate.py │ │ ├── base_delegate_test.py │ │ ├── base_test.py │ │ ├── constants.py │ │ ├── converter.py │ │ ├── data_structures.py │ │ ├── data_structures_test.py │ │ ├── layer_utils.py │ │ ├── python_state.py │ │ ├── python_state_test.py │ │ ├── resource.py │ │ ├── resource_test.py │ │ ├── trackable_utils.py │ │ └── trackable_utils_test.py │ ├── training │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── adadelta.py │ │ ├── adadelta_test.py │ │ ├── adagrad.py │ │ ├── adagrad_da.py │ │ ├── adagrad_da_test.py │ │ ├── adagrad_test.py │ │ ├── adam.py │ │ ├── adam_test.py │ │ ├── basic_loops.py │ │ ├── basic_loops_test.py │ │ ├── basic_session_run_hooks.py │ │ ├── basic_session_run_hooks_test.py │ │ ├── checkpoint_management.py │ │ ├── checkpoint_ops.py │ │ ├── checkpoint_ops_test.py │ │ ├── checkpoint_state.proto │ │ ├── checkpoint_utils.py │ │ ├── checkpoint_utils_test.py │ │ ├── coordinator.py │ │ ├── coordinator_test.py │ │ ├── device_setter.py │ │ ├── device_setter_test.py │ │ ├── evaluation.py │ │ ├── evaluation_test.py │ │ ├── experimental │ │ │ ├── BUILD │ │ │ ├── loss_scale.py │ │ │ ├── loss_scale_optimizer.py │ │ │ ├── loss_scale_optimizer_test.py │ │ │ ├── loss_scale_test.py │ │ │ ├── mixed_precision.py │ │ │ ├── mixed_precision_global_state.py │ │ │ └── mixed_precision_test.py │ │ ├── ftrl.py │ │ ├── ftrl_test.py │ │ ├── gradient_descent.py │ │ ├── gradient_descent_test.py │ │ ├── input.py │ │ ├── input_test.py │ │ ├── learning_rate_decay.py │ │ ├── localhost_cluster_performance_test.py │ │ ├── momentum.py │ │ ├── momentum_test.py │ │ ├── monitored_session.py │ │ ├── monitored_session_test.py │ │ ├── moving_averages.py │ │ ├── moving_averages_test.py │ │ ├── optimizer.py │ │ ├── optimizer_test.py │ │ ├── proximal_adagrad.py │ │ ├── proximal_adagrad_test.py │ │ ├── proximal_gradient_descent.py │ │ ├── proximal_gradient_descent_test.py │ │ ├── py_checkpoint_reader.py │ │ ├── quantize_training.py │ │ ├── quantize_training_test.py │ │ ├── quantize_training_wrapper.cc │ │ ├── queue_runner.py │ │ ├── queue_runner_impl.py │ │ ├── queue_runner_test.py │ │ ├── rmsprop.py │ │ ├── rmsprop_test.py │ │ ├── saver.py │ │ ├── saver_large_partitioned_variable_test.py │ │ ├── saver_large_variable_test.py │ │ ├── saver_test.py │ │ ├── saver_test_utils.py │ │ ├── saving │ │ │ ├── BUILD │ │ │ ├── checkpoint_options.py │ │ │ ├── functional_saver.py │ │ │ ├── saveable_object.py │ │ │ ├── saveable_object_util.py │ │ │ ├── saveable_object_util_test.py │ │ │ └── trace_saveable_util.py │ │ ├── server_lib.py │ │ ├── server_lib_multiple_containers_test.py │ │ ├── server_lib_same_variables_clear_container_test.py │ │ ├── server_lib_same_variables_clear_test.py │ │ ├── server_lib_same_variables_no_clear_test.py │ │ ├── server_lib_sparse_job_test.py │ │ ├── server_lib_test.py │ │ ├── session_manager.py │ │ ├── session_manager_test.py │ │ ├── session_run_hook.py │ │ ├── slot_creator.py │ │ ├── slot_creator_test.py │ │ ├── summary_io.py │ │ ├── supervisor.py │ │ ├── supervisor_test.py │ │ ├── sync_replicas_optimizer.py │ │ ├── sync_replicas_optimizer_test.py │ │ ├── training.py │ │ ├── training_ops_test.py │ │ ├── training_util.py │ │ ├── training_util_test.py │ │ ├── warm_starting_util.py │ │ └── warm_starting_util_test.py │ ├── types │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── core.py │ │ ├── data.py │ │ ├── distribute.py │ │ ├── doc_typealias.py │ │ ├── internal.py │ │ └── trace.py │ ├── user_ops │ │ ├── BUILD │ │ ├── __init__.py │ │ └── user_ops.py │ └── util │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── _function_parameter_canonicalizer_binding_for_test.pyi │ │ ├── _pywrap_checkpoint_reader.pyi │ │ ├── _pywrap_determinism.pyi │ │ ├── _pywrap_kernel_registry.pyi │ │ ├── _pywrap_nest.pyi │ │ ├── _pywrap_stat_summarizer.pyi │ │ ├── _pywrap_tensor_float_32_execution.pyi │ │ ├── _pywrap_tfprof.pyi │ │ ├── _pywrap_transform_graph.pyi │ │ ├── _pywrap_util_port.pyi │ │ ├── _pywrap_utils.pyi │ │ ├── _tf_stack.pyi │ │ ├── all_util.py │ │ ├── compat.py │ │ ├── compat_test.py │ │ ├── custom_nest_protocol.py │ │ ├── decorator_utils.py │ │ ├── decorator_utils_test.py │ │ ├── deprecated_module.py │ │ ├── deprecated_module_new.py │ │ ├── deprecation.py │ │ ├── deprecation_test.py │ │ ├── deprecation_wrapper.py │ │ ├── determinism.cc │ │ ├── dispatch.py │ │ ├── dispatch_test.py │ │ ├── example_parser_configuration.py │ │ ├── example_parser_configuration_test.py │ │ ├── fast_module_type.cc │ │ ├── fast_module_type.pyi │ │ ├── fast_module_type_test.py │ │ ├── function_parameter_canonicalizer.cc │ │ ├── function_parameter_canonicalizer.h │ │ ├── function_parameter_canonicalizer_binding_for_test.cc │ │ ├── function_parameter_canonicalizer_test.py │ │ ├── function_utils.py │ │ ├── function_utils_test.py │ │ ├── is_in_graph_mode.py │ │ ├── keras_deps.py │ │ ├── kernel_registry.cc │ │ ├── kernel_registry.h │ │ ├── kernel_registry_wrapper.cc │ │ ├── keyword_args.py │ │ ├── keyword_args_test.py │ │ ├── lazy_loader.py │ │ ├── lazy_loader_test.py │ │ ├── lock_util.py │ │ ├── lock_util_test.py │ │ ├── module_wrapper.py │ │ ├── module_wrapper_test.py │ │ ├── nest.cc │ │ ├── nest.h │ │ ├── nest.py │ │ ├── nest_test.py │ │ ├── nest_util.py │ │ ├── nest_wrapper.cc │ │ ├── object_identity.py │ │ ├── object_identity_test.py │ │ ├── port_wrapper.cc │ │ ├── protobuf │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── compare.py │ │ ├── compare_test.proto │ │ └── compare_test.py │ │ ├── py_checkpoint_reader_wrapper.cc │ │ ├── pywrap_xla_ops.pyi │ │ ├── pywrap_xla_ops_test.py │ │ ├── serialization.py │ │ ├── serialization_test.py │ │ ├── stack_trace.cc │ │ ├── stack_trace.h │ │ ├── stat_summarizer_wrapper.cc │ │ ├── tensor_float_32.cc │ │ ├── tf2xla_opset_wrapper.cc │ │ ├── tf_contextlib.py │ │ ├── tf_contextlib_test.py │ │ ├── tf_decorator.py │ │ ├── tf_decorator_export.py │ │ ├── tf_decorator_test.py │ │ ├── tf_export.py │ │ ├── tf_export_test.py │ │ ├── tf_inspect.py │ │ ├── tf_inspect_test.py │ │ ├── tf_should_use.py │ │ ├── tf_should_use_test.py │ │ ├── tf_stack.cc │ │ ├── tf_stack.py │ │ ├── tf_stack_test.py │ │ ├── tfprof_wrapper.cc │ │ ├── traceback_utils.py │ │ ├── traceback_utils_test.py │ │ ├── transform_graph_wrapper.cc │ │ ├── type_annotations.py │ │ ├── type_annotations_test.py │ │ ├── util.cc │ │ ├── util.h │ │ ├── util_wrapper.cc │ │ ├── variable_utils.py │ │ ├── variable_utils_test.py │ │ └── vlog_test.py ├── pytype.default.bzl ├── security │ ├── README.md │ ├── advisory │ │ ├── tfsa-2018-001.md │ │ ├── tfsa-2018-002.md │ │ ├── tfsa-2018-003.md │ │ ├── tfsa-2018-004.md │ │ ├── tfsa-2018-005.md │ │ ├── tfsa-2018-006.md │ │ ├── tfsa-2019-001.md │ │ ├── tfsa-2019-002.md │ │ ├── tfsa-2020-001.md │ │ ├── tfsa-2020-002.md │ │ ├── tfsa-2020-003.md │ │ ├── tfsa-2020-004.md │ │ ├── tfsa-2020-005.md │ │ ├── tfsa-2020-006.md │ │ ├── tfsa-2020-007.md │ │ ├── tfsa-2020-008.md │ │ ├── tfsa-2020-009.md │ │ ├── tfsa-2020-010.md │ │ ├── tfsa-2020-011.md │ │ ├── tfsa-2020-012.md │ │ ├── tfsa-2020-013.md │ │ ├── tfsa-2020-014.md │ │ ├── tfsa-2020-015.md │ │ ├── tfsa-2020-016.md │ │ ├── tfsa-2020-017.md │ │ ├── tfsa-2020-018.md │ │ ├── tfsa-2020-019.md │ │ ├── tfsa-2020-020.md │ │ ├── tfsa-2020-021.md │ │ ├── tfsa-2020-022.md │ │ ├── tfsa-2020-023.md │ │ ├── tfsa-2020-024.md │ │ ├── tfsa-2020-025.md │ │ ├── tfsa-2020-026.md │ │ ├── tfsa-2020-027.md │ │ ├── tfsa-2020-028.md │ │ ├── tfsa-2020-029.md │ │ ├── tfsa-2020-030.md │ │ ├── tfsa-2020-031.md │ │ ├── tfsa-2020-032.md │ │ ├── tfsa-2020-033.md │ │ ├── tfsa-2020-034.md │ │ ├── tfsa-2021-001.md │ │ ├── tfsa-2021-002.md │ │ ├── tfsa-2021-003.md │ │ ├── tfsa-2021-004.md │ │ ├── tfsa-2021-005.md │ │ ├── tfsa-2021-006.md │ │ ├── tfsa-2021-007.md │ │ ├── tfsa-2021-008.md │ │ ├── tfsa-2021-009.md │ │ ├── tfsa-2021-010.md │ │ ├── tfsa-2021-011.md │ │ ├── tfsa-2021-012.md │ │ ├── tfsa-2021-013.md │ │ ├── tfsa-2021-014.md │ │ ├── tfsa-2021-015.md │ │ ├── tfsa-2021-016.md │ │ ├── tfsa-2021-017.md │ │ ├── tfsa-2021-018.md │ │ ├── tfsa-2021-019.md │ │ ├── tfsa-2021-020.md │ │ ├── tfsa-2021-021.md │ │ ├── tfsa-2021-022.md │ │ ├── tfsa-2021-023.md │ │ ├── tfsa-2021-024.md │ │ ├── tfsa-2021-025.md │ │ ├── tfsa-2021-026.md │ │ ├── tfsa-2021-027.md │ │ ├── tfsa-2021-028.md │ │ ├── tfsa-2021-029.md │ │ ├── tfsa-2021-030.md │ │ ├── tfsa-2021-031.md │ │ ├── tfsa-2021-032.md │ │ ├── tfsa-2021-033.md │ │ ├── tfsa-2021-034.md │ │ ├── tfsa-2021-035.md │ │ ├── tfsa-2021-036.md │ │ ├── tfsa-2021-037.md │ │ ├── tfsa-2021-038.md │ │ ├── tfsa-2021-039.md │ │ ├── tfsa-2021-040.md │ │ ├── tfsa-2021-041.md │ │ ├── tfsa-2021-042.md │ │ ├── tfsa-2021-043.md │ │ ├── tfsa-2021-044.md │ │ ├── tfsa-2021-045.md │ │ ├── tfsa-2021-046.md │ │ ├── tfsa-2021-047.md │ │ ├── tfsa-2021-048.md │ │ ├── tfsa-2021-049.md │ │ ├── tfsa-2021-050.md │ │ ├── tfsa-2021-051.md │ │ ├── tfsa-2021-052.md │ │ ├── tfsa-2021-053.md │ │ ├── tfsa-2021-054.md │ │ ├── tfsa-2021-055.md │ │ ├── tfsa-2021-056.md │ │ ├── tfsa-2021-057.md │ │ ├── tfsa-2021-058.md │ │ ├── tfsa-2021-059.md │ │ ├── tfsa-2021-060.md │ │ ├── tfsa-2021-061.md │ │ ├── tfsa-2021-062.md │ │ ├── tfsa-2021-063.md │ │ ├── tfsa-2021-064.md │ │ ├── tfsa-2021-065.md │ │ ├── tfsa-2021-066.md │ │ ├── tfsa-2021-067.md │ │ ├── tfsa-2021-068.md │ │ ├── tfsa-2021-069.md │ │ ├── tfsa-2021-070.md │ │ ├── tfsa-2021-071.md │ │ ├── tfsa-2021-072.md │ │ ├── tfsa-2021-073.md │ │ ├── tfsa-2021-074.md │ │ ├── tfsa-2021-075.md │ │ ├── tfsa-2021-076.md │ │ ├── tfsa-2021-077.md │ │ ├── tfsa-2021-078.md │ │ ├── tfsa-2021-079.md │ │ ├── tfsa-2021-080.md │ │ ├── tfsa-2021-081.md │ │ ├── tfsa-2021-082.md │ │ ├── tfsa-2021-083.md │ │ ├── tfsa-2021-084.md │ │ ├── tfsa-2021-085.md │ │ ├── tfsa-2021-086.md │ │ ├── tfsa-2021-087.md │ │ ├── tfsa-2021-088.md │ │ ├── tfsa-2021-089.md │ │ ├── tfsa-2021-090.md │ │ ├── tfsa-2021-091.md │ │ ├── tfsa-2021-092.md │ │ ├── tfsa-2021-093.md │ │ ├── tfsa-2021-094.md │ │ ├── tfsa-2021-095.md │ │ ├── tfsa-2021-096.md │ │ ├── tfsa-2021-097.md │ │ ├── tfsa-2021-098.md │ │ ├── tfsa-2021-099.md │ │ ├── tfsa-2021-100.md │ │ ├── tfsa-2021-101.md │ │ ├── tfsa-2021-102.md │ │ ├── tfsa-2021-103.md │ │ ├── tfsa-2021-104.md │ │ ├── tfsa-2021-105.md │ │ ├── tfsa-2021-106.md │ │ ├── tfsa-2021-107.md │ │ ├── tfsa-2021-108.md │ │ ├── tfsa-2021-109.md │ │ ├── tfsa-2021-110.md │ │ ├── tfsa-2021-111.md │ │ ├── tfsa-2021-112.md │ │ ├── tfsa-2021-113.md │ │ ├── tfsa-2021-114.md │ │ ├── tfsa-2021-115.md │ │ ├── tfsa-2021-116.md │ │ ├── tfsa-2021-117.md │ │ ├── tfsa-2021-118.md │ │ ├── tfsa-2021-119.md │ │ ├── tfsa-2021-120.md │ │ ├── tfsa-2021-121.md │ │ ├── tfsa-2021-122.md │ │ ├── tfsa-2021-123.md │ │ ├── tfsa-2021-124.md │ │ ├── tfsa-2021-125.md │ │ ├── tfsa-2021-126.md │ │ ├── tfsa-2021-127.md │ │ ├── tfsa-2021-128.md │ │ ├── tfsa-2021-129.md │ │ ├── tfsa-2021-130.md │ │ ├── tfsa-2021-131.md │ │ ├── tfsa-2021-132.md │ │ ├── tfsa-2021-133.md │ │ ├── tfsa-2021-134.md │ │ ├── tfsa-2021-135.md │ │ ├── tfsa-2021-136.md │ │ ├── tfsa-2021-137.md │ │ ├── tfsa-2021-138.md │ │ ├── tfsa-2021-139.md │ │ ├── tfsa-2021-140.md │ │ ├── tfsa-2021-141.md │ │ ├── tfsa-2021-142.md │ │ ├── tfsa-2021-143.md │ │ ├── tfsa-2021-144.md │ │ ├── tfsa-2021-145.md │ │ ├── tfsa-2021-146.md │ │ ├── tfsa-2021-147.md │ │ ├── tfsa-2021-148.md │ │ ├── tfsa-2021-149.md │ │ ├── tfsa-2021-150.md │ │ ├── tfsa-2021-151.md │ │ ├── tfsa-2021-152.md │ │ ├── tfsa-2021-153.md │ │ ├── tfsa-2021-154.md │ │ ├── tfsa-2021-155.md │ │ ├── tfsa-2021-156.md │ │ ├── tfsa-2021-157.md │ │ ├── tfsa-2021-158.md │ │ ├── tfsa-2021-159.md │ │ ├── tfsa-2021-160.md │ │ ├── tfsa-2021-161.md │ │ ├── tfsa-2021-162.md │ │ ├── tfsa-2021-163.md │ │ ├── tfsa-2021-164.md │ │ ├── tfsa-2021-165.md │ │ ├── tfsa-2021-166.md │ │ ├── tfsa-2021-167.md │ │ ├── tfsa-2021-168.md │ │ ├── tfsa-2021-169.md │ │ ├── tfsa-2021-170.md │ │ ├── tfsa-2021-171.md │ │ ├── tfsa-2021-172.md │ │ ├── tfsa-2021-173.md │ │ ├── tfsa-2021-174.md │ │ ├── tfsa-2021-175.md │ │ ├── tfsa-2021-176.md │ │ ├── tfsa-2021-177.md │ │ ├── tfsa-2021-178.md │ │ ├── tfsa-2021-179.md │ │ ├── tfsa-2021-180.md │ │ ├── tfsa-2021-181.md │ │ ├── tfsa-2021-182.md │ │ ├── tfsa-2021-183.md │ │ ├── tfsa-2021-184.md │ │ ├── tfsa-2021-185.md │ │ ├── tfsa-2021-186.md │ │ ├── tfsa-2021-187.md │ │ ├── tfsa-2021-188.md │ │ ├── tfsa-2021-189.md │ │ ├── tfsa-2021-190.md │ │ ├── tfsa-2021-191.md │ │ ├── tfsa-2021-192.md │ │ ├── tfsa-2021-193.md │ │ ├── tfsa-2021-194.md │ │ ├── tfsa-2021-195.md │ │ ├── tfsa-2021-196.md │ │ ├── tfsa-2021-197.md │ │ ├── tfsa-2021-198.md │ │ ├── tfsa-2021-199.md │ │ ├── tfsa-2021-200.md │ │ ├── tfsa-2022-001.md │ │ ├── tfsa-2022-002.md │ │ ├── tfsa-2022-003.md │ │ ├── tfsa-2022-004.md │ │ ├── tfsa-2022-005.md │ │ ├── tfsa-2022-006.md │ │ ├── tfsa-2022-007.md │ │ ├── tfsa-2022-008.md │ │ ├── tfsa-2022-009.md │ │ ├── tfsa-2022-010.md │ │ ├── tfsa-2022-011.md │ │ ├── tfsa-2022-012.md │ │ ├── tfsa-2022-013.md │ │ ├── tfsa-2022-014.md │ │ ├── tfsa-2022-015.md │ │ ├── tfsa-2022-016.md │ │ ├── tfsa-2022-017.md │ │ ├── tfsa-2022-018.md │ │ ├── tfsa-2022-019.md │ │ ├── tfsa-2022-020.md │ │ ├── tfsa-2022-021.md │ │ ├── tfsa-2022-022.md │ │ ├── tfsa-2022-023.md │ │ ├── tfsa-2022-024.md │ │ ├── tfsa-2022-025.md │ │ ├── tfsa-2022-026.md │ │ ├── tfsa-2022-027.md │ │ ├── tfsa-2022-028.md │ │ ├── tfsa-2022-029.md │ │ ├── tfsa-2022-030.md │ │ ├── tfsa-2022-031.md │ │ ├── tfsa-2022-032.md │ │ ├── tfsa-2022-033.md │ │ ├── tfsa-2022-034.md │ │ ├── tfsa-2022-035.md │ │ ├── tfsa-2022-036.md │ │ ├── tfsa-2022-037.md │ │ ├── tfsa-2022-038.md │ │ ├── tfsa-2022-039.md │ │ ├── tfsa-2022-040.md │ │ ├── tfsa-2022-041.md │ │ ├── tfsa-2022-042.md │ │ ├── tfsa-2022-043.md │ │ ├── tfsa-2022-044.md │ │ ├── tfsa-2022-045.md │ │ ├── tfsa-2022-046.md │ │ ├── tfsa-2022-047.md │ │ ├── tfsa-2022-048.md │ │ ├── tfsa-2022-049.md │ │ ├── tfsa-2022-050.md │ │ ├── tfsa-2022-051.md │ │ ├── tfsa-2022-052.md │ │ ├── tfsa-2022-053.md │ │ ├── tfsa-2022-054.md │ │ ├── tfsa-2022-055.md │ │ ├── tfsa-2022-056.md │ │ ├── tfsa-2022-057.md │ │ ├── tfsa-2022-058.md │ │ ├── tfsa-2022-059.md │ │ ├── tfsa-2022-060.md │ │ ├── tfsa-2022-061.md │ │ ├── tfsa-2022-062.md │ │ ├── tfsa-2022-063.md │ │ ├── tfsa-2022-064.md │ │ ├── tfsa-2022-065.md │ │ ├── tfsa-2022-066.md │ │ ├── tfsa-2022-067.md │ │ ├── tfsa-2022-068.md │ │ ├── tfsa-2022-069.md │ │ ├── tfsa-2022-070.md │ │ ├── tfsa-2022-071.md │ │ ├── tfsa-2022-072.md │ │ ├── tfsa-2022-073.md │ │ ├── tfsa-2022-074.md │ │ ├── tfsa-2022-075.md │ │ ├── tfsa-2022-076.md │ │ ├── tfsa-2022-077.md │ │ ├── tfsa-2022-078.md │ │ ├── tfsa-2022-079.md │ │ ├── tfsa-2022-080.md │ │ ├── tfsa-2022-081.md │ │ ├── tfsa-2022-082.md │ │ ├── tfsa-2022-083.md │ │ ├── tfsa-2022-084.md │ │ ├── tfsa-2022-085.md │ │ ├── tfsa-2022-086.md │ │ ├── tfsa-2022-087.md │ │ ├── tfsa-2022-088.md │ │ ├── tfsa-2022-089.md │ │ ├── tfsa-2022-090.md │ │ ├── tfsa-2022-091.md │ │ ├── tfsa-2022-092.md │ │ ├── tfsa-2022-093.md │ │ ├── tfsa-2022-094.md │ │ ├── tfsa-2022-095.md │ │ ├── tfsa-2022-096.md │ │ ├── tfsa-2022-097.md │ │ ├── tfsa-2022-098.md │ │ ├── tfsa-2022-099.md │ │ ├── tfsa-2022-100.md │ │ ├── tfsa-2022-101.md │ │ ├── tfsa-2022-102.md │ │ ├── tfsa-2022-103.md │ │ ├── tfsa-2022-104.md │ │ ├── tfsa-2022-105.md │ │ ├── tfsa-2022-106.md │ │ ├── tfsa-2022-107.md │ │ ├── tfsa-2022-108.md │ │ ├── tfsa-2022-109.md │ │ ├── tfsa-2022-110.md │ │ ├── tfsa-2022-111.md │ │ ├── tfsa-2022-112.md │ │ ├── tfsa-2022-113.md │ │ ├── tfsa-2022-114.md │ │ ├── tfsa-2022-115.md │ │ ├── tfsa-2022-116.md │ │ ├── tfsa-2022-117.md │ │ ├── tfsa-2022-118.md │ │ ├── tfsa-2022-119.md │ │ ├── tfsa-2022-120.md │ │ ├── tfsa-2022-121.md │ │ ├── tfsa-2022-122.md │ │ ├── tfsa-2022-123.md │ │ ├── tfsa-2022-124.md │ │ ├── tfsa-2022-125.md │ │ ├── tfsa-2022-126.md │ │ ├── tfsa-2022-127.md │ │ ├── tfsa-2022-128.md │ │ ├── tfsa-2022-129.md │ │ ├── tfsa-2022-130.md │ │ ├── tfsa-2022-131.md │ │ ├── tfsa-2022-132.md │ │ ├── tfsa-2022-133.md │ │ ├── tfsa-2022-134.md │ │ ├── tfsa-2022-135.md │ │ ├── tfsa-2022-136.md │ │ ├── tfsa-2022-137.md │ │ ├── tfsa-2022-138.md │ │ ├── tfsa-2022-139.md │ │ ├── tfsa-2022-140.md │ │ ├── tfsa-2022-141.md │ │ ├── tfsa-2022-142.md │ │ ├── tfsa-2022-143.md │ │ ├── tfsa-2022-144.md │ │ ├── tfsa-2022-145.md │ │ ├── tfsa-2022-146.md │ │ ├── tfsa-2022-147.md │ │ ├── tfsa-2022-148.md │ │ ├── tfsa-2022-149.md │ │ ├── tfsa-2022-150.md │ │ ├── tfsa-2022-151.md │ │ ├── tfsa-2022-152.md │ │ ├── tfsa-2022-153.md │ │ ├── tfsa-2022-154.md │ │ ├── tfsa-2022-155.md │ │ ├── tfsa-2022-156.md │ │ ├── tfsa-2022-157.md │ │ ├── tfsa-2022-158.md │ │ ├── tfsa-2022-159.md │ │ ├── tfsa-2022-160.md │ │ ├── tfsa-2022-161.md │ │ ├── tfsa-2022-162.md │ │ ├── tfsa-2022-163.md │ │ ├── tfsa-2022-164.md │ │ ├── tfsa-2022-165.md │ │ ├── tfsa-2022-166.md │ │ ├── tfsa-2022-167.md │ │ ├── tfsa-2022-168.md │ │ ├── tfsa-2022-169.md │ │ ├── tfsa-2022-170.md │ │ ├── tfsa-2023-001.md │ │ ├── tfsa-2023-002.md │ │ ├── tfsa-2023-003.md │ │ ├── tfsa-2023-004.md │ │ ├── tfsa-2023-005.md │ │ ├── tfsa-2023-006.md │ │ ├── tfsa-2023-007.md │ │ ├── tfsa-2023-008.md │ │ ├── tfsa-2023-009.md │ │ ├── tfsa-2023-010.md │ │ ├── tfsa-2023-011.md │ │ ├── tfsa-2023-012.md │ │ ├── tfsa-2023-013.md │ │ ├── tfsa-2023-014.md │ │ ├── tfsa-2023-015.md │ │ ├── tfsa-2023-016.md │ │ ├── tfsa-2023-017.md │ │ ├── tfsa-2023-018.md │ │ ├── tfsa-2023-019.md │ │ └── tfsa-2023-020.md │ └── fuzzing │ │ ├── BUILD │ │ ├── abs_fuzz.py │ │ ├── acos_fuzz.py │ │ ├── acosh_fuzz.py │ │ ├── add_fuzz.py │ │ ├── cc │ │ ├── AreAttrValuesEqual_fuzz.cc │ │ ├── BUILD │ │ ├── ParseAttrValue_fuzz.cc │ │ ├── arg_def_case_fuzz.cc │ │ ├── base64_fuzz.cc │ │ ├── bfloat16_fuzz.cc │ │ ├── checkpoint_reader_fuzz.cc │ │ ├── checkpoint_reader_fuzz_input.proto │ │ ├── checkpoint_reader_testdata │ │ │ ├── extent_end_overflow.textproto │ │ │ ├── missing_extent_length.textproto │ │ │ ├── missing_tensor_data.textproto │ │ │ ├── missing_tensor_type.textproto │ │ │ ├── negative_dimension.textproto │ │ │ ├── negative_extent_start.textproto │ │ │ ├── tensor_size_overflow.textproto │ │ │ ├── unsupported_tensor_type.textproto │ │ │ └── valid.textproto │ │ ├── cleanpath_fuzz.cc │ │ ├── consume_leading_digits_fuzz.cc │ │ ├── core │ │ │ ├── framework │ │ │ │ ├── BUILD │ │ │ │ ├── datatype_domains.cc │ │ │ │ ├── datatype_domains.h │ │ │ │ ├── tensor_domains.cc │ │ │ │ ├── tensor_domains.h │ │ │ │ ├── tensor_shape_domains.cc │ │ │ │ └── tensor_shape_domains.h │ │ │ └── function │ │ │ │ ├── BUILD │ │ │ │ └── runtime_client_fuzz.cc │ │ ├── end_to_end_fuzz.cc │ │ ├── fuzz_domains.h │ │ ├── fuzz_session.h │ │ ├── joinpath_fuzz.cc │ │ ├── ops │ │ │ ├── BUILD │ │ │ ├── add_fuzz.cc │ │ │ ├── bincount_fuzz.cc │ │ │ ├── concat_fuzz.cc │ │ │ ├── general_ops_fuzz.cc │ │ │ ├── identity_fuzz.cc │ │ │ ├── matmul_fuzz.cc │ │ │ ├── string_ops_fuzz.cc │ │ │ └── string_to_number_fuzz.cc │ │ ├── parseURI_fuzz.cc │ │ ├── status_fuzz.cc │ │ ├── status_group_fuzz.cc │ │ ├── string_replace_fuzz.cc │ │ ├── stringprintf_fuzz.cc │ │ ├── text_literal_reader_fuzz.cc │ │ └── tstring_fuzz.cc │ │ ├── constant_fuzz.py │ │ ├── dataFormatVecPermute_fuzz.py │ │ ├── immutableConst_fuzz.py │ │ ├── py │ │ ├── BUILD │ │ └── annotation_types.py │ │ ├── python_fuzzing.py │ │ ├── raggedCountSparseOutput_fuzz.py │ │ ├── sparseCountSparseOutput_fuzz.py │ │ ├── tf2migration_fuzz.py │ │ └── tf_fuzzing.bzl ├── strict.default.bzl ├── tensorflow.bzl ├── tensorflow.default.bzl ├── tf_exported_symbols.lds ├── tf_framework_version_script.lds ├── tf_private_symbols.lds ├── tf_version_script.lds ├── tools │ ├── __init__.py │ ├── android │ │ ├── README.md │ │ ├── inference_interface │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── asset_manager_filesystem.cc │ │ │ ├── asset_manager_filesystem.h │ │ │ ├── java │ │ │ │ └── org │ │ │ │ │ └── tensorflow │ │ │ │ │ └── contrib │ │ │ │ │ └── android │ │ │ │ │ ├── RunStats.java │ │ │ │ │ └── TensorFlowInferenceInterface.java │ │ │ └── jni │ │ │ │ ├── run_stats_jni.cc │ │ │ │ ├── run_stats_jni.h │ │ │ │ └── version_script.lds │ │ └── test │ │ │ ├── .gitignore │ │ │ ├── AndroidManifest.xml │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── assets │ │ │ └── BUILD │ │ │ ├── bin │ │ │ └── AndroidManifest.xml │ │ │ ├── build.gradle │ │ │ ├── download-models.gradle │ │ │ ├── gradlew │ │ │ ├── gradlew.bat │ │ │ ├── jni │ │ │ ├── __init__.py │ │ │ ├── imageutils_jni.cc │ │ │ ├── object_tracking │ │ │ │ ├── config.h │ │ │ │ ├── flow_cache.h │ │ │ │ ├── frame_pair.cc │ │ │ │ ├── frame_pair.h │ │ │ │ ├── geom.h │ │ │ │ ├── gl_utils.h │ │ │ │ ├── image-inl.h │ │ │ │ ├── image.h │ │ │ │ ├── image_data.h │ │ │ │ ├── image_neon.cc │ │ │ │ ├── image_utils.h │ │ │ │ ├── integral_image.h │ │ │ │ ├── jni_utils.h │ │ │ │ ├── keypoint.h │ │ │ │ ├── keypoint_detector.cc │ │ │ │ ├── keypoint_detector.h │ │ │ │ ├── logging.cc │ │ │ │ ├── logging.h │ │ │ │ ├── object_detector.cc │ │ │ │ ├── object_detector.h │ │ │ │ ├── object_model.h │ │ │ │ ├── object_tracker.cc │ │ │ │ ├── object_tracker.h │ │ │ │ ├── object_tracker_jni.cc │ │ │ │ ├── optical_flow.cc │ │ │ │ ├── optical_flow.h │ │ │ │ ├── sprite.h │ │ │ │ ├── time_log.cc │ │ │ │ ├── time_log.h │ │ │ │ ├── tracked_object.cc │ │ │ │ ├── tracked_object.h │ │ │ │ ├── utils.h │ │ │ │ └── utils_neon.cc │ │ │ ├── rgb2yuv.cc │ │ │ ├── rgb2yuv.h │ │ │ ├── version_script.lds │ │ │ ├── yuv2rgb.cc │ │ │ └── yuv2rgb.h │ │ │ ├── res │ │ │ ├── animator │ │ │ │ └── color_animation.xml │ │ │ ├── drawable-hdpi │ │ │ │ ├── ic_action_info.png │ │ │ │ ├── ic_launcher.png │ │ │ │ └── tile.9.png │ │ │ ├── drawable-mdpi │ │ │ │ ├── ic_action_info.png │ │ │ │ └── ic_launcher.png │ │ │ ├── drawable-xhdpi │ │ │ │ ├── ic_action_info.png │ │ │ │ └── ic_launcher.png │ │ │ ├── drawable-xxhdpi │ │ │ │ ├── ic_action_info.png │ │ │ │ └── ic_launcher.png │ │ │ ├── drawable │ │ │ │ └── border.xml │ │ │ ├── layout │ │ │ │ ├── activity_camera.xml │ │ │ │ ├── activity_speech.xml │ │ │ │ ├── camera_connection_fragment.xml │ │ │ │ ├── camera_connection_fragment_stylize.xml │ │ │ │ ├── camera_connection_fragment_tracking.xml │ │ │ │ └── list_text_item.xml │ │ │ ├── values-sw600dp │ │ │ │ ├── template-dimens.xml │ │ │ │ └── template-styles.xml │ │ │ ├── values-v11 │ │ │ │ ├── styles.xml │ │ │ │ └── template-styles.xml │ │ │ ├── values-v14 │ │ │ │ └── styles.xml │ │ │ ├── values-v21 │ │ │ │ ├── base-colors.xml │ │ │ │ └── base-template-styles.xml │ │ │ └── values │ │ │ │ ├── attrs.xml │ │ │ │ ├── base-strings.xml │ │ │ │ ├── colors.xml │ │ │ │ ├── strings.xml │ │ │ │ ├── styles.xml │ │ │ │ ├── template-dimens.xml │ │ │ │ └── template-styles.xml │ │ │ ├── sample_images │ │ │ ├── classify1.jpg │ │ │ ├── detect1.jpg │ │ │ └── stylize1.jpg │ │ │ ├── settings.gradle │ │ │ └── src │ │ │ └── org │ │ │ └── tensorflow │ │ │ └── demo │ │ │ ├── AutoFitTextureView.java │ │ │ ├── CameraActivity.java │ │ │ ├── CameraConnectionFragment.java │ │ │ ├── Classifier.java │ │ │ ├── ClassifierActivity.java │ │ │ ├── DetectorActivity.java │ │ │ ├── LegacyCameraConnectionFragment.java │ │ │ ├── OverlayView.java │ │ │ ├── RecognitionScoreView.java │ │ │ ├── RecognizeCommands.java │ │ │ ├── ResultsView.java │ │ │ ├── SpeechActivity.java │ │ │ ├── StylizeActivity.java │ │ │ ├── TensorFlowImageClassifier.java │ │ │ ├── TensorFlowMultiBoxDetector.java │ │ │ ├── TensorFlowObjectDetectionAPIModel.java │ │ │ ├── TensorFlowYoloDetector.java │ │ │ ├── env │ │ │ ├── BorderedText.java │ │ │ ├── ImageUtils.java │ │ │ ├── Logger.java │ │ │ ├── Size.java │ │ │ └── SplitTimer.java │ │ │ └── tracking │ │ │ ├── MultiBoxTracker.java │ │ │ └── ObjectTracker.java │ ├── api │ │ ├── golden │ │ │ ├── BUILD │ │ │ ├── v1 │ │ │ │ ├── tensorflow.-aggregation-method.pbtxt │ │ │ │ ├── tensorflow.-attr-value.-list-value.pbtxt │ │ │ │ ├── tensorflow.-attr-value.pbtxt │ │ │ │ ├── tensorflow.-conditional-accumulator-base.pbtxt │ │ │ │ ├── tensorflow.-conditional-accumulator.pbtxt │ │ │ │ ├── tensorflow.-config-proto.-device-count-entry.pbtxt │ │ │ │ ├── tensorflow.-config-proto.-experimental.pbtxt │ │ │ │ ├── tensorflow.-config-proto.pbtxt │ │ │ │ ├── tensorflow.-critical-section.pbtxt │ │ │ │ ├── tensorflow.-d-type.pbtxt │ │ │ │ ├── tensorflow.-device-spec.pbtxt │ │ │ │ ├── tensorflow.-dimension.pbtxt │ │ │ │ ├── tensorflow.-event.pbtxt │ │ │ │ ├── tensorflow.-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.-fixed-len-feature.pbtxt │ │ │ │ ├── tensorflow.-fixed-len-sequence-feature.pbtxt │ │ │ │ ├── tensorflow.-fixed-length-record-reader.pbtxt │ │ │ │ ├── tensorflow.-g-p-u-options.pbtxt │ │ │ │ ├── tensorflow.-gradient-tape.pbtxt │ │ │ │ ├── tensorflow.-graph-def.pbtxt │ │ │ │ ├── tensorflow.-graph-keys.pbtxt │ │ │ │ ├── tensorflow.-graph-options.pbtxt │ │ │ │ ├── tensorflow.-graph.pbtxt │ │ │ │ ├── tensorflow.-histogram-proto.pbtxt │ │ │ │ ├── tensorflow.-identity-reader.pbtxt │ │ │ │ ├── tensorflow.-indexed-slices-spec.pbtxt │ │ │ │ ├── tensorflow.-indexed-slices.pbtxt │ │ │ │ ├── tensorflow.-interactive-session.pbtxt │ │ │ │ ├── tensorflow.-l-m-d-b-reader.pbtxt │ │ │ │ ├── tensorflow.-log-message.pbtxt │ │ │ │ ├── tensorflow.-meta-graph-def.-collection-def-entry.pbtxt │ │ │ │ ├── tensorflow.-meta-graph-def.-meta-info-def.-function-aliases-entry.pbtxt │ │ │ │ ├── tensorflow.-meta-graph-def.-meta-info-def.pbtxt │ │ │ │ ├── tensorflow.-meta-graph-def.-signature-def-entry.pbtxt │ │ │ │ ├── tensorflow.-meta-graph-def.pbtxt │ │ │ │ ├── tensorflow.-module.pbtxt │ │ │ │ ├── tensorflow.-name-attr-list.-attr-entry.pbtxt │ │ │ │ ├── tensorflow.-name-attr-list.pbtxt │ │ │ │ ├── tensorflow.-node-def.-attr-entry.pbtxt │ │ │ │ ├── tensorflow.-node-def.-experimental-debug-info.pbtxt │ │ │ │ ├── tensorflow.-node-def.pbtxt │ │ │ │ ├── tensorflow.-op-error.pbtxt │ │ │ │ ├── tensorflow.-operation.pbtxt │ │ │ │ ├── tensorflow.-optimizer-options.pbtxt │ │ │ │ ├── tensorflow.-optional-spec.pbtxt │ │ │ │ ├── tensorflow.-padding-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.-priority-queue.pbtxt │ │ │ │ ├── tensorflow.-queue-base.pbtxt │ │ │ │ ├── tensorflow.-ragged-tensor-spec.pbtxt │ │ │ │ ├── tensorflow.-ragged-tensor.pbtxt │ │ │ │ ├── tensorflow.-random-shuffle-queue.pbtxt │ │ │ │ ├── tensorflow.-reader-base.pbtxt │ │ │ │ ├── tensorflow.-register-gradient.pbtxt │ │ │ │ ├── tensorflow.-run-metadata.-function-graphs.pbtxt │ │ │ │ ├── tensorflow.-run-metadata.pbtxt │ │ │ │ ├── tensorflow.-run-options.-experimental.-run-handler-pool-options.pbtxt │ │ │ │ ├── tensorflow.-run-options.-experimental.pbtxt │ │ │ │ ├── tensorflow.-run-options.pbtxt │ │ │ │ ├── tensorflow.-session-log.pbtxt │ │ │ │ ├── tensorflow.-session.pbtxt │ │ │ │ ├── tensorflow.-sparse-conditional-accumulator.pbtxt │ │ │ │ ├── tensorflow.-sparse-feature.pbtxt │ │ │ │ ├── tensorflow.-sparse-tensor-spec.pbtxt │ │ │ │ ├── tensorflow.-sparse-tensor-value.pbtxt │ │ │ │ ├── tensorflow.-sparse-tensor.pbtxt │ │ │ │ ├── tensorflow.-summary-metadata.-plugin-data.pbtxt │ │ │ │ ├── tensorflow.-summary-metadata.pbtxt │ │ │ │ ├── tensorflow.-summary.-audio.pbtxt │ │ │ │ ├── tensorflow.-summary.-image.pbtxt │ │ │ │ ├── tensorflow.-summary.-value.pbtxt │ │ │ │ ├── tensorflow.-summary.pbtxt │ │ │ │ ├── tensorflow.-t-f-record-reader.pbtxt │ │ │ │ ├── tensorflow.-tensor-array-spec.pbtxt │ │ │ │ ├── tensorflow.-tensor-array.pbtxt │ │ │ │ ├── tensorflow.-tensor-info.-composite-tensor.pbtxt │ │ │ │ ├── tensorflow.-tensor-info.-coo-sparse.pbtxt │ │ │ │ ├── tensorflow.-tensor-info.pbtxt │ │ │ │ ├── tensorflow.-tensor-shape.pbtxt │ │ │ │ ├── tensorflow.-tensor-spec.pbtxt │ │ │ │ ├── tensorflow.-tensor.pbtxt │ │ │ │ ├── tensorflow.-text-line-reader.pbtxt │ │ │ │ ├── tensorflow.-type-spec.pbtxt │ │ │ │ ├── tensorflow.-unconnected-gradients.pbtxt │ │ │ │ ├── tensorflow.-var-len-feature.pbtxt │ │ │ │ ├── tensorflow.-variable-aggregation.pbtxt │ │ │ │ ├── tensorflow.-variable-scope.pbtxt │ │ │ │ ├── tensorflow.-variable-synchronization.pbtxt │ │ │ │ ├── tensorflow.-variable.-save-slice-info.pbtxt │ │ │ │ ├── tensorflow.-variable.pbtxt │ │ │ │ ├── tensorflow.-whole-file-reader.pbtxt │ │ │ │ ├── tensorflow.__internal__.-symbolic-tensor.pbtxt │ │ │ │ ├── tensorflow.__internal__.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.data.-dataset.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.data.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.pbtxt │ │ │ │ ├── tensorflow.app.pbtxt │ │ │ │ ├── tensorflow.audio.pbtxt │ │ │ │ ├── tensorflow.autograph.experimental.-feature.pbtxt │ │ │ │ ├── tensorflow.autograph.experimental.pbtxt │ │ │ │ ├── tensorflow.autograph.pbtxt │ │ │ │ ├── tensorflow.bitwise.pbtxt │ │ │ │ ├── tensorflow.compat.pbtxt │ │ │ │ ├── tensorflow.config.-logical-device-configuration.pbtxt │ │ │ │ ├── tensorflow.config.-logical-device.pbtxt │ │ │ │ ├── tensorflow.config.-physical-device.pbtxt │ │ │ │ ├── tensorflow.config.experimental.-cluster-device-filters.pbtxt │ │ │ │ ├── tensorflow.config.experimental.-virtual-device-configuration.pbtxt │ │ │ │ ├── tensorflow.config.experimental.pbtxt │ │ │ │ ├── tensorflow.config.optimizer.pbtxt │ │ │ │ ├── tensorflow.config.pbtxt │ │ │ │ ├── tensorflow.config.threading.pbtxt │ │ │ │ ├── tensorflow.constant_initializer.pbtxt │ │ │ │ ├── tensorflow.data.-dataset-spec.pbtxt │ │ │ │ ├── tensorflow.data.-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-fixed-length-record-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-iterator.pbtxt │ │ │ │ ├── tensorflow.data.-numpy-iterator.pbtxt │ │ │ │ ├── tensorflow.data.-options.pbtxt │ │ │ │ ├── tensorflow.data.-t-f-record-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-text-line-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-threading-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-auto-shard-policy.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-autotune-algorithm.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-autotune-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-checkpoint-input-pipeline-hook.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-csv-dataset.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-dataset-initializer.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-dataset-structure.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-distribute-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-external-state-policy.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-optimization-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-optional-structure.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-optional.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-random-dataset.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-reducer.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-sql-dataset.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-structure.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-t-f-record-writer.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-threading-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-cross-trainer-cache.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-dispatcher-config.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-sharding-policy.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-worker-config.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.pbtxt │ │ │ │ ├── tensorflow.data.pbtxt │ │ │ │ ├── tensorflow.debugging.experimental.pbtxt │ │ │ │ ├── tensorflow.debugging.pbtxt │ │ │ │ ├── tensorflow.distribute.-cross-device-ops.pbtxt │ │ │ │ ├── tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt │ │ │ │ ├── tensorflow.distribute.-input-context.pbtxt │ │ │ │ ├── tensorflow.distribute.-input-replication-mode.pbtxt │ │ │ │ ├── tensorflow.distribute.-mirrored-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-nccl-all-reduce.pbtxt │ │ │ │ ├── tensorflow.distribute.-one-device-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-reduce-op.pbtxt │ │ │ │ ├── tensorflow.distribute.-reduction-to-one-device.pbtxt │ │ │ │ ├── tensorflow.distribute.-replica-context.pbtxt │ │ │ │ ├── tensorflow.distribute.-run-options.pbtxt │ │ │ │ ├── tensorflow.distribute.-server.pbtxt │ │ │ │ ├── tensorflow.distribute.-strategy-extended.pbtxt │ │ │ │ ├── tensorflow.distribute.-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-central-storage-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-collective-communication.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-collective-hints.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-communication-implementation.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-communication-options.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.pbtxt │ │ │ │ ├── tensorflow.distribute.pbtxt │ │ │ │ ├── tensorflow.distributions.-bernoulli.pbtxt │ │ │ │ ├── tensorflow.distributions.-beta.pbtxt │ │ │ │ ├── tensorflow.distributions.-categorical.pbtxt │ │ │ │ ├── tensorflow.distributions.-dirichlet-multinomial.pbtxt │ │ │ │ ├── tensorflow.distributions.-dirichlet.pbtxt │ │ │ │ ├── tensorflow.distributions.-distribution.pbtxt │ │ │ │ ├── tensorflow.distributions.-exponential.pbtxt │ │ │ │ ├── tensorflow.distributions.-gamma.pbtxt │ │ │ │ ├── tensorflow.distributions.-laplace.pbtxt │ │ │ │ ├── tensorflow.distributions.-multinomial.pbtxt │ │ │ │ ├── tensorflow.distributions.-normal.pbtxt │ │ │ │ ├── tensorflow.distributions.-register-k-l.pbtxt │ │ │ │ ├── tensorflow.distributions.-reparameterization-type.pbtxt │ │ │ │ ├── tensorflow.distributions.-student-t.pbtxt │ │ │ │ ├── tensorflow.distributions.-uniform.pbtxt │ │ │ │ ├── tensorflow.distributions.pbtxt │ │ │ │ ├── tensorflow.dtypes.-d-type.pbtxt │ │ │ │ ├── tensorflow.dtypes.experimental.pbtxt │ │ │ │ ├── tensorflow.dtypes.pbtxt │ │ │ │ ├── tensorflow.errors.-aborted-error.pbtxt │ │ │ │ ├── tensorflow.errors.-already-exists-error.pbtxt │ │ │ │ ├── tensorflow.errors.-cancelled-error.pbtxt │ │ │ │ ├── tensorflow.errors.-data-loss-error.pbtxt │ │ │ │ ├── tensorflow.errors.-deadline-exceeded-error.pbtxt │ │ │ │ ├── tensorflow.errors.-failed-precondition-error.pbtxt │ │ │ │ ├── tensorflow.errors.-internal-error.pbtxt │ │ │ │ ├── tensorflow.errors.-invalid-argument-error.pbtxt │ │ │ │ ├── tensorflow.errors.-not-found-error.pbtxt │ │ │ │ ├── tensorflow.errors.-op-error.pbtxt │ │ │ │ ├── tensorflow.errors.-out-of-range-error.pbtxt │ │ │ │ ├── tensorflow.errors.-permission-denied-error.pbtxt │ │ │ │ ├── tensorflow.errors.-resource-exhausted-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unauthenticated-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unavailable-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unimplemented-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unknown-error.pbtxt │ │ │ │ ├── tensorflow.errors.pbtxt │ │ │ │ ├── tensorflow.errors.raise_exception_on_not_ok_status.pbtxt │ │ │ │ ├── tensorflow.estimator.-baseline-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-baseline-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-baseline-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-best-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-binary-class-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-checkpoint-saver-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-checkpoint-saver-listener.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-linear-combined-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-estimator-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-eval-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-feed-fn-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-final-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-final-ops-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-global-step-waiter-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-latest-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-linear-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-linear-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-linear-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-logging-tensor-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-logistic-regression-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-mode-keys.pbtxt │ │ │ │ ├── tensorflow.estimator.-multi-class-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-multi-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-multi-label-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-nan-loss-during-training-error.pbtxt │ │ │ │ ├── tensorflow.estimator.-nan-tensor-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-poisson-regression-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-profiler-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-regression-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-run-config.pbtxt │ │ │ │ ├── tensorflow.estimator.-second-or-step-timer.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-args.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-context.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-values.pbtxt │ │ │ │ ├── tensorflow.estimator.-step-counter-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-stop-at-step-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-summary-saver-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-train-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.-vocab-info.pbtxt │ │ │ │ ├── tensorflow.estimator.-warm-start-settings.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-in-memory-evaluator-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-k-means.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-linear-s-d-c-a.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-classification-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-classification-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-eval-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-eval-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-export-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-export-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-predict-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-predict-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-regression-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-regression-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-serving-input-receiver.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-tensor-serving-input-receiver.pbtxt │ │ │ │ ├── tensorflow.estimator.export.pbtxt │ │ │ │ ├── tensorflow.estimator.inputs.pbtxt │ │ │ │ ├── tensorflow.estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.-input-pipeline-config.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.-run-config.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.-t-p-u-config.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.-t-p-u-estimator-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.-t-p-u-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.experimental.-embedding-config-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.experimental.pbtxt │ │ │ │ ├── tensorflow.estimator.tpu.pbtxt │ │ │ │ ├── tensorflow.experimental.-batchable-extension-type.pbtxt │ │ │ │ ├── tensorflow.experimental.-dynamic-ragged-shape.-spec.pbtxt │ │ │ │ ├── tensorflow.experimental.-dynamic-ragged-shape.pbtxt │ │ │ │ ├── tensorflow.experimental.-extension-type-batch-encoder.pbtxt │ │ │ │ ├── tensorflow.experimental.-extension-type-spec.pbtxt │ │ │ │ ├── tensorflow.experimental.-extension-type.pbtxt │ │ │ │ ├── tensorflow.experimental.-optional.pbtxt │ │ │ │ ├── tensorflow.experimental.-row-partition.pbtxt │ │ │ │ ├── tensorflow.experimental.-structured-tensor.-spec.pbtxt │ │ │ │ ├── tensorflow.experimental.-structured-tensor.pbtxt │ │ │ │ ├── tensorflow.experimental.extension_type.pbtxt │ │ │ │ ├── tensorflow.experimental.pbtxt │ │ │ │ ├── tensorflow.feature_column.pbtxt │ │ │ │ ├── tensorflow.gfile.-fast-g-file.pbtxt │ │ │ │ ├── tensorflow.gfile.-g-file.pbtxt │ │ │ │ ├── tensorflow.gfile.-open.pbtxt │ │ │ │ ├── tensorflow.gfile.pbtxt │ │ │ │ ├── tensorflow.glorot_normal_initializer.pbtxt │ │ │ │ ├── tensorflow.glorot_uniform_initializer.pbtxt │ │ │ │ ├── tensorflow.graph_util.pbtxt │ │ │ │ ├── tensorflow.image.-resize-method.pbtxt │ │ │ │ ├── tensorflow.image.pbtxt │ │ │ │ ├── tensorflow.initializers.constant.pbtxt │ │ │ │ ├── tensorflow.initializers.glorot_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.glorot_uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.identity.pbtxt │ │ │ │ ├── tensorflow.initializers.ones.pbtxt │ │ │ │ ├── tensorflow.initializers.orthogonal.pbtxt │ │ │ │ ├── tensorflow.initializers.pbtxt │ │ │ │ ├── tensorflow.initializers.random_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.random_uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.truncated_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.uniform_unit_scaling.pbtxt │ │ │ │ ├── tensorflow.initializers.variance_scaling.pbtxt │ │ │ │ ├── tensorflow.initializers.zeros.pbtxt │ │ │ │ ├── tensorflow.io.-fixed-len-feature.pbtxt │ │ │ │ ├── tensorflow.io.-fixed-len-sequence-feature.pbtxt │ │ │ │ ├── tensorflow.io.-padding-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.io.-priority-queue.pbtxt │ │ │ │ ├── tensorflow.io.-queue-base.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-lengths.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-limits.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-splits.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-starts.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-uniform-row-length.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-value-row-ids.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.pbtxt │ │ │ │ ├── tensorflow.io.-random-shuffle-queue.pbtxt │ │ │ │ ├── tensorflow.io.-sparse-feature.pbtxt │ │ │ │ ├── tensorflow.io.-t-f-record-compression-type.pbtxt │ │ │ │ ├── tensorflow.io.-t-f-record-options.pbtxt │ │ │ │ ├── tensorflow.io.-t-f-record-writer.pbtxt │ │ │ │ ├── tensorflow.io.-var-len-feature.pbtxt │ │ │ │ ├── tensorflow.io.gfile.-g-file.pbtxt │ │ │ │ ├── tensorflow.io.gfile.pbtxt │ │ │ │ ├── tensorflow.io.pbtxt │ │ │ │ ├── tensorflow.layers.-average-pooling1-d.pbtxt │ │ │ │ ├── tensorflow.layers.-average-pooling2-d.pbtxt │ │ │ │ ├── tensorflow.layers.-average-pooling3-d.pbtxt │ │ │ │ ├── tensorflow.layers.-batch-normalization.pbtxt │ │ │ │ ├── tensorflow.layers.-conv1-d.pbtxt │ │ │ │ ├── tensorflow.layers.-conv2-d-transpose.pbtxt │ │ │ │ ├── tensorflow.layers.-conv2-d.pbtxt │ │ │ │ ├── tensorflow.layers.-conv3-d-transpose.pbtxt │ │ │ │ ├── tensorflow.layers.-conv3-d.pbtxt │ │ │ │ ├── tensorflow.layers.-dense.pbtxt │ │ │ │ ├── tensorflow.layers.-dropout.pbtxt │ │ │ │ ├── tensorflow.layers.-flatten.pbtxt │ │ │ │ ├── tensorflow.layers.-input-spec.pbtxt │ │ │ │ ├── tensorflow.layers.-layer.pbtxt │ │ │ │ ├── tensorflow.layers.-max-pooling1-d.pbtxt │ │ │ │ ├── tensorflow.layers.-max-pooling2-d.pbtxt │ │ │ │ ├── tensorflow.layers.-max-pooling3-d.pbtxt │ │ │ │ ├── tensorflow.layers.-separable-conv1-d.pbtxt │ │ │ │ ├── tensorflow.layers.-separable-conv2-d.pbtxt │ │ │ │ ├── tensorflow.layers.experimental.pbtxt │ │ │ │ ├── tensorflow.layers.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-adjoint.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-block-diag.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-block-lower-triangular.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-circulant.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-circulant2-d.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-circulant3-d.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-composition.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-diag.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-full-matrix.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-householder.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-identity.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-inversion.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-kronecker.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-low-rank-update.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-lower-triangular.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-permutation.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-scaled-identity.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-toeplitz.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-tridiag.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-zeros.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator.pbtxt │ │ │ │ ├── tensorflow.linalg.experimental.pbtxt │ │ │ │ ├── tensorflow.linalg.pbtxt │ │ │ │ ├── tensorflow.lite.-interpreter.pbtxt │ │ │ │ ├── tensorflow.lite.-op-hint.-op-hint-argument-tracker.pbtxt │ │ │ │ ├── tensorflow.lite.-op-hint.pbtxt │ │ │ │ ├── tensorflow.lite.-ops-set.pbtxt │ │ │ │ ├── tensorflow.lite.-optimize.pbtxt │ │ │ │ ├── tensorflow.lite.-representative-dataset.pbtxt │ │ │ │ ├── tensorflow.lite.-t-f-lite-converter.pbtxt │ │ │ │ ├── tensorflow.lite.-target-spec.pbtxt │ │ │ │ ├── tensorflow.lite.-toco-converter.pbtxt │ │ │ │ ├── tensorflow.lite.constants.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-analyzer.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-op-resolver-type.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-quantization-debug-options.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-quantization-debugger.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.authoring.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.pbtxt │ │ │ │ ├── tensorflow.lite.pbtxt │ │ │ │ ├── tensorflow.logging.pbtxt │ │ │ │ ├── tensorflow.lookup.-key-value-tensor-initializer.pbtxt │ │ │ │ ├── tensorflow.lookup.-static-hash-table.pbtxt │ │ │ │ ├── tensorflow.lookup.-static-vocabulary-table.pbtxt │ │ │ │ ├── tensorflow.lookup.-text-file-index.pbtxt │ │ │ │ ├── tensorflow.lookup.-text-file-initializer.pbtxt │ │ │ │ ├── tensorflow.lookup.experimental.-dense-hash-table.pbtxt │ │ │ │ ├── tensorflow.lookup.experimental.-mutable-hash-table.pbtxt │ │ │ │ ├── tensorflow.lookup.experimental.pbtxt │ │ │ │ ├── tensorflow.lookup.pbtxt │ │ │ │ ├── tensorflow.losses.-reduction.pbtxt │ │ │ │ ├── tensorflow.losses.pbtxt │ │ │ │ ├── tensorflow.manip.pbtxt │ │ │ │ ├── tensorflow.math.pbtxt │ │ │ │ ├── tensorflow.math.special.pbtxt │ │ │ │ ├── tensorflow.metrics.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.-dynamic-loss-scale.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.-fixed-loss-scale.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.-loss-scale.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.-mixed-precision-loss-scale-optimizer.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.experimental.-dynamic-loss-scale.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.experimental.-fixed-loss-scale.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.experimental.-loss-scale.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.experimental.pbtxt │ │ │ │ ├── tensorflow.mixed_precision.pbtxt │ │ │ │ ├── tensorflow.mlir.experimental.pbtxt │ │ │ │ ├── tensorflow.mlir.pbtxt │ │ │ │ ├── tensorflow.name_scope.pbtxt │ │ │ │ ├── tensorflow.nest.pbtxt │ │ │ │ ├── tensorflow.nn.experimental.pbtxt │ │ │ │ ├── tensorflow.nn.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-basic-l-s-t-m-cell.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-basic-r-n-n-cell.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-device-wrapper.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-dropout-wrapper.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-g-r-u-cell.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-l-s-t-m-state-tuple.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-multi-r-n-n-cell.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-r-n-n-cell.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.-residual-wrapper.pbtxt │ │ │ │ ├── tensorflow.nn.rnn_cell.pbtxt │ │ │ │ ├── tensorflow.ones_initializer.pbtxt │ │ │ │ ├── tensorflow.orthogonal_initializer.pbtxt │ │ │ │ ├── tensorflow.pbtxt │ │ │ │ ├── tensorflow.profiler.-advice-proto.-checker.pbtxt │ │ │ │ ├── tensorflow.profiler.-advice-proto.-checkers-entry.pbtxt │ │ │ │ ├── tensorflow.profiler.-advice-proto.pbtxt │ │ │ │ ├── tensorflow.profiler.-graph-node-proto.-input-shapes-entry.pbtxt │ │ │ │ ├── tensorflow.profiler.-graph-node-proto.pbtxt │ │ │ │ ├── tensorflow.profiler.-multi-graph-node-proto.pbtxt │ │ │ │ ├── tensorflow.profiler.-op-log-proto.-id-to-string-entry.pbtxt │ │ │ │ ├── tensorflow.profiler.-op-log-proto.pbtxt │ │ │ │ ├── tensorflow.profiler.-profile-option-builder.pbtxt │ │ │ │ ├── tensorflow.profiler.-profiler.pbtxt │ │ │ │ ├── tensorflow.profiler.pbtxt │ │ │ │ ├── tensorflow.python_io.-t-f-record-compression-type.pbtxt │ │ │ │ ├── tensorflow.python_io.-t-f-record-options.pbtxt │ │ │ │ ├── tensorflow.python_io.-t-f-record-writer.pbtxt │ │ │ │ ├── tensorflow.python_io.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-component-spec.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-method.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-options.-representative-datasets-entry.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-options.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-tf-record-representative-dataset-saver.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-unit-wise-quantization-spec.-quantization-unit.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-unit-wise-quantization-spec.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.pbtxt │ │ │ │ ├── tensorflow.quantization.pbtxt │ │ │ │ ├── tensorflow.queue.-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.queue.-padding-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.queue.-priority-queue.pbtxt │ │ │ │ ├── tensorflow.queue.-queue-base.pbtxt │ │ │ │ ├── tensorflow.queue.-random-shuffle-queue.pbtxt │ │ │ │ ├── tensorflow.queue.pbtxt │ │ │ │ ├── tensorflow.ragged.-ragged-tensor-value.pbtxt │ │ │ │ ├── tensorflow.ragged.pbtxt │ │ │ │ ├── tensorflow.random.-algorithm.pbtxt │ │ │ │ ├── tensorflow.random.-generator.pbtxt │ │ │ │ ├── tensorflow.random.experimental.-algorithm.pbtxt │ │ │ │ ├── tensorflow.random.experimental.-generator.pbtxt │ │ │ │ ├── tensorflow.random.experimental.pbtxt │ │ │ │ ├── tensorflow.random.pbtxt │ │ │ │ ├── tensorflow.random_normal_initializer.pbtxt │ │ │ │ ├── tensorflow.random_uniform_initializer.pbtxt │ │ │ │ ├── tensorflow.raw_ops.pbtxt │ │ │ │ ├── tensorflow.resource_loader.pbtxt │ │ │ │ ├── tensorflow.saved_model.-asset.pbtxt │ │ │ │ ├── tensorflow.saved_model.-builder.pbtxt │ │ │ │ ├── tensorflow.saved_model.-save-options.pbtxt │ │ │ │ ├── tensorflow.saved_model.builder.-saved-model-builder.pbtxt │ │ │ │ ├── tensorflow.saved_model.builder.pbtxt │ │ │ │ ├── tensorflow.saved_model.constants.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.-trackable-resource.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.-variable-policy.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.pbtxt │ │ │ │ ├── tensorflow.saved_model.loader.pbtxt │ │ │ │ ├── tensorflow.saved_model.main_op.pbtxt │ │ │ │ ├── tensorflow.saved_model.pbtxt │ │ │ │ ├── tensorflow.saved_model.signature_constants.pbtxt │ │ │ │ ├── tensorflow.saved_model.signature_def_utils.-method-name-updater.pbtxt │ │ │ │ ├── tensorflow.saved_model.signature_def_utils.pbtxt │ │ │ │ ├── tensorflow.saved_model.tag_constants.pbtxt │ │ │ │ ├── tensorflow.saved_model.utils.pbtxt │ │ │ │ ├── tensorflow.sets.pbtxt │ │ │ │ ├── tensorflow.signal.pbtxt │ │ │ │ ├── tensorflow.sparse.-sparse-conditional-accumulator.pbtxt │ │ │ │ ├── tensorflow.sparse.-sparse-tensor.pbtxt │ │ │ │ ├── tensorflow.sparse.pbtxt │ │ │ │ ├── tensorflow.spectral.pbtxt │ │ │ │ ├── tensorflow.strings.pbtxt │ │ │ │ ├── tensorflow.summary.-event.pbtxt │ │ │ │ ├── tensorflow.summary.-file-writer-cache.pbtxt │ │ │ │ ├── tensorflow.summary.-file-writer.pbtxt │ │ │ │ ├── tensorflow.summary.-session-log.pbtxt │ │ │ │ ├── tensorflow.summary.-summary-description.pbtxt │ │ │ │ ├── tensorflow.summary.-summary.-audio.pbtxt │ │ │ │ ├── tensorflow.summary.-summary.-image.pbtxt │ │ │ │ ├── tensorflow.summary.-summary.-value.pbtxt │ │ │ │ ├── tensorflow.summary.-summary.pbtxt │ │ │ │ ├── tensorflow.summary.-tagged-run-metadata.pbtxt │ │ │ │ ├── tensorflow.summary.pbtxt │ │ │ │ ├── tensorflow.sysconfig.pbtxt │ │ │ │ ├── tensorflow.test.-benchmark.pbtxt │ │ │ │ ├── tensorflow.test.-stub-out-for-testing.pbtxt │ │ │ │ ├── tensorflow.test.experimental.pbtxt │ │ │ │ ├── tensorflow.test.pbtxt │ │ │ │ ├── tensorflow.tpu.-cross-shard-optimizer.pbtxt │ │ │ │ ├── tensorflow.tpu.-padding-spec.pbtxt │ │ │ │ ├── tensorflow.tpu.-x-l-a-options.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-adagrad-parameters.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-adam-parameters.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-device-assignment.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-device-order-mode.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-ftrl-parameters.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-hardware-feature.-embedding-feature.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-hardware-feature.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-stochastic-gradient-descent-parameters.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-t-p-u-system-metadata.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-topology.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-adagrad-momentum.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-adagrad.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-adam.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-f-t-r-l.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-feature-config.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-quantization-config.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-row-id-initializer.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-s-g-d.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding-for-serving.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding-v0.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding-v2.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-table-config.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.pbtxt │ │ │ │ ├── tensorflow.tpu.pbtxt │ │ │ │ ├── tensorflow.train.-adadelta-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-adagrad-d-a-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-adagrad-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-adam-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-bytes-list.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-manager.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-options.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-saver-hook.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-saver-listener.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint.pbtxt │ │ │ │ ├── tensorflow.train.-chief-session-creator.pbtxt │ │ │ │ ├── tensorflow.train.-cluster-def.pbtxt │ │ │ │ ├── tensorflow.train.-cluster-spec.pbtxt │ │ │ │ ├── tensorflow.train.-coordinator.pbtxt │ │ │ │ ├── tensorflow.train.-example.pbtxt │ │ │ │ ├── tensorflow.train.-exponential-moving-average.pbtxt │ │ │ │ ├── tensorflow.train.-feature-list.pbtxt │ │ │ │ ├── tensorflow.train.-feature-lists.-feature-list-entry.pbtxt │ │ │ │ ├── tensorflow.train.-feature-lists.pbtxt │ │ │ │ ├── tensorflow.train.-feature.pbtxt │ │ │ │ ├── tensorflow.train.-features.-feature-entry.pbtxt │ │ │ │ ├── tensorflow.train.-features.pbtxt │ │ │ │ ├── tensorflow.train.-feed-fn-hook.pbtxt │ │ │ │ ├── tensorflow.train.-final-ops-hook.pbtxt │ │ │ │ ├── tensorflow.train.-float-list.pbtxt │ │ │ │ ├── tensorflow.train.-ftrl-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-global-step-waiter-hook.pbtxt │ │ │ │ ├── tensorflow.train.-gradient-descent-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-int64-list.pbtxt │ │ │ │ ├── tensorflow.train.-job-def.-tasks-entry.pbtxt │ │ │ │ ├── tensorflow.train.-job-def.pbtxt │ │ │ │ ├── tensorflow.train.-logging-tensor-hook.pbtxt │ │ │ │ ├── tensorflow.train.-looper-thread.pbtxt │ │ │ │ ├── tensorflow.train.-momentum-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-monitored-session.-step-context.pbtxt │ │ │ │ ├── tensorflow.train.-monitored-session.pbtxt │ │ │ │ ├── tensorflow.train.-nan-loss-during-training-error.pbtxt │ │ │ │ ├── tensorflow.train.-nan-tensor-hook.pbtxt │ │ │ │ ├── tensorflow.train.-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-profiler-hook.pbtxt │ │ │ │ ├── tensorflow.train.-proximal-adagrad-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-proximal-gradient-descent-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-queue-runner.pbtxt │ │ │ │ ├── tensorflow.train.-r-m-s-prop-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-saver-def.pbtxt │ │ │ │ ├── tensorflow.train.-saver.pbtxt │ │ │ │ ├── tensorflow.train.-scaffold.pbtxt │ │ │ │ ├── tensorflow.train.-second-or-step-timer.pbtxt │ │ │ │ ├── tensorflow.train.-sequence-example.pbtxt │ │ │ │ ├── tensorflow.train.-server-def.pbtxt │ │ │ │ ├── tensorflow.train.-server.pbtxt │ │ │ │ ├── tensorflow.train.-session-creator.pbtxt │ │ │ │ ├── tensorflow.train.-session-manager.pbtxt │ │ │ │ ├── tensorflow.train.-session-run-args.pbtxt │ │ │ │ ├── tensorflow.train.-session-run-context.pbtxt │ │ │ │ ├── tensorflow.train.-session-run-hook.pbtxt │ │ │ │ ├── tensorflow.train.-session-run-values.pbtxt │ │ │ │ ├── tensorflow.train.-singular-monitored-session.-step-context.pbtxt │ │ │ │ ├── tensorflow.train.-singular-monitored-session.pbtxt │ │ │ │ ├── tensorflow.train.-step-counter-hook.pbtxt │ │ │ │ ├── tensorflow.train.-stop-at-step-hook.pbtxt │ │ │ │ ├── tensorflow.train.-summary-saver-hook.pbtxt │ │ │ │ ├── tensorflow.train.-supervisor.pbtxt │ │ │ │ ├── tensorflow.train.-sync-replicas-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.-vocab-info.pbtxt │ │ │ │ ├── tensorflow.train.-worker-session-creator.pbtxt │ │ │ │ ├── tensorflow.train.experimental.-dynamic-loss-scale.pbtxt │ │ │ │ ├── tensorflow.train.experimental.-fixed-loss-scale.pbtxt │ │ │ │ ├── tensorflow.train.experimental.-loss-scale.pbtxt │ │ │ │ ├── tensorflow.train.experimental.-mixed-precision-loss-scale-optimizer.pbtxt │ │ │ │ ├── tensorflow.train.experimental.-python-state.pbtxt │ │ │ │ ├── tensorflow.train.experimental.pbtxt │ │ │ │ ├── tensorflow.train.pbtxt │ │ │ │ ├── tensorflow.train.queue_runner.-queue-runner.pbtxt │ │ │ │ ├── tensorflow.train.queue_runner.pbtxt │ │ │ │ ├── tensorflow.truncated_normal_initializer.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-function-type.empty.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-function-type.pbtxt │ │ │ │ ├── tensorflow.types.experimental.pbtxt │ │ │ │ ├── tensorflow.types.pbtxt │ │ │ │ ├── tensorflow.uniform_unit_scaling_initializer.pbtxt │ │ │ │ ├── tensorflow.variable_scope.pbtxt │ │ │ │ ├── tensorflow.variance_scaling_initializer.pbtxt │ │ │ │ ├── tensorflow.version.pbtxt │ │ │ │ ├── tensorflow.xla.experimental.pbtxt │ │ │ │ ├── tensorflow.xla.pbtxt │ │ │ │ └── tensorflow.zeros_initializer.pbtxt │ │ │ └── v2 │ │ │ │ ├── tensorflow.-aggregation-method.pbtxt │ │ │ │ ├── tensorflow.-critical-section.pbtxt │ │ │ │ ├── tensorflow.-d-type.pbtxt │ │ │ │ ├── tensorflow.-device-spec.pbtxt │ │ │ │ ├── tensorflow.-gradient-tape.pbtxt │ │ │ │ ├── tensorflow.-graph.pbtxt │ │ │ │ ├── tensorflow.-indexed-slices-spec.pbtxt │ │ │ │ ├── tensorflow.-indexed-slices.pbtxt │ │ │ │ ├── tensorflow.-module.pbtxt │ │ │ │ ├── tensorflow.-operation.pbtxt │ │ │ │ ├── tensorflow.-optional-spec.pbtxt │ │ │ │ ├── tensorflow.-ragged-tensor-spec.pbtxt │ │ │ │ ├── tensorflow.-ragged-tensor.pbtxt │ │ │ │ ├── tensorflow.-register-gradient.pbtxt │ │ │ │ ├── tensorflow.-sparse-tensor-spec.pbtxt │ │ │ │ ├── tensorflow.-sparse-tensor.pbtxt │ │ │ │ ├── tensorflow.-tensor-array-spec.pbtxt │ │ │ │ ├── tensorflow.-tensor-array.pbtxt │ │ │ │ ├── tensorflow.-tensor-shape.pbtxt │ │ │ │ ├── tensorflow.-tensor-spec.pbtxt │ │ │ │ ├── tensorflow.-tensor.pbtxt │ │ │ │ ├── tensorflow.-type-spec.pbtxt │ │ │ │ ├── tensorflow.-unconnected-gradients.pbtxt │ │ │ │ ├── tensorflow.-variable-aggregation.pbtxt │ │ │ │ ├── tensorflow.-variable-synchronization.pbtxt │ │ │ │ ├── tensorflow.-variable.-save-slice-info.pbtxt │ │ │ │ ├── tensorflow.-variable.pbtxt │ │ │ │ ├── tensorflow.__internal__.-composite-tensor.pbtxt │ │ │ │ ├── tensorflow.__internal__.-eager-tensor.pbtxt │ │ │ │ ├── tensorflow.__internal__.-func-graph.pbtxt │ │ │ │ ├── tensorflow.__internal__.-symbolic-tensor.pbtxt │ │ │ │ ├── tensorflow.__internal__.autograph.pbtxt │ │ │ │ ├── tensorflow.__internal__.decorator.pbtxt │ │ │ │ ├── tensorflow.__internal__.dispatch.-global-op-dispatcher.pbtxt │ │ │ │ ├── tensorflow.__internal__.dispatch.-op-dispatcher.pbtxt │ │ │ │ ├── tensorflow.__internal__.dispatch.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.-sharded-variable.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.combinations.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.interim.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.multi_process_runner.-not-initialized-error.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.multi_process_runner.-subprocess-timeout-error.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.multi_process_runner.-unexpected-subprocess-exit-error.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.multi_process_runner.pbtxt │ │ │ │ ├── tensorflow.__internal__.distribute.pbtxt │ │ │ │ ├── tensorflow.__internal__.eager_context.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.-dense-column.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.-feature-column.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.-feature-transformation-cache.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.-sequence-dense-column.-tensor-sequence-length-pair.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.-sequence-dense-column.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.-state-manager.pbtxt │ │ │ │ ├── tensorflow.__internal__.feature_column.pbtxt │ │ │ │ ├── tensorflow.__internal__.function.-function.pbtxt │ │ │ │ ├── tensorflow.__internal__.function.pbtxt │ │ │ │ ├── tensorflow.__internal__.graph_util.pbtxt │ │ │ │ ├── tensorflow.__internal__.mixed_precision.pbtxt │ │ │ │ ├── tensorflow.__internal__.monitoring.-bool-gauge.pbtxt │ │ │ │ ├── tensorflow.__internal__.monitoring.pbtxt │ │ │ │ ├── tensorflow.__internal__.nest.pbtxt │ │ │ │ ├── tensorflow.__internal__.ops.pbtxt │ │ │ │ ├── tensorflow.__internal__.pbtxt │ │ │ │ ├── tensorflow.__internal__.saved_model.-saved-model-builder.pbtxt │ │ │ │ ├── tensorflow.__internal__.saved_model.load.-versioned-type-registration.pbtxt │ │ │ │ ├── tensorflow.__internal__.saved_model.load.pbtxt │ │ │ │ ├── tensorflow.__internal__.saved_model.pbtxt │ │ │ │ ├── tensorflow.__internal__.smart_cond.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.-parameterized-benchmark.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.combinations.-named-object.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.combinations.-optional-parameter.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.combinations.-parameter-modifier.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.combinations.-test-combination.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.combinations.pbtxt │ │ │ │ ├── tensorflow.__internal__.test.pbtxt │ │ │ │ ├── tensorflow.__internal__.tf2.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-auto-trackable.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-checkpoint-initial-value-callable.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-checkpoint-initial-value.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-delegating-trackable-mixin.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-object-graph-view.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-trackable-data-structure.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-trackable-reference.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.-trackable.pbtxt │ │ │ │ ├── tensorflow.__internal__.tracking.pbtxt │ │ │ │ ├── tensorflow.__internal__.train.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.-tensor.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.data.-dataset.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.data.pbtxt │ │ │ │ ├── tensorflow.__internal__.types.pbtxt │ │ │ │ ├── tensorflow.__operators__.pbtxt │ │ │ │ ├── tensorflow.audio.pbtxt │ │ │ │ ├── tensorflow.autodiff.-forward-accumulator.pbtxt │ │ │ │ ├── tensorflow.autodiff.-gradient-tape.pbtxt │ │ │ │ ├── tensorflow.autodiff.pbtxt │ │ │ │ ├── tensorflow.autograph.experimental.-feature.pbtxt │ │ │ │ ├── tensorflow.autograph.experimental.pbtxt │ │ │ │ ├── tensorflow.autograph.pbtxt │ │ │ │ ├── tensorflow.bitwise.pbtxt │ │ │ │ ├── tensorflow.compat.pbtxt │ │ │ │ ├── tensorflow.config.-logical-device-configuration.pbtxt │ │ │ │ ├── tensorflow.config.-logical-device.pbtxt │ │ │ │ ├── tensorflow.config.-physical-device.pbtxt │ │ │ │ ├── tensorflow.config.experimental.-cluster-device-filters.pbtxt │ │ │ │ ├── tensorflow.config.experimental.-virtual-device-configuration.pbtxt │ │ │ │ ├── tensorflow.config.experimental.pbtxt │ │ │ │ ├── tensorflow.config.optimizer.pbtxt │ │ │ │ ├── tensorflow.config.pbtxt │ │ │ │ ├── tensorflow.config.threading.pbtxt │ │ │ │ ├── tensorflow.constant_initializer.pbtxt │ │ │ │ ├── tensorflow.data.-dataset-spec.pbtxt │ │ │ │ ├── tensorflow.data.-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-fixed-length-record-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-iterator-spec.pbtxt │ │ │ │ ├── tensorflow.data.-iterator.pbtxt │ │ │ │ ├── tensorflow.data.-numpy-iterator.pbtxt │ │ │ │ ├── tensorflow.data.-options.pbtxt │ │ │ │ ├── tensorflow.data.-t-f-record-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-text-line-dataset.pbtxt │ │ │ │ ├── tensorflow.data.-threading-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-auto-shard-policy.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-autotune-algorithm.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-autotune-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-checkpoint-input-pipeline-hook.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-csv-dataset.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-dataset-initializer.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-distribute-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-external-state-policy.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-optimization-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-optional.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-random-dataset.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-reducer.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-sql-dataset.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-t-f-record-writer.pbtxt │ │ │ │ ├── tensorflow.data.experimental.-threading-options.pbtxt │ │ │ │ ├── tensorflow.data.experimental.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-cross-trainer-cache.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-dispatch-server.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-dispatcher-config.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-sharding-policy.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-worker-config.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.-worker-server.pbtxt │ │ │ │ ├── tensorflow.data.experimental.service.pbtxt │ │ │ │ ├── tensorflow.data.pbtxt │ │ │ │ ├── tensorflow.debugging.experimental.pbtxt │ │ │ │ ├── tensorflow.debugging.pbtxt │ │ │ │ ├── tensorflow.distribute.-cross-device-ops.pbtxt │ │ │ │ ├── tensorflow.distribute.-distributed-dataset.pbtxt │ │ │ │ ├── tensorflow.distribute.-distributed-iterator.pbtxt │ │ │ │ ├── tensorflow.distribute.-distributed-values.pbtxt │ │ │ │ ├── tensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt │ │ │ │ ├── tensorflow.distribute.-input-context.pbtxt │ │ │ │ ├── tensorflow.distribute.-input-options.pbtxt │ │ │ │ ├── tensorflow.distribute.-input-replication-mode.pbtxt │ │ │ │ ├── tensorflow.distribute.-mirrored-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-nccl-all-reduce.pbtxt │ │ │ │ ├── tensorflow.distribute.-one-device-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-parameter-server-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-reduce-op.pbtxt │ │ │ │ ├── tensorflow.distribute.-reduction-to-one-device.pbtxt │ │ │ │ ├── tensorflow.distribute.-replica-context.pbtxt │ │ │ │ ├── tensorflow.distribute.-run-options.pbtxt │ │ │ │ ├── tensorflow.distribute.-server.pbtxt │ │ │ │ ├── tensorflow.distribute.-strategy-extended.pbtxt │ │ │ │ ├── tensorflow.distribute.-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.-t-p-u-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-g-c-e-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-kubernetes-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-simple-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-slurm-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-t-f-config-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-t-p-u-cluster-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.-union-resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.cluster_resolver.pbtxt │ │ │ │ ├── tensorflow.distribute.coordinator.-cluster-coordinator.pbtxt │ │ │ │ ├── tensorflow.distribute.coordinator.-per-worker-value.pbtxt │ │ │ │ ├── tensorflow.distribute.coordinator.-remote-value.pbtxt │ │ │ │ ├── tensorflow.distribute.coordinator.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-central-storage-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-collective-communication.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-collective-hints.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-communication-implementation.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-communication-options.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-preemption-checkpoint-handler.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-preemption-watcher.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-termination-config.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.-value-context.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.coordinator.-cluster-coordinator.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.coordinator.-per-worker-values.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.coordinator.-remote-value.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.coordinator.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.partitioners.-fixed-shards-partitioner.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.partitioners.-max-size-partitioner.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.partitioners.-min-size-partitioner.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.partitioners.-partitioner.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.partitioners.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.rpc.-client.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.rpc.-server.pbtxt │ │ │ │ ├── tensorflow.distribute.experimental.rpc.pbtxt │ │ │ │ ├── tensorflow.distribute.pbtxt │ │ │ │ ├── tensorflow.dtypes.-d-type.pbtxt │ │ │ │ ├── tensorflow.dtypes.experimental.pbtxt │ │ │ │ ├── tensorflow.dtypes.pbtxt │ │ │ │ ├── tensorflow.errors.-aborted-error.pbtxt │ │ │ │ ├── tensorflow.errors.-already-exists-error.pbtxt │ │ │ │ ├── tensorflow.errors.-cancelled-error.pbtxt │ │ │ │ ├── tensorflow.errors.-data-loss-error.pbtxt │ │ │ │ ├── tensorflow.errors.-deadline-exceeded-error.pbtxt │ │ │ │ ├── tensorflow.errors.-failed-precondition-error.pbtxt │ │ │ │ ├── tensorflow.errors.-internal-error.pbtxt │ │ │ │ ├── tensorflow.errors.-invalid-argument-error.pbtxt │ │ │ │ ├── tensorflow.errors.-not-found-error.pbtxt │ │ │ │ ├── tensorflow.errors.-op-error.pbtxt │ │ │ │ ├── tensorflow.errors.-operator-not-allowed-in-graph-error.pbtxt │ │ │ │ ├── tensorflow.errors.-out-of-range-error.pbtxt │ │ │ │ ├── tensorflow.errors.-permission-denied-error.pbtxt │ │ │ │ ├── tensorflow.errors.-resource-exhausted-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unauthenticated-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unavailable-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unimplemented-error.pbtxt │ │ │ │ ├── tensorflow.errors.-unknown-error.pbtxt │ │ │ │ ├── tensorflow.errors.pbtxt │ │ │ │ ├── tensorflow.estimator.-baseline-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-baseline-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-baseline-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-best-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-binary-class-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-checkpoint-saver-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-checkpoint-saver-listener.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-linear-combined-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-linear-combined-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-linear-combined-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-d-n-n-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-estimator-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-eval-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-feed-fn-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-final-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-final-ops-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-global-step-waiter-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-latest-exporter.pbtxt │ │ │ │ ├── tensorflow.estimator.-linear-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.-linear-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.-linear-regressor.pbtxt │ │ │ │ ├── tensorflow.estimator.-logging-tensor-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-logistic-regression-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-mode-keys.pbtxt │ │ │ │ ├── tensorflow.estimator.-multi-class-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-multi-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-multi-label-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-nan-loss-during-training-error.pbtxt │ │ │ │ ├── tensorflow.estimator.-nan-tensor-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-poisson-regression-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-profiler-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-regression-head.pbtxt │ │ │ │ ├── tensorflow.estimator.-run-config.pbtxt │ │ │ │ ├── tensorflow.estimator.-second-or-step-timer.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-args.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-context.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-session-run-values.pbtxt │ │ │ │ ├── tensorflow.estimator.-step-counter-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-stop-at-step-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-summary-saver-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.-train-spec.pbtxt │ │ │ │ ├── tensorflow.estimator.-vocab-info.pbtxt │ │ │ │ ├── tensorflow.estimator.-warm-start-settings.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-in-memory-evaluator-hook.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-linear-s-d-c-a.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-r-n-n-classifier.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.-r-n-n-estimator.pbtxt │ │ │ │ ├── tensorflow.estimator.experimental.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-classification-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-classification-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-eval-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-eval-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-export-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-export-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-predict-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-predict-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-regression-output.__metaclass__.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-regression-output.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-serving-input-receiver.pbtxt │ │ │ │ ├── tensorflow.estimator.export.-tensor-serving-input-receiver.pbtxt │ │ │ │ ├── tensorflow.estimator.export.pbtxt │ │ │ │ ├── tensorflow.estimator.pbtxt │ │ │ │ ├── tensorflow.experimental.-batchable-extension-type.pbtxt │ │ │ │ ├── tensorflow.experimental.-dynamic-ragged-shape.-spec.pbtxt │ │ │ │ ├── tensorflow.experimental.-dynamic-ragged-shape.pbtxt │ │ │ │ ├── tensorflow.experimental.-extension-type-batch-encoder.pbtxt │ │ │ │ ├── tensorflow.experimental.-extension-type-spec.pbtxt │ │ │ │ ├── tensorflow.experimental.-extension-type.pbtxt │ │ │ │ ├── tensorflow.experimental.-optional.pbtxt │ │ │ │ ├── tensorflow.experimental.-row-partition.pbtxt │ │ │ │ ├── tensorflow.experimental.-structured-tensor.-spec.pbtxt │ │ │ │ ├── tensorflow.experimental.-structured-tensor.pbtxt │ │ │ │ ├── tensorflow.experimental.dlpack.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.-d-tensor-checkpoint.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.-d-tensor-dataset.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.-d-variable.-save-slice-info.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.-d-variable.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.-layout.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.-mesh.pbtxt │ │ │ │ ├── tensorflow.experimental.dtensor.pbtxt │ │ │ │ ├── tensorflow.experimental.extension_type.pbtxt │ │ │ │ ├── tensorflow.experimental.pbtxt │ │ │ │ ├── tensorflow.experimental.tensorrt.-conversion-params.pbtxt │ │ │ │ ├── tensorflow.experimental.tensorrt.-converter.pbtxt │ │ │ │ ├── tensorflow.experimental.tensorrt.pbtxt │ │ │ │ ├── tensorflow.feature_column.pbtxt │ │ │ │ ├── tensorflow.graph_util.pbtxt │ │ │ │ ├── tensorflow.image.-resize-method.pbtxt │ │ │ │ ├── tensorflow.image.pbtxt │ │ │ │ ├── tensorflow.initializers.-constant.pbtxt │ │ │ │ ├── tensorflow.initializers.-glorot-normal.pbtxt │ │ │ │ ├── tensorflow.initializers.-glorot-uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.-he-normal.pbtxt │ │ │ │ ├── tensorflow.initializers.-he-uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.-identity.pbtxt │ │ │ │ ├── tensorflow.initializers.-initializer.pbtxt │ │ │ │ ├── tensorflow.initializers.-lecun-normal.pbtxt │ │ │ │ ├── tensorflow.initializers.-lecun-uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.-ones.pbtxt │ │ │ │ ├── tensorflow.initializers.-orthogonal.pbtxt │ │ │ │ ├── tensorflow.initializers.-random-normal.pbtxt │ │ │ │ ├── tensorflow.initializers.-random-uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.-truncated-normal.pbtxt │ │ │ │ ├── tensorflow.initializers.-variance-scaling.pbtxt │ │ │ │ ├── tensorflow.initializers.-zeros.pbtxt │ │ │ │ ├── tensorflow.initializers.constant.pbtxt │ │ │ │ ├── tensorflow.initializers.glorot_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.glorot_uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.he_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.he_uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.identity.pbtxt │ │ │ │ ├── tensorflow.initializers.lecun_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.lecun_uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.ones.pbtxt │ │ │ │ ├── tensorflow.initializers.orthogonal.pbtxt │ │ │ │ ├── tensorflow.initializers.pbtxt │ │ │ │ ├── tensorflow.initializers.random_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.random_uniform.pbtxt │ │ │ │ ├── tensorflow.initializers.truncated_normal.pbtxt │ │ │ │ ├── tensorflow.initializers.variance_scaling.pbtxt │ │ │ │ ├── tensorflow.initializers.zeros.pbtxt │ │ │ │ ├── tensorflow.io.-fixed-len-feature.pbtxt │ │ │ │ ├── tensorflow.io.-fixed-len-sequence-feature.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-lengths.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-limits.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-splits.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-row-starts.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-uniform-row-length.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.-value-row-ids.pbtxt │ │ │ │ ├── tensorflow.io.-ragged-feature.pbtxt │ │ │ │ ├── tensorflow.io.-sparse-feature.pbtxt │ │ │ │ ├── tensorflow.io.-t-f-record-options.pbtxt │ │ │ │ ├── tensorflow.io.-t-f-record-writer.pbtxt │ │ │ │ ├── tensorflow.io.-var-len-feature.pbtxt │ │ │ │ ├── tensorflow.io.gfile.-g-file.pbtxt │ │ │ │ ├── tensorflow.io.gfile.pbtxt │ │ │ │ ├── tensorflow.io.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-adjoint.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-block-diag.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-block-lower-triangular.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-circulant.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-circulant2-d.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-circulant3-d.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-composition.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-diag.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-full-matrix.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-householder.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-identity.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-inversion.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-kronecker.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-low-rank-update.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-lower-triangular.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-permutation.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-scaled-identity.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-toeplitz.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-tridiag.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator-zeros.pbtxt │ │ │ │ ├── tensorflow.linalg.-linear-operator.pbtxt │ │ │ │ ├── tensorflow.linalg.experimental.pbtxt │ │ │ │ ├── tensorflow.linalg.pbtxt │ │ │ │ ├── tensorflow.lite.-interpreter.pbtxt │ │ │ │ ├── tensorflow.lite.-ops-set.pbtxt │ │ │ │ ├── tensorflow.lite.-optimize.pbtxt │ │ │ │ ├── tensorflow.lite.-representative-dataset.pbtxt │ │ │ │ ├── tensorflow.lite.-t-f-lite-converter.pbtxt │ │ │ │ ├── tensorflow.lite.-target-spec.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-analyzer.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-op-resolver-type.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-quantization-debug-options.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.-quantization-debugger.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.authoring.pbtxt │ │ │ │ ├── tensorflow.lite.experimental.pbtxt │ │ │ │ ├── tensorflow.lite.pbtxt │ │ │ │ ├── tensorflow.lookup.-key-value-tensor-initializer.pbtxt │ │ │ │ ├── tensorflow.lookup.-static-hash-table.pbtxt │ │ │ │ ├── tensorflow.lookup.-static-vocabulary-table.pbtxt │ │ │ │ ├── tensorflow.lookup.-text-file-index.pbtxt │ │ │ │ ├── tensorflow.lookup.-text-file-initializer.pbtxt │ │ │ │ ├── tensorflow.lookup.experimental.-dense-hash-table.pbtxt │ │ │ │ ├── tensorflow.lookup.experimental.-mutable-hash-table.pbtxt │ │ │ │ ├── tensorflow.lookup.experimental.pbtxt │ │ │ │ ├── tensorflow.lookup.pbtxt │ │ │ │ ├── tensorflow.losses.-binary-crossentropy.pbtxt │ │ │ │ ├── tensorflow.losses.-binary-focal-crossentropy.pbtxt │ │ │ │ ├── tensorflow.losses.-categorical-crossentropy.pbtxt │ │ │ │ ├── tensorflow.losses.-categorical-focal-crossentropy.pbtxt │ │ │ │ ├── tensorflow.losses.-categorical-hinge.pbtxt │ │ │ │ ├── tensorflow.losses.-cosine-similarity.pbtxt │ │ │ │ ├── tensorflow.losses.-hinge.pbtxt │ │ │ │ ├── tensorflow.losses.-huber.pbtxt │ │ │ │ ├── tensorflow.losses.-k-l-divergence.pbtxt │ │ │ │ ├── tensorflow.losses.-log-cosh.pbtxt │ │ │ │ ├── tensorflow.losses.-loss.pbtxt │ │ │ │ ├── tensorflow.losses.-mean-absolute-error.pbtxt │ │ │ │ ├── tensorflow.losses.-mean-absolute-percentage-error.pbtxt │ │ │ │ ├── tensorflow.losses.-mean-squared-error.pbtxt │ │ │ │ ├── tensorflow.losses.-mean-squared-logarithmic-error.pbtxt │ │ │ │ ├── tensorflow.losses.-poisson.pbtxt │ │ │ │ ├── tensorflow.losses.-reduction.pbtxt │ │ │ │ ├── tensorflow.losses.-sparse-categorical-crossentropy.pbtxt │ │ │ │ ├── tensorflow.losses.-squared-hinge.pbtxt │ │ │ │ ├── tensorflow.losses.pbtxt │ │ │ │ ├── tensorflow.math.pbtxt │ │ │ │ ├── tensorflow.math.special.pbtxt │ │ │ │ ├── tensorflow.metrics.-a-u-c.pbtxt │ │ │ │ ├── tensorflow.metrics.-accuracy.pbtxt │ │ │ │ ├── tensorflow.metrics.-binary-accuracy.pbtxt │ │ │ │ ├── tensorflow.metrics.-binary-crossentropy.pbtxt │ │ │ │ ├── tensorflow.metrics.-binary-io-u.pbtxt │ │ │ │ ├── tensorflow.metrics.-categorical-accuracy.pbtxt │ │ │ │ ├── tensorflow.metrics.-categorical-crossentropy.pbtxt │ │ │ │ ├── tensorflow.metrics.-categorical-hinge.pbtxt │ │ │ │ ├── tensorflow.metrics.-cosine-similarity.pbtxt │ │ │ │ ├── tensorflow.metrics.-f-beta-score.pbtxt │ │ │ │ ├── tensorflow.metrics.-f1-score.pbtxt │ │ │ │ ├── tensorflow.metrics.-false-negatives.pbtxt │ │ │ │ ├── tensorflow.metrics.-false-positives.pbtxt │ │ │ │ ├── tensorflow.metrics.-hinge.pbtxt │ │ │ │ ├── tensorflow.metrics.-io-u.pbtxt │ │ │ │ ├── tensorflow.metrics.-k-l-divergence.pbtxt │ │ │ │ ├── tensorflow.metrics.-log-cosh-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-absolute-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-absolute-percentage-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-io-u.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-metric-wrapper.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-relative-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-squared-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-squared-logarithmic-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean-tensor.pbtxt │ │ │ │ ├── tensorflow.metrics.-mean.pbtxt │ │ │ │ ├── tensorflow.metrics.-metric.pbtxt │ │ │ │ ├── tensorflow.metrics.-one-hot-io-u.pbtxt │ │ │ │ ├── tensorflow.metrics.-one-hot-mean-io-u.pbtxt │ │ │ │ ├── tensorflow.metrics.-poisson.pbtxt │ │ │ │ ├── tensorflow.metrics.-precision-at-recall.pbtxt │ │ │ │ ├── tensorflow.metrics.-precision.pbtxt │ │ │ │ ├── tensorflow.metrics.-r2-score.pbtxt │ │ │ │ ├── tensorflow.metrics.-recall-at-precision.pbtxt │ │ │ │ ├── tensorflow.metrics.-recall.pbtxt │ │ │ │ ├── tensorflow.metrics.-root-mean-squared-error.pbtxt │ │ │ │ ├── tensorflow.metrics.-sensitivity-at-specificity.pbtxt │ │ │ │ ├── tensorflow.metrics.-sparse-categorical-accuracy.pbtxt │ │ │ │ ├── tensorflow.metrics.-sparse-categorical-crossentropy.pbtxt │ │ │ │ ├── tensorflow.metrics.-sparse-top-k-categorical-accuracy.pbtxt │ │ │ │ ├── tensorflow.metrics.-specificity-at-sensitivity.pbtxt │ │ │ │ ├── tensorflow.metrics.-squared-hinge.pbtxt │ │ │ │ ├── tensorflow.metrics.-sum.pbtxt │ │ │ │ ├── tensorflow.metrics.-top-k-categorical-accuracy.pbtxt │ │ │ │ ├── tensorflow.metrics.-true-negatives.pbtxt │ │ │ │ ├── tensorflow.metrics.-true-positives.pbtxt │ │ │ │ ├── tensorflow.metrics.experimental.-py-metric.pbtxt │ │ │ │ ├── tensorflow.metrics.experimental.pbtxt │ │ │ │ ├── tensorflow.metrics.pbtxt │ │ │ │ ├── tensorflow.mlir.experimental.pbtxt │ │ │ │ ├── tensorflow.mlir.pbtxt │ │ │ │ ├── tensorflow.name_scope.pbtxt │ │ │ │ ├── tensorflow.nest.pbtxt │ │ │ │ ├── tensorflow.nn.-r-n-n-cell-device-wrapper.pbtxt │ │ │ │ ├── tensorflow.nn.-r-n-n-cell-dropout-wrapper.pbtxt │ │ │ │ ├── tensorflow.nn.-r-n-n-cell-residual-wrapper.pbtxt │ │ │ │ ├── tensorflow.nn.experimental.pbtxt │ │ │ │ ├── tensorflow.nn.pbtxt │ │ │ │ ├── tensorflow.ones_initializer.pbtxt │ │ │ │ ├── tensorflow.optimizers.-adadelta.pbtxt │ │ │ │ ├── tensorflow.optimizers.-adafactor.pbtxt │ │ │ │ ├── tensorflow.optimizers.-adagrad.pbtxt │ │ │ │ ├── tensorflow.optimizers.-adam-w.pbtxt │ │ │ │ ├── tensorflow.optimizers.-adam.pbtxt │ │ │ │ ├── tensorflow.optimizers.-adamax.pbtxt │ │ │ │ ├── tensorflow.optimizers.-ftrl.pbtxt │ │ │ │ ├── tensorflow.optimizers.-lion.pbtxt │ │ │ │ ├── tensorflow.optimizers.-nadam.pbtxt │ │ │ │ ├── tensorflow.optimizers.-optimizer.pbtxt │ │ │ │ ├── tensorflow.optimizers.-r-m-sprop.pbtxt │ │ │ │ ├── tensorflow.optimizers.-s-g-d.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-adadelta.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-adafactor.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-adagrad.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-adam-w.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-adam.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-adamax.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-ftrl.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-nadam.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-optimizer.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-r-m-sprop.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.-s-g-d.pbtxt │ │ │ │ ├── tensorflow.optimizers.experimental.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-adadelta.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-adagrad.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-adam.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-adamax.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-ftrl.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-nadam.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-optimizer.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-r-m-sprop.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.-s-g-d.pbtxt │ │ │ │ ├── tensorflow.optimizers.legacy.pbtxt │ │ │ │ ├── tensorflow.optimizers.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-cosine-decay-restarts.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-cosine-decay.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-exponential-decay.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-inverse-time-decay.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-learning-rate-schedule.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-piecewise-constant-decay.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.-polynomial-decay.pbtxt │ │ │ │ ├── tensorflow.optimizers.schedules.pbtxt │ │ │ │ ├── tensorflow.pbtxt │ │ │ │ ├── tensorflow.profiler.experimental.-profile.pbtxt │ │ │ │ ├── tensorflow.profiler.experimental.-profiler-options.pbtxt │ │ │ │ ├── tensorflow.profiler.experimental.-trace.pbtxt │ │ │ │ ├── tensorflow.profiler.experimental.client.pbtxt │ │ │ │ ├── tensorflow.profiler.experimental.pbtxt │ │ │ │ ├── tensorflow.profiler.experimental.server.pbtxt │ │ │ │ ├── tensorflow.profiler.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-component-spec.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-method.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-options.-representative-datasets-entry.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-quantization-options.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-tf-record-representative-dataset-saver.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-unit-wise-quantization-spec.-quantization-unit.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.-unit-wise-quantization-spec.pbtxt │ │ │ │ ├── tensorflow.quantization.experimental.pbtxt │ │ │ │ ├── tensorflow.quantization.pbtxt │ │ │ │ ├── tensorflow.queue.-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.queue.-padding-f-i-f-o-queue.pbtxt │ │ │ │ ├── tensorflow.queue.-priority-queue.pbtxt │ │ │ │ ├── tensorflow.queue.-queue-base.pbtxt │ │ │ │ ├── tensorflow.queue.-random-shuffle-queue.pbtxt │ │ │ │ ├── tensorflow.queue.pbtxt │ │ │ │ ├── tensorflow.ragged.pbtxt │ │ │ │ ├── tensorflow.random.-algorithm.pbtxt │ │ │ │ ├── tensorflow.random.-generator.pbtxt │ │ │ │ ├── tensorflow.random.experimental.-algorithm.pbtxt │ │ │ │ ├── tensorflow.random.experimental.-generator.pbtxt │ │ │ │ ├── tensorflow.random.experimental.pbtxt │ │ │ │ ├── tensorflow.random.pbtxt │ │ │ │ ├── tensorflow.random_normal_initializer.pbtxt │ │ │ │ ├── tensorflow.random_uniform_initializer.pbtxt │ │ │ │ ├── tensorflow.raw_ops.pbtxt │ │ │ │ ├── tensorflow.saved_model.-asset.pbtxt │ │ │ │ ├── tensorflow.saved_model.-load-options.pbtxt │ │ │ │ ├── tensorflow.saved_model.-save-options.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.-fingerprint.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.-trackable-resource.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.-variable-policy.pbtxt │ │ │ │ ├── tensorflow.saved_model.experimental.pbtxt │ │ │ │ ├── tensorflow.saved_model.pbtxt │ │ │ │ ├── tensorflow.sets.pbtxt │ │ │ │ ├── tensorflow.signal.pbtxt │ │ │ │ ├── tensorflow.sparse.-sparse-tensor.pbtxt │ │ │ │ ├── tensorflow.sparse.pbtxt │ │ │ │ ├── tensorflow.strings.pbtxt │ │ │ │ ├── tensorflow.summary.-summary-writer.pbtxt │ │ │ │ ├── tensorflow.summary.experimental.pbtxt │ │ │ │ ├── tensorflow.summary.pbtxt │ │ │ │ ├── tensorflow.sysconfig.pbtxt │ │ │ │ ├── tensorflow.test.-benchmark.pbtxt │ │ │ │ ├── tensorflow.test.experimental.pbtxt │ │ │ │ ├── tensorflow.test.pbtxt │ │ │ │ ├── tensorflow.tpu.-x-l-a-options.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-device-assignment.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-device-order-mode.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-hardware-feature.-embedding-feature.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-hardware-feature.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-t-p-u-system-metadata.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.-topology.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-adagrad-momentum.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-adagrad.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-adam.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-f-t-r-l.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-feature-config.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-quantization-config.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-row-id-initializer.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-s-g-d.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding-for-serving.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding-v0.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding-v2.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.-table-config.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.embedding.pbtxt │ │ │ │ ├── tensorflow.tpu.experimental.pbtxt │ │ │ │ ├── tensorflow.tpu.pbtxt │ │ │ │ ├── tensorflow.train.-bytes-list.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-manager.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-options.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint-view.pbtxt │ │ │ │ ├── tensorflow.train.-checkpoint.pbtxt │ │ │ │ ├── tensorflow.train.-cluster-def.pbtxt │ │ │ │ ├── tensorflow.train.-cluster-spec.pbtxt │ │ │ │ ├── tensorflow.train.-coordinator.pbtxt │ │ │ │ ├── tensorflow.train.-example.pbtxt │ │ │ │ ├── tensorflow.train.-exponential-moving-average.pbtxt │ │ │ │ ├── tensorflow.train.-feature-list.pbtxt │ │ │ │ ├── tensorflow.train.-feature-lists.-feature-list-entry.pbtxt │ │ │ │ ├── tensorflow.train.-feature-lists.pbtxt │ │ │ │ ├── tensorflow.train.-feature.pbtxt │ │ │ │ ├── tensorflow.train.-features.-feature-entry.pbtxt │ │ │ │ ├── tensorflow.train.-features.pbtxt │ │ │ │ ├── tensorflow.train.-float-list.pbtxt │ │ │ │ ├── tensorflow.train.-int64-list.pbtxt │ │ │ │ ├── tensorflow.train.-job-def.-tasks-entry.pbtxt │ │ │ │ ├── tensorflow.train.-job-def.pbtxt │ │ │ │ ├── tensorflow.train.-sequence-example.pbtxt │ │ │ │ ├── tensorflow.train.-server-def.pbtxt │ │ │ │ ├── tensorflow.train.-trackable-view.pbtxt │ │ │ │ ├── tensorflow.train.experimental.-python-state.pbtxt │ │ │ │ ├── tensorflow.train.experimental.pbtxt │ │ │ │ ├── tensorflow.train.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-atomic-function.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-callable.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-concrete-function.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-function-type.empty.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-function-type.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-generic-function.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-polymorphic-function.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-supports-tracing-protocol.pbtxt │ │ │ │ ├── tensorflow.types.experimental.-trace-type.pbtxt │ │ │ │ ├── tensorflow.types.experimental.distributed.-mirrored.pbtxt │ │ │ │ ├── tensorflow.types.experimental.distributed.-per-replica.pbtxt │ │ │ │ ├── tensorflow.types.experimental.distributed.pbtxt │ │ │ │ ├── tensorflow.types.experimental.pbtxt │ │ │ │ ├── tensorflow.types.pbtxt │ │ │ │ ├── tensorflow.version.pbtxt │ │ │ │ ├── tensorflow.xla.experimental.pbtxt │ │ │ │ ├── tensorflow.xla.pbtxt │ │ │ │ └── tensorflow.zeros_initializer.pbtxt │ │ ├── lib │ │ │ ├── BUILD │ │ │ ├── api_objects.proto │ │ │ └── python_object_to_proto_visitor.py │ │ └── tests │ │ │ ├── API_UPDATE_WARNING.txt │ │ │ ├── BUILD │ │ │ ├── README.txt │ │ │ ├── api_compatibility_test.py │ │ │ ├── convert_from_multiline.cc │ │ │ └── module_test.py │ ├── benchmark │ │ ├── BUILD │ │ ├── README.md │ │ ├── benchmark_model.cc │ │ ├── benchmark_model.h │ │ ├── benchmark_model_main.cc │ │ ├── benchmark_model_test.cc │ │ ├── download_models.sh │ │ ├── onednn_benchmark_config.sh │ │ ├── parse_onednn_benchmarks.py │ │ ├── run_models.sh │ │ └── run_onednn_benchmarks.sh │ ├── build_info │ │ ├── BUILD │ │ └── gen_build_info.py │ ├── ci_build │ │ ├── BUILD │ │ ├── Dockerfile.android │ │ ├── Dockerfile.cmake │ │ ├── Dockerfile.cpu │ │ ├── Dockerfile.cpu-py36 │ │ ├── Dockerfile.cpu.arm64 │ │ ├── Dockerfile.cpu.ppc64le │ │ ├── Dockerfile.cuda-clang │ │ ├── Dockerfile.custom_op_gpu │ │ ├── Dockerfile.custom_op_ubuntu_16 │ │ ├── Dockerfile.custom_op_ubuntu_16_cuda10.0 │ │ ├── Dockerfile.custom_op_ubuntu_16_cuda10.1 │ │ ├── Dockerfile.custom_op_ubuntu_16_cuda11.2 │ │ ├── Dockerfile.debian.bullseye.cpu │ │ ├── Dockerfile.gpu │ │ ├── Dockerfile.gpu.ppc64le │ │ ├── Dockerfile.local-toolchain-ubuntu18.04-manylinux2010 │ │ ├── Dockerfile.local-toolchain-ubuntu20.04-manylinux2014 │ │ ├── Dockerfile.micro │ │ ├── Dockerfile.pi │ │ ├── Dockerfile.rbe.cpu │ │ ├── Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010 │ │ ├── Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010-multipython │ │ ├── Dockerfile.rbe.cuda10.1-cudnn7-ubuntu18.04-manylinux2010-multipython │ │ ├── Dockerfile.rbe.cuda10.2-cudnn7-ubuntu18.04-manylinux2010-multipython │ │ ├── Dockerfile.rbe.cuda11.0-cudnn8-ubuntu18.04-manylinux2010-multipython │ │ ├── Dockerfile.rbe.cuda11.2-cudnn8.1-ubuntu20.04-manylinux2014-multipython │ │ ├── Dockerfile.rbe.cuda11.8-cudnn8.6-ubuntu20.04-manylinux2014-multipython │ │ ├── Dockerfile.rbe.cuda12.2-cudnn8.9-ubuntu20.04-manylinux2014-multipython │ │ ├── Dockerfile.rbe.gpu │ │ ├── Dockerfile.rbe.rocm-ubuntu18.04-manylinux2010-multipython │ │ ├── Dockerfile.rbe.rocm-ubuntu20.04-manylinux2014-multipython │ │ ├── Dockerfile.rbe.ubuntu16.04-manylinux2010 │ │ ├── Dockerfile.rocm │ │ ├── README.md │ │ ├── a100 │ │ │ └── nightly.sh │ │ ├── build_rbe.sh │ │ ├── build_scripts │ │ │ ├── ARM_SKIP_TESTS.sh │ │ │ ├── ARM_SKIP_TESTS_EXTENDED.sh │ │ │ ├── DEFAULT_TEST_TARGETS.sh │ │ │ └── README.md │ │ ├── builds │ │ │ ├── android.sh │ │ │ ├── android_full.sh │ │ │ ├── benchmark.sh │ │ │ ├── builds_common.sh │ │ │ ├── check_system_libs.py │ │ │ ├── cmake.sh │ │ │ ├── configured │ │ │ ├── docker_cpu_pip.sh │ │ │ ├── docker_test.sh │ │ │ ├── integration_tests.sh │ │ │ ├── libtensorflow.sh │ │ │ ├── libtensorflow_nightly_symlink.sh │ │ │ ├── nightly_release_smoke_test.sh │ │ │ ├── pip_new.sh │ │ │ ├── print_build_info.sh │ │ │ ├── py_test_delegate.sh │ │ │ ├── run_pip_tests.sh │ │ │ ├── test_tutorials.sh │ │ │ ├── test_user_ops.sh │ │ │ ├── user_ops │ │ │ │ ├── cuda_op_kernel.cc │ │ │ │ ├── cuda_op_kernel.cu.cc │ │ │ │ └── zero_out_op_kernel_1.cc │ │ │ └── with_the_same_user │ │ ├── ci_build.sh │ │ ├── cloudbuild.yaml │ │ ├── code_link_check.sh │ │ ├── copy_binary.py │ │ ├── ctpu │ │ │ ├── BUILD │ │ │ └── ctpu.sh │ │ ├── cuda-clang.patch │ │ ├── devtoolset │ │ │ ├── build_devtoolset.sh │ │ │ ├── fixlinks.sh │ │ │ ├── platlib.patch │ │ │ └── rpm-patch.sh │ │ ├── gpu_build │ │ │ ├── BUILD │ │ │ └── parallel_gpu_execute.sh │ │ ├── install │ │ │ ├── .bazelrc │ │ │ ├── BUILD │ │ │ ├── build_and_install_python.sh │ │ │ ├── install_auditwheel.sh │ │ │ ├── install_bazel.sh │ │ │ ├── install_bazel_from_source.sh │ │ │ ├── install_bootstrap_deb_packages.sh │ │ │ ├── install_buildifier.sh │ │ │ ├── install_buildifier_from_source.sh │ │ │ ├── install_centos_pip_packages.sh │ │ │ ├── install_centos_python36.sh │ │ │ ├── install_clang.sh │ │ │ ├── install_clang_17.sh │ │ │ ├── install_cmake.sh │ │ │ ├── install_deb_packages.sh │ │ │ ├── install_gcc6.sh │ │ │ ├── install_golang.sh │ │ │ ├── install_golang_centos.sh │ │ │ ├── install_golang_ppc64le.sh │ │ │ ├── install_latest_clang.sh │ │ │ ├── install_mpi.sh │ │ │ ├── install_openblas_ppc64le.sh │ │ │ ├── install_patchelf.sh │ │ │ ├── install_pi_python3.9_toolchain.sh │ │ │ ├── install_pi_python3x_toolchain.sh │ │ │ ├── install_pi_toolchain.sh │ │ │ ├── install_pip_packages.sh │ │ │ ├── install_pip_packages_by_version.sh │ │ │ ├── install_pip_packages_remote.sh │ │ │ ├── install_proto3.sh │ │ │ ├── install_python3.6_pip_packages.sh │ │ │ ├── install_python37.sh │ │ │ └── install_yum_packages.sh │ │ ├── linux │ │ │ ├── cmake │ │ │ │ └── run.sh │ │ │ ├── cpu │ │ │ │ ├── run_cc_core.sh │ │ │ │ ├── run_mkl.sh │ │ │ │ ├── run_py2_core.sh │ │ │ │ ├── run_py3_contrib.sh │ │ │ │ └── run_py3_core.sh │ │ │ ├── gpu │ │ │ │ ├── run_cc_core.sh │ │ │ │ ├── run_mkl.sh │ │ │ │ └── run_py3_core.sh │ │ │ ├── libtensorflow.sh │ │ │ ├── libtensorflow_cpu.sh │ │ │ ├── libtensorflow_docker.sh │ │ │ ├── libtensorflow_gpu.sh │ │ │ ├── libtensorflow_rocm.sh │ │ │ ├── mkl │ │ │ │ ├── Dockerfile.devel-mkl │ │ │ │ ├── basic-mkl-gpu-test.sh │ │ │ │ ├── basic-mkl-test.sh │ │ │ │ ├── build-dev-container.sh │ │ │ │ ├── install_openmpi_horovod.sh │ │ │ │ └── set-build-env.py │ │ │ ├── ppc64le │ │ │ │ ├── cpu │ │ │ │ │ ├── run_py2.sh │ │ │ │ │ └── run_py3.sh │ │ │ │ └── gpu │ │ │ │ │ ├── run_py2.sh │ │ │ │ │ └── run_py3.sh │ │ │ └── rocm │ │ │ │ ├── rocm_py310_pip.sh │ │ │ │ ├── rocm_py36_pip.sh │ │ │ │ ├── rocm_py37_pip.sh │ │ │ │ ├── rocm_py38_pip.sh │ │ │ │ ├── rocm_py39_pip.sh │ │ │ │ ├── run_cpu.sh │ │ │ │ ├── run_gpu_multi.sh │ │ │ │ └── run_gpu_single.sh │ │ ├── osx │ │ │ ├── arm64 │ │ │ │ ├── .macos.bazelrc │ │ │ │ ├── tensorflow_as_build_nightly.Jenkinsfile │ │ │ │ ├── tensorflow_as_build_release.Jenkinsfile │ │ │ │ ├── tensorflow_as_test_ci.Jenkinsfile │ │ │ │ ├── tensorflow_as_test_nightly.Jenkinsfile │ │ │ │ ├── tensorflow_as_test_release.Jenkinsfile │ │ │ │ ├── tensorflow_metal_plugin_test.py │ │ │ │ ├── tensorflow_nightly_kickoff.Jenkinsfile │ │ │ │ ├── tensorflow_release_kickoff.Jenkinsfile │ │ │ │ ├── upload_nightly.Jenkinsfile │ │ │ │ └── upload_release.Jenkinsfile │ │ │ ├── cpu │ │ │ │ ├── run_contrib.sh │ │ │ │ ├── run_py2_cc_core.sh │ │ │ │ └── run_py3_cc_core.sh │ │ │ ├── libtensorflow_cpu.sh │ │ │ ├── libtensorflow_gpu.sh │ │ │ └── libtensorflow_rocm.sh │ │ ├── pep8 │ │ ├── pi │ │ │ └── build_raspberry_pi.sh │ │ ├── presubmit │ │ │ ├── ubuntu_16 │ │ │ │ ├── android │ │ │ │ │ └── build.sh │ │ │ │ ├── cpu_py39_full │ │ │ │ │ └── build.sh │ │ │ │ └── gpu_py39_full │ │ │ │ │ └── build.sh │ │ │ └── windows │ │ │ │ └── cpu_py39_full │ │ │ │ └── build.bat │ │ ├── protobuf │ │ │ └── protobuf_optimized_pip.sh │ │ ├── pylintrc │ │ ├── rel │ │ │ ├── macos │ │ │ │ ├── cpu_libtensorflow.sh │ │ │ │ ├── cpu_py310_nonpip.sh │ │ │ │ ├── cpu_py310_pip.sh │ │ │ │ ├── cpu_py39_nonpip.sh │ │ │ │ └── cpu_py39_pip.sh │ │ │ ├── ubuntu │ │ │ │ ├── cpu_arm64_test.sh │ │ │ │ ├── cpu_arm64_test_build.sh │ │ │ │ ├── cpu_arm64_test_cpp.sh │ │ │ │ ├── cpu_libtensorflow.sh │ │ │ │ └── gpu_libtensorflow.sh │ │ │ └── windows │ │ │ │ ├── cpu_libtensorflow.bat │ │ │ │ ├── cpu_py310.bat │ │ │ │ └── cpu_py39.bat │ │ ├── release │ │ │ ├── common.sh │ │ │ ├── common_win.bat │ │ │ ├── mac_build_utils.sh │ │ │ ├── requirements_common.txt │ │ │ ├── requirements_mac.txt │ │ │ ├── requirements_ubuntu.txt │ │ │ ├── ubuntu_16 │ │ │ │ └── custom_op │ │ │ │ │ ├── nightly.sh │ │ │ │ │ └── release.sh │ │ │ └── windows │ │ │ │ ├── cpu_py310_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── cpu_py37_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── cpu_py38_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── cpu_py39_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── gpu_py310_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── gpu_py36_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── gpu_py37_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ ├── gpu_py38_full │ │ │ │ └── release_pip_rename.sh │ │ │ │ └── gpu_py39_full │ │ │ │ └── release_pip_rename.sh │ │ ├── update_version.py │ │ ├── windows │ │ │ ├── bazel │ │ │ │ ├── bazel_test_lib.sh │ │ │ │ ├── common_env.sh │ │ │ │ └── cpu_win_test.sh │ │ │ ├── cpu │ │ │ │ ├── bazel │ │ │ │ │ ├── run_cc_test_windows.bat │ │ │ │ │ ├── run_cc_test_windows.sh │ │ │ │ │ └── run_libtensorflow.bat │ │ │ │ ├── cmake │ │ │ │ │ ├── run_build.bat │ │ │ │ │ └── run_py.bat │ │ │ │ └── pip │ │ │ │ │ ├── build_tf_windows.sh │ │ │ │ │ ├── build_tf_windows_clang-cl.sh │ │ │ │ │ └── run.bat │ │ │ └── libtensorflow_cpu.sh │ │ └── xla │ │ │ └── linux │ │ │ └── gpu │ │ │ └── run_py3.sh │ ├── common │ │ ├── BUILD │ │ ├── public_api.py │ │ ├── public_api_test.py │ │ ├── test_module1.py │ │ ├── test_module2.py │ │ ├── traverse.py │ │ └── traverse_test.py │ ├── compatibility │ │ ├── BUILD │ │ ├── README.md │ │ ├── all_renames_v2.py │ │ ├── all_renames_v2_test.py │ │ ├── ast_edits.py │ │ ├── ast_edits_test.py │ │ ├── ipynb.py │ │ ├── module_deprecations_v2.py │ │ ├── renames_v2.py │ │ ├── reorders_v2.py │ │ ├── testdata │ │ │ ├── test_file_v0_11.py │ │ │ └── test_file_v1_12.py │ │ ├── tf_upgrade.py │ │ ├── tf_upgrade_test.py │ │ ├── tf_upgrade_v2.py │ │ ├── tf_upgrade_v2_main.py │ │ ├── tf_upgrade_v2_safety.py │ │ ├── tf_upgrade_v2_safety_test.py │ │ ├── tf_upgrade_v2_test.py │ │ └── update │ │ │ ├── BUILD │ │ │ ├── generate_v2_renames_map.py │ │ │ └── generate_v2_reorders_map.py │ ├── def_file_filter │ │ ├── BUILD │ │ ├── BUILD.tpl │ │ ├── def_file_filter.py.tpl │ │ ├── def_file_filter_configure.bzl │ │ └── symbols_pybind.txt │ ├── docs │ │ ├── BUILD │ │ ├── base_dir.py │ │ ├── build_cc_api_headers.py │ │ ├── build_java_api_docs.py │ │ ├── doc_controls.py │ │ ├── fenced_doctest_lib.py │ │ ├── fenced_doctest_test.py │ │ ├── generate2.py │ │ ├── generate2_test.py │ │ ├── tf_doctest.py │ │ ├── tf_doctest_lib.py │ │ └── tf_doctest_test.py │ ├── gcs_test │ │ ├── Dockerfile │ │ ├── gcs_smoke.sh │ │ ├── gcs_smoke_wrapper.sh │ │ └── python │ │ │ └── gcs_smoke.py │ ├── git │ │ ├── .gitignore │ │ ├── BUILD │ │ └── gen_git_source.py │ ├── graph_transforms │ │ ├── BUILD │ │ ├── README.md │ │ ├── __init__.py │ │ ├── add_default_attributes.cc │ │ ├── add_default_attributes_test.cc │ │ ├── backports.cc │ │ ├── backports_test.cc │ │ ├── compare_graphs.cc │ │ ├── file_utils.cc │ │ ├── file_utils.h │ │ ├── file_utils_test.cc │ │ ├── flatten_atrous.cc │ │ ├── flatten_atrous_test.cc │ │ ├── fold_batch_norms.cc │ │ ├── fold_batch_norms_test.cc │ │ ├── fold_constants_lib.cc │ │ ├── fold_constants_lib.h │ │ ├── fold_constants_test.cc │ │ ├── fold_old_batch_norms.cc │ │ ├── fold_old_batch_norms_test.cc │ │ ├── freeze_requantization_ranges.cc │ │ ├── freeze_requantization_ranges_test.cc │ │ ├── fuse_convolutions.cc │ │ ├── fuse_convolutions_test.cc │ │ ├── inline_partitionedcall.cc │ │ ├── inline_partitionedcall_test.cc │ │ ├── insert_logging.cc │ │ ├── insert_logging_test.cc │ │ ├── obfuscate_names.cc │ │ ├── obfuscate_names_test.cc │ │ ├── python │ │ │ └── transform_graph_test.py │ │ ├── quantize_nodes.cc │ │ ├── quantize_nodes_test.cc │ │ ├── quantize_weights.cc │ │ ├── quantize_weights_test.cc │ │ ├── remove_attribute.cc │ │ ├── remove_attribute_test.cc │ │ ├── remove_control_dependencies.cc │ │ ├── remove_device.cc │ │ ├── remove_device_test.cc │ │ ├── remove_nodes.cc │ │ ├── remove_nodes_test.cc │ │ ├── rename_attribute.cc │ │ ├── rename_attribute_test.cc │ │ ├── rename_node.cc │ │ ├── rename_node_test.cc │ │ ├── rename_op.cc │ │ ├── rename_op_test.cc │ │ ├── round_weights.cc │ │ ├── round_weights_test.cc │ │ ├── set_device.cc │ │ ├── set_device_test.cc │ │ ├── sort_by_execution_order.cc │ │ ├── sort_by_execution_order_test.cc │ │ ├── sparsify_gather.cc │ │ ├── sparsify_gather_test.cc │ │ ├── strip_unused_nodes.cc │ │ ├── strip_unused_nodes_test.cc │ │ ├── summarize_graph_main.cc │ │ ├── transform_graph.cc │ │ ├── transform_graph.h │ │ ├── transform_graph_main.cc │ │ ├── transform_graph_test.cc │ │ ├── transform_utils.cc │ │ ├── transform_utils.h │ │ └── transform_utils_test.cc │ ├── lib_package │ │ ├── BUILD │ │ ├── LibTensorFlowTest.java │ │ ├── README.md │ │ ├── concat_licenses.sh │ │ ├── libtensorflow_java_test.sh │ │ ├── libtensorflow_test.c │ │ └── libtensorflow_test.sh │ ├── mlpbtxt │ │ ├── BUILD │ │ ├── frommlpbtxt.cc │ │ └── tomlpbtxt.cc │ ├── optimization │ │ ├── BUILD │ │ ├── gpu_optimization_pass_runner_main.cc │ │ ├── optimization_pass_runner.cc │ │ └── optimization_pass_runner.h │ ├── pip_package │ │ ├── BUILD │ │ ├── MANIFEST.in │ │ ├── README │ │ ├── THIRD_PARTY_NOTICES.txt │ │ ├── build_pip_package.sh │ │ ├── check_load_py_test.py │ │ ├── pip_smoke_test.py │ │ ├── redundant_tensorflow_gpu │ │ │ ├── README.md │ │ │ ├── setup.cfg │ │ │ └── setup.py │ │ ├── redundant_tf_nightly_gpu │ │ │ ├── README.md │ │ │ ├── setup.cfg │ │ │ └── setup.py │ │ ├── setup.py │ │ ├── simple_console.py │ │ ├── simple_console_for_windows.py │ │ └── xla_build │ │ │ ├── CMakeLists.txt │ │ │ └── pip_test │ │ │ ├── CMakeLists.txt │ │ │ └── run_xla_aot_test.sh │ ├── proto_splitter │ │ ├── BUILD │ │ ├── README.md │ │ ├── cc │ │ │ ├── BUILD │ │ │ ├── composable_splitter.h │ │ │ ├── composable_splitter_base.cc │ │ │ ├── composable_splitter_base.h │ │ │ ├── composable_splitter_test.cc │ │ │ ├── graph_def_splitter.cc │ │ │ ├── graph_def_splitter.h │ │ │ ├── graph_def_splitter_test.cc │ │ │ ├── large_node_splitter.cc │ │ │ ├── large_node_splitter.h │ │ │ ├── max_size.cc │ │ │ ├── max_size.h │ │ │ ├── repeated_field_splitter.cc │ │ │ ├── repeated_field_splitter.h │ │ │ ├── saved_model_splitter.cc │ │ │ ├── saved_model_splitter.h │ │ │ ├── saved_model_splitter_test.cc │ │ │ ├── size_splitter.h │ │ │ ├── split.h │ │ │ ├── test_util.h │ │ │ ├── util.cc │ │ │ ├── util.h │ │ │ └── util_test.cc │ │ ├── chunk.proto │ │ ├── constants.py │ │ ├── g3doc │ │ │ └── in-depth-guide.md │ │ ├── merge.cc │ │ ├── merge.h │ │ ├── merge_test.cc │ │ ├── python │ │ │ ├── BUILD │ │ │ ├── saved_model.py │ │ │ ├── saved_model_test.py │ │ │ ├── test_util.py │ │ │ └── test_util_test.py │ │ ├── split.py │ │ ├── split_graph_def.py │ │ ├── split_graph_def_test.py │ │ ├── split_test.py │ │ ├── testdata │ │ │ ├── BUILD │ │ │ ├── bf-split-tree.cpb │ │ │ ├── bf-split-tree.pbtxt │ │ │ ├── df-split-tree.cpb │ │ │ ├── df-split-tree.pbtxt │ │ │ ├── function-large-nodes.pb │ │ │ ├── function-lots-of-nodes.pb │ │ │ ├── graph-def-and-function.pb │ │ │ ├── many-field.cpb │ │ │ ├── many-field.pbtxt │ │ │ ├── many_field_gen.py │ │ │ ├── split-large-constant.cpb │ │ │ ├── split-large-constant.pb │ │ │ ├── split-large-constant.pbtxt │ │ │ ├── split-large-nodes.cpb │ │ │ ├── split-large-nodes.pb │ │ │ ├── split-large-nodes.pbtxt │ │ │ ├── split-lots-nodes.cpb │ │ │ ├── split-lots-nodes.pb │ │ │ ├── split-lots-nodes.pbtxt │ │ │ ├── split-standard.cpb │ │ │ ├── split-standard.pb │ │ │ ├── split-standard.pbtxt │ │ │ ├── split-tree.pbtxt │ │ │ ├── split_gen.py │ │ │ ├── split_graph_def_gen.py │ │ │ ├── split_saved_model_gen.py │ │ │ └── test_message.proto │ │ ├── util.py │ │ ├── util_test.py │ │ ├── version.py │ │ └── versions.proto │ ├── proto_text │ │ ├── BUILD │ │ ├── gen_proto_text_functions.cc │ │ ├── gen_proto_text_functions_lib.cc │ │ ├── gen_proto_text_functions_lib.h │ │ ├── gen_proto_text_functions_lib_test.cc │ │ ├── placeholder.txt │ │ └── test.proto │ ├── tensorflow_builder │ │ ├── compat_checker │ │ │ ├── BUILD │ │ │ ├── compat_checker.py │ │ │ ├── compat_checker_test.py │ │ │ ├── sample_config.ini │ │ │ └── test_config.ini │ │ └── config_detector │ │ │ ├── BUILD │ │ │ ├── config_detector.py │ │ │ └── data │ │ │ ├── __init__.py │ │ │ ├── cuda_compute_capability.py │ │ │ └── golden │ │ │ ├── BUILD │ │ │ └── compute_capability_golden.csv │ ├── test │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── file_name_test.py │ │ ├── gpu_info_lib.py │ │ ├── performance.bzl │ │ ├── run_and_gather_logs.py │ │ ├── run_and_gather_logs_lib.py │ │ ├── system_info.py │ │ ├── system_info_lib.py │ │ ├── upload_test_benchmarks.py │ │ └── upload_test_benchmarks_index.yaml │ ├── tf_sig_build_dockerfiles │ │ ├── Dockerfile │ │ ├── README.md │ │ ├── builder.devtoolset │ │ │ ├── build_devtoolset.sh │ │ │ ├── fixlinks.sh │ │ │ ├── glibc2.17-inline.patch │ │ │ └── rpm-patch.sh │ │ ├── builder.packages.txt │ │ ├── devel.bashrc │ │ ├── devel.packages.txt │ │ ├── devel.requirements.txt │ │ ├── devel.usertools │ │ │ ├── code_check_changed_files.bats │ │ │ ├── code_check_full.bats │ │ │ ├── cpu.bazelrc │ │ │ ├── cpu_gcc.bazelrc │ │ │ ├── get_test_list.sh │ │ │ ├── gpu.bazelrc │ │ │ ├── nvidia.bazelrc │ │ │ ├── rename_and_verify_wheels.sh │ │ │ ├── repack_libtensorflow.sh │ │ │ ├── squash_testlogs.py │ │ │ └── wheel_verification.bats │ │ ├── setup.cuda.sh │ │ ├── setup.packages.sh │ │ ├── setup.python.sh │ │ └── setup.sources.sh │ ├── tfg_graph_transforms │ │ ├── BUILD │ │ ├── tfg_graph_transforms_main.cc │ │ ├── utils.cc │ │ └── utils.h │ └── toolchains │ │ ├── BUILD │ │ ├── archives.bzl │ │ ├── clang6 │ │ ├── BUILD │ │ ├── CROSSTOOL.tpl │ │ ├── README.md │ │ ├── clang.BUILD │ │ └── repo.bzl │ │ ├── cpus │ │ ├── aarch64 │ │ │ ├── README.md │ │ │ ├── aarch64.bzl │ │ │ ├── aarch64_compiler_configure.bzl │ │ │ └── crosstool │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ └── cc_toolchain_config.bzl.tpl │ │ ├── arm │ │ │ ├── arm_compiler_configure.bzl │ │ │ ├── cc_config.bzl.tpl │ │ │ └── template.BUILD │ │ ├── py │ │ │ └── BUILD │ │ └── py3 │ │ │ └── BUILD │ │ ├── cross_compile │ │ ├── cc │ │ │ └── BUILD │ │ └── config │ │ │ └── BUILD │ │ ├── embedded │ │ ├── README.md │ │ └── arm-linux │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── aarch64-linux-toolchain.BUILD │ │ │ ├── arm_linux_toolchain_configure.bzl │ │ │ ├── armhf-linux-toolchain.BUILD │ │ │ ├── cc_config.bzl.tpl │ │ │ └── template.BUILD │ │ ├── java │ │ └── BUILD │ │ ├── python │ │ ├── BUILD │ │ └── python_repo.bzl │ │ ├── remote │ │ ├── BUILD │ │ ├── BUILD.tpl │ │ ├── configure.bzl │ │ └── execution.bzl.tpl │ │ ├── remote_config │ │ ├── BUILD │ │ ├── configs.bzl │ │ ├── containers.bzl │ │ └── rbe_config.bzl │ │ ├── win │ │ ├── BUILD │ │ ├── bazel_211 │ │ │ ├── BUILD │ │ │ ├── armeabi_cc_toolchain_config.bzl │ │ │ ├── builtin_include_directory_paths_msvc │ │ │ └── windows_cc_toolchain_config.bzl │ │ └── tf_win_05022023 │ │ │ ├── BUILD │ │ │ ├── armeabi_cc_toolchain_config.bzl │ │ │ ├── builtin_include_directory_paths_msvc │ │ │ ├── toolchain_image_info │ │ │ └── windows_cc_toolchain_config.bzl │ │ └── win_1803 │ │ ├── py38 │ │ └── BUILD │ │ └── py39 │ │ └── BUILD ├── virtual_root_template_v1.__init__.py ├── virtual_root_template_v2.__init__.py ├── workspace0.bzl ├── workspace1.bzl ├── workspace2.bzl └── workspace3.bzl ├── third_party ├── BUILD ├── FP16 │ ├── BUILD │ ├── FP16.BUILD │ └── workspace.bzl ├── __init__.py ├── absl │ ├── BUILD │ ├── com_google_absl.BUILD │ ├── invert_the_is_inline_bin.patch │ ├── system.BUILD │ ├── system.absl.algorithm.BUILD │ ├── system.absl.base.BUILD │ ├── system.absl.cleanup.BUILD │ ├── system.absl.container.BUILD │ ├── system.absl.debugging.BUILD │ ├── system.absl.flags.BUILD │ ├── system.absl.functional.BUILD │ ├── system.absl.hash.BUILD │ ├── system.absl.memory.BUILD │ ├── system.absl.meta.BUILD │ ├── system.absl.numeric.BUILD │ ├── system.absl.random.BUILD │ ├── system.absl.status.BUILD │ ├── system.absl.strings.BUILD │ ├── system.absl.synchronization.BUILD │ ├── system.absl.time.BUILD │ ├── system.absl.types.BUILD │ ├── system.absl.utility.BUILD │ └── workspace.bzl ├── android │ ├── BUILD │ ├── android.bzl.tpl │ ├── android_configure.BUILD.tpl │ └── android_configure.bzl ├── arm_neon_2_x86_sse.BUILD ├── benchmark │ ├── BUILD │ └── workspace.bzl ├── boringssl │ └── BUILD ├── clang_toolchain │ ├── BUILD │ ├── cc_configure_clang.bzl │ └── download_clang.bzl ├── codegen.BUILD ├── compute_library │ ├── BUILD │ ├── acl_thread_local_scheduler.patch │ ├── build_defs.bzl │ └── compute_library.patch ├── coremltools.BUILD ├── cub.BUILD ├── cudnn_frontend.BUILD ├── cudnn_frontend_header_fix.patch ├── curl.BUILD ├── cython.BUILD ├── dlpack │ ├── BUILD │ ├── dlpack.BUILD │ └── workspace.bzl ├── ducc │ ├── BUILD │ ├── ducc.BUILD │ ├── ducc0_custom_lowlevel_threading.h │ ├── fft.cc │ ├── fft.h │ ├── threading.cc │ ├── threading.h │ └── workspace.bzl ├── eigen3 │ ├── BUILD │ ├── LICENSE │ ├── eigen_archive.BUILD │ └── workspace.bzl ├── farmhash │ ├── BUILD │ ├── farmhash.BUILD │ ├── farmhash_gpu.BUILD │ ├── farmhash_support_cuda.patch │ └── workspace.bzl ├── fft2d │ ├── BUILD │ ├── LICENSE │ ├── fft.h │ ├── fft2d.BUILD │ └── fft2d.h ├── flatbuffers │ ├── BUILD │ ├── BUILD.system │ ├── build_defs.bzl │ ├── flatbuffers.BUILD │ └── workspace.bzl ├── gemmlowp │ ├── BUILD │ └── workspace.bzl ├── gif.BUILD ├── gif_fix_image_counter.patch ├── gif_fix_strtok_r.patch ├── git │ ├── BUILD │ ├── BUILD.tpl │ └── git_configure.bzl ├── googleapis │ ├── BUILD │ ├── build_rules.bzl │ ├── googleapis.BUILD │ └── repository_rules.bzl ├── gpus │ ├── BUILD │ ├── check_cuda_libs.py │ ├── crosstool │ │ ├── BUILD │ │ ├── BUILD.rocm.tpl │ │ ├── BUILD.tpl │ │ ├── LICENSE │ │ ├── cc_toolchain_config.bzl.tpl │ │ ├── clang │ │ │ └── bin │ │ │ │ ├── crosstool_wrapper_driver_is_not_gcc.tpl │ │ │ │ └── crosstool_wrapper_driver_rocm.tpl │ │ ├── hipcc_cc_toolchain_config.bzl.tpl │ │ └── windows │ │ │ └── msvc_wrapper_for_nvcc.py.tpl │ ├── cuda │ │ ├── BUILD │ │ ├── BUILD.tpl │ │ ├── BUILD.windows.tpl │ │ ├── LICENSE │ │ ├── build_defs.bzl.tpl │ │ ├── cuda_config.h.tpl │ │ └── cuda_config.py.tpl │ ├── cuda_configure.bzl │ ├── find_cuda_config.py │ ├── find_rocm_config.py │ ├── local_config_cuda.BUILD │ ├── rocm │ │ ├── BUILD │ │ ├── BUILD.tpl │ │ ├── build_defs.bzl.tpl │ │ └── rocm_config.h.tpl │ └── rocm_configure.bzl ├── grpc │ ├── BUILD │ ├── generate_cc_env_fix.patch │ ├── register_go_toolchain.patch │ └── upb_platform_fix.patch ├── hexagon │ ├── BUILD │ ├── hexagon.BUILD │ └── workspace.bzl ├── highwayhash │ ├── BUILD │ ├── highwayhash.BUILD │ └── workspace.bzl ├── hwloc │ ├── BUILD │ ├── BUILD.system │ ├── hwloc.BUILD │ ├── static-components.h │ └── workspace.bzl ├── icu │ ├── BUILD │ ├── BUILD.system │ ├── data │ │ ├── BUILD.bazel │ │ ├── LICENSE │ │ ├── icu_conversion_data.c.gz.aaa │ │ ├── icu_conversion_data.c.gz.aab │ │ ├── icu_conversion_data.c.gz.aac │ │ ├── icu_conversion_data.c.gz.aad │ │ ├── icu_conversion_data.c.gz.aae │ │ ├── icu_conversion_data.c.gz.aaf │ │ ├── icu_conversion_data.c.gz.aag │ │ ├── icu_conversion_data.c.gz.aah │ │ ├── icu_conversion_data.c.gz.aai │ │ ├── icu_conversion_data.c.gz.aaj │ │ ├── icu_conversion_data.c.gz.aak │ │ ├── icu_conversion_data.c.gz.aal │ │ ├── icu_conversion_data.c.gz.aam │ │ ├── icu_conversion_data.c.gz.aan │ │ ├── icu_conversion_data.c.gz.aao │ │ ├── icu_conversion_data.c.gz.aap │ │ ├── icu_conversion_data.c.gz.aaq │ │ ├── icu_conversion_data.c.gz.aar │ │ ├── icu_conversion_data.c.gz.aas │ │ ├── icu_conversion_data.c.gz.aat │ │ ├── icu_conversion_data.c.gz.aau │ │ ├── icu_conversion_data.c.gz.aav │ │ ├── icu_conversion_data.c.gz.aaw │ │ ├── icu_conversion_data.c.gz.aax │ │ ├── icu_conversion_data.c.gz.aay │ │ ├── icu_conversion_data.c.gz.aaz │ │ ├── icu_conversion_data.c.gz.aba │ │ ├── icu_conversion_data.c.gz.abb │ │ ├── icu_conversion_data.c.gz.abc │ │ ├── icu_conversion_data.c.gz.abd │ │ ├── icu_conversion_data.c.gz.abe │ │ ├── icu_conversion_data.c.gz.abf │ │ ├── icu_conversion_data.c.gz.abg │ │ ├── icu_conversion_data.c.gz.abh │ │ ├── icu_conversion_data.c.gz.abi │ │ ├── icu_conversion_data.c.gz.abj │ │ ├── icu_conversion_data.c.gz.abk │ │ ├── icu_conversion_data.c.gz.abl │ │ ├── icu_conversion_data.c.gz.abm │ │ ├── icu_conversion_data.c.gz.abn │ │ ├── icu_conversion_data.c.gz.abo │ │ ├── icu_conversion_data.c.gz.abp │ │ ├── icu_conversion_data.c.gz.abq │ │ ├── icu_conversion_data.c.gz.abr │ │ ├── icu_conversion_data.c.gz.abs │ │ ├── icu_conversion_data.c.gz.abt │ │ ├── icu_conversion_data.c.gz.abu │ │ ├── icu_conversion_data.c.gz.abv │ │ ├── icu_conversion_data.c.gz.abw │ │ ├── icu_conversion_data.c.gz.abx │ │ ├── icu_conversion_data.c.gz.aby │ │ ├── icu_conversion_data.c.gz.abz │ │ ├── icu_conversion_data.c.gz.aca │ │ ├── icu_conversion_data.c.gz.acb │ │ ├── icu_conversion_data.c.gz.acc │ │ ├── icu_conversion_data.c.gz.acd │ │ ├── icu_conversion_data.c.gz.ace │ │ ├── icu_conversion_data.c.gz.acf │ │ ├── icu_conversion_data.c.gz.acg │ │ ├── icu_conversion_data.c.gz.ach │ │ ├── icu_conversion_data.c.gz.aci │ │ ├── icu_conversion_data.c.gz.acj │ │ ├── icu_conversion_data.c.gz.ack │ │ ├── icu_conversion_data.c.gz.acl │ │ ├── icu_conversion_data.c.gz.acm │ │ ├── icu_conversion_data.c.gz.acn │ │ ├── icu_conversion_data.c.gz.aco │ │ ├── icu_conversion_data.c.gz.acp │ │ ├── icu_conversion_data.c.gz.acq │ │ ├── icu_conversion_data.c.gz.acr │ │ ├── icu_conversion_data.c.gz.acs │ │ ├── icu_conversion_data.c.gz.act │ │ ├── icu_conversion_data.c.gz.acu │ │ ├── icu_conversion_data.c.gz.acv │ │ ├── icu_conversion_data.c.gz.acw │ │ ├── icu_conversion_data.c.gz.acx │ │ ├── icu_conversion_data.c.gz.acy │ │ ├── icu_conversion_data.c.gz.acz │ │ ├── icu_conversion_data.c.gz.ada │ │ ├── icu_conversion_data.c.gz.adb │ │ ├── icu_conversion_data.c.gz.adc │ │ ├── icu_conversion_data.c.gz.add │ │ ├── icu_conversion_data.c.gz.ade │ │ ├── icu_conversion_data.c.gz.adf │ │ ├── icu_conversion_data.c.gz.adg │ │ └── icu_conversion_data.c.gz.adh │ ├── icu.BUILD │ ├── udata.patch │ └── workspace.bzl ├── implib_so │ ├── BUILD │ ├── get_symbols.py │ ├── implib_so.BUILD │ ├── make_stub.py │ └── workspace.bzl ├── jpeg │ ├── BUILD │ ├── BUILD.system │ ├── jpeg.BUILD │ ├── jpeg_helpers.BUILD.bazel │ └── workspace.bzl ├── kissfft │ ├── BUILD │ ├── kissfft.BUILD │ └── workspace.bzl ├── libprotobuf_mutator │ ├── BUILD │ ├── libprotobuf_mutator.BUILD │ └── workspace.bzl ├── linenoise.BUILD ├── llvm │ ├── BUILD │ ├── build.patch │ ├── generated.patch │ ├── llvm.BUILD │ ├── mathextras.patch │ ├── run_lit.sh │ ├── setup.bzl │ ├── toolchains.patch │ ├── workspace.bzl │ └── zstd.patch ├── llvm_openmp │ ├── BUILD │ ├── cmake_vars.bzl │ ├── expand_cmake_vars.py │ ├── openmp.bzl │ └── openmp_switch_default_patch.patch ├── mkl │ ├── BUILD │ └── build_defs.bzl ├── mkl_dnn │ ├── BUILD │ ├── LICENSE │ ├── build_defs.bzl │ ├── mkldnn_acl.BUILD │ ├── mkldnn_v1.BUILD │ ├── onednn_acl_bf16_capability_detection_for_ubuntu20.04.patch │ ├── onednn_acl_fp32_bf16_reorder.patch │ ├── onednn_acl_reorder.patch │ ├── onednn_acl_thread_local_scheduler.patch │ └── onednn_acl_threadcap.patch ├── mpi │ └── .gitignore ├── nasm │ ├── BUILD │ ├── BUILD.system │ ├── config.h │ ├── nasm.BUILD │ └── workspace.bzl ├── nccl │ ├── BUILD │ ├── LICENSE │ ├── archive.BUILD │ ├── archive.patch │ ├── build_defs.bzl.tpl │ ├── nccl_configure.bzl │ └── system.BUILD.tpl ├── net_zstd.BUILD ├── nlohmann_json.BUILD ├── nsync.patch ├── nvtx.BUILD ├── opencl_headers │ ├── BUILD │ ├── opencl_headers.BUILD │ └── workspace.bzl ├── ortools │ ├── BUILD │ ├── bliss.BUILD │ ├── glpk.BUILD │ ├── ortools.patch │ ├── scip.BUILD │ └── scip.patch ├── pasta │ ├── BUILD │ ├── BUILD.system │ ├── build_defs.bzl │ ├── pasta.BUILD │ └── workspace.bzl ├── png.BUILD ├── png_fix_rpi.patch ├── pprof.BUILD ├── protobuf │ ├── BUILD │ └── protobuf.patch ├── py │ ├── BUILD │ ├── BUILD.tpl │ ├── ml_dtypes │ │ ├── BUILD │ │ ├── LICENSE │ │ ├── ml_dtypes.BUILD │ │ ├── ml_dtypes.tests.BUILD │ │ └── workspace.bzl │ ├── non_hermetic │ │ ├── BUILD │ │ ├── BUILD.tpl │ │ ├── README │ │ ├── ml_dtypes │ │ │ ├── BUILD │ │ │ ├── LICENSE │ │ │ ├── ml_dtypes.BUILD │ │ │ ├── ml_dtypes.tests.BUILD │ │ │ └── workspace.bzl │ │ ├── numpy │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ └── tf_numpy_api │ │ │ │ ├── BUILD │ │ │ │ ├── tensorflow.experimental.numpy.ndarray.pbtxt │ │ │ │ ├── tensorflow.experimental.numpy.pbtxt │ │ │ │ └── tensorflow.experimental.numpy.random.pbtxt │ │ └── python_configure.bzl │ ├── numpy │ │ ├── BUILD │ │ ├── LICENSE │ │ ├── README.md │ │ └── tf_numpy_api │ │ │ ├── BUILD │ │ │ ├── tensorflow.experimental.numpy.ndarray.pbtxt │ │ │ ├── tensorflow.experimental.numpy.pbtxt │ │ │ └── tensorflow.experimental.numpy.random.pbtxt │ └── python_configure.bzl ├── pybind11.BUILD ├── pybind11_abseil │ ├── BUILD │ ├── remove_license.patch │ └── workspace.bzl ├── pybind11_bazel │ ├── BUILD │ ├── pybind11_bazel.patch │ └── workspace.bzl ├── pybind11_protobuf │ ├── BUILD │ └── remove_license.patch ├── python_runtime │ └── BUILD ├── remote_config │ ├── BUILD │ ├── BUILD.tpl │ ├── common.bzl │ └── remote_platform_configure.bzl ├── repo.bzl ├── riegeli_fix.patch ├── ruy │ ├── BUILD │ └── workspace.bzl ├── six.BUILD ├── snappy.BUILD ├── sobol_data │ ├── BUILD │ ├── sobol_data.BUILD │ └── workspace.bzl ├── sqlite.BUILD ├── stablehlo │ ├── BUILD │ ├── temporary.patch │ └── workspace.bzl ├── systemlibs │ ├── BUILD │ ├── BUILD.tpl │ ├── absl_py.BUILD │ ├── absl_py.absl.BUILD │ ├── absl_py.absl.flags.BUILD │ ├── absl_py.absl.logging.BUILD │ ├── absl_py.absl.testing.BUILD │ ├── boringssl.BUILD │ ├── build_defs.bzl.tpl │ ├── curl.BUILD │ ├── cython.BUILD │ ├── double_conversion.BUILD │ ├── gif.BUILD │ ├── google_cloud_cpp.BUILD │ ├── google_cloud_cpp.google.cloud.bigtable.BUILD │ ├── grpc.BUILD │ ├── grpc.bazel.cc_grpc_library.bzl │ ├── grpc.bazel.generate_cc.bzl │ ├── grpc.bazel.grpc_deps.bzl │ ├── grpc.bazel.grpc_extra_deps.bzl │ ├── grpc.bazel.protobuf.bzl │ ├── jsoncpp.BUILD │ ├── lmdb.BUILD │ ├── nsync.BUILD │ ├── png.BUILD │ ├── protobuf.BUILD │ ├── protobuf.bzl │ ├── protobuf_deps.bzl │ ├── pybind11.BUILD │ ├── re2.BUILD │ ├── six.BUILD │ ├── snappy.BUILD │ ├── sqlite.BUILD │ ├── syslibs_configure.bzl │ ├── tblib.BUILD │ ├── typing_extensions.BUILD │ ├── wrapt.BUILD │ └── zlib.BUILD ├── tensorrt │ ├── BUILD │ ├── BUILD.tpl │ ├── LICENSE │ ├── build_defs.bzl.tpl │ ├── plugin.BUILD.tpl │ ├── plugin │ │ ├── BUILD │ │ └── tensorrt_oss.patch │ ├── tensorrt │ │ ├── include │ │ │ └── tensorrt_config.h.tpl │ │ └── tensorrt_config.py.tpl │ ├── tensorrt_configure.bzl │ └── workspace.bzl ├── tf_runtime │ ├── BUILD │ └── workspace.bzl ├── tf_toolchains.BUILD ├── tflite_mobilenet.BUILD ├── tflite_mobilenet_float.BUILD ├── tflite_mobilenet_quant.BUILD ├── tflite_ovic_testdata.BUILD ├── tflite_smartreply.BUILD ├── triton │ ├── BUILD │ ├── b304456327.patch │ ├── cl568176943.patch │ └── workspace.bzl ├── vulkan_headers │ ├── BUILD │ ├── tensorflow │ │ └── vulkan_hpp_dispatch_loader_dynamic.cc │ ├── vulkan_headers.BUILD │ └── workspace.bzl ├── xla │ ├── .bazelrc │ ├── .bazelversion │ ├── .clang-tidy │ ├── .github │ │ ├── CODEOWNERS │ │ └── workflows │ │ │ ├── benchmark.yml │ │ │ ├── buildifier.yml │ │ │ ├── check_contents.yml │ │ │ ├── github_api.py │ │ │ ├── rollback_notification.py │ │ │ ├── rollback_notification.yml │ │ │ ├── scorecards-analysis.yml │ │ │ ├── trusted-partners.yml │ │ │ └── trusted_partners.js │ ├── .gitignore │ ├── .kokoro │ │ ├── generate_index_html.sh │ │ ├── jax │ │ │ └── build.sh │ │ ├── linux │ │ │ └── build.sh │ │ ├── macos │ │ │ └── build.sh │ │ └── windows │ │ │ ├── build.bat │ │ │ └── windows_build.sh │ ├── BUILD.bazel │ ├── CONTRIBUTING.md │ ├── LICENSE │ ├── README.md │ ├── WORKSPACE │ ├── build_tools │ │ ├── docker │ │ │ ├── context │ │ │ │ ├── .bazelversion │ │ │ │ ├── install_bazel.sh │ │ │ │ ├── install_python_deps.sh │ │ │ │ └── python_build_requirements.txt │ │ │ └── dockerfiles │ │ │ │ └── benchmarking.Dockerfile │ │ ├── github_actions │ │ │ └── build_xla.sh │ │ └── lint │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── check_contents.py │ │ │ ├── check_contents_test.py │ │ │ ├── diff_parser.py │ │ │ ├── diff_parser_test.py │ │ │ ├── generate_compile_commands.py │ │ │ └── testdata │ │ │ ├── bad_cc.diff │ │ │ ├── crosstool.diff │ │ │ └── important_cc.diff │ ├── configure │ ├── configure.cmd │ ├── configure.py │ ├── docs │ │ ├── _book.yaml │ │ ├── aliasing.md │ │ ├── architecture.md │ │ ├── async_ops.md │ │ ├── broadcasting.md │ │ ├── build_from_source.md │ │ ├── code_reviews.md │ │ ├── copybara.md │ │ ├── custom_call.md │ │ ├── developer_guide.md │ │ ├── developing_new_backend.md │ │ ├── images │ │ │ ├── batch_group_counts.svg │ │ │ ├── how-does-xla-work.png │ │ │ ├── ops_2d_matrix.png │ │ │ ├── ops_alltoall.png │ │ │ ├── ops_concatenate.png │ │ │ ├── ops_pad.png │ │ │ ├── ops_reduce_from_2d_matrix.png │ │ │ ├── ops_reduce_from_3d_matrix.png │ │ │ ├── ops_reduce_window.png │ │ │ ├── ops_reduce_window_stride.png │ │ │ ├── ops_scatter_to_selected_window_element.png │ │ │ ├── ops_while.png │ │ │ ├── ops_xla_gather_0.svg │ │ │ ├── ops_xla_gather_1.svg │ │ │ ├── ops_xla_gather_2.svg │ │ │ ├── send_recv_order.png │ │ │ ├── send_recv_schedule.png │ │ │ ├── tf_xla_performance.png │ │ │ ├── xla_array_layout_figure1.png │ │ │ └── xla_array_layout_figure2.png │ │ ├── index.md │ │ ├── operation_semantics.md │ │ ├── shapes.md │ │ └── tiled_layout.md │ ├── opensource_only.files │ ├── third_party │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── compute_library │ │ │ ├── BUILD │ │ │ ├── acl_thread_local_scheduler.patch │ │ │ ├── build_defs.bzl │ │ │ └── compute_library.patch │ │ ├── cudnn_frontend.BUILD │ │ ├── cudnn_frontend_header_fix.patch │ │ ├── dlpack │ │ │ ├── BUILD │ │ │ ├── dlpack.BUILD │ │ │ └── workspace.bzl │ │ ├── llvm_openmp │ │ │ ├── BUILD │ │ │ ├── cmake_vars.bzl │ │ │ ├── expand_cmake_vars.py │ │ │ ├── openmp.bzl │ │ │ └── openmp_switch_default_patch.patch │ │ ├── ortools │ │ │ ├── BUILD │ │ │ ├── bliss.BUILD │ │ │ ├── glpk.BUILD │ │ │ ├── ortools.patch │ │ │ ├── scip.BUILD │ │ │ └── scip.patch │ │ ├── py │ │ │ ├── BUILD │ │ │ ├── BUILD.tpl │ │ │ ├── ml_dtypes │ │ │ │ ├── BUILD │ │ │ │ ├── LICENSE │ │ │ │ ├── ml_dtypes.BUILD │ │ │ │ ├── ml_dtypes.tests.BUILD │ │ │ │ └── workspace.bzl │ │ │ ├── non_hermetic │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── README │ │ │ │ ├── ml_dtypes │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── ml_dtypes.BUILD │ │ │ │ │ ├── ml_dtypes.tests.BUILD │ │ │ │ │ └── workspace.bzl │ │ │ │ ├── numpy │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ └── tf_numpy_api │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── tensorflow.experimental.numpy.ndarray.pbtxt │ │ │ │ │ │ ├── tensorflow.experimental.numpy.pbtxt │ │ │ │ │ │ └── tensorflow.experimental.numpy.random.pbtxt │ │ │ │ └── python_configure.bzl │ │ │ ├── numpy │ │ │ │ ├── BUILD │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ └── tf_numpy_api │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── tensorflow.experimental.numpy.ndarray.pbtxt │ │ │ │ │ ├── tensorflow.experimental.numpy.pbtxt │ │ │ │ │ └── tensorflow.experimental.numpy.random.pbtxt │ │ │ └── python_configure.bzl │ │ ├── python_runtime │ │ │ └── BUILD │ │ ├── repo.bzl │ │ ├── stablehlo │ │ │ ├── BUILD │ │ │ ├── temporary.patch │ │ │ └── workspace.bzl │ │ ├── triton │ │ │ ├── BUILD │ │ │ ├── b304456327.patch │ │ │ ├── cl568176943.patch │ │ │ └── workspace.bzl │ │ └── tsl │ │ │ ├── .bazelrc │ │ │ ├── .bazelversion │ │ │ ├── .github │ │ │ └── workflows │ │ │ │ └── scorecards-analysis.yml │ │ │ ├── .kokoro │ │ │ ├── generate_index_html.sh │ │ │ ├── linux │ │ │ │ ├── build.sh │ │ │ │ └── cpu │ │ │ │ │ ├── build_cpu.cfg │ │ │ │ │ └── common.cfg │ │ │ ├── macos │ │ │ │ ├── build.sh │ │ │ │ └── cpu │ │ │ │ │ ├── common.cfg │ │ │ │ │ └── cpu_py39_full.cfg │ │ │ └── windows │ │ │ │ ├── build.bat │ │ │ │ ├── cpu │ │ │ │ ├── build_cpu_py39.cfg │ │ │ │ └── common.cfg │ │ │ │ └── windows_build.sh │ │ │ ├── BUILD.bazel │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── WORKSPACE │ │ │ ├── opensource_only.files │ │ │ ├── third_party │ │ │ ├── BUILD │ │ │ ├── __init__.py │ │ │ ├── absl │ │ │ │ ├── BUILD │ │ │ │ ├── com_google_absl.BUILD │ │ │ │ ├── invert_the_is_inline_bin.patch │ │ │ │ ├── system.BUILD │ │ │ │ ├── system.absl.algorithm.BUILD │ │ │ │ ├── system.absl.base.BUILD │ │ │ │ ├── system.absl.cleanup.BUILD │ │ │ │ ├── system.absl.container.BUILD │ │ │ │ ├── system.absl.debugging.BUILD │ │ │ │ ├── system.absl.flags.BUILD │ │ │ │ ├── system.absl.functional.BUILD │ │ │ │ ├── system.absl.hash.BUILD │ │ │ │ ├── system.absl.memory.BUILD │ │ │ │ ├── system.absl.meta.BUILD │ │ │ │ ├── system.absl.numeric.BUILD │ │ │ │ ├── system.absl.random.BUILD │ │ │ │ ├── system.absl.status.BUILD │ │ │ │ ├── system.absl.strings.BUILD │ │ │ │ ├── system.absl.synchronization.BUILD │ │ │ │ ├── system.absl.time.BUILD │ │ │ │ ├── system.absl.types.BUILD │ │ │ │ ├── system.absl.utility.BUILD │ │ │ │ └── workspace.bzl │ │ │ ├── benchmark │ │ │ │ ├── BUILD │ │ │ │ └── workspace.bzl │ │ │ ├── clang_toolchain │ │ │ │ ├── BUILD │ │ │ │ ├── cc_configure_clang.bzl │ │ │ │ └── download_clang.bzl │ │ │ ├── compute_library │ │ │ │ ├── BUILD │ │ │ │ ├── acl_thread_local_scheduler.patch │ │ │ │ ├── build_defs.bzl │ │ │ │ └── compute_library.patch │ │ │ ├── curl.BUILD │ │ │ ├── cython.BUILD │ │ │ ├── ducc │ │ │ │ ├── BUILD │ │ │ │ ├── ducc.BUILD │ │ │ │ ├── ducc0_custom_lowlevel_threading.h │ │ │ │ ├── fft.cc │ │ │ │ ├── fft.h │ │ │ │ ├── threading.cc │ │ │ │ ├── threading.h │ │ │ │ └── workspace.bzl │ │ │ ├── eigen3 │ │ │ │ ├── BUILD │ │ │ │ ├── LICENSE │ │ │ │ ├── eigen_archive.BUILD │ │ │ │ └── workspace.bzl │ │ │ ├── farmhash │ │ │ │ ├── BUILD │ │ │ │ ├── farmhash.BUILD │ │ │ │ ├── farmhash_gpu.BUILD │ │ │ │ ├── farmhash_support_cuda.patch │ │ │ │ └── workspace.bzl │ │ │ ├── gemmlowp │ │ │ │ ├── BUILD │ │ │ │ └── workspace.bzl │ │ │ ├── gif.BUILD │ │ │ ├── gif_fix_strtok_r.patch │ │ │ ├── git │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ └── git_configure.bzl │ │ │ ├── gpus │ │ │ │ ├── BUILD │ │ │ │ ├── check_cuda_libs.py │ │ │ │ ├── crosstool │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── BUILD.rocm.tpl │ │ │ │ │ ├── BUILD.tpl │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── cc_toolchain_config.bzl.tpl │ │ │ │ │ ├── clang │ │ │ │ │ │ └── bin │ │ │ │ │ │ │ ├── crosstool_wrapper_driver_is_not_gcc.tpl │ │ │ │ │ │ │ └── crosstool_wrapper_driver_rocm.tpl │ │ │ │ │ ├── hipcc_cc_toolchain_config.bzl.tpl │ │ │ │ │ └── windows │ │ │ │ │ │ └── msvc_wrapper_for_nvcc.py.tpl │ │ │ │ ├── cuda │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── BUILD.tpl │ │ │ │ │ ├── BUILD.windows.tpl │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── build_defs.bzl.tpl │ │ │ │ │ ├── cuda_config.h.tpl │ │ │ │ │ └── cuda_config.py.tpl │ │ │ │ ├── cuda_configure.bzl │ │ │ │ ├── find_cuda_config.py │ │ │ │ ├── find_rocm_config.py │ │ │ │ ├── local_config_cuda.BUILD │ │ │ │ ├── rocm │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── BUILD.tpl │ │ │ │ │ ├── build_defs.bzl.tpl │ │ │ │ │ └── rocm_config.h.tpl │ │ │ │ └── rocm_configure.bzl │ │ │ ├── grpc │ │ │ │ ├── BUILD │ │ │ │ ├── generate_cc_env_fix.patch │ │ │ │ ├── register_go_toolchain.patch │ │ │ │ └── upb_platform_fix.patch │ │ │ ├── hwloc │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.system │ │ │ │ ├── hwloc.BUILD │ │ │ │ ├── static-components.h │ │ │ │ └── workspace.bzl │ │ │ ├── implib_so │ │ │ │ ├── BUILD │ │ │ │ ├── get_symbols.py │ │ │ │ ├── implib_so.BUILD │ │ │ │ ├── make_stub.py │ │ │ │ └── workspace.bzl │ │ │ ├── jpeg │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.system │ │ │ │ ├── jpeg.BUILD │ │ │ │ ├── jpeg_helpers.BUILD.bazel │ │ │ │ └── workspace.bzl │ │ │ ├── llvm_openmp │ │ │ │ ├── BUILD │ │ │ │ ├── cmake_vars.bzl │ │ │ │ ├── expand_cmake_vars.py │ │ │ │ ├── openmp.bzl │ │ │ │ └── openmp_switch_default_patch.patch │ │ │ ├── mkl │ │ │ │ ├── BUILD │ │ │ │ └── build_defs.bzl │ │ │ ├── mkl_dnn │ │ │ │ ├── BUILD │ │ │ │ ├── LICENSE │ │ │ │ ├── build_defs.bzl │ │ │ │ ├── mkldnn_acl.BUILD │ │ │ │ ├── mkldnn_v1.BUILD │ │ │ │ ├── onednn_acl_bf16_capability_detection_for_ubuntu20.04.patch │ │ │ │ ├── onednn_acl_fp32_bf16_reorder.patch │ │ │ │ ├── onednn_acl_reorder.patch │ │ │ │ ├── onednn_acl_thread_local_scheduler.patch │ │ │ │ └── onednn_acl_threadcap.patch │ │ │ ├── nasm │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.system │ │ │ │ ├── config.h │ │ │ │ ├── nasm.BUILD │ │ │ │ └── workspace.bzl │ │ │ ├── nccl │ │ │ │ ├── BUILD │ │ │ │ ├── LICENSE │ │ │ │ ├── archive.BUILD │ │ │ │ ├── archive.patch │ │ │ │ ├── build_defs.bzl.tpl │ │ │ │ ├── nccl_configure.bzl │ │ │ │ └── system.BUILD.tpl │ │ │ ├── nsync.patch │ │ │ ├── nvtx.BUILD │ │ │ ├── png.BUILD │ │ │ ├── png_fix_rpi.patch │ │ │ ├── protobuf │ │ │ │ ├── BUILD │ │ │ │ └── protobuf.patch │ │ │ ├── py │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── ml_dtypes │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── ml_dtypes.BUILD │ │ │ │ │ ├── ml_dtypes.tests.BUILD │ │ │ │ │ └── workspace.bzl │ │ │ │ ├── non_hermetic │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── BUILD.tpl │ │ │ │ │ ├── README │ │ │ │ │ ├── ml_dtypes │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── LICENSE │ │ │ │ │ │ ├── ml_dtypes.BUILD │ │ │ │ │ │ ├── ml_dtypes.tests.BUILD │ │ │ │ │ │ └── workspace.bzl │ │ │ │ │ ├── numpy │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ └── tf_numpy_api │ │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ │ ├── tensorflow.experimental.numpy.ndarray.pbtxt │ │ │ │ │ │ │ ├── tensorflow.experimental.numpy.pbtxt │ │ │ │ │ │ │ └── tensorflow.experimental.numpy.random.pbtxt │ │ │ │ │ └── python_configure.bzl │ │ │ │ ├── numpy │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── README.md │ │ │ │ │ └── tf_numpy_api │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── tensorflow.experimental.numpy.ndarray.pbtxt │ │ │ │ │ │ ├── tensorflow.experimental.numpy.pbtxt │ │ │ │ │ │ └── tensorflow.experimental.numpy.random.pbtxt │ │ │ │ └── python_configure.bzl │ │ │ ├── pybind11.BUILD │ │ │ ├── pybind11_abseil │ │ │ │ ├── BUILD │ │ │ │ ├── remove_license.patch │ │ │ │ └── workspace.bzl │ │ │ ├── pybind11_bazel │ │ │ │ ├── BUILD │ │ │ │ ├── pybind11_bazel.patch │ │ │ │ └── workspace.bzl │ │ │ ├── python_runtime │ │ │ │ └── BUILD │ │ │ ├── remote_config │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── common.bzl │ │ │ │ └── remote_platform_configure.bzl │ │ │ ├── repo.bzl │ │ │ ├── six.BUILD │ │ │ ├── snappy.BUILD │ │ │ ├── systemlibs │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── absl_py.BUILD │ │ │ │ ├── absl_py.absl.BUILD │ │ │ │ ├── absl_py.absl.flags.BUILD │ │ │ │ ├── absl_py.absl.logging.BUILD │ │ │ │ ├── absl_py.absl.testing.BUILD │ │ │ │ ├── boringssl.BUILD │ │ │ │ ├── build_defs.bzl.tpl │ │ │ │ ├── curl.BUILD │ │ │ │ ├── cython.BUILD │ │ │ │ ├── double_conversion.BUILD │ │ │ │ ├── gif.BUILD │ │ │ │ ├── google_cloud_cpp.BUILD │ │ │ │ ├── google_cloud_cpp.google.cloud.bigtable.BUILD │ │ │ │ ├── grpc.BUILD │ │ │ │ ├── grpc.bazel.cc_grpc_library.bzl │ │ │ │ ├── grpc.bazel.generate_cc.bzl │ │ │ │ ├── grpc.bazel.grpc_deps.bzl │ │ │ │ ├── grpc.bazel.grpc_extra_deps.bzl │ │ │ │ ├── grpc.bazel.protobuf.bzl │ │ │ │ ├── jsoncpp.BUILD │ │ │ │ ├── lmdb.BUILD │ │ │ │ ├── nsync.BUILD │ │ │ │ ├── png.BUILD │ │ │ │ ├── protobuf.BUILD │ │ │ │ ├── protobuf.bzl │ │ │ │ ├── protobuf_deps.bzl │ │ │ │ ├── pybind11.BUILD │ │ │ │ ├── re2.BUILD │ │ │ │ ├── six.BUILD │ │ │ │ ├── snappy.BUILD │ │ │ │ ├── sqlite.BUILD │ │ │ │ ├── syslibs_configure.bzl │ │ │ │ ├── tblib.BUILD │ │ │ │ ├── typing_extensions.BUILD │ │ │ │ ├── wrapt.BUILD │ │ │ │ └── zlib.BUILD │ │ │ ├── tensorrt │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── LICENSE │ │ │ │ ├── build_defs.bzl.tpl │ │ │ │ ├── plugin.BUILD.tpl │ │ │ │ ├── plugin │ │ │ │ │ ├── BUILD │ │ │ │ │ └── tensorrt_oss.patch │ │ │ │ ├── tensorrt │ │ │ │ │ ├── include │ │ │ │ │ │ └── tensorrt_config.h.tpl │ │ │ │ │ └── tensorrt_config.py.tpl │ │ │ │ ├── tensorrt_configure.bzl │ │ │ │ └── workspace.bzl │ │ │ ├── tf_runtime │ │ │ │ ├── BUILD │ │ │ │ └── workspace.bzl │ │ │ └── zlib.BUILD │ │ │ ├── tools │ │ │ ├── def_file_filter │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── def_file_filter.py.tpl │ │ │ │ ├── def_file_filter_configure.bzl │ │ │ │ └── symbols_pybind.txt │ │ │ └── toolchains │ │ │ │ ├── BUILD │ │ │ │ ├── archives.bzl │ │ │ │ ├── clang6 │ │ │ │ ├── BUILD │ │ │ │ ├── CROSSTOOL.tpl │ │ │ │ ├── README.md │ │ │ │ ├── clang.BUILD │ │ │ │ └── repo.bzl │ │ │ │ ├── cpus │ │ │ │ ├── aarch64 │ │ │ │ │ ├── README.md │ │ │ │ │ ├── aarch64.bzl │ │ │ │ │ ├── aarch64_compiler_configure.bzl │ │ │ │ │ └── crosstool │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── BUILD.tpl │ │ │ │ │ │ └── cc_toolchain_config.bzl.tpl │ │ │ │ ├── arm │ │ │ │ │ ├── arm_compiler_configure.bzl │ │ │ │ │ ├── cc_config.bzl.tpl │ │ │ │ │ └── template.BUILD │ │ │ │ ├── py │ │ │ │ │ └── BUILD │ │ │ │ └── py3 │ │ │ │ │ └── BUILD │ │ │ │ ├── cross_compile │ │ │ │ ├── cc │ │ │ │ │ └── BUILD │ │ │ │ └── config │ │ │ │ │ └── BUILD │ │ │ │ ├── embedded │ │ │ │ ├── README.md │ │ │ │ └── arm-linux │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── README.md │ │ │ │ │ ├── aarch64-linux-toolchain.BUILD │ │ │ │ │ ├── arm_linux_toolchain_configure.bzl │ │ │ │ │ ├── armhf-linux-toolchain.BUILD │ │ │ │ │ ├── cc_config.bzl.tpl │ │ │ │ │ └── template.BUILD │ │ │ │ ├── java │ │ │ │ └── BUILD │ │ │ │ ├── python │ │ │ │ ├── BUILD │ │ │ │ └── python_repo.bzl │ │ │ │ ├── remote │ │ │ │ ├── BUILD │ │ │ │ ├── BUILD.tpl │ │ │ │ ├── configure.bzl │ │ │ │ └── execution.bzl.tpl │ │ │ │ ├── remote_config │ │ │ │ ├── BUILD │ │ │ │ ├── configs.bzl │ │ │ │ ├── containers.bzl │ │ │ │ └── rbe_config.bzl │ │ │ │ ├── win │ │ │ │ ├── BUILD │ │ │ │ ├── bazel_211 │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── armeabi_cc_toolchain_config.bzl │ │ │ │ │ ├── builtin_include_directory_paths_msvc │ │ │ │ │ └── windows_cc_toolchain_config.bzl │ │ │ │ └── tf_win_05022023 │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── armeabi_cc_toolchain_config.bzl │ │ │ │ │ ├── builtin_include_directory_paths_msvc │ │ │ │ │ ├── toolchain_image_info │ │ │ │ │ └── windows_cc_toolchain_config.bzl │ │ │ │ └── win_1803 │ │ │ │ ├── py38 │ │ │ │ └── BUILD │ │ │ │ └── py39 │ │ │ │ └── BUILD │ │ │ ├── tsl │ │ │ ├── BUILD │ │ │ ├── c │ │ │ │ ├── BUILD │ │ │ │ ├── tsl_status.cc │ │ │ │ ├── tsl_status.h │ │ │ │ ├── tsl_status_helper.cc │ │ │ │ ├── tsl_status_helper.h │ │ │ │ ├── tsl_status_internal.h │ │ │ │ └── tsl_status_test.cc │ │ │ ├── concurrency │ │ │ │ ├── BUILD │ │ │ │ ├── async_value.cc │ │ │ │ ├── async_value.h │ │ │ │ ├── async_value_ptr_test.cc │ │ │ │ ├── async_value_ref.cc │ │ │ │ ├── async_value_ref.h │ │ │ │ ├── async_value_ref_test.cc │ │ │ │ ├── async_value_test.cc │ │ │ │ ├── chain.h │ │ │ │ ├── concurrent_vector.h │ │ │ │ ├── concurrent_vector_test.cc │ │ │ │ └── ref_count.h │ │ │ ├── cuda │ │ │ │ ├── BUILD.bazel │ │ │ │ ├── cublas.symbols │ │ │ │ ├── cublasLt.symbols │ │ │ │ ├── cublasLt_stub.cc │ │ │ │ ├── cublas_stub.cc │ │ │ │ ├── cuda.symbols │ │ │ │ ├── cuda_stub.cc │ │ │ │ ├── cudart.symbols │ │ │ │ ├── cudart_stub.cc │ │ │ │ ├── cudnn.symbols │ │ │ │ ├── cudnn_stub.cc │ │ │ │ ├── cufft.symbols │ │ │ │ ├── cufft_stub.cc │ │ │ │ ├── cupti.symbols │ │ │ │ ├── cupti_stub.cc │ │ │ │ ├── cusolver.symbols │ │ │ │ ├── cusolver_stub.cc │ │ │ │ ├── cusparse.symbols │ │ │ │ ├── cusparse_stub.cc │ │ │ │ ├── nccl.symbols │ │ │ │ ├── nccl_stub.cc │ │ │ │ └── stub.bzl │ │ │ ├── distributed_runtime │ │ │ │ ├── BUILD │ │ │ │ ├── call_options.cc │ │ │ │ ├── call_options.h │ │ │ │ ├── coordination │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── coordination_client.h │ │ │ │ │ ├── coordination_service.cc │ │ │ │ │ ├── coordination_service.h │ │ │ │ │ ├── coordination_service_agent.cc │ │ │ │ │ ├── coordination_service_agent.h │ │ │ │ │ ├── coordination_service_agent_test.cc │ │ │ │ │ ├── coordination_service_error_util.h │ │ │ │ │ ├── coordination_service_error_util_test.cc │ │ │ │ │ ├── coordination_service_recoverable_job_test.cc │ │ │ │ │ ├── coordination_service_rpc_handler.cc │ │ │ │ │ ├── coordination_service_rpc_handler.h │ │ │ │ │ ├── coordination_service_test.cc │ │ │ │ │ └── test_device.proto │ │ │ │ ├── preemption │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── preemption_notifier.cc │ │ │ │ │ ├── preemption_notifier.h │ │ │ │ │ ├── preemption_notifier_test.cc │ │ │ │ │ ├── preemption_sync_manager.cc │ │ │ │ │ ├── preemption_sync_manager.h │ │ │ │ │ └── preemption_sync_manager_test.cc │ │ │ │ └── rpc │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── async_service_interface.h │ │ │ │ │ ├── coordination │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── grpc_coordination_client.cc │ │ │ │ │ ├── grpc_coordination_client.h │ │ │ │ │ ├── grpc_coordination_service_impl.cc │ │ │ │ │ └── grpc_coordination_service_impl.h │ │ │ │ │ ├── grpc_call.h │ │ │ │ │ ├── grpc_channel.cc │ │ │ │ │ ├── grpc_channel.h │ │ │ │ │ ├── grpc_channel_common.h │ │ │ │ │ ├── grpc_channel_test.cc │ │ │ │ │ ├── grpc_client_cq_tag.h │ │ │ │ │ ├── grpc_state.h │ │ │ │ │ ├── grpc_util.cc │ │ │ │ │ ├── grpc_util.h │ │ │ │ │ ├── grpc_util_test.cc │ │ │ │ │ └── test_request.proto │ │ │ ├── framework │ │ │ │ ├── BUILD │ │ │ │ ├── allocator.cc │ │ │ │ ├── allocator.h │ │ │ │ ├── allocator_registry.cc │ │ │ │ ├── allocator_registry.h │ │ │ │ ├── allocator_retry.cc │ │ │ │ ├── allocator_retry.h │ │ │ │ ├── bfc_allocator.cc │ │ │ │ ├── bfc_allocator.h │ │ │ │ ├── cancellation.cc │ │ │ │ ├── cancellation.h │ │ │ │ ├── cancellation_test.cc │ │ │ │ ├── contraction │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── build_defs.bzl │ │ │ │ │ ├── eigen_contraction_kernel.cc │ │ │ │ │ └── eigen_contraction_kernel.h │ │ │ │ ├── convolution │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── eigen_convolution_helpers.h │ │ │ │ │ ├── eigen_spatial_convolutions-inl.h │ │ │ │ │ ├── eigen_spatial_convolutions.h │ │ │ │ │ └── eigen_spatial_convolutions_test.cc │ │ │ │ ├── cpu_allocator_impl.cc │ │ │ │ ├── device_id.h │ │ │ │ ├── device_id_manager.cc │ │ │ │ ├── device_id_manager.h │ │ │ │ ├── device_id_utils.cc │ │ │ │ ├── device_id_utils.h │ │ │ │ ├── device_id_utils_test.cc │ │ │ │ ├── device_type.cc │ │ │ │ ├── device_type.h │ │ │ │ ├── fixedpoint │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── FixedPoint.h │ │ │ │ │ ├── MatMatProduct.h │ │ │ │ │ ├── MatMatProductAVX2.h │ │ │ │ │ ├── MatMatProductNEON.h │ │ │ │ │ ├── MatVecProduct.h │ │ │ │ │ ├── PacketMathAVX.h │ │ │ │ │ ├── PacketMathAVX2.h │ │ │ │ │ ├── PacketMathAVX512.h │ │ │ │ │ ├── TypeCastingAVX2.h │ │ │ │ │ └── TypeCastingAVX512.h │ │ │ │ ├── fixedpoint_types.h │ │ │ │ ├── metrics.cc │ │ │ │ ├── metrics.h │ │ │ │ ├── numeric_types.h │ │ │ │ ├── shared_counter.h │ │ │ │ ├── tracking_allocator.cc │ │ │ │ ├── tracking_allocator.h │ │ │ │ └── type_traits.h │ │ │ ├── lib │ │ │ │ ├── core │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── bitmap.cc │ │ │ │ │ ├── bitmap.h │ │ │ │ │ ├── bitmap_test.cc │ │ │ │ │ ├── bits.h │ │ │ │ │ ├── bits_test.cc │ │ │ │ │ └── status_test_util.h │ │ │ │ ├── gtl │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── compactptrset.h │ │ │ │ │ ├── compactptrset_test.cc │ │ │ │ │ ├── flatmap.h │ │ │ │ │ ├── flatmap_test.cc │ │ │ │ │ ├── flatrep.h │ │ │ │ │ ├── flatset.h │ │ │ │ │ ├── flatset_test.cc │ │ │ │ │ ├── inlined_vector.h │ │ │ │ │ ├── int_type.h │ │ │ │ │ ├── int_type_test.cc │ │ │ │ │ ├── iterator_range.h │ │ │ │ │ ├── iterator_range_test.cc │ │ │ │ │ ├── map_util.h │ │ │ │ │ ├── map_util_test.cc │ │ │ │ │ └── subtle │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ └── map_traits.h │ │ │ │ ├── hash │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── crc32c.cc │ │ │ │ │ ├── crc32c.h │ │ │ │ │ ├── crc32c_accelerate.cc │ │ │ │ │ └── crc32c_test.cc │ │ │ │ ├── histogram │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── histogram.cc │ │ │ │ │ ├── histogram.h │ │ │ │ │ └── histogram_test.cc │ │ │ │ ├── io │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── block.cc │ │ │ │ │ ├── block.h │ │ │ │ │ ├── block_builder.cc │ │ │ │ │ ├── block_builder.h │ │ │ │ │ ├── buffered_file.h │ │ │ │ │ ├── buffered_file_test.cc │ │ │ │ │ ├── buffered_inputstream.cc │ │ │ │ │ ├── buffered_inputstream.h │ │ │ │ │ ├── buffered_inputstream_test.cc │ │ │ │ │ ├── cache.cc │ │ │ │ │ ├── cache.h │ │ │ │ │ ├── cache_test.cc │ │ │ │ │ ├── compression.cc │ │ │ │ │ ├── compression.h │ │ │ │ │ ├── format.cc │ │ │ │ │ ├── format.h │ │ │ │ │ ├── inputbuffer.cc │ │ │ │ │ ├── inputbuffer.h │ │ │ │ │ ├── inputbuffer_test.cc │ │ │ │ │ ├── inputstream_interface.cc │ │ │ │ │ ├── inputstream_interface.h │ │ │ │ │ ├── inputstream_interface_test.cc │ │ │ │ │ ├── iterator.cc │ │ │ │ │ ├── iterator.h │ │ │ │ │ ├── proto_encode_helper.h │ │ │ │ │ ├── random_inputstream.cc │ │ │ │ │ ├── random_inputstream.h │ │ │ │ │ ├── random_inputstream_test.cc │ │ │ │ │ ├── record_reader.cc │ │ │ │ │ ├── record_reader.h │ │ │ │ │ ├── record_reader_writer_test.cc │ │ │ │ │ ├── record_writer.cc │ │ │ │ │ ├── record_writer.h │ │ │ │ │ ├── recordio_test.cc │ │ │ │ │ ├── snappy │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── snappy_compression_options.h │ │ │ │ │ │ ├── snappy_inputbuffer.cc │ │ │ │ │ │ ├── snappy_inputbuffer.h │ │ │ │ │ │ ├── snappy_inputstream.cc │ │ │ │ │ │ ├── snappy_inputstream.h │ │ │ │ │ │ ├── snappy_outputbuffer.cc │ │ │ │ │ │ ├── snappy_outputbuffer.h │ │ │ │ │ │ └── snappy_test.cc │ │ │ │ │ ├── table.cc │ │ │ │ │ ├── table.h │ │ │ │ │ ├── table_builder.cc │ │ │ │ │ ├── table_builder.h │ │ │ │ │ ├── table_format.txt │ │ │ │ │ ├── table_options.h │ │ │ │ │ ├── table_test.cc │ │ │ │ │ ├── two_level_iterator.cc │ │ │ │ │ ├── two_level_iterator.h │ │ │ │ │ ├── zlib_buffers_test.cc │ │ │ │ │ ├── zlib_compression_options.cc │ │ │ │ │ ├── zlib_compression_options.h │ │ │ │ │ ├── zlib_inputstream.cc │ │ │ │ │ ├── zlib_inputstream.h │ │ │ │ │ ├── zlib_outputbuffer.cc │ │ │ │ │ └── zlib_outputbuffer.h │ │ │ │ ├── math │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── math_util.h │ │ │ │ │ └── math_util_test.cc │ │ │ │ ├── monitoring │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── cell_reader-inl.cc │ │ │ │ │ ├── cell_reader-inl.h │ │ │ │ │ ├── cell_reader.h │ │ │ │ │ ├── collected_metrics.h │ │ │ │ │ ├── collection_registry.cc │ │ │ │ │ ├── collection_registry.h │ │ │ │ │ ├── counter.h │ │ │ │ │ ├── gauge.h │ │ │ │ │ ├── metric_def.h │ │ │ │ │ ├── percentile_sampler.cc │ │ │ │ │ ├── percentile_sampler.h │ │ │ │ │ ├── sampler.cc │ │ │ │ │ ├── sampler.h │ │ │ │ │ ├── test_utils.cc │ │ │ │ │ ├── test_utils.h │ │ │ │ │ ├── timed.h │ │ │ │ │ └── types.h │ │ │ │ ├── random │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── distribution_sampler.cc │ │ │ │ │ ├── distribution_sampler.h │ │ │ │ │ ├── distribution_sampler_test.cc │ │ │ │ │ ├── exact_uniform_int.h │ │ │ │ │ ├── philox_random.h │ │ │ │ │ ├── philox_random_test.cc │ │ │ │ │ ├── philox_random_test_utils.h │ │ │ │ │ ├── random_distributions.cc │ │ │ │ │ ├── random_distributions.h │ │ │ │ │ ├── random_distributions_test.cc │ │ │ │ │ ├── random_distributions_utils.h │ │ │ │ │ ├── simple_philox.cc │ │ │ │ │ ├── simple_philox.h │ │ │ │ │ ├── simple_philox_test.cc │ │ │ │ │ ├── weighted_picker.cc │ │ │ │ │ ├── weighted_picker.h │ │ │ │ │ └── weighted_picker_test.cc │ │ │ │ └── strings │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── proto_serialization.cc │ │ │ │ │ └── proto_serialization.h │ │ │ ├── mkl │ │ │ │ ├── BUILD │ │ │ │ ├── LICENSE │ │ │ │ ├── MKL_LICENSE │ │ │ │ └── build_defs.bzl │ │ │ ├── platform │ │ │ │ ├── BUILD │ │ │ │ ├── abi.cc │ │ │ │ ├── abi.h │ │ │ │ ├── abi_test.cc │ │ │ │ ├── base64.cc │ │ │ │ ├── base64.h │ │ │ │ ├── bfloat16.h │ │ │ │ ├── blocking_counter.h │ │ │ │ ├── build_config.bzl │ │ │ │ ├── build_config_root.bzl │ │ │ │ ├── byte_order.h │ │ │ │ ├── casts.h │ │ │ │ ├── cloud │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── auth_provider.h │ │ │ │ │ ├── compute_engine_metadata_client.cc │ │ │ │ │ ├── compute_engine_metadata_client.h │ │ │ │ │ ├── compute_engine_metadata_client_test.cc │ │ │ │ │ ├── compute_engine_zone_provider.cc │ │ │ │ │ ├── compute_engine_zone_provider.h │ │ │ │ │ ├── compute_engine_zone_provider_test.cc │ │ │ │ │ ├── curl_http_request.cc │ │ │ │ │ ├── curl_http_request.h │ │ │ │ │ ├── curl_http_request_test.cc │ │ │ │ │ ├── expiring_lru_cache.h │ │ │ │ │ ├── expiring_lru_cache_test.cc │ │ │ │ │ ├── file_block_cache.h │ │ │ │ │ ├── gcs_dns_cache.cc │ │ │ │ │ ├── gcs_dns_cache.h │ │ │ │ │ ├── gcs_dns_cache_test.cc │ │ │ │ │ ├── gcs_file_system.cc │ │ │ │ │ ├── gcs_file_system.h │ │ │ │ │ ├── gcs_file_system_test.cc │ │ │ │ │ ├── gcs_throttle.cc │ │ │ │ │ ├── gcs_throttle.h │ │ │ │ │ ├── gcs_throttle_test.cc │ │ │ │ │ ├── google_auth_provider.cc │ │ │ │ │ ├── google_auth_provider.h │ │ │ │ │ ├── google_auth_provider_test.cc │ │ │ │ │ ├── http_request.h │ │ │ │ │ ├── http_request_fake.h │ │ │ │ │ ├── now_seconds_env.h │ │ │ │ │ ├── oauth_client.cc │ │ │ │ │ ├── oauth_client.h │ │ │ │ │ ├── oauth_client_test.cc │ │ │ │ │ ├── ram_file_block_cache.cc │ │ │ │ │ ├── ram_file_block_cache.h │ │ │ │ │ ├── ram_file_block_cache_test.cc │ │ │ │ │ ├── testdata │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── application_default_credentials.json │ │ │ │ │ │ ├── service_account_credentials.json │ │ │ │ │ │ └── service_account_public_key.txt │ │ │ │ │ ├── time_util.cc │ │ │ │ │ ├── time_util.h │ │ │ │ │ ├── time_util_test.cc │ │ │ │ │ └── zone_provider.h │ │ │ │ ├── coding.cc │ │ │ │ ├── coding.h │ │ │ │ ├── context.h │ │ │ │ ├── cord.h │ │ │ │ ├── cpu_info.cc │ │ │ │ ├── cpu_info.h │ │ │ │ ├── crash_analysis.h │ │ │ │ ├── criticality.h │ │ │ │ ├── criticality_test.cc │ │ │ │ ├── ctstring.h │ │ │ │ ├── ctstring_internal.h │ │ │ │ ├── ctstring_test.cc │ │ │ │ ├── cuda_libdevice_path.h │ │ │ │ ├── default │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── build_config.bzl │ │ │ │ │ ├── build_config │ │ │ │ │ │ └── BUILD │ │ │ │ │ ├── build_config_root.bzl │ │ │ │ │ ├── casts.h │ │ │ │ │ ├── context.h │ │ │ │ │ ├── crash_analysis.cc │ │ │ │ │ ├── crash_analysis.h │ │ │ │ │ ├── criticality.h │ │ │ │ │ ├── cuda_build_defs.bzl │ │ │ │ │ ├── cuda_libdevice_path.cc │ │ │ │ │ ├── dlopen_checker.cc │ │ │ │ │ ├── dlopen_checker_stub.cc │ │ │ │ │ ├── dso_loader.cc │ │ │ │ │ ├── dso_loader.h │ │ │ │ │ ├── dynamic_annotations.h │ │ │ │ │ ├── env.cc │ │ │ │ │ ├── env_time.cc │ │ │ │ │ ├── error_logging.cc │ │ │ │ │ ├── human_readable_json.cc │ │ │ │ │ ├── integral_types.h │ │ │ │ │ ├── load_library.cc │ │ │ │ │ ├── logging.cc │ │ │ │ │ ├── logging.h │ │ │ │ │ ├── mutex.cc │ │ │ │ │ ├── mutex.h │ │ │ │ │ ├── mutex_data.h │ │ │ │ │ ├── net.cc │ │ │ │ │ ├── notification.h │ │ │ │ │ ├── platform.bzl │ │ │ │ │ ├── port.cc │ │ │ │ │ ├── posix_file_system.cc │ │ │ │ │ ├── posix_file_system.h │ │ │ │ │ ├── resource.cc │ │ │ │ │ ├── resource_loader.cc │ │ │ │ │ ├── rocm_rocdl_path.cc │ │ │ │ │ ├── rules_cc.bzl │ │ │ │ │ ├── stacktrace.h │ │ │ │ │ ├── stacktrace_handler.cc │ │ │ │ │ ├── status.h │ │ │ │ │ ├── statusor.h │ │ │ │ │ ├── subprocess.cc │ │ │ │ │ ├── subprocess.h │ │ │ │ │ ├── tracing.cc │ │ │ │ │ ├── tracing_impl.h │ │ │ │ │ ├── unbounded_work_queue.cc │ │ │ │ │ └── unbounded_work_queue.h │ │ │ │ ├── demangle.h │ │ │ │ ├── denormal.cc │ │ │ │ ├── denormal.h │ │ │ │ ├── denormal_test.cc │ │ │ │ ├── dso_loader.h │ │ │ │ ├── dynamic_annotations.h │ │ │ │ ├── env.cc │ │ │ │ ├── env.h │ │ │ │ ├── env_time.h │ │ │ │ ├── error_logging.h │ │ │ │ ├── errors.cc │ │ │ │ ├── errors.h │ │ │ │ ├── errors_test.cc │ │ │ │ ├── file_statistics.h │ │ │ │ ├── file_system.cc │ │ │ │ ├── file_system.h │ │ │ │ ├── file_system_helper.cc │ │ │ │ ├── file_system_helper.h │ │ │ │ ├── fingerprint.h │ │ │ │ ├── fingerprint_test.cc │ │ │ │ ├── float8.h │ │ │ │ ├── gif.h │ │ │ │ ├── hash.cc │ │ │ │ ├── hash.h │ │ │ │ ├── hash_test.cc │ │ │ │ ├── host_info.h │ │ │ │ ├── human_readable_json.h │ │ │ │ ├── init_main.h │ │ │ │ ├── integral_types_test.cc │ │ │ │ ├── intrusive_ptr.h │ │ │ │ ├── intrusive_ptr_test.cc │ │ │ │ ├── jpeg.h │ │ │ │ ├── load_library.h │ │ │ │ ├── logger.cc │ │ │ │ ├── logger.h │ │ │ │ ├── logging.h │ │ │ │ ├── logging_test.cc │ │ │ │ ├── macros.h │ │ │ │ ├── mem.h │ │ │ │ ├── ml_dtypes.h │ │ │ │ ├── mutex.h │ │ │ │ ├── mutex_test.cc │ │ │ │ ├── net.h │ │ │ │ ├── net_test.cc │ │ │ │ ├── notification.h │ │ │ │ ├── null_file_system.h │ │ │ │ ├── numa.h │ │ │ │ ├── numa_test.cc │ │ │ │ ├── numbers.cc │ │ │ │ ├── numbers.h │ │ │ │ ├── numbers_test.cc │ │ │ │ ├── path.cc │ │ │ │ ├── path.h │ │ │ │ ├── path_test.cc │ │ │ │ ├── platform.h │ │ │ │ ├── platform_strings_computed.h │ │ │ │ ├── png.h │ │ │ │ ├── port_test.cc │ │ │ │ ├── prefetch.h │ │ │ │ ├── profile_utils │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── android_armv7a_cpu_utils_helper.cc │ │ │ │ │ ├── android_armv7a_cpu_utils_helper.h │ │ │ │ │ ├── clock_cycle_profiler.cc │ │ │ │ │ ├── clock_cycle_profiler.h │ │ │ │ │ ├── cpu_utils.cc │ │ │ │ │ ├── cpu_utils.h │ │ │ │ │ ├── cpu_utils_test.cc │ │ │ │ │ └── i_cpu_utils_helper.h │ │ │ │ ├── protobuf.cc │ │ │ │ ├── protobuf.h │ │ │ │ ├── protobuf_compiler.h │ │ │ │ ├── protobuf_util.cc │ │ │ │ ├── ram_file_system.h │ │ │ │ ├── random.cc │ │ │ │ ├── random.h │ │ │ │ ├── random_test.cc │ │ │ │ ├── raw_coding.h │ │ │ │ ├── refcount.h │ │ │ │ ├── refcount_test.cc │ │ │ │ ├── regexp.h │ │ │ │ ├── resource.h │ │ │ │ ├── resource_loader.h │ │ │ │ ├── retrying_file_system.h │ │ │ │ ├── retrying_file_system_test.cc │ │ │ │ ├── retrying_utils.cc │ │ │ │ ├── retrying_utils.h │ │ │ │ ├── retrying_utils_test.cc │ │ │ │ ├── rocm_rocdl_path.h │ │ │ │ ├── rules_cc.bzl │ │ │ │ ├── scanner.cc │ │ │ │ ├── scanner.h │ │ │ │ ├── scanner_test.cc │ │ │ │ ├── setround.cc │ │ │ │ ├── setround.h │ │ │ │ ├── setround_test.cc │ │ │ │ ├── snappy.h │ │ │ │ ├── stack_frame.h │ │ │ │ ├── stacktrace.h │ │ │ │ ├── stacktrace_handler.h │ │ │ │ ├── stacktrace_handler_test.cc │ │ │ │ ├── stacktrace_test.cc │ │ │ │ ├── status.cc │ │ │ │ ├── status.h │ │ │ │ ├── status_matchers.cc │ │ │ │ ├── status_matchers.h │ │ │ │ ├── status_matchers_test.cc │ │ │ │ ├── status_test.cc │ │ │ │ ├── status_to_from_proto.cc │ │ │ │ ├── status_to_from_proto.h │ │ │ │ ├── statusor.h │ │ │ │ ├── statusor_test.cc │ │ │ │ ├── str_util.cc │ │ │ │ ├── str_util.h │ │ │ │ ├── str_util_test.cc │ │ │ │ ├── strcat.cc │ │ │ │ ├── strcat.h │ │ │ │ ├── strcat_test.cc │ │ │ │ ├── stringpiece.h │ │ │ │ ├── stringpiece_test.cc │ │ │ │ ├── stringprintf.cc │ │ │ │ ├── stringprintf.h │ │ │ │ ├── stringprintf_test.cc │ │ │ │ ├── subprocess.h │ │ │ │ ├── subprocess_test.cc │ │ │ │ ├── tensor_float_32_utils.cc │ │ │ │ ├── tensor_float_32_utils.h │ │ │ │ ├── test.cc │ │ │ │ ├── test.h │ │ │ │ ├── test_benchmark.h │ │ │ │ ├── test_main.cc │ │ │ │ ├── testdata │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── test_echo.cc │ │ │ │ │ ├── test_echo_argv_1.cc │ │ │ │ │ ├── test_noop.cc │ │ │ │ │ └── test_stderr.cc │ │ │ │ ├── thread_annotations.h │ │ │ │ ├── threadpool.cc │ │ │ │ ├── threadpool.h │ │ │ │ ├── threadpool_interface.h │ │ │ │ ├── threadpool_options.h │ │ │ │ ├── tracing.cc │ │ │ │ ├── tracing.h │ │ │ │ ├── tstring.h │ │ │ │ ├── tstring_test.cc │ │ │ │ ├── types.h │ │ │ │ ├── unbounded_work_queue.h │ │ │ │ ├── unbounded_work_queue_test.cc │ │ │ │ └── windows │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── env.cc │ │ │ │ │ ├── env_time.cc │ │ │ │ │ ├── error_windows.cc │ │ │ │ │ ├── error_windows.h │ │ │ │ │ ├── intrinsics_port.h │ │ │ │ │ ├── load_library.cc │ │ │ │ │ ├── net.cc │ │ │ │ │ ├── port.cc │ │ │ │ │ ├── stacktrace.cc │ │ │ │ │ ├── stacktrace.h │ │ │ │ │ ├── stacktrace_handler.cc │ │ │ │ │ ├── subprocess.cc │ │ │ │ │ ├── subprocess.h │ │ │ │ │ ├── wide_char.cc │ │ │ │ │ ├── wide_char.h │ │ │ │ │ ├── windows_file_system.cc │ │ │ │ │ └── windows_file_system.h │ │ │ ├── profiler │ │ │ │ ├── BUILD │ │ │ │ ├── backends │ │ │ │ │ └── cpu │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── annotation_stack.cc │ │ │ │ │ │ ├── annotation_stack.h │ │ │ │ │ │ ├── host_tracer_utils.cc │ │ │ │ │ │ ├── host_tracer_utils.h │ │ │ │ │ │ ├── traceme_recorder.cc │ │ │ │ │ │ ├── traceme_recorder.h │ │ │ │ │ │ └── traceme_recorder_test.cc │ │ │ │ ├── builds │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── build_config.bzl │ │ │ │ │ └── oss │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ └── build_config.bzl │ │ │ │ ├── convert │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── post_process_single_host_xplane.cc │ │ │ │ │ ├── post_process_single_host_xplane.h │ │ │ │ │ ├── trace_container.cc │ │ │ │ │ ├── trace_container.h │ │ │ │ │ ├── trace_container_test.cc │ │ │ │ │ ├── trace_events_to_json.cc │ │ │ │ │ ├── trace_events_to_json.h │ │ │ │ │ ├── trace_events_to_json_test.cc │ │ │ │ │ ├── xla_op_utils.h │ │ │ │ │ ├── xla_op_utils_test.cc │ │ │ │ │ ├── xplane_to_trace_events.cc │ │ │ │ │ ├── xplane_to_trace_events.h │ │ │ │ │ └── xplane_to_trace_events_test.cc │ │ │ │ ├── lib │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── connected_traceme.h │ │ │ │ │ ├── context_types.cc │ │ │ │ │ ├── context_types.h │ │ │ │ │ ├── nvtx_utils.h │ │ │ │ │ ├── profiler_collection.cc │ │ │ │ │ ├── profiler_collection.h │ │ │ │ │ ├── profiler_controller.cc │ │ │ │ │ ├── profiler_controller.h │ │ │ │ │ ├── profiler_factory.cc │ │ │ │ │ ├── profiler_factory.h │ │ │ │ │ ├── profiler_factory_test.cc │ │ │ │ │ ├── profiler_interface.h │ │ │ │ │ ├── profiler_lock.cc │ │ │ │ │ ├── profiler_lock.h │ │ │ │ │ ├── profiler_lock_test.cc │ │ │ │ │ ├── profiler_session.cc │ │ │ │ │ ├── profiler_session.h │ │ │ │ │ ├── scoped_annotation.h │ │ │ │ │ ├── scoped_annotation_stack.h │ │ │ │ │ ├── scoped_annotation_test.cc │ │ │ │ │ ├── scoped_memory_debug_annotation.cc │ │ │ │ │ ├── scoped_memory_debug_annotation.h │ │ │ │ │ ├── traceme.h │ │ │ │ │ ├── traceme_encode.h │ │ │ │ │ └── traceme_encode_test.cc │ │ │ │ ├── protobuf │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── profile.proto │ │ │ │ │ ├── profiled_instructions.proto │ │ │ │ │ ├── profiler_analysis.proto │ │ │ │ │ ├── profiler_options.proto │ │ │ │ │ ├── profiler_service.proto │ │ │ │ │ ├── profiler_service_monitor_result.proto │ │ │ │ │ ├── trace_events.proto │ │ │ │ │ └── xplane.proto │ │ │ │ ├── rpc │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── client │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── capture_profile.cc │ │ │ │ │ │ ├── capture_profile.h │ │ │ │ │ │ ├── profiler_client.cc │ │ │ │ │ │ ├── profiler_client.h │ │ │ │ │ │ ├── profiler_client_test.cc │ │ │ │ │ │ ├── profiler_client_test_util.h │ │ │ │ │ │ ├── remote_profiler_session_manager.cc │ │ │ │ │ │ ├── remote_profiler_session_manager.h │ │ │ │ │ │ ├── remote_profiler_session_manager_test.cc │ │ │ │ │ │ ├── save_profile.cc │ │ │ │ │ │ └── save_profile.h │ │ │ │ │ ├── profiler_server.cc │ │ │ │ │ ├── profiler_server.h │ │ │ │ │ ├── profiler_service_impl.cc │ │ │ │ │ └── profiler_service_impl.h │ │ │ │ └── utils │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── buffer_pool.cc │ │ │ │ │ ├── buffer_pool.h │ │ │ │ │ ├── buffer_pool_test.cc │ │ │ │ │ ├── file_system_utils.h │ │ │ │ │ ├── format_utils.h │ │ │ │ │ ├── group_events.cc │ │ │ │ │ ├── group_events.h │ │ │ │ │ ├── group_events_test.cc │ │ │ │ │ ├── math_utils.h │ │ │ │ │ ├── parse_annotation.cc │ │ │ │ │ ├── parse_annotation.h │ │ │ │ │ ├── parse_annotation_test.cc │ │ │ │ │ ├── preprocess_xplane.cc │ │ │ │ │ ├── preprocess_xplane.h │ │ │ │ │ ├── preprocess_xplane_test.cc │ │ │ │ │ ├── session_manager.cc │ │ │ │ │ ├── session_manager.h │ │ │ │ │ ├── tf_op_utils.cc │ │ │ │ │ ├── tf_op_utils.h │ │ │ │ │ ├── tf_op_utils_test.cc │ │ │ │ │ ├── tf_xplane_visitor.h │ │ │ │ │ ├── time_utils.cc │ │ │ │ │ ├── time_utils.h │ │ │ │ │ ├── timespan.h │ │ │ │ │ ├── timespan_test.cc │ │ │ │ │ ├── tpu_xplane_utils.cc │ │ │ │ │ ├── tpu_xplane_utils.h │ │ │ │ │ ├── tpu_xplane_utils_test.cc │ │ │ │ │ ├── trace_utils.h │ │ │ │ │ ├── xplane_builder.cc │ │ │ │ │ ├── xplane_builder.h │ │ │ │ │ ├── xplane_builder_test.cc │ │ │ │ │ ├── xplane_schema.cc │ │ │ │ │ ├── xplane_schema.h │ │ │ │ │ ├── xplane_test_utils.cc │ │ │ │ │ ├── xplane_test_utils.h │ │ │ │ │ ├── xplane_utils.cc │ │ │ │ │ ├── xplane_utils.h │ │ │ │ │ ├── xplane_utils_test.cc │ │ │ │ │ ├── xplane_visitor.cc │ │ │ │ │ └── xplane_visitor.h │ │ │ ├── protobuf │ │ │ │ ├── BUILD │ │ │ │ ├── bfc_memory_map.proto │ │ │ │ ├── coordination_config.proto │ │ │ │ ├── coordination_service.proto │ │ │ │ ├── distributed_runtime_payloads.proto │ │ │ │ ├── dnn.proto │ │ │ │ ├── error_codes.proto │ │ │ │ ├── histogram.proto │ │ │ │ ├── rpc_options.proto │ │ │ │ ├── status.proto │ │ │ │ └── test_log.proto │ │ │ ├── python │ │ │ │ └── lib │ │ │ │ │ └── core │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── ml_dtypes.cc │ │ │ │ │ ├── ml_dtypes.h │ │ │ │ │ ├── numpy.cc │ │ │ │ │ └── numpy.h │ │ │ ├── tsl.bzl │ │ │ ├── tsl.default.bzl │ │ │ └── util │ │ │ │ ├── BUILD │ │ │ │ ├── byte_swap_array.cc │ │ │ │ ├── byte_swap_array.h │ │ │ │ ├── command_line_flags.cc │ │ │ │ ├── command_line_flags.h │ │ │ │ ├── determinism.cc │ │ │ │ ├── determinism.h │ │ │ │ ├── determinism_test_util.h │ │ │ │ ├── device_name_utils.cc │ │ │ │ ├── device_name_utils.h │ │ │ │ ├── device_name_utils_test.cc │ │ │ │ ├── env_var.cc │ │ │ │ ├── env_var.h │ │ │ │ ├── onednn_threadpool.h │ │ │ │ ├── proto │ │ │ │ ├── BUILD │ │ │ │ └── proto_utils.h │ │ │ │ ├── reporter.cc │ │ │ │ ├── reporter.h │ │ │ │ ├── stat_summarizer_options.h │ │ │ │ ├── stats_calculator.cc │ │ │ │ ├── stats_calculator.h │ │ │ │ ├── stats_calculator_test.cc │ │ │ │ ├── use_cudnn.cc │ │ │ │ └── use_cudnn.h │ │ │ ├── workspace0.bzl │ │ │ ├── workspace1.bzl │ │ │ ├── workspace2.bzl │ │ │ └── workspace3.bzl │ ├── tools │ │ ├── ci_build │ │ │ └── gpu_build │ │ │ │ ├── BUILD │ │ │ │ └── parallel_gpu_execute.sh │ │ └── toolchains │ │ │ ├── BUILD │ │ │ ├── archives.bzl │ │ │ ├── clang6 │ │ │ ├── BUILD │ │ │ ├── CROSSTOOL.tpl │ │ │ ├── README.md │ │ │ ├── clang.BUILD │ │ │ └── repo.bzl │ │ │ ├── cpus │ │ │ ├── aarch64 │ │ │ │ ├── README.md │ │ │ │ ├── aarch64.bzl │ │ │ │ ├── aarch64_compiler_configure.bzl │ │ │ │ └── crosstool │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── BUILD.tpl │ │ │ │ │ └── cc_toolchain_config.bzl.tpl │ │ │ ├── arm │ │ │ │ ├── arm_compiler_configure.bzl │ │ │ │ ├── cc_config.bzl.tpl │ │ │ │ └── template.BUILD │ │ │ ├── py │ │ │ │ └── BUILD │ │ │ └── py3 │ │ │ │ └── BUILD │ │ │ ├── cross_compile │ │ │ ├── cc │ │ │ │ └── BUILD │ │ │ └── config │ │ │ │ └── BUILD │ │ │ ├── embedded │ │ │ ├── README.md │ │ │ └── arm-linux │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── aarch64-linux-toolchain.BUILD │ │ │ │ ├── arm_linux_toolchain_configure.bzl │ │ │ │ ├── armhf-linux-toolchain.BUILD │ │ │ │ ├── cc_config.bzl.tpl │ │ │ │ └── template.BUILD │ │ │ ├── java │ │ │ └── BUILD │ │ │ ├── python │ │ │ ├── BUILD │ │ │ └── python_repo.bzl │ │ │ ├── remote │ │ │ ├── BUILD │ │ │ ├── BUILD.tpl │ │ │ ├── configure.bzl │ │ │ └── execution.bzl.tpl │ │ │ ├── remote_config │ │ │ ├── BUILD │ │ │ ├── configs.bzl │ │ │ ├── containers.bzl │ │ │ └── rbe_config.bzl │ │ │ ├── win │ │ │ ├── BUILD │ │ │ ├── bazel_211 │ │ │ │ ├── BUILD │ │ │ │ ├── armeabi_cc_toolchain_config.bzl │ │ │ │ ├── builtin_include_directory_paths_msvc │ │ │ │ └── windows_cc_toolchain_config.bzl │ │ │ └── tf_win_05022023 │ │ │ │ ├── BUILD │ │ │ │ ├── armeabi_cc_toolchain_config.bzl │ │ │ │ ├── builtin_include_directory_paths_msvc │ │ │ │ ├── toolchain_image_info │ │ │ │ └── windows_cc_toolchain_config.bzl │ │ │ └── win_1803 │ │ │ ├── py38 │ │ │ └── BUILD │ │ │ └── py39 │ │ │ └── BUILD │ ├── workspace0.bzl │ ├── workspace1.bzl │ ├── workspace2.bzl │ ├── workspace3.bzl │ ├── workspace4.bzl │ └── xla │ │ ├── .clang-format │ │ ├── BUILD │ │ ├── README.md │ │ ├── array.cc │ │ ├── array.h │ │ ├── array2d.h │ │ ├── array2d_test.cc │ │ ├── array3d.h │ │ ├── array3d_test.cc │ │ ├── array4d.h │ │ ├── array4d_test.cc │ │ ├── array_test.cc │ │ ├── autotune_results.proto │ │ ├── autotuning.proto │ │ ├── backends │ │ ├── interpreter │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── compiler.cc │ │ │ ├── compiler.h │ │ │ ├── executable.cc │ │ │ ├── executable.h │ │ │ ├── executable_base.cc │ │ │ ├── executable_base.h │ │ │ ├── executor.cc │ │ │ ├── executor.h │ │ │ ├── interpreter_transfer_manager.cc │ │ │ ├── interpreter_transfer_manager.h │ │ │ ├── platform.cc │ │ │ ├── platform.h │ │ │ ├── platform_id.cc │ │ │ └── platform_id.h │ │ └── profiler │ │ │ ├── BUILD │ │ │ ├── cpu │ │ │ ├── BUILD │ │ │ ├── host_tracer.cc │ │ │ ├── host_tracer.h │ │ │ ├── host_tracer_factory.cc │ │ │ ├── metadata_collector.cc │ │ │ ├── metadata_utils.h │ │ │ ├── python_tracer.cc │ │ │ ├── python_tracer.h │ │ │ └── python_tracer_factory.cc │ │ │ ├── gpu │ │ │ ├── BUILD │ │ │ ├── cuda_test.cu.cc │ │ │ ├── cuda_test.h │ │ │ ├── cupti_collector.cc │ │ │ ├── cupti_collector.h │ │ │ ├── cupti_error_manager.cc │ │ │ ├── cupti_error_manager.h │ │ │ ├── cupti_error_manager_test.cc │ │ │ ├── cupti_interface.h │ │ │ ├── cupti_profiler.cc │ │ │ ├── cupti_profiler.h │ │ │ ├── cupti_tracer.cc │ │ │ ├── cupti_tracer.h │ │ │ ├── cupti_utils.cc │ │ │ ├── cupti_wrapper.cc │ │ │ ├── cupti_wrapper.h │ │ │ ├── device_tracer_cuda.cc │ │ │ ├── device_tracer_rocm.cc │ │ │ ├── mock_cupti.h │ │ │ ├── nvtx_utils.cc │ │ │ ├── nvtx_utils.h │ │ │ ├── rocm_tracer.cc │ │ │ └── rocm_tracer.h │ │ │ ├── plugin │ │ │ ├── BUILD │ │ │ ├── plugin_tracer.cc │ │ │ ├── plugin_tracer.h │ │ │ ├── plugin_tracer_impl.cc │ │ │ ├── plugin_tracer_impl.h │ │ │ ├── plugin_tracer_impl_test.cc │ │ │ ├── profiler_c_api.h │ │ │ ├── profiler_error.cc │ │ │ └── profiler_error.h │ │ │ └── tpu │ │ │ ├── BUILD │ │ │ └── tpu_tracer.cc │ │ ├── bit_cast.h │ │ ├── bit_cast_test.cc │ │ ├── c │ │ ├── BUILD │ │ └── c_api_decl.h │ │ ├── client │ │ ├── BUILD │ │ ├── client.cc │ │ ├── client.h │ │ ├── client_library.cc │ │ ├── client_library.h │ │ ├── compile_only_client.cc │ │ ├── compile_only_client.h │ │ ├── executable_build_options.cc │ │ ├── executable_build_options.h │ │ ├── global_data.cc │ │ ├── global_data.h │ │ ├── lib │ │ │ ├── BUILD │ │ │ ├── approx_topk.cc │ │ │ ├── approx_topk.h │ │ │ ├── approx_topk_shape.cc │ │ │ ├── approx_topk_shape.h │ │ │ ├── arithmetic.cc │ │ │ ├── arithmetic.h │ │ │ ├── arithmetic_test.cc │ │ │ ├── broadcast.cc │ │ │ ├── broadcast.h │ │ │ ├── comparators.cc │ │ │ ├── comparators.h │ │ │ ├── comparators_test.cc │ │ │ ├── constants.cc │ │ │ ├── constants.h │ │ │ ├── constants_test.cc │ │ │ ├── conv_grad_size_util.cc │ │ │ ├── conv_grad_size_util.h │ │ │ ├── dynamic_shaped_ops.cc │ │ │ ├── dynamic_shaped_ops.h │ │ │ ├── logdet.cc │ │ │ ├── logdet.h │ │ │ ├── logdet_test.cc │ │ │ ├── loops.cc │ │ │ ├── loops.h │ │ │ ├── lu_decomposition.cc │ │ │ ├── lu_decomposition.h │ │ │ ├── math.cc │ │ │ ├── math.h │ │ │ ├── math_test.cc │ │ │ ├── matrix.cc │ │ │ ├── matrix.h │ │ │ ├── matrix_test.cc │ │ │ ├── pooling.cc │ │ │ ├── pooling.h │ │ │ ├── pooling_test.cc │ │ │ ├── prng.cc │ │ │ ├── prng.h │ │ │ ├── prng_test.cc │ │ │ ├── qr.cc │ │ │ ├── qr.h │ │ │ ├── qr_test.cc │ │ │ ├── quantize.h │ │ │ ├── quantize_test.cc │ │ │ ├── self_adjoint_eig.cc │ │ │ ├── self_adjoint_eig.h │ │ │ ├── self_adjoint_eig_test.cc │ │ │ ├── slicing.cc │ │ │ ├── slicing.h │ │ │ ├── slicing_test.cc │ │ │ ├── sorting.cc │ │ │ ├── sorting.h │ │ │ ├── sorting_test.cc │ │ │ ├── svd.cc │ │ │ ├── svd.h │ │ │ ├── svd_test.cc │ │ │ ├── testing.cc │ │ │ ├── testing.h │ │ │ ├── tridiagonal.cc │ │ │ ├── tridiagonal.h │ │ │ ├── tridiagonal_test.cc │ │ │ ├── tuple.cc │ │ │ ├── tuple.h │ │ │ └── tuple_test.cc │ │ ├── local_client.cc │ │ ├── local_client.h │ │ ├── padding.cc │ │ ├── padding.h │ │ ├── padding_test.cc │ │ ├── sharding_builder.cc │ │ ├── sharding_builder.h │ │ ├── value_inference.cc │ │ ├── value_inference.h │ │ ├── xla_builder.cc │ │ ├── xla_builder.h │ │ ├── xla_builder_test.cc │ │ ├── xla_computation.cc │ │ └── xla_computation.h │ │ ├── comparison_util.cc │ │ ├── comparison_util.h │ │ ├── comparison_util_test.cc │ │ ├── cpu_function_runtime.cc │ │ ├── cpu_function_runtime.h │ │ ├── debug_options_flags.cc │ │ ├── debug_options_flags.h │ │ ├── debug_options_parsers.h │ │ ├── debug_options_parsers_test.cc │ │ ├── device_util.h │ │ ├── error_spec.h │ │ ├── examples │ │ └── axpy │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── stablehlo_axpy.mlir │ │ │ └── stablehlo_compile_test.cc │ │ ├── executable_run_options.cc │ │ ├── executable_run_options.h │ │ ├── execution_options_util.cc │ │ ├── execution_options_util.h │ │ ├── experiments │ │ ├── BUILD │ │ ├── README.md │ │ ├── sm_bandwidth_benchmark │ │ │ ├── BUILD │ │ │ ├── sm_bw_kernels.cu.cc │ │ │ ├── sm_bw_kernels.h │ │ │ ├── sm_bw_test.cc │ │ │ └── sm_bw_utils.h │ │ └── triton_autotuning │ │ │ ├── __init__.py │ │ │ ├── check_csv.py │ │ │ ├── check_data.py │ │ │ ├── matmul_lib.py │ │ │ ├── run_single_matmul.py │ │ │ ├── search.py │ │ │ └── tune_single_matmul.py │ │ ├── ffi │ │ ├── BUILD │ │ ├── README.md │ │ ├── api │ │ │ ├── BUILD │ │ │ ├── api.h │ │ │ ├── c_api.h │ │ │ ├── c_api_internal.h │ │ │ ├── ffi.h │ │ │ └── ffi_test.cc │ │ ├── call_frame.cc │ │ ├── call_frame.h │ │ ├── ffi.h │ │ ├── ffi_api.cc │ │ ├── ffi_api.h │ │ └── ffi_test.cc │ │ ├── frontend_attributes.cc │ │ ├── frontend_attributes.h │ │ ├── glob_lit_test.bzl │ │ ├── hlo │ │ ├── evaluator │ │ │ ├── BUILD │ │ │ ├── hlo_evaluator.cc │ │ │ ├── hlo_evaluator.h │ │ │ ├── hlo_evaluator_test.cc │ │ │ ├── hlo_evaluator_typed_visitor.h │ │ │ ├── hlo_evaluator_typed_visitor_bfloat16.cc │ │ │ ├── hlo_evaluator_typed_visitor_bool.cc │ │ │ ├── hlo_evaluator_typed_visitor_complex128.cc │ │ │ ├── hlo_evaluator_typed_visitor_complex64.cc │ │ │ ├── hlo_evaluator_typed_visitor_double.cc │ │ │ ├── hlo_evaluator_typed_visitor_float.cc │ │ │ ├── hlo_evaluator_typed_visitor_float8.cc │ │ │ ├── hlo_evaluator_typed_visitor_half.cc │ │ │ ├── hlo_evaluator_typed_visitor_int16.cc │ │ │ ├── hlo_evaluator_typed_visitor_int32.cc │ │ │ ├── hlo_evaluator_typed_visitor_int4.cc │ │ │ ├── hlo_evaluator_typed_visitor_int64.cc │ │ │ ├── hlo_evaluator_typed_visitor_int8.cc │ │ │ ├── hlo_evaluator_typed_visitor_uint16.cc │ │ │ ├── hlo_evaluator_typed_visitor_uint32.cc │ │ │ ├── hlo_evaluator_typed_visitor_uint64.cc │ │ │ └── hlo_evaluator_typed_visitor_uint8.cc │ │ ├── experimental │ │ │ └── auto_sharding │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── auto_sharding.cc │ │ │ │ ├── auto_sharding.h │ │ │ │ ├── auto_sharding.proto │ │ │ │ ├── auto_sharding_cost_graph.h │ │ │ │ ├── auto_sharding_dot_handler.cc │ │ │ │ ├── auto_sharding_impl.cc │ │ │ │ ├── auto_sharding_option.cc │ │ │ │ ├── auto_sharding_option.h │ │ │ │ ├── auto_sharding_runner.cc │ │ │ │ ├── auto_sharding_solver.cc │ │ │ │ ├── auto_sharding_solver.h │ │ │ │ ├── auto_sharding_solver_impl.cc │ │ │ │ ├── auto_sharding_solver_test.cc │ │ │ │ ├── auto_sharding_strategy.cc │ │ │ │ ├── auto_sharding_strategy.h │ │ │ │ ├── auto_sharding_test.cc │ │ │ │ ├── auto_sharding_util.cc │ │ │ │ ├── auto_sharding_util.h │ │ │ │ ├── auto_sharding_wrapper.h │ │ │ │ ├── cluster_environment.cc │ │ │ │ ├── cluster_environment.h │ │ │ │ ├── matrix.h │ │ │ │ ├── metrics.cc │ │ │ │ ├── metrics.h │ │ │ │ └── profiling_result.h │ │ ├── ir │ │ │ ├── BUILD │ │ │ ├── dfs_hlo_visitor.cc │ │ │ ├── dfs_hlo_visitor.h │ │ │ ├── dfs_hlo_visitor_with_default.h │ │ │ ├── dynamic_parameter_binding.cc │ │ │ ├── dynamic_parameter_binding.h │ │ │ ├── hlo_casting_utils.h │ │ │ ├── hlo_clone_context.h │ │ │ ├── hlo_computation.cc │ │ │ ├── hlo_computation.h │ │ │ ├── hlo_domain_metadata.h │ │ │ ├── hlo_frontend_attributes.cc │ │ │ ├── hlo_frontend_attributes.h │ │ │ ├── hlo_input_output_alias_config.cc │ │ │ ├── hlo_input_output_alias_config.h │ │ │ ├── hlo_instruction.cc │ │ │ ├── hlo_instruction.h │ │ │ ├── hlo_instructions.cc │ │ │ ├── hlo_instructions.h │ │ │ ├── hlo_module.cc │ │ │ ├── hlo_module.h │ │ │ ├── hlo_module_group.cc │ │ │ ├── hlo_module_group.h │ │ │ ├── hlo_module_metadata.cc │ │ │ ├── hlo_module_metadata.h │ │ │ ├── hlo_op_metadata.cc │ │ │ ├── hlo_op_metadata.h │ │ │ ├── hlo_opcode.cc │ │ │ ├── hlo_opcode.h │ │ │ ├── hlo_reachability.cc │ │ │ ├── hlo_reachability.h │ │ │ ├── hlo_schedule.cc │ │ │ ├── hlo_schedule.h │ │ │ ├── hlo_sharding.cc │ │ │ ├── hlo_sharding.h │ │ │ ├── hlo_sharding_metadata.cc │ │ │ ├── hlo_sharding_metadata.h │ │ │ ├── tile_assignment.cc │ │ │ └── tile_assignment.h │ │ ├── transforms │ │ │ ├── BUILD │ │ │ ├── hlo_constant_splitter.cc │ │ │ ├── hlo_constant_splitter.h │ │ │ └── hlo_constant_splitter_test.cc │ │ └── utils │ │ │ ├── BUILD │ │ │ ├── hlo_live_range.cc │ │ │ ├── hlo_live_range.h │ │ │ ├── hlo_live_range_test.cc │ │ │ ├── hlo_matchers.cc │ │ │ ├── hlo_matchers.h │ │ │ ├── hlo_matchers_test.cc │ │ │ ├── hlo_query.cc │ │ │ ├── hlo_query.h │ │ │ ├── hlo_sharding_util.cc │ │ │ ├── hlo_sharding_util.h │ │ │ └── hlo_sharding_util_test.cc │ │ ├── index_util.cc │ │ ├── index_util.h │ │ ├── index_util_test.cc │ │ ├── iterator_util.h │ │ ├── iterator_util_test.cc │ │ ├── layout.cc │ │ ├── layout.h │ │ ├── layout_test.cc │ │ ├── layout_util.cc │ │ ├── layout_util.h │ │ ├── layout_util_test.cc │ │ ├── lazy.h │ │ ├── literal.cc │ │ ├── literal.h │ │ ├── literal_comparison.cc │ │ ├── literal_comparison.h │ │ ├── literal_test.cc │ │ ├── literal_util.cc │ │ ├── literal_util.h │ │ ├── map_util.h │ │ ├── metric_table_report.cc │ │ ├── metric_table_report.h │ │ ├── mlir │ │ ├── backends │ │ │ ├── cpu │ │ │ │ ├── BUILD │ │ │ │ ├── transforms │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── legalize_i1_vector_transfers.cc │ │ │ │ │ ├── legalize_library_ops.cc │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── passes.td │ │ │ │ │ ├── remove_copies_to_out_params.cc │ │ │ │ │ ├── sparse_rewrite_passes.cc │ │ │ │ │ ├── tests │ │ │ │ │ │ ├── BUILD │ │ │ │ │ │ ├── collective_ops.mlir │ │ │ │ │ │ ├── fft.mlir │ │ │ │ │ │ ├── legalize_i1_vector_transfers.mlir │ │ │ │ │ │ ├── library_ops_to_cpu_runtime.mlir │ │ │ │ │ │ ├── lmhlo_custom_call.mlir │ │ │ │ │ │ ├── remove_copies_to_out_params.mlir │ │ │ │ │ │ ├── rng_bit_generator.mlir │ │ │ │ │ │ ├── xla_abi_legalization.mlir │ │ │ │ │ │ ├── xla_cpu_infeed.mlir │ │ │ │ │ │ ├── xla_cpu_memref_element_cast_to_llvm.mlir │ │ │ │ │ │ └── xla_cpu_outfeed.mlir │ │ │ │ │ ├── xla_abi_legalization.cc │ │ │ │ │ ├── xla_cpu_memref_element_cast_to_llvm.cc │ │ │ │ │ ├── xla_cpu_to_cpu_runtime.cc │ │ │ │ │ └── xla_rewrite_realloc_to_alloc.cc │ │ │ │ └── xla-cpu-opt.cc │ │ │ └── gpu │ │ │ │ ├── BUILD │ │ │ │ ├── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── add_concurrent_regions.cc │ │ │ │ ├── add_hlo_trace_annotations.cc │ │ │ │ ├── dataflow_analysis.cc │ │ │ │ ├── dataflow_analysis.h │ │ │ │ ├── gpu_to_gpu_runtime.cc │ │ │ │ ├── lmhlo_gpu_to_gpu_runtime.cc │ │ │ │ ├── lmhlo_to_gpu_launch.cc │ │ │ │ ├── lmhlo_to_gpu_runtime.cc │ │ │ │ ├── memref_get_global_to_arg.cc │ │ │ │ ├── outline_cuda_graphs.cc │ │ │ │ ├── passes.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ ├── stream_assignment.cc │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── add_concurrent_regions.mlir │ │ │ │ │ ├── add_hlo_trace.mlir │ │ │ │ │ ├── gpu_launch.mlir │ │ │ │ │ ├── gpu_memcpy.mlir │ │ │ │ │ ├── gpu_memset.mlir │ │ │ │ │ ├── lmhlo_case.mlir │ │ │ │ │ ├── lmhlo_custom_call.mlir │ │ │ │ │ ├── lmhlo_fft.mlir │ │ │ │ │ ├── lmhlo_gpu_cholesky.mlir │ │ │ │ │ ├── lmhlo_gpu_conv.mlir │ │ │ │ │ ├── lmhlo_gpu_cublas_lt_matmul.mlir │ │ │ │ │ ├── lmhlo_gpu_gemm.mlir │ │ │ │ │ ├── lmhlo_infeed.mlir │ │ │ │ │ ├── lmhlo_outfeed.mlir │ │ │ │ │ ├── lmhlo_send_recv.mlir │ │ │ │ │ ├── lmhlo_while.mlir │ │ │ │ │ ├── memref_get_global_to_arg.mlir │ │ │ │ │ ├── outline_cuda_graphs.mlir │ │ │ │ │ └── stream_assignment.mlir │ │ │ │ └── uid_generator.h │ │ │ │ └── xla-gpu-opt.cc │ │ ├── framework │ │ │ ├── ir │ │ │ │ ├── BUILD │ │ │ │ ├── xla_framework.cc │ │ │ │ ├── xla_framework.h │ │ │ │ └── xla_framework_ops.td │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── legalize-xla-framework.mlir │ │ │ │ ├── outline-with-xla-framework.mlir │ │ │ │ └── xla-framework.mlir │ │ │ └── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── outline_with_xla_framework.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ └── xla_framework_to_llvm_pass.cc │ │ ├── math │ │ │ ├── BUILD │ │ │ └── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── math_approximation.cc │ │ │ │ ├── math_optimization.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ └── tests │ │ │ │ ├── BUILD │ │ │ │ └── math_optimization.mlir │ │ ├── memref │ │ │ ├── BUILD │ │ │ └── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── aligned_allocations.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ └── tests │ │ │ │ ├── BUILD │ │ │ │ └── aligned_allocations.mlir │ │ ├── runtime │ │ │ ├── BUILD │ │ │ ├── ir │ │ │ │ ├── BUILD │ │ │ │ ├── rt_dialect.cc │ │ │ │ ├── rt_dialect.h │ │ │ │ ├── rt_dialect.td │ │ │ │ ├── rt_interfaces.cc │ │ │ │ ├── rt_interfaces.h │ │ │ │ ├── rt_interfaces.td │ │ │ │ ├── rt_ops.cc │ │ │ │ ├── rt_ops.h │ │ │ │ ├── rt_ops.td │ │ │ │ └── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── ops.mlir │ │ │ │ │ ├── ops_verify.mlir │ │ │ │ │ ├── testlib.cc │ │ │ │ │ ├── testlib.h │ │ │ │ │ ├── testlib.mlir │ │ │ │ │ ├── testlib.td │ │ │ │ │ ├── testlib_attrs.td │ │ │ │ │ ├── testlib_enums.td │ │ │ │ │ └── testlib_types.td │ │ │ ├── transforms │ │ │ │ ├── BUILD │ │ │ │ ├── add_initializations.cc │ │ │ │ ├── calling_convention.cc │ │ │ │ ├── calling_convention.h │ │ │ │ ├── calling_convention_test.cc │ │ │ │ ├── compilation_pipeline_cpu.cc │ │ │ │ ├── compilation_pipeline_cpu.h │ │ │ │ ├── compilation_pipeline_gpu.cc │ │ │ │ ├── compilation_pipeline_gpu.h │ │ │ │ ├── compilation_pipeline_options.h │ │ │ │ ├── compiler.h │ │ │ │ ├── convert_asserts.cc │ │ │ │ ├── convert_custom_calls.cc │ │ │ │ ├── custom_call_encoding.cc │ │ │ │ ├── custom_call_encoding.h │ │ │ │ ├── export_functions.cc │ │ │ │ ├── jit_compiler.cc │ │ │ │ ├── jit_compiler.h │ │ │ │ ├── move_allocas_to_entry_block.cc │ │ │ │ ├── ordinal_assignment.cc │ │ │ │ ├── passes.h │ │ │ │ ├── passes.td │ │ │ │ ├── rt_to_llvm.cc │ │ │ │ ├── specialization.cc │ │ │ │ ├── specialization.h │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── compilation_pipeline.mlir │ │ │ │ │ ├── convert_asserts.mlir │ │ │ │ │ ├── convert_custom_calls.mlir │ │ │ │ │ ├── export_functions.mlir │ │ │ │ │ ├── move_allocas_to_entry_block.mlir │ │ │ │ │ ├── ordinal_assignment.mlir │ │ │ │ │ ├── rt_to_llvm.mlir │ │ │ │ │ ├── testlib_pipeline.cc │ │ │ │ │ └── testlib_pipeline.h │ │ │ │ ├── type_converter.cc │ │ │ │ ├── type_converter.h │ │ │ │ └── type_converter_test.cc │ │ │ ├── utils │ │ │ │ ├── BUILD │ │ │ │ ├── async_runtime_api.cc │ │ │ │ ├── async_runtime_api.h │ │ │ │ ├── c_runner_utils.h │ │ │ │ ├── constraints.cc │ │ │ │ ├── constraints.h │ │ │ │ ├── custom_calls.cc │ │ │ │ ├── custom_calls.h │ │ │ │ └── float_16bits.h │ │ │ └── xla-runtime-opt.cc │ │ ├── utils │ │ │ ├── BUILD │ │ │ ├── error_util.cc │ │ │ ├── error_util.h │ │ │ └── error_util_test.cc │ │ └── xla_cpu │ │ │ ├── ir │ │ │ ├── BUILD │ │ │ ├── xla_cpu.cc │ │ │ ├── xla_cpu.h │ │ │ ├── xla_cpu_dialect.td │ │ │ ├── xla_cpu_enums.td │ │ │ └── xla_cpu_ops.td │ │ │ └── tests │ │ │ ├── BUILD │ │ │ ├── bufferize.mlir │ │ │ ├── invalid.mlir │ │ │ └── ops.mlir │ │ ├── mlir_hlo │ │ ├── .bazelignore │ │ ├── .bazelrc │ │ ├── .clang-format │ │ ├── .clang-tidy │ │ ├── .gitignore │ │ ├── BUILD │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── WORKSPACE │ │ ├── analysis │ │ │ ├── CMakeLists.txt │ │ │ ├── test_userange_analysis.cc │ │ │ ├── userange_analysis.cc │ │ │ └── userange_analysis.h │ │ ├── bindings │ │ │ ├── CMakeLists.txt │ │ │ ├── c │ │ │ │ ├── Attributes.cc │ │ │ │ ├── Attributes.h │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── Dialects.cc │ │ │ │ ├── Dialects.h │ │ │ │ ├── Passes.cc │ │ │ │ ├── Passes.h │ │ │ │ ├── Types.cc │ │ │ │ └── Types.h │ │ │ └── python │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── MlirHloModule.cc │ │ │ │ └── mlir │ │ │ │ └── dialects │ │ │ │ ├── MhloOps.td │ │ │ │ └── mhlo.py │ │ ├── build_tools │ │ │ ├── build_mlir.sh │ │ │ └── llvm_version.txt │ │ ├── cmake │ │ │ └── modules │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── MHLOConfig.cmake.in │ │ ├── deallocation │ │ │ ├── CMakeLists.txt │ │ │ ├── transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── buffer_reuse.cc │ │ │ │ ├── passes.h │ │ │ │ └── passes.td │ │ │ └── utils │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── util.cc │ │ │ │ └── util.h │ │ ├── lhlo │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── lhlo_dialect.td │ │ │ │ ├── lhlo_ops.cc │ │ │ │ ├── lhlo_ops.h │ │ │ │ ├── lhlo_ops.td │ │ │ │ ├── lhlo_ops_base.td │ │ │ │ ├── lhlo_ops_structs.h │ │ │ │ ├── lhlo_ops_structs.td │ │ │ │ ├── lhlo_structured_interface.cc │ │ │ │ ├── lhlo_structured_interface.h │ │ │ │ └── lhlo_structured_interface.td │ │ │ ├── transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── legalize_to_tensor_op │ │ │ │ │ └── legalize_to_tensor_op.cc │ │ │ │ ├── lhlo_elemental_utils.cc │ │ │ │ ├── lhlo_elemental_utils.h │ │ │ │ ├── lhlo_legalize_to_affine │ │ │ │ │ └── lhlo_legalize_to_affine.cc │ │ │ │ ├── lhlo_legalize_to_gpu │ │ │ │ │ └── lhlo_legalize_to_gpu.cc │ │ │ │ ├── lhlo_legalize_to_parallel_loops │ │ │ │ │ └── lhlo_legalize_to_parallel_loops.cc │ │ │ │ ├── lmhlo_passes.td │ │ │ │ ├── map_hlo_to_lhlo_op.h │ │ │ │ ├── map_lhlo_to_hlo_op.h │ │ │ │ ├── map_lmhlo_to_scalar_op.h │ │ │ │ └── passes.h │ │ │ └── utils │ │ │ │ └── lhlo_utils.h │ │ ├── lhlo_gpu │ │ │ ├── CMakeLists.txt │ │ │ └── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── lhlo_gpu_ops.cc │ │ │ │ ├── lhlo_gpu_ops.h │ │ │ │ ├── lhlo_gpu_ops.td │ │ │ │ ├── lhlo_gpu_ops_base.td │ │ │ │ └── lhlo_gpu_ops_enums.td │ │ ├── mhlo │ │ │ ├── CMakeLists.txt │ │ │ ├── IR │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── chlo_canonicalize.td │ │ │ │ ├── hlo_base.td │ │ │ │ ├── hlo_ops.cc │ │ │ │ ├── hlo_ops.h │ │ │ │ ├── hlo_ops.td │ │ │ │ ├── hlo_ops_attrs.td │ │ │ │ ├── hlo_ops_common.cc │ │ │ │ ├── hlo_ops_common.h │ │ │ │ ├── hlo_ops_common.td │ │ │ │ ├── hlo_ops_enums.td │ │ │ │ ├── hlo_ops_typedefs.td │ │ │ │ ├── hlo_patterns.td │ │ │ │ ├── hlo_utils.td │ │ │ │ ├── init.cc │ │ │ │ ├── mhlo_bytecode.cc │ │ │ │ ├── mhlo_bytecode.h │ │ │ │ ├── mhlo_canonicalize.td │ │ │ │ └── register.h │ │ │ ├── analysis │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── shape_component_analysis.cc │ │ │ │ ├── shape_component_analysis.h │ │ │ │ └── test_shape_component_analysis.cc │ │ │ ├── interfaces │ │ │ │ └── bufferizable_op_interface_impl.h │ │ │ ├── transforms │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── broadcast_propagation │ │ │ │ │ └── broadcast_propagation.cc │ │ │ │ ├── chlo_legalize_to_hlo │ │ │ │ │ ├── chlo_legalize_to_hlo.cc │ │ │ │ │ ├── chlo_legalize_to_hlo_pass.cc │ │ │ │ │ └── chlo_legalize_to_hlo_patterns.td │ │ │ │ ├── collapse_elementwise_map │ │ │ │ │ └── collapse_elementwise_map.cc │ │ │ │ ├── constraint_fusion │ │ │ │ │ └── constraint_fusion_pass.cc │ │ │ │ ├── convert_to_signless │ │ │ │ │ └── convert_to_signless_pass.cc │ │ │ │ ├── expand_hlo_tuples │ │ │ │ │ └── expand_hlo_tuples.cc │ │ │ │ ├── expand_ops_simplifier │ │ │ │ │ └── expand_ops_simplifier.cc │ │ │ │ ├── group_reduction_dimensions │ │ │ │ │ └── group_reduction_dimensions.cc │ │ │ │ ├── hlo_legalize_shape_ops_to_standard │ │ │ │ │ └── hlo_legalize_shape_ops_to_standard.cc │ │ │ │ ├── hlo_legalize_to_arithmetic │ │ │ │ │ └── hlo_legalize_to_arithmetic.cc │ │ │ │ ├── hlo_legalize_to_memref │ │ │ │ │ └── hlo_legalize_to_memref.cc │ │ │ │ ├── hlo_legalize_to_stablehlo │ │ │ │ │ ├── hlo_legalize_to_stablehlo.cc │ │ │ │ │ └── hlo_legalize_to_stablehlo_pass.cc │ │ │ │ ├── legalize_broadcast_to_broadcast_in_dim │ │ │ │ │ └── legalize_broadcast_to_broadcast_in_dim.cc │ │ │ │ ├── legalize_control_flow │ │ │ │ │ └── legalize_control_flow.cc │ │ │ │ ├── legalize_create_token_to_after_all │ │ │ │ │ └── legalize_create_token_to_after_all.cc │ │ │ │ ├── legalize_cross_replica_sum_to_all_reduce │ │ │ │ │ └── legalize_cross_replica_sum_to_all_reduce.cc │ │ │ │ ├── legalize_dot_general_to_dot │ │ │ │ │ └── legalize_dot_general_to_dot.cc │ │ │ │ ├── legalize_dot_to_dot_general │ │ │ │ │ └── legalize_dot_to_dot_general.cc │ │ │ │ ├── legalize_einsum_to_dot_general │ │ │ │ │ └── legalize_einsum_to_dot_general.cc │ │ │ │ ├── legalize_gather_to_torch_index_select │ │ │ │ │ └── legalize_gather_to_torch_index_select.cc │ │ │ │ ├── legalize_shape_computations │ │ │ │ │ └── legalize_shape_computations.cc │ │ │ │ ├── legalize_sort │ │ │ │ │ └── legalize_sort.cc │ │ │ │ ├── legalize_sparse_ops │ │ │ │ │ ├── legalize_sparse_ops.cc │ │ │ │ │ └── sparse_ops_to_custom_calls.cc │ │ │ │ ├── legalize_to_linalg │ │ │ │ │ └── legalize_to_linalg.cc │ │ │ │ ├── legalize_to_standard │ │ │ │ │ ├── legalize_to_standard.cc │ │ │ │ │ └── legalize_to_standard_patterns.td │ │ │ │ ├── legalize_torch_index_select_to_gather │ │ │ │ │ └── legalize_torch_index_select_to_gather.cc │ │ │ │ ├── legalize_trigonometric_to_approximation │ │ │ │ │ └── legalize_trigonometric_to_approximation.cc │ │ │ │ ├── lower_complex │ │ │ │ │ ├── lower_complex.cc │ │ │ │ │ └── lower_complex_patterns.td │ │ │ │ ├── lower_general_dot │ │ │ │ │ └── lower_general_dot.cc │ │ │ │ ├── map_chlo_to_hlo_op.h │ │ │ │ ├── map_mhlo_to_scalar_op.h │ │ │ │ ├── map_stablehlo_to_hlo_op.h │ │ │ │ ├── materialize_broadcasts │ │ │ │ │ ├── materialize_broadcasts.cc │ │ │ │ │ └── materialize_broadcasts_pass.cc │ │ │ │ ├── merge_assuming_ops │ │ │ │ │ └── merge_assuming_ops.cc │ │ │ │ ├── mhlo_canonicalize_dot │ │ │ │ │ └── mhlo_canonicalize_dot.cc │ │ │ │ ├── mhlo_canonicalize_gather │ │ │ │ │ └── mhlo_canonicalize_gather.cc │ │ │ │ ├── mhlo_canonicalize_reduction │ │ │ │ │ └── mhlo_canonicalize_reduction.cc │ │ │ │ ├── mhlo_canonicalize_scatter │ │ │ │ │ └── mhlo_canonicalize_scatter.cc │ │ │ │ ├── mhlo_flatten_tuple │ │ │ │ │ └── mhlo_flatten_tuple.cc │ │ │ │ ├── mhlo_passes.td │ │ │ │ ├── optimize_mhlo │ │ │ │ │ ├── optimize_mhlo.cc │ │ │ │ │ └── optimize_mhlo_pass.cc │ │ │ │ ├── passes.h │ │ │ │ ├── prepare_for_export │ │ │ │ │ └── prepare_for_export.cc │ │ │ │ ├── rank_specialization │ │ │ │ │ └── rank_specialization.cc │ │ │ │ ├── restrict_max_rank │ │ │ │ │ └── restrict_max_rank.cc │ │ │ │ ├── rewriters.h │ │ │ │ ├── shape_legalize_to_hlo │ │ │ │ │ └── shape_legalize_to_hlo.cc │ │ │ │ ├── shape_reification │ │ │ │ │ └── shape_reification_pass.cc │ │ │ │ ├── shape_simplification │ │ │ │ │ └── shape_simplification.cc │ │ │ │ ├── sink_constants_to_control_flow │ │ │ │ │ └── sink_constants_to_control_flow.cc │ │ │ │ ├── sparse_rewriting │ │ │ │ │ └── sparse_rewriting.cc │ │ │ │ ├── stablehlo_legalize_to_hlo │ │ │ │ │ ├── stablehlo_legalize_to_hlo.cc │ │ │ │ │ └── stablehlo_legalize_to_hlo_pass.cc │ │ │ │ ├── symbolic_shape_optimization │ │ │ │ │ └── symbolic_shape_optimization.cc │ │ │ │ ├── test_infer_shaped_type │ │ │ │ │ └── test_infer_shaped_type_pass.cc │ │ │ │ └── unfuse_batch_norm │ │ │ │ │ ├── unfuse_batch_norm.cc │ │ │ │ │ └── unfuse_batch_norm_pass.cc │ │ │ └── utils │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── legalize_to_linalg_utils.cc │ │ │ │ ├── legalize_to_linalg_utils.h │ │ │ │ ├── mhlo_rng_utils.cc │ │ │ │ ├── mhlo_rng_utils.h │ │ │ │ ├── mhlo_scatter_gather_utils.cc │ │ │ │ ├── mhlo_scatter_gather_utils.h │ │ │ │ ├── type_conversion.cc │ │ │ │ └── type_conversion.h │ │ ├── tests │ │ │ ├── BUILD │ │ │ ├── CMakeLists.txt │ │ │ ├── Dialect │ │ │ │ ├── bufferization │ │ │ │ │ └── hlo_one_shot_bufferize.mlir │ │ │ │ ├── chlo │ │ │ │ │ ├── chlo_legalize_to_hlo_broadcasts.mlir │ │ │ │ │ ├── chlo_legalize_to_hlo_no_broadcasts.mlir │ │ │ │ │ ├── chlo_legalize_to_mhlo.mlir │ │ │ │ │ └── sparse_chlo_legalize_to_linalg.mlir │ │ │ │ ├── deallocation │ │ │ │ │ └── buffer_reuse.mlir │ │ │ │ ├── lhlo │ │ │ │ │ ├── lhlo-legalize-select-and-scatter.mlir │ │ │ │ │ ├── lhlo-legalize-to-affine.mlir │ │ │ │ │ ├── lhlo-legalize-to-gpu.mlir │ │ │ │ │ ├── lhlo-legalize-to-parallel-loops.mlir │ │ │ │ │ ├── lhlo-legalize-to-tensor-op.mlir │ │ │ │ │ └── ops.mlir │ │ │ │ ├── lhlo_gpu │ │ │ │ │ └── lhlo_gpu_ops.mlir │ │ │ │ └── mhlo │ │ │ │ │ ├── attrs.mlir │ │ │ │ │ ├── broadcast_propagation.mlir │ │ │ │ │ ├── canonicalize │ │ │ │ │ ├── bitcast.mlir │ │ │ │ │ ├── canonicalize.mlir │ │ │ │ │ ├── concatenate.mlir │ │ │ │ │ ├── convert.mlir │ │ │ │ │ ├── convolution.mlir │ │ │ │ │ ├── custom_call.mlir │ │ │ │ │ ├── folder_limit.mlir │ │ │ │ │ ├── reduce.mlir │ │ │ │ │ ├── reshape.mlir │ │ │ │ │ ├── reverse.mlir │ │ │ │ │ ├── scatter.mlir │ │ │ │ │ ├── transpose.mlir │ │ │ │ │ ├── tuple.mlir │ │ │ │ │ └── while.mlir │ │ │ │ │ ├── constraint_fusion.mlir │ │ │ │ │ ├── convert_to_signless.mlir │ │ │ │ │ ├── expand_hlo_tuples.mlir │ │ │ │ │ ├── expand_ops_simplifier.mlir │ │ │ │ │ ├── group_reduction_dimensions.mlir │ │ │ │ │ ├── hlo-collapse-elementwise-map.mlir │ │ │ │ │ ├── hlo-legalize-broadcast-to-broadcast-in-dim.mlir │ │ │ │ │ ├── hlo-legalize-create-token-to-after-all.mlir │ │ │ │ │ ├── hlo-legalize-cross-replica-sum-to-all-reduce.mlir │ │ │ │ │ ├── hlo-legalize-dot-general-to-dot.mlir │ │ │ │ │ ├── hlo-legalize-dot-to-dot-general.mlir │ │ │ │ │ ├── hlo-legalize-einsum-to-dot-general.mlir │ │ │ │ │ ├── hlo-legalize-gather-to-torch-index-select.mlir │ │ │ │ │ ├── hlo-legalize-rng-to-linalg.mlir │ │ │ │ │ ├── hlo-legalize-shape-ops-to-standard.mlir │ │ │ │ │ ├── hlo-legalize-sort.mlir │ │ │ │ │ ├── hlo-legalize-to-arithmetic.mlir │ │ │ │ │ ├── hlo-legalize-to-linalg.mlir │ │ │ │ │ ├── hlo-legalize-to-memref-unranked.mlir │ │ │ │ │ ├── hlo-legalize-to-memref.mlir │ │ │ │ │ ├── hlo-legalize-to-stablehlo-experimental.mlir │ │ │ │ │ ├── hlo-legalize-to-stablehlo.mlir │ │ │ │ │ ├── hlo-legalize-torch-index-select-to-gather.mlir │ │ │ │ │ ├── inlining.mlir │ │ │ │ │ ├── invalid.mlir │ │ │ │ │ ├── legalize-control-flow.mlir │ │ │ │ │ ├── legalize-hlo-shape-computations.mlir │ │ │ │ │ ├── legalize-to-std.mlir │ │ │ │ │ ├── lower-complex.mlir │ │ │ │ │ ├── lower-general-dot.mlir │ │ │ │ │ ├── materialize-broadcasts.mlir │ │ │ │ │ ├── merge_assuming_ops.mlir │ │ │ │ │ ├── mhlo_bytecode_customizations.mlir │ │ │ │ │ ├── mhlo_canonicalize_dot.mlir │ │ │ │ │ ├── mhlo_canonicalize_gather.mlir │ │ │ │ │ ├── mhlo_canonicalize_reduction.mlir │ │ │ │ │ ├── mhlo_canonicalize_scatter.mlir │ │ │ │ │ ├── mhlo_flatten_tuple.mlir │ │ │ │ │ ├── mhlo_infer_shape_type_methods.mlir │ │ │ │ │ ├── mhlo_ops_prettyprint.mlir │ │ │ │ │ ├── mhlo_reduce_pretty_print.mlir │ │ │ │ │ ├── ops.mlir │ │ │ │ │ ├── optimize-hlo.mlir │ │ │ │ │ ├── prepare-for-export.mlir │ │ │ │ │ ├── reify-result-types.mlir │ │ │ │ │ ├── restrict_max_rank.mlir │ │ │ │ │ ├── shape_cstr_legalize_to_hlo.mlir │ │ │ │ │ ├── shape_legalize_to_hlo.mlir │ │ │ │ │ ├── shape_reification.mlir │ │ │ │ │ ├── sink-constants-to-control-flow.mlir │ │ │ │ │ ├── sparse_gendot_lower.mlir │ │ │ │ │ ├── sparse_lower.mlir │ │ │ │ │ ├── sparse_ops.mlir │ │ │ │ │ ├── sparse_rewriting.mlir │ │ │ │ │ ├── sparse_transpose.mlir │ │ │ │ │ ├── stablehlo-legalize-to-hlo.mlir │ │ │ │ │ ├── symbolic-shape-optimization.mlir │ │ │ │ │ ├── unfuse_batch_norm.mlir │ │ │ │ │ ├── verifier_bounds.mlir │ │ │ │ │ ├── verifier_conv_op.mlir │ │ │ │ │ ├── verifier_reduce_op.mlir │ │ │ │ │ ├── verifier_reduce_window_op.mlir │ │ │ │ │ ├── verifier_scatter_op.mlir │ │ │ │ │ ├── verifier_select_and_scatter_op.mlir │ │ │ │ │ ├── verifier_while_op.mlir │ │ │ │ │ └── while_prettyprint.mlir │ │ │ ├── alloc_to_arg.mlir │ │ │ ├── buffer_packing.mlir │ │ │ ├── bufferize.mlir │ │ │ ├── bufferize_one_shot.mlir │ │ │ ├── capi_test.c │ │ │ ├── collapse_parallel_loops_to_1d_pass.mlir │ │ │ ├── detensorize_scf_ops.mlir │ │ │ ├── index_type_llvm_lowering.mlir │ │ │ ├── legalize-trigonometric-to-approximation.mlir │ │ │ ├── lit.cfg.py │ │ │ ├── lit.site.cfg.py.in │ │ │ ├── lower_index_cast.mlir │ │ │ ├── naive_copy_removal.mlir │ │ │ ├── propagate_static_shapes.mlir │ │ │ ├── python │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── attributes.py │ │ │ │ ├── smoketest.py │ │ │ │ └── types.py │ │ │ ├── rank-specialization.mlir │ │ │ ├── shape-component-analysis.mlir │ │ │ ├── shape_simplification.mlir │ │ │ ├── test_userange.mlir │ │ │ ├── tile_loops.mlir │ │ │ ├── unbufferize.mlir │ │ │ ├── unroll-loops.mlir │ │ │ └── vectorize_copy.mlir │ │ ├── tools │ │ │ ├── CMakeLists.txt │ │ │ └── mlir-hlo-opt │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── mlir-hlo-opt.cc │ │ ├── transforms │ │ │ ├── CMakeLists.txt │ │ │ ├── alloc_to_arg_pass.cc │ │ │ ├── buffer_packing.cc │ │ │ ├── bufferize.cc │ │ │ ├── bufferize_pass.cc │ │ │ ├── collapse_parallel_loops_to_1d_pass.cc │ │ │ ├── detensorize_scf_ops.cc │ │ │ ├── generic_host_to_llvm.cc │ │ │ ├── gpu_kernel_lowering_passes.cc │ │ │ ├── gpu_passes.cc │ │ │ ├── gpu_passes.h │ │ │ ├── gpu_passes.td │ │ │ ├── lower_index_cast_pass.cc │ │ │ ├── naive_copy_removal.cc │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── propagate_static_shapes_to_kernel.cc │ │ │ ├── rewriters.h │ │ │ ├── test_hlo_transform_dialect_interpreter.cc │ │ │ ├── tile_loops_pass.cc │ │ │ ├── unbufferize_pass.cc │ │ │ ├── unroll_loops.cc │ │ │ └── vectorize_copy.cc │ │ └── utils │ │ │ ├── CMakeLists.txt │ │ │ ├── codegen_utils.cc │ │ │ ├── codegen_utils.h │ │ │ ├── convert_op_folder.cc │ │ │ ├── convert_op_folder.h │ │ │ ├── cycle_detector.cc │ │ │ ├── cycle_detector.h │ │ │ ├── cycle_detector_test.cc │ │ │ ├── hlo_utils.cc │ │ │ ├── hlo_utils.h │ │ │ └── placement_utils.h │ │ ├── overflow_util.h │ │ ├── packed_literal_reader.cc │ │ ├── packed_literal_reader.h │ │ ├── parse_flags_from_env.cc │ │ ├── parse_flags_from_env.h │ │ ├── parse_flags_from_env_test.cc │ │ ├── permutation_util.cc │ │ ├── permutation_util.h │ │ ├── permutation_util_test.cc │ │ ├── pjrt │ │ ├── BUILD │ │ ├── c │ │ │ ├── BUILD │ │ │ ├── CHANGELOG.md │ │ │ ├── README.md │ │ │ ├── docs │ │ │ │ └── pjrt_integration_guide.md │ │ │ ├── pjrt_c_api.h │ │ │ ├── pjrt_c_api_cpu.cc │ │ │ ├── pjrt_c_api_cpu.h │ │ │ ├── pjrt_c_api_cpu_internal.cc │ │ │ ├── pjrt_c_api_cpu_internal.h │ │ │ ├── pjrt_c_api_cpu_test.cc │ │ │ ├── pjrt_c_api_gpu.cc │ │ │ ├── pjrt_c_api_gpu.h │ │ │ ├── pjrt_c_api_gpu_extension.h │ │ │ ├── pjrt_c_api_gpu_internal.cc │ │ │ ├── pjrt_c_api_gpu_internal.h │ │ │ ├── pjrt_c_api_gpu_test.cc │ │ │ ├── pjrt_c_api_gpu_version_script.lds │ │ │ ├── pjrt_c_api_helpers.cc │ │ │ ├── pjrt_c_api_helpers.h │ │ │ ├── pjrt_c_api_helpers_test.cc │ │ │ ├── pjrt_c_api_macros.h │ │ │ ├── pjrt_c_api_profiler_extension.h │ │ │ ├── pjrt_c_api_test.cc │ │ │ ├── pjrt_c_api_test.h │ │ │ ├── pjrt_c_api_test_base.cc │ │ │ ├── pjrt_c_api_test_base.h │ │ │ ├── pjrt_c_api_tpu.h │ │ │ ├── pjrt_c_api_wrapper_impl.cc │ │ │ └── pjrt_c_api_wrapper_impl.h │ │ ├── compile_options.proto │ │ ├── cpu │ │ │ ├── BUILD │ │ │ ├── abstract_tfrt_cpu_buffer.cc │ │ │ ├── abstract_tfrt_cpu_buffer.h │ │ │ ├── cpu_client.cc │ │ │ ├── cpu_client.h │ │ │ ├── cpu_client_test.cc │ │ │ ├── pjrt_client_test_cpu.cc │ │ │ ├── tracked_tfrt_cpu_device_buffer.cc │ │ │ ├── tracked_tfrt_cpu_device_buffer.h │ │ │ └── tracked_tfrt_cpu_device_buffer_test.cc │ │ ├── distributed │ │ │ ├── BUILD │ │ │ ├── client.cc │ │ │ ├── client.h │ │ │ ├── client_server_test.cc │ │ │ ├── distributed.cc │ │ │ ├── distributed.h │ │ │ ├── protocol.proto │ │ │ ├── service.cc │ │ │ ├── service.h │ │ │ ├── topology_util.cc │ │ │ ├── topology_util.h │ │ │ ├── topology_util_test.cc │ │ │ └── util.h │ │ ├── event_pool.cc │ │ ├── event_pool.h │ │ ├── executable_metadata.proto │ │ ├── execute_options.proto │ │ ├── gpu │ │ │ ├── BUILD │ │ │ ├── gpu_helpers.cc │ │ │ ├── gpu_helpers.h │ │ │ ├── gpu_topology.cc │ │ │ ├── gpu_topology.h │ │ │ ├── gpu_topology.proto │ │ │ ├── nccl_id_store.cc │ │ │ ├── nccl_id_store.h │ │ │ ├── pjrt_client_test_se_gpu.cc │ │ │ ├── se_gpu_pjrt_client.cc │ │ │ ├── se_gpu_pjrt_client.h │ │ │ ├── se_gpu_pjrt_client_test.cc │ │ │ ├── se_gpu_pjrt_compiler.cc │ │ │ ├── se_gpu_pjrt_compiler.h │ │ │ ├── se_gpu_pjrt_compiler_aot_test.cc │ │ │ └── se_gpu_pjrt_compiler_test.cc │ │ ├── host_callback.cc │ │ ├── host_callback.h │ │ ├── host_callback_test.cc │ │ ├── interpreter_device.cc │ │ ├── interpreter_device.h │ │ ├── layout_mode.cc │ │ ├── layout_mode.h │ │ ├── local_device_state.cc │ │ ├── local_device_state.h │ │ ├── lru_cache.h │ │ ├── lru_cache_test.cc │ │ ├── metrics.cc │ │ ├── metrics.h │ │ ├── mlir_to_hlo.cc │ │ ├── mlir_to_hlo.h │ │ ├── pjrt_api.cc │ │ ├── pjrt_api.h │ │ ├── pjrt_api_test.cc │ │ ├── pjrt_c_api_client.cc │ │ ├── pjrt_c_api_client.h │ │ ├── pjrt_c_api_client_test.cc │ │ ├── pjrt_client.cc │ │ ├── pjrt_client.h │ │ ├── pjrt_client_test.cc │ │ ├── pjrt_client_test.h │ │ ├── pjrt_common.h │ │ ├── pjrt_compiler.cc │ │ ├── pjrt_compiler.h │ │ ├── pjrt_compiler_test.cc │ │ ├── pjrt_device_description.h │ │ ├── pjrt_executable.cc │ │ ├── pjrt_executable.h │ │ ├── pjrt_executable_test.cc │ │ ├── pjrt_future.h │ │ ├── pjrt_stream_executor_client.cc │ │ ├── pjrt_stream_executor_client.h │ │ ├── pjrt_stream_executor_client_test.cc │ │ ├── plugin │ │ │ └── BUILD │ │ ├── semaphore.cc │ │ ├── semaphore.h │ │ ├── semaphore_test.cc │ │ ├── stream_executor_executable.cc │ │ ├── stream_executor_executable.h │ │ ├── stream_executor_executable.proto │ │ ├── tf_pjrt_client.cc │ │ ├── tf_pjrt_client.h │ │ ├── tf_pjrt_client_test.cc │ │ ├── tfrt_cpu_pjrt_client.h │ │ ├── tracked_device_buffer.cc │ │ ├── tracked_device_buffer.h │ │ ├── tracked_device_buffer_test.cc │ │ ├── transpose.cc │ │ ├── transpose.h │ │ ├── transpose_kernels.h │ │ ├── transpose_test.cc │ │ ├── utils.cc │ │ ├── utils.h │ │ ├── worker_thread.cc │ │ └── worker_thread.h │ │ ├── primitive_util.cc │ │ ├── primitive_util.h │ │ ├── primitive_util_test.cc │ │ ├── printer.cc │ │ ├── printer.h │ │ ├── protobuf_util.cc │ │ ├── protobuf_util.h │ │ ├── python │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── callback.cc │ │ ├── callback.h │ │ ├── custom_call_for_test.pyx │ │ ├── custom_call_sharding.cc │ │ ├── custom_call_sharding.h │ │ ├── dlpack.cc │ │ ├── dlpack.h │ │ ├── exceptions.h │ │ ├── ifrt │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── array.cc │ │ │ ├── array.h │ │ │ ├── array_impl_test_lib.cc │ │ │ ├── array_test.cc │ │ │ ├── client.cc │ │ │ ├── client.h │ │ │ ├── client_impl_test_lib.cc │ │ │ ├── compiler.cc │ │ │ ├── compiler.h │ │ │ ├── device.cc │ │ │ ├── device.h │ │ │ ├── dtype.cc │ │ │ ├── dtype.h │ │ │ ├── executable.cc │ │ │ ├── executable.h │ │ │ ├── future.cc │ │ │ ├── future.h │ │ │ ├── future_test.cc │ │ │ ├── host_callback.cc │ │ │ ├── host_callback.h │ │ │ ├── index.cc │ │ │ ├── index.h │ │ │ ├── index_domain.cc │ │ │ ├── index_domain.h │ │ │ ├── index_domain_test.cc │ │ │ ├── index_test.cc │ │ │ ├── ir │ │ │ │ ├── BUILD │ │ │ │ ├── compiler.cc │ │ │ │ ├── compiler.h │ │ │ │ ├── constants.h │ │ │ │ ├── ifrt_dialect.cc │ │ │ │ ├── ifrt_dialect.h │ │ │ │ ├── ifrt_dialect.td │ │ │ │ ├── ifrt_interfaces.cc │ │ │ │ ├── ifrt_interfaces.h │ │ │ │ ├── ifrt_interfaces.td │ │ │ │ ├── ifrt_ops.cc │ │ │ │ ├── ifrt_ops.h │ │ │ │ ├── ifrt_ops.td │ │ │ │ ├── sharding_param.cc │ │ │ │ ├── sharding_param.h │ │ │ │ ├── tests │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── executable_impl_test_base.cc │ │ │ │ │ ├── executable_impl_test_base.h │ │ │ │ │ ├── executable_impl_test_lib.cc │ │ │ │ │ ├── ifrt-opt.cc │ │ │ │ │ ├── ifrt_duplicated_callee_elimination.mlir │ │ │ │ │ ├── spmd_expansion.mlir │ │ │ │ │ ├── spmd_interface_verification.mlir │ │ │ │ │ ├── verify_array.mlir │ │ │ │ │ ├── verify_assemble.mlir │ │ │ │ │ ├── verify_attrs.mlir │ │ │ │ │ ├── verify_call.mlir │ │ │ │ │ ├── verify_call_loaded_executable.mlir │ │ │ │ │ ├── verify_disassemble.mlir │ │ │ │ │ ├── verify_loaded_executable.mlir │ │ │ │ │ └── verify_reshard.mlir │ │ │ │ └── transforms │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── built_in_spmd_expansions.cc │ │ │ │ │ ├── built_in_spmd_expansions.h │ │ │ │ │ ├── constants.h │ │ │ │ │ ├── ifrt_duplicated_callee_elimination_pass.cc │ │ │ │ │ ├── passes.h │ │ │ │ │ ├── passes.td │ │ │ │ │ ├── spmd_expandable_interface_verification_pass.cc │ │ │ │ │ ├── spmd_expanders │ │ │ │ │ ├── BUILD │ │ │ │ │ ├── noop_ifrt_spmd_expander.h │ │ │ │ │ ├── terminator_ifrt_spmd_expander.h │ │ │ │ │ └── unimplemented_ifrt_spmd_expander.h │ │ │ │ │ └── spmd_expansion_pass.cc │ │ │ ├── memory.cc │ │ │ ├── memory.h │ │ │ ├── memory_test.cc │ │ │ ├── mock.cc │ │ │ ├── mock.h │ │ │ ├── no_impl_test_main.cc │ │ │ ├── serdes.cc │ │ │ ├── serdes.h │ │ │ ├── serdes.proto │ │ │ ├── serdes_test.cc │ │ │ ├── shape.cc │ │ │ ├── shape.h │ │ │ ├── shape_test.cc │ │ │ ├── sharding.cc │ │ │ ├── sharding.h │ │ │ ├── sharding.proto │ │ │ ├── sharding_serdes.cc │ │ │ ├── sharding_serdes.h │ │ │ ├── sharding_serdes_test.cc │ │ │ ├── sharding_test.cc │ │ │ ├── sharding_test_util.cc │ │ │ ├── sharding_test_util.h │ │ │ ├── support │ │ │ │ ├── BUILD │ │ │ │ ├── sharding_param_to_op_sharding.cc │ │ │ │ ├── sharding_param_to_op_sharding.h │ │ │ │ └── sharding_param_to_op_sharding_test.cc │ │ │ ├── test_util.cc │ │ │ ├── test_util.h │ │ │ ├── tuple.cc │ │ │ ├── tuple.h │ │ │ ├── tuple_impl_test_lib.cc │ │ │ ├── types.proto │ │ │ ├── value.cc │ │ │ └── value.h │ │ ├── inspect_sharding.cc │ │ ├── inspect_sharding.h │ │ ├── jax_jit.cc │ │ ├── jax_jit.h │ │ ├── logging.cc │ │ ├── logging.h │ │ ├── mlir.cc │ │ ├── mlir.h │ │ ├── ops.cc │ │ ├── ops.h │ │ ├── outfeed_receiver.cc │ │ ├── outfeed_receiver.h │ │ ├── outfeed_receiver_py.cc │ │ ├── outfeed_receiver_py.h │ │ ├── outfeed_receiver_test.cc │ │ ├── pjit.cc │ │ ├── pjit.h │ │ ├── pjrt_ifrt │ │ │ ├── BUILD │ │ │ ├── pjrt_array.cc │ │ │ ├── pjrt_array.h │ │ │ ├── pjrt_array_impl_test_tfrt_cpu.cc │ │ │ ├── pjrt_client.cc │ │ │ ├── pjrt_client.h │ │ │ ├── pjrt_compiler.cc │ │ │ ├── pjrt_compiler.h │ │ │ ├── pjrt_executable.cc │ │ │ ├── pjrt_executable.h │ │ │ ├── pjrt_executable_impl_test_tfrt_cpu.cc │ │ │ ├── pjrt_host_callback.cc │ │ │ ├── pjrt_host_callback.h │ │ │ ├── pjrt_tuple.cc │ │ │ ├── pjrt_tuple.h │ │ │ ├── tfrt_cpu_client_test_lib.cc │ │ │ ├── xla_compiler.cc │ │ │ ├── xla_compiler.h │ │ │ ├── xla_compiler.proto │ │ │ ├── xla_executable_impl_test_lib.cc │ │ │ ├── xla_host_callback.proto │ │ │ ├── xla_program_serdes.cc │ │ │ ├── xla_program_serdes_test.cc │ │ │ ├── xla_sharding.cc │ │ │ ├── xla_sharding.h │ │ │ ├── xla_sharding.proto │ │ │ ├── xla_sharding_serdes.cc │ │ │ ├── xla_sharding_serdes_test.cc │ │ │ └── xla_sharding_test.cc │ │ ├── pmap_lib.cc │ │ ├── pmap_lib.h │ │ ├── pprof_profile_builder.cc │ │ ├── pprof_profile_builder.h │ │ ├── profiler.cc │ │ ├── profiler.h │ │ ├── profiler │ │ │ └── internal │ │ │ │ ├── BUILD │ │ │ │ ├── python_hooks.cc │ │ │ │ ├── python_hooks.h │ │ │ │ └── traceme_wrapper.h │ │ ├── py_array.cc │ │ ├── py_array.h │ │ ├── py_buffer.cc │ │ ├── py_buffer.h │ │ ├── py_client.cc │ │ ├── py_client.h │ │ ├── py_client_gpu.cc │ │ ├── py_client_gpu.h │ │ ├── py_compile_only_client.cc │ │ ├── py_compile_only_client.h │ │ ├── py_device_list.cc │ │ ├── py_device_list.h │ │ ├── py_executable.cc │ │ ├── py_executable.h │ │ ├── py_host_callback.cc │ │ ├── py_host_callback.h │ │ ├── py_host_callback.proto │ │ ├── py_values.cc │ │ ├── py_values.h │ │ ├── python_ref_manager.cc │ │ ├── python_ref_manager.h │ │ ├── python_utils.h │ │ ├── pytree.cc │ │ ├── pytree.h │ │ ├── pytree.proto │ │ ├── pytree_test.py │ │ ├── refine_polymorphic_shapes.cc │ │ ├── refine_polymorphic_shapes.h │ │ ├── sharded_device_array.h │ │ ├── sharding.cc │ │ ├── sharding.h │ │ ├── status_casters.h │ │ ├── status_casters_ext.cc │ │ ├── status_casters_test.py │ │ ├── tpu_driver │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── client │ │ │ │ ├── BUILD │ │ │ │ ├── libtpu.h │ │ │ │ └── libtpu_client.c │ │ │ ├── event_id.h │ │ │ ├── grpc_tpu_driver.cc │ │ │ ├── grpc_tpu_driver.h │ │ │ ├── platform │ │ │ │ └── external │ │ │ │ │ ├── compat.h │ │ │ │ │ └── tools.bzl │ │ │ ├── pod_tpu_driver.cc │ │ │ ├── recording_tpu_driver.cc │ │ │ ├── tpu_driver.cc │ │ │ ├── tpu_driver.h │ │ │ ├── tpu_driver.proto │ │ │ └── tpu_service.proto │ │ ├── traceback.cc │ │ ├── traceback.h │ │ ├── transfer_guard_lib.cc │ │ ├── transfer_guard_lib.h │ │ ├── types.cc │ │ ├── types.h │ │ ├── util.cc │ │ ├── util.h │ │ ├── weakref_lru_cache.cc │ │ ├── weakref_lru_cache.h │ │ ├── weakref_lru_cache_test.py │ │ ├── xla.cc │ │ ├── xla.h │ │ ├── xla_client.py │ │ ├── xla_client.pyi │ │ ├── xla_client_backend_independent_test.py │ │ ├── xla_client_test.py │ │ ├── xla_compiler.cc │ │ ├── xla_compiler.h │ │ ├── xla_extension.cc │ │ ├── xla_extension │ │ │ ├── __init__.pyi │ │ │ ├── jax_jit.pyi │ │ │ ├── mlir.pyi │ │ │ ├── ops.pyi │ │ │ ├── outfeed_receiver.pyi │ │ │ ├── pmap_lib.pyi │ │ │ ├── profiler.pyi │ │ │ ├── pytree.pyi │ │ │ └── transfer_guard_lib.pyi │ │ ├── xplane_to_profile_instructions.cc │ │ ├── xplane_to_profile_instructions.h │ │ └── xplane_to_profile_instructions_test.cc │ │ ├── python_api │ │ ├── BUILD │ │ ├── types_.py │ │ ├── xla_literal.py │ │ ├── xla_literal_test.py │ │ ├── xla_shape.py │ │ └── xla_shape_test.py │ │ ├── pytype.default.bzl │ │ ├── refcounting_hash_map.h │ │ ├── refcounting_hash_map_test.cc │ │ ├── reference_util.cc │ │ ├── reference_util.h │ │ ├── reference_util_test.cc │ │ ├── runlit.cfg.py │ │ ├── runlit.site.cfg.py │ │ ├── runtime │ │ ├── BUILD │ │ ├── README.md │ │ ├── aot_ffi.cc │ │ ├── aot_ffi.h │ │ ├── aot_ffi_c_symbols.cc │ │ ├── aot_ffi_c_symbols.h │ │ ├── aot_ffi_execution_context.h │ │ ├── arguments.cc │ │ ├── arguments.h │ │ ├── arguments_test.cc │ │ ├── async_runtime.cc │ │ ├── async_runtime.h │ │ ├── async_runtime_test.cc │ │ ├── async_values_cache.h │ │ ├── compiler.h │ │ ├── constraints.cc │ │ ├── constraints.h │ │ ├── cpu_event.h │ │ ├── custom_call.cc │ │ ├── custom_call.h │ │ ├── custom_call_registry.cc │ │ ├── custom_call_registry.h │ │ ├── custom_call_test.cc │ │ ├── default │ │ │ ├── BUILD │ │ │ ├── async_values_cache.h │ │ │ └── memory_mapper.h │ │ ├── diagnostics.cc │ │ ├── diagnostics.h │ │ ├── diagnostics_test.cc │ │ ├── errors.h │ │ ├── executable.cc │ │ ├── executable.h │ │ ├── executable_test.cc │ │ ├── execution_engine.cc │ │ ├── execution_engine.h │ │ ├── ffi │ │ │ ├── BUILD │ │ │ ├── ffi_abi.h │ │ │ ├── ffi_api.h │ │ │ └── ffi_c_api.h │ │ ├── jit_executable.cc │ │ ├── jit_executable.h │ │ ├── logical_result.h │ │ ├── map_by_type.h │ │ ├── map_by_type_test.cc │ │ ├── memory_mapper.cc │ │ ├── memory_mapper.h │ │ ├── memref_view.h │ │ ├── module.h │ │ ├── module_registry.cc │ │ ├── module_registry.h │ │ ├── module_test.cc │ │ ├── results.h │ │ ├── results_test.cc │ │ ├── runtime.h │ │ ├── state.h │ │ ├── state_test.cc │ │ ├── symbolic_shape.cc │ │ ├── symbolic_shape.h │ │ ├── symbolic_shape_test.cc │ │ ├── tracing.h │ │ ├── type_id.cc │ │ ├── type_id.h │ │ ├── type_id_test.cc │ │ ├── types.cc │ │ └── types.h │ │ ├── service │ │ ├── BUILD │ │ ├── algebraic_simplifier.cc │ │ ├── algebraic_simplifier.h │ │ ├── algebraic_simplifier_alive2_proofs.md │ │ ├── algebraic_simplifier_overflow_test.cc │ │ ├── algebraic_simplifier_proof_distributive_property.py │ │ ├── algebraic_simplifier_test.cc │ │ ├── all_gather_broadcast_reorder.cc │ │ ├── all_gather_broadcast_reorder.h │ │ ├── all_gather_broadcast_reorder_test.cc │ │ ├── all_gather_combiner.cc │ │ ├── all_gather_combiner.h │ │ ├── all_gather_combiner_test.cc │ │ ├── all_gather_decomposer.cc │ │ ├── all_gather_decomposer.h │ │ ├── all_gather_decomposer_test.cc │ │ ├── all_reduce_combiner.cc │ │ ├── all_reduce_combiner.h │ │ ├── all_reduce_combiner_test.cc │ │ ├── all_reduce_contiguous.cc │ │ ├── all_reduce_contiguous.h │ │ ├── all_reduce_contiguous_test.cc │ │ ├── all_reduce_folder.cc │ │ ├── all_reduce_folder.h │ │ ├── all_reduce_folder_test.cc │ │ ├── all_reduce_key.cc │ │ ├── all_reduce_key.h │ │ ├── all_reduce_promotion.cc │ │ ├── all_reduce_promotion.h │ │ ├── all_reduce_promotion_test.cc │ │ ├── all_reduce_reassociate.cc │ │ ├── all_reduce_reassociate.h │ │ ├── all_reduce_reassociate_test.cc │ │ ├── all_reduce_simplifier.cc │ │ ├── all_reduce_simplifier.h │ │ ├── all_reduce_simplifier_test.cc │ │ ├── all_to_all_decomposer.cc │ │ ├── all_to_all_decomposer.h │ │ ├── allocation_tracker.cc │ │ ├── allocation_tracker.h │ │ ├── ar_crs_combiner.cc │ │ ├── ar_crs_combiner.h │ │ ├── ar_crs_combiner_test.cc │ │ ├── async_collective_creator.cc │ │ ├── async_collective_creator.h │ │ ├── async_collective_creator_test.cc │ │ ├── async_op_canonicalizer.cc │ │ ├── async_op_canonicalizer.h │ │ ├── async_op_canonicalizer_test.cc │ │ ├── backend.cc │ │ ├── backend.h │ │ ├── batch_dot_simplification.cc │ │ ├── batch_dot_simplification.h │ │ ├── batch_dot_simplification_test.cc │ │ ├── batchnorm_expander.cc │ │ ├── batchnorm_expander.h │ │ ├── batchnorm_expander_test.cc │ │ ├── bfloat16_conversion_folding.cc │ │ ├── bfloat16_conversion_folding.h │ │ ├── bfloat16_conversion_folding_test.cc │ │ ├── bfloat16_propagation.cc │ │ ├── bfloat16_propagation.h │ │ ├── bfloat16_propagation_test.cc │ │ ├── bitcast_dtypes_expander.cc │ │ ├── bitcast_dtypes_expander.h │ │ ├── bitcast_dtypes_expander_test.cc │ │ ├── broadcast_canonicalizer.cc │ │ ├── broadcast_canonicalizer.h │ │ ├── broadcast_canonicalizer_test.cc │ │ ├── buffer_assignment.cc │ │ ├── buffer_assignment.h │ │ ├── buffer_assignment.proto │ │ ├── buffer_assignment_test.cc │ │ ├── buffer_value.cc │ │ ├── buffer_value.h │ │ ├── buffer_value_containers.h │ │ ├── call_graph.cc │ │ ├── call_graph.h │ │ ├── call_graph_test.cc │ │ ├── call_inliner.cc │ │ ├── call_inliner.h │ │ ├── call_inliner_test.cc │ │ ├── change_op_data_type.cc │ │ ├── change_op_data_type.h │ │ ├── change_op_data_type_test.cc │ │ ├── channel_tracker.cc │ │ ├── channel_tracker.h │ │ ├── cholesky_expander.cc │ │ ├── cholesky_expander.h │ │ ├── collective_combiner_utils.h │ │ ├── collective_decomposer_utils.cc │ │ ├── collective_decomposer_utils.h │ │ ├── collective_ops_utils.cc │ │ ├── collective_ops_utils.h │ │ ├── collective_ops_utils_test.cc │ │ ├── collective_opt_utils.cc │ │ ├── collective_opt_utils.h │ │ ├── collective_permute_decomposer.cc │ │ ├── collective_permute_decomposer.h │ │ ├── collective_permute_decomposer_test.cc │ │ ├── collective_pipeliner.cc │ │ ├── collective_pipeliner.h │ │ ├── collective_pipeliner_test.cc │ │ ├── collective_transformation_reorderer.cc │ │ ├── collective_transformation_reorderer.h │ │ ├── collective_transformation_reorderer_test.cc │ │ ├── collectives_schedule_linearizer.cc │ │ ├── collectives_schedule_linearizer.h │ │ ├── collectives_schedule_linearizer_test.cc │ │ ├── comparison_expander.cc │ │ ├── comparison_expander.h │ │ ├── compilation_cache.cc │ │ ├── compilation_cache.h │ │ ├── compilation_environments.cc │ │ ├── compilation_environments.h │ │ ├── compilation_environments_test.cc │ │ ├── compilation_stats.cc │ │ ├── compilation_stats.h │ │ ├── compile_only_service.cc │ │ ├── compile_only_service.h │ │ ├── compile_time_cap.h │ │ ├── compiler.cc │ │ ├── compiler.h │ │ ├── computation_layout.cc │ │ ├── computation_layout.h │ │ ├── computation_placer.cc │ │ ├── computation_placer.h │ │ ├── conditional_canonicalizer.cc │ │ ├── conditional_canonicalizer.h │ │ ├── conditional_canonicalizer_test.cc │ │ ├── conditional_code_motion.cc │ │ ├── conditional_code_motion.h │ │ ├── conditional_code_motion_test.cc │ │ ├── conditional_simplifier.cc │ │ ├── conditional_simplifier.h │ │ ├── conditional_simplifier_test.cc │ │ ├── conditional_to_select.cc │ │ ├── conditional_to_select.h │ │ ├── conditional_to_select_test.cc │ │ ├── constant_value.cc │ │ ├── constant_value.h │ │ ├── constant_value_test.cc │ │ ├── convert_async_collectives_to_sync.cc │ │ ├── convert_async_collectives_to_sync.h │ │ ├── convert_async_collectives_to_sync_test.cc │ │ ├── convert_mover.cc │ │ ├── convert_mover.h │ │ ├── convert_mover_test.cc │ │ ├── convert_operand_folding.cc │ │ ├── convert_operand_folding.h │ │ ├── convert_operand_folding_test.cc │ │ ├── convolution_4d_expander.cc │ │ ├── convolution_4d_expander.h │ │ ├── convolution_4d_expander_test.cc │ │ ├── convolution_group_converter.cc │ │ ├── convolution_group_converter.h │ │ ├── convolution_group_converter_test.cc │ │ ├── convolution_pred_expander.cc │ │ ├── convolution_pred_expander.h │ │ ├── convolution_pred_expander_test.cc │ │ ├── copy_insertion.cc │ │ ├── copy_insertion.h │ │ ├── copy_insertion_test.cc │ │ ├── cpu │ │ │ ├── BUILD │ │ │ ├── backend_config.proto │ │ │ ├── buffer_desc.h │ │ │ ├── buffer_info_util.cc │ │ │ ├── buffer_info_util.h │ │ │ ├── build_defs.bzl │ │ │ ├── collectives_interface.h │ │ │ ├── compiler_functor.cc │ │ │ ├── compiler_functor.h │ │ │ ├── conv_canonicalization.cc │ │ │ ├── conv_canonicalization.h │ │ │ ├── conv_canonicalization_test.cc │ │ │ ├── cpu_compiler.cc │ │ │ ├── cpu_compiler.h │ │ │ ├── cpu_compiler_registerer.cc │ │ │ ├── cpu_eigen_tensor_alignment_test.cc │ │ │ ├── cpu_executable.cc │ │ │ ├── cpu_executable.h │ │ │ ├── cpu_instruction_fusion.cc │ │ │ ├── cpu_instruction_fusion.h │ │ │ ├── cpu_instruction_fusion_test.cc │ │ │ ├── cpu_layout_assignment.cc │ │ │ ├── cpu_layout_assignment.h │ │ │ ├── cpu_layout_assignment_test.cc │ │ │ ├── cpu_options.cc │ │ │ ├── cpu_options.h │ │ │ ├── cpu_runtime.cc │ │ │ ├── cpu_runtime.h │ │ │ ├── cpu_runtime_test.cc │ │ │ ├── cpu_symbol_repository.h │ │ │ ├── cpu_transfer_manager.cc │ │ │ ├── cpu_transfer_manager.h │ │ │ ├── cpu_xfeed.cc │ │ │ ├── cpu_xfeed.h │ │ │ ├── dot_op_emitter.cc │ │ │ ├── dot_op_emitter.h │ │ │ ├── dot_op_emitter_internal.h │ │ │ ├── elemental_ir_emitter.cc │ │ │ ├── elemental_ir_emitter.h │ │ │ ├── executable.proto │ │ │ ├── hlo_xla_runtime_pipeline.cc │ │ │ ├── hlo_xla_runtime_pipeline.h │ │ │ ├── in_process_collectives.cc │ │ │ ├── in_process_collectives.h │ │ │ ├── ir_emission_utils.cc │ │ │ ├── ir_emission_utils.h │ │ │ ├── ir_emission_utils_test.cc │ │ │ ├── ir_emitter.cc │ │ │ ├── ir_emitter.h │ │ │ ├── ir_function.cc │ │ │ ├── ir_function.h │ │ │ ├── llvm_ir_runtime.cc │ │ │ ├── llvm_ir_runtime.h │ │ │ ├── mlir_emitter.cc │ │ │ ├── mlir_emitter.h │ │ │ ├── onednn_matmul.cc │ │ │ ├── onednn_matmul.h │ │ │ ├── onednn_memory_util.cc │ │ │ ├── onednn_memory_util.h │ │ │ ├── onednn_rewriter.cc │ │ │ ├── onednn_rewriter.h │ │ │ ├── orc_jit_memory_mapper.cc │ │ │ ├── orc_jit_memory_mapper.h │ │ │ ├── parallel_loop_emitter.cc │ │ │ ├── parallel_loop_emitter.h │ │ │ ├── parallel_task_assignment.cc │ │ │ ├── parallel_task_assignment.h │ │ │ ├── parallel_task_assignment_test.cc │ │ │ ├── runtime │ │ │ │ ├── BUILD │ │ │ │ ├── collectives.cc │ │ │ │ ├── collectives.h │ │ │ │ ├── convolution.cc │ │ │ │ ├── convolution.h │ │ │ │ ├── convolution_call.cc │ │ │ │ ├── convolution_call.h │ │ │ │ ├── convolution_ffi.cc │ │ │ │ ├── convolution_ffi.h │ │ │ │ ├── custom_call.cc │ │ │ │ ├── custom_call.h │ │ │ │ ├── fft_call.cc │ │ │ │ ├── fft_call.h │ │ │ │ ├── retain.cc │ │ │ │ ├── rng.cc │ │ │ │ ├── rng.h │ │ │ │ ├── rng_call.cc │ │ │ │ ├── rng_call.h │ │ │ │ ├── rng_ffi.cc │ │ │ │ ├── rng_ffi.h │ │ │ │ ├── xfeed.cc │ │ │ │ └── xfeed.h │ │ │ ├── runtime_conv2d.cc │ │ │ ├── runtime_conv2d.h │ │ │ ├── runtime_conv2d_acl.cc │ │ │ ├── runtime_conv2d_acl.h │ │ │ ├── runtime_conv2d_mkl.cc │ │ │ ├── runtime_conv2d_mkl.h │ │ │ ├── runtime_conv3d.cc │ │ │ ├── runtime_conv3d.h │ │ │ ├── runtime_conv_impl.h │ │ │ ├── runtime_custom_call_status.cc │ │ │ ├── runtime_custom_call_status.h │ │ │ ├── runtime_fft.cc │ │ │ ├── runtime_fft.h │ │ │ ├── runtime_fork_join.cc │ │ │ ├── runtime_fork_join.h │ │ │ ├── runtime_fp16.cc │ │ │ ├── runtime_fp16.h │ │ │ ├── runtime_key_value_sort.cc │ │ │ ├── runtime_key_value_sort.h │ │ │ ├── runtime_lightweight_check.h │ │ │ ├── runtime_matmul.h │ │ │ ├── runtime_matmul_acl.cc │ │ │ ├── runtime_matmul_acl.h │ │ │ ├── runtime_matmul_c128.cc │ │ │ ├── runtime_matmul_c64.cc │ │ │ ├── runtime_matmul_common.h │ │ │ ├── runtime_matmul_f16.cc │ │ │ ├── runtime_matmul_f32.cc │ │ │ ├── runtime_matmul_f64.cc │ │ │ ├── runtime_matmul_s32.cc │ │ │ ├── runtime_pow.cc │ │ │ ├── runtime_pow.h │ │ │ ├── runtime_single_threaded_conv2d.cc │ │ │ ├── runtime_single_threaded_conv2d.h │ │ │ ├── runtime_single_threaded_conv3d.cc │ │ │ ├── runtime_single_threaded_conv3d.h │ │ │ ├── runtime_single_threaded_fft.cc │ │ │ ├── runtime_single_threaded_fft.h │ │ │ ├── runtime_single_threaded_matmul.h │ │ │ ├── runtime_single_threaded_matmul_c128.cc │ │ │ ├── runtime_single_threaded_matmul_c64.cc │ │ │ ├── runtime_single_threaded_matmul_common.h │ │ │ ├── runtime_single_threaded_matmul_f16.cc │ │ │ ├── runtime_single_threaded_matmul_f32.cc │ │ │ ├── runtime_single_threaded_matmul_f64.cc │ │ │ ├── runtime_single_threaded_matmul_s32.cc │ │ │ ├── runtime_topk.cc │ │ │ ├── runtime_topk.h │ │ │ ├── sample_harness.cc │ │ │ ├── shape_partition.cc │ │ │ ├── shape_partition.h │ │ │ ├── shape_partition_test.cc │ │ │ ├── simple_orc_jit.cc │ │ │ ├── simple_orc_jit.h │ │ │ ├── target_machine_features.cc │ │ │ ├── target_machine_features.h │ │ │ ├── target_machine_features_fake.h │ │ │ ├── test_target_triple_helper.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── cpu_bytesizeof_test.cc │ │ │ │ ├── cpu_codegen_test.h │ │ │ │ ├── cpu_dyn_shape_test.cc │ │ │ │ ├── cpu_eigen_dot_operation_test.cc │ │ │ │ ├── cpu_external_constants_test.cc │ │ │ │ ├── cpu_fusion_test.cc │ │ │ │ ├── cpu_infeed_test.cc │ │ │ │ ├── cpu_intrinsic_test.cc │ │ │ │ ├── cpu_key_value_sort_test.cc │ │ │ │ ├── cpu_literal_caching_test.cc │ │ │ │ ├── cpu_noalias_test.cc │ │ │ │ ├── cpu_outfeed_test.cc │ │ │ │ ├── cpu_profiling_test.cc │ │ │ │ ├── cpu_spmd_compile_test.cc │ │ │ │ ├── cpu_topk_test.cc │ │ │ │ ├── cpu_vectorization_test.cc │ │ │ │ ├── cpu_while_test.cc │ │ │ │ └── tree_reduction_rewriter_test.cc │ │ │ ├── tiled_dot_emitter.cc │ │ │ ├── tiled_dot_emitter.h │ │ │ ├── vector_support_library.cc │ │ │ ├── vector_support_library.h │ │ │ ├── vectorized_reduce_with_no_vector_registers_test.cc │ │ │ ├── windows_compatibility.cc │ │ │ ├── windows_compatibility.h │ │ │ ├── xfeed_manager.cc │ │ │ ├── xfeed_manager.h │ │ │ ├── xfeed_manager_test.cc │ │ │ ├── xla_framework.h │ │ │ └── xla_framework.proto │ │ ├── cpu_gpu_shape_verifier.cc │ │ ├── cpu_gpu_shape_verifier.h │ │ ├── cpu_gpu_shape_verifier_test.cc │ │ ├── custom_call_sharding_helper.cc │ │ ├── custom_call_sharding_helper.h │ │ ├── custom_call_status.cc │ │ ├── custom_call_status.h │ │ ├── custom_call_status_internal.h │ │ ├── custom_call_status_test.cc │ │ ├── custom_call_status_test_c_caller.c │ │ ├── custom_call_status_test_c_caller.h │ │ ├── custom_call_target_registry.cc │ │ ├── custom_call_target_registry.h │ │ ├── defuser.cc │ │ ├── defuser.h │ │ ├── defuser_test.cc │ │ ├── despecializer.cc │ │ ├── despecializer.h │ │ ├── despecializer_test.cc │ │ ├── dfs_hlo_visitor_with_default_test.cc │ │ ├── dot_as_convolution_util.cc │ │ ├── dot_as_convolution_util.h │ │ ├── dot_decomposer.cc │ │ ├── dot_decomposer.h │ │ ├── dot_decomposer_test.cc │ │ ├── dot_dimension_merger.cc │ │ ├── dot_dimension_merger.h │ │ ├── dot_dimension_merger_test.cc │ │ ├── dot_merger.cc │ │ ├── dot_merger.h │ │ ├── dot_merger_test.cc │ │ ├── dump.cc │ │ ├── dump.h │ │ ├── dynamic_dimension_inference.cc │ │ ├── dynamic_dimension_inference.h │ │ ├── dynamic_dimension_inference_test.cc │ │ ├── dynamic_dimension_simplifier.cc │ │ ├── dynamic_dimension_simplifier.h │ │ ├── dynamic_dimension_simplifier_test.cc │ │ ├── dynamic_index_splitter.cc │ │ ├── dynamic_index_splitter.h │ │ ├── dynamic_index_splitter_test.cc │ │ ├── dynamic_padder.cc │ │ ├── dynamic_padder.h │ │ ├── dynamic_padder_test.cc │ │ ├── dynamic_parameter_binding_test.cc │ │ ├── dynamic_update_slice_test.cc │ │ ├── dynamic_window_utils.cc │ │ ├── dynamic_window_utils.h │ │ ├── eigh_expander.cc │ │ ├── eigh_expander.h │ │ ├── elemental_ir_emitter.cc │ │ ├── elemental_ir_emitter.h │ │ ├── elemental_ir_emitter_test.cc │ │ ├── executable.cc │ │ ├── executable.h │ │ ├── execution_tracker.cc │ │ ├── execution_tracker.h │ │ ├── export_hlo.h │ │ ├── flatten_call_graph.cc │ │ ├── flatten_call_graph.h │ │ ├── flatten_call_graph_test.cc │ │ ├── float8_fnuz_ir_emitter.cc │ │ ├── float8_fnuz_ir_emitter.h │ │ ├── float_normalization.cc │ │ ├── float_normalization.h │ │ ├── float_normalization_test.cc │ │ ├── float_support.cc │ │ ├── float_support.h │ │ ├── fusion_node_indexing_evaluation.cc │ │ ├── fusion_node_indexing_evaluation.h │ │ ├── fusion_node_indexing_evaluation_test.cc │ │ ├── fusion_queue.h │ │ ├── gather_expander.cc │ │ ├── gather_expander.h │ │ ├── gather_expander_test.cc │ │ ├── gather_scatter_utils.cc │ │ ├── gather_scatter_utils.h │ │ ├── gather_simplifier.cc │ │ ├── gather_simplifier.h │ │ ├── gather_simplifier_test.cc │ │ ├── generate_test_hlo_checks.py │ │ ├── generate_test_hlo_checks_test.py │ │ ├── generic_transfer_manager.cc │ │ ├── generic_transfer_manager.h │ │ ├── generic_transfer_manager_test.cc │ │ ├── global_device_id.cc │ │ ├── global_device_id.h │ │ ├── gpu │ │ │ ├── BUILD │ │ │ ├── alias_passthrough_params.cc │ │ │ ├── alias_passthrough_params.h │ │ │ ├── alias_passthrough_params_test.cc │ │ │ ├── all_reduce_blueconnect.cc │ │ │ ├── all_reduce_blueconnect.h │ │ │ ├── all_reduce_blueconnect_test.cc │ │ │ ├── amdgpu_compiler.cc │ │ │ ├── amdgpu_compiler.h │ │ │ ├── amdgpu_compiler_registration.cc │ │ │ ├── auto_sharding_gpu_compiler_test.cc │ │ │ ├── autotuner_compile_util.cc │ │ │ ├── autotuner_compile_util.h │ │ │ ├── autotuner_util.cc │ │ │ ├── autotuner_util.h │ │ │ ├── autotuner_util_test.cc │ │ │ ├── backend_configs.proto │ │ │ ├── backend_configs_test.cc │ │ │ ├── buffer_allocations.cc │ │ │ ├── buffer_allocations.h │ │ │ ├── buffer_comparator.cc │ │ │ ├── buffer_comparator.cu.cc │ │ │ ├── buffer_comparator.h │ │ │ ├── buffer_comparator_test.cc │ │ │ ├── buffer_sharing.cc │ │ │ ├── buffer_sharing.h │ │ │ ├── build_defs.bzl │ │ │ ├── command_buffer_scheduling.cc │ │ │ ├── command_buffer_scheduling.h │ │ │ ├── command_buffer_scheduling_test.cc │ │ │ ├── compile_module_to_llvm_ir.cc │ │ │ ├── compile_module_to_llvm_ir.h │ │ │ ├── conditional_thunk.cc │ │ │ ├── conditional_thunk.h │ │ │ ├── conv_algorithm_picker.cc │ │ │ ├── conv_algorithm_picker.h │ │ │ ├── conv_algorithm_picker_test.cc │ │ │ ├── conv_layout_normalization.cc │ │ │ ├── conv_layout_normalization.h │ │ │ ├── conv_layout_normalization_test.cc │ │ │ ├── convolution_thunk.cc │ │ │ ├── convolution_thunk.h │ │ │ ├── copy_fusion.cc │ │ │ ├── copy_fusion.h │ │ │ ├── copy_fusion_test.cc │ │ │ ├── copy_thunk.cc │ │ │ ├── copy_thunk.h │ │ │ ├── cub_sort_kernel.cu.cc │ │ │ ├── cub_sort_kernel.h │ │ │ ├── cub_sort_thunk.cc │ │ │ ├── cub_sort_thunk.h │ │ │ ├── cublas_cudnn.cc │ │ │ ├── cublas_cudnn.h │ │ │ ├── cublas_pad_for_gemms.cc │ │ │ ├── cublas_pad_for_gemms.h │ │ │ ├── cublas_pad_for_gemms_test.cc │ │ │ ├── cublas_padding_requirements.cc │ │ │ ├── cublas_padding_requirements.h │ │ │ ├── cudnn_fused_conv_rewriter.cc │ │ │ ├── cudnn_fused_conv_rewriter.h │ │ │ ├── cudnn_fused_conv_rewriter_test.cc │ │ │ ├── cudnn_fused_mha_rewriter.cc │ │ │ ├── cudnn_fused_mha_rewriter.h │ │ │ ├── cudnn_fused_mha_rewriter_test.cc │ │ │ ├── cudnn_fused_mha_transpose_fusion.cc │ │ │ ├── cudnn_fused_mha_transpose_fusion.h │ │ │ ├── cudnn_norm_rewriter.cc │ │ │ ├── cudnn_norm_rewriter.h │ │ │ ├── cudnn_norm_rewriter_test.cc │ │ │ ├── cudnn_pad_for_convolutions.cc │ │ │ ├── cudnn_pad_for_convolutions.h │ │ │ ├── cudnn_pad_for_convolutions_test.cc │ │ │ ├── cudnn_simplify_padding.cc │ │ │ ├── cudnn_simplify_padding.h │ │ │ ├── cudnn_simplify_padding_test.cc │ │ │ ├── cudnn_support_utils.cc │ │ │ ├── cudnn_support_utils.h │ │ │ ├── cudnn_support_utils_test.cc │ │ │ ├── cudnn_vectorize_convolutions.cc │ │ │ ├── cudnn_vectorize_convolutions.h │ │ │ ├── cudnn_vectorize_convolutions_test.cc │ │ │ ├── cusolver_context.cc │ │ │ ├── cusolver_context.h │ │ │ ├── cusolver_rewriter.cc │ │ │ ├── cusolver_rewriter.h │ │ │ ├── custom_call_test.cc │ │ │ ├── custom_fusion_rewriter.cc │ │ │ ├── custom_fusion_rewriter.h │ │ │ ├── custom_fusion_rewriter_test.cc │ │ │ ├── data │ │ │ │ └── hlo_algorithm_denylist.pbtxt │ │ │ ├── determinism_test.cc │ │ │ ├── dot_dimension_sorter.cc │ │ │ ├── dot_dimension_sorter.h │ │ │ ├── dot_dimension_sorter_test.cc │ │ │ ├── elemental_ir_emitter.cc │ │ │ ├── elemental_ir_emitter.h │ │ │ ├── executable.proto │ │ │ ├── float_support_test.cc │ │ │ ├── for_thunk.cc │ │ │ ├── for_thunk.h │ │ │ ├── fused_mha_thunk.cc │ │ │ ├── fused_mha_thunk.h │ │ │ ├── fusion_merger.cc │ │ │ ├── fusion_merger.h │ │ │ ├── fusion_merger_test.cc │ │ │ ├── fusion_merger_triton.cc │ │ │ ├── fusion_merger_triton.h │ │ │ ├── fusion_merger_triton_test.cc │ │ │ ├── fusion_pipeline.cc │ │ │ ├── fusion_pipeline.h │ │ │ ├── fusion_process_dump.proto │ │ │ ├── fusion_wrapper.cc │ │ │ ├── fusion_wrapper.h │ │ │ ├── fusion_wrapper_test.cc │ │ │ ├── fusions │ │ │ │ ├── BUILD │ │ │ │ ├── copy.cc │ │ │ │ ├── copy.h │ │ │ │ ├── fusion_emitter.cc │ │ │ │ ├── fusion_emitter.h │ │ │ │ ├── fusions.cc │ │ │ │ ├── fusions.h │ │ │ │ ├── in_place_dynamic_update_slice.cc │ │ │ │ ├── in_place_dynamic_update_slice.h │ │ │ │ ├── input_slices.cc │ │ │ │ ├── input_slices.h │ │ │ │ ├── loop.cc │ │ │ │ ├── loop.h │ │ │ │ ├── reduction.cc │ │ │ │ ├── reduction.h │ │ │ │ ├── thunk_util.cc │ │ │ │ ├── thunk_util.h │ │ │ │ ├── tiling_util.cc │ │ │ │ ├── tiling_util.h │ │ │ │ ├── transpose.cc │ │ │ │ └── transpose.h │ │ │ ├── gemm_algorithm_picker.cc │ │ │ ├── gemm_algorithm_picker.h │ │ │ ├── gemm_algorithm_picker_test.cc │ │ │ ├── gemm_broadcast_folding_rewriter.cc │ │ │ ├── gemm_broadcast_folding_rewriter.h │ │ │ ├── gemm_rewriter.cc │ │ │ ├── gemm_rewriter.h │ │ │ ├── gemm_rewriter_triton.cc │ │ │ ├── gemm_rewriter_triton.h │ │ │ ├── gemm_rewriter_triton_test.cc │ │ │ ├── gemm_thunk.cc │ │ │ ├── gemm_thunk.h │ │ │ ├── gpu_all_gather_optimizer.cc │ │ │ ├── gpu_all_gather_optimizer.h │ │ │ ├── gpu_aot_compilation_test.cc │ │ │ ├── gpu_asm_opts_util.cc │ │ │ ├── gpu_asm_opts_util.h │ │ │ ├── gpu_async_collective_annotator.cc │ │ │ ├── gpu_async_collective_annotator.h │ │ │ ├── gpu_async_collective_annotator_test.cc │ │ │ ├── gpu_autotuning.proto │ │ │ ├── gpu_compiler.cc │ │ │ ├── gpu_compiler.h │ │ │ ├── gpu_compiler_test.cc │ │ │ ├── gpu_constants.h │ │ │ ├── gpu_conv_padding_legalization.cc │ │ │ ├── gpu_conv_padding_legalization.h │ │ │ ├── gpu_conv_padding_legalization_test.cc │ │ │ ├── gpu_conv_rewriter.cc │ │ │ ├── gpu_conv_rewriter.h │ │ │ ├── gpu_conv_rewriter_test.cc │ │ │ ├── gpu_conv_runner.cc │ │ │ ├── gpu_conv_runner.h │ │ │ ├── gpu_convert_async_collectives_to_sync.cc │ │ │ ├── gpu_convert_async_collectives_to_sync.h │ │ │ ├── gpu_convert_async_collectives_to_sync_test.cc │ │ │ ├── gpu_copy_insertion_test.cc │ │ │ ├── gpu_device_info_for_tests.cc │ │ │ ├── gpu_device_info_for_tests.h │ │ │ ├── gpu_device_info_test.cc │ │ │ ├── gpu_executable.cc │ │ │ ├── gpu_executable.h │ │ │ ├── gpu_executable_run_options.cc │ │ │ ├── gpu_executable_run_options.h │ │ │ ├── gpu_float_support.cc │ │ │ ├── gpu_float_support.h │ │ │ ├── gpu_fused_mha_runner.cc │ │ │ ├── gpu_fused_mha_runner.h │ │ │ ├── gpu_fusible.cc │ │ │ ├── gpu_fusible.h │ │ │ ├── gpu_fusible_test.cc │ │ │ ├── gpu_hlo_schedule.cc │ │ │ ├── gpu_hlo_schedule.h │ │ │ ├── gpu_hlo_schedule_test.cc │ │ │ ├── gpu_layout_assignment.cc │ │ │ ├── gpu_layout_assignment.h │ │ │ ├── gpu_layout_assignment_test.cc │ │ │ ├── gpu_norm_runner.cc │ │ │ ├── gpu_norm_runner.h │ │ │ ├── gpu_prim_cuda.h │ │ │ ├── gpu_prim_rocm.h │ │ │ ├── gpu_reduce_scatter_creator.cc │ │ │ ├── gpu_reduce_scatter_creator.h │ │ │ ├── gpu_sanitize_constant_names.cc │ │ │ ├── gpu_sanitize_constant_names.h │ │ │ ├── gpu_sanitize_constant_names_test.cc │ │ │ ├── gpu_scatter_expander.cc │ │ │ ├── gpu_scatter_expander.h │ │ │ ├── gpu_schedule_postprocessing.cc │ │ │ ├── gpu_schedule_postprocessing.h │ │ │ ├── gpu_schedule_postprocessing_test.cc │ │ │ ├── gpu_sort_rewriter.cc │ │ │ ├── gpu_sort_rewriter.h │ │ │ ├── gpu_sort_rewriter_test.cc │ │ │ ├── gpu_symbol_repository.h │ │ │ ├── gpu_transfer_manager.cc │ │ │ ├── gpu_transfer_manager.h │ │ │ ├── gpublas_lt_matmul_thunk.cc │ │ │ ├── gpublas_lt_matmul_thunk.h │ │ │ ├── hlo_algorithm_denylist.cc │ │ │ ├── hlo_algorithm_denylist.h │ │ │ ├── hlo_algorithm_denylist_test.cc │ │ │ ├── hlo_fusion_analysis.cc │ │ │ ├── hlo_fusion_analysis.h │ │ │ ├── hlo_fusion_analysis_test.cc │ │ │ ├── hlo_fusion_stats.cc │ │ │ ├── hlo_fusion_stats.h │ │ │ ├── hlo_fusion_stats_test.cc │ │ │ ├── hlo_to_ir_bindings.cc │ │ │ ├── hlo_to_ir_bindings.h │ │ │ ├── hlo_traversal.cc │ │ │ ├── hlo_traversal.h │ │ │ ├── hlo_traversal_test.cc │ │ │ ├── horizontal_input_fusion.cc │ │ │ ├── horizontal_input_fusion.h │ │ │ ├── horizontal_input_fusion_test.cc │ │ │ ├── horizontal_loop_fusion.cc │ │ │ ├── horizontal_loop_fusion.h │ │ │ ├── horizontal_loop_fusion_test.cc │ │ │ ├── infeed_manager.cc │ │ │ ├── infeed_manager.h │ │ │ ├── infeed_thunk.cc │ │ │ ├── infeed_thunk.h │ │ │ ├── instruction_fusion.cc │ │ │ ├── instruction_fusion.h │ │ │ ├── instruction_fusion_test.cc │ │ │ ├── ir_emission_utils.cc │ │ │ ├── ir_emission_utils.h │ │ │ ├── ir_emission_utils_test.cc │ │ │ ├── ir_emitter.cc │ │ │ ├── ir_emitter.h │ │ │ ├── ir_emitter_context.cc │ │ │ ├── ir_emitter_context.h │ │ │ ├── ir_emitter_nested.cc │ │ │ ├── ir_emitter_nested.h │ │ │ ├── ir_emitter_triton.cc │ │ │ ├── ir_emitter_triton.h │ │ │ ├── ir_emitter_triton_large_test.cc │ │ │ ├── ir_emitter_triton_parametrized_test.cc │ │ │ ├── ir_emitter_triton_test.cc │ │ │ ├── ir_emitter_unnested.cc │ │ │ ├── ir_emitter_unnested.h │ │ │ ├── kernel_arguments.cc │ │ │ ├── kernel_arguments.h │ │ │ ├── kernel_mapping_scheme.h │ │ │ ├── kernel_reuse_cache.cc │ │ │ ├── kernel_reuse_cache.h │ │ │ ├── kernel_thunk.cc │ │ │ ├── kernel_thunk.h │ │ │ ├── kernels │ │ │ │ ├── BUILD │ │ │ │ ├── custom_fusion.cc │ │ │ │ ├── custom_fusion.h │ │ │ │ ├── custom_fusion_pattern.cc │ │ │ │ ├── custom_fusion_pattern.h │ │ │ │ ├── custom_kernel.cc │ │ │ │ ├── custom_kernel.h │ │ │ │ ├── cutlass_gemm.h │ │ │ │ ├── cutlass_gemm_custom_kernel.cu.cc │ │ │ │ ├── cutlass_gemm_custom_kernel.h │ │ │ │ ├── cutlass_gemm_custom_kernel_test.cc │ │ │ │ ├── cutlass_gemm_fusion.cc │ │ │ │ ├── cutlass_gemm_fusion.h │ │ │ │ ├── cutlass_gemm_fusion_test.cc │ │ │ │ ├── cutlass_gemm_kernel.cu.h │ │ │ │ ├── cutlass_gemm_kernels.cu.h │ │ │ │ ├── cutlass_gemm_kernels_bf16xbf16_to_bf16.cu.cc │ │ │ │ └── cutlass_gemm_kernels_f32xf32_to_f32.cu.cc │ │ │ ├── launch_dimensions.cc │ │ │ ├── launch_dimensions.h │ │ │ ├── llvm_gpu_backend │ │ │ │ ├── BUILD │ │ │ │ ├── gpu_backend_lib.cc │ │ │ │ ├── gpu_backend_lib.h │ │ │ │ ├── tests_data │ │ │ │ │ └── saxpy.ll │ │ │ │ ├── utils.cc │ │ │ │ ├── utils.h │ │ │ │ └── utils_test.cc │ │ │ ├── loop_double_buffer_transformer.cc │ │ │ ├── loop_double_buffer_transformer.h │ │ │ ├── loop_double_buffer_transformer_test.cc │ │ │ ├── make_batch_pointers.cc │ │ │ ├── make_batch_pointers.cu.cc │ │ │ ├── make_batch_pointers.h │ │ │ ├── matmul_utils.cc │ │ │ ├── matmul_utils.h │ │ │ ├── matmul_utils_test.cc │ │ │ ├── memset_thunk.cc │ │ │ ├── memset_thunk.h │ │ │ ├── metrics.cc │ │ │ ├── metrics.h │ │ │ ├── model │ │ │ │ ├── BUILD │ │ │ │ ├── analytical_latency_estimator.cc │ │ │ │ ├── analytical_latency_estimator.h │ │ │ │ ├── analytical_latency_estimator_test.cc │ │ │ │ ├── fusion_analysis_cache.cc │ │ │ │ ├── fusion_analysis_cache.h │ │ │ │ ├── fusion_analysis_cache_test.cc │ │ │ │ ├── gpu_cost_model_stats_collection.cc │ │ │ │ ├── gpu_cost_model_stats_collection.h │ │ │ │ ├── gpu_cost_model_stats_collection_test.cc │ │ │ │ ├── gpu_hlo_cost_analysis.cc │ │ │ │ ├── gpu_hlo_cost_analysis.h │ │ │ │ ├── gpu_hlo_cost_analysis_test.cc │ │ │ │ ├── gpu_performance_model.cc │ │ │ │ ├── gpu_performance_model.h │ │ │ │ ├── gpu_performance_model_test.cc │ │ │ │ ├── hlo_op_profile.proto │ │ │ │ ├── hlo_op_profiler.cc │ │ │ │ ├── hlo_op_profiler.h │ │ │ │ ├── hlo_op_profiler_run.cc │ │ │ │ ├── hlo_op_profiler_test.cc │ │ │ │ ├── hlo_op_profiles.h │ │ │ │ ├── tile_analysis.cc │ │ │ │ ├── tile_analysis.h │ │ │ │ └── tile_analysis_test.cc │ │ │ ├── move_copy_to_users.cc │ │ │ ├── move_copy_to_users.h │ │ │ ├── move_copy_to_users_test.cc │ │ │ ├── multi_output_fusion.cc │ │ │ ├── multi_output_fusion.h │ │ │ ├── multi_output_fusion_test.cc │ │ │ ├── nccl_all_gather_thunk.cc │ │ │ ├── nccl_all_gather_thunk.h │ │ │ ├── nccl_all_reduce_thunk.cc │ │ │ ├── nccl_all_reduce_thunk.h │ │ │ ├── nccl_all_to_all_thunk.cc │ │ │ ├── nccl_all_to_all_thunk.h │ │ │ ├── nccl_collective_permute_thunk.cc │ │ │ ├── nccl_collective_permute_thunk.h │ │ │ ├── nccl_collective_thunk.cc │ │ │ ├── nccl_collective_thunk.h │ │ │ ├── nccl_p2p_thunk_common.cc │ │ │ ├── nccl_p2p_thunk_common.h │ │ │ ├── nccl_recv_thunk.cc │ │ │ ├── nccl_recv_thunk.h │ │ │ ├── nccl_send_thunk.cc │ │ │ ├── nccl_send_thunk.h │ │ │ ├── nccl_utils.cc │ │ │ ├── nccl_utils.h │ │ │ ├── non_atomically_upgradeable_rw_lock.h │ │ │ ├── non_atomically_upgradeable_rw_lock_test.cc │ │ │ ├── norm_thunk.cc │ │ │ ├── norm_thunk.h │ │ │ ├── nvptx_compiler.cc │ │ │ ├── nvptx_compiler.h │ │ │ ├── nvptx_compiler_registration.cc │ │ │ ├── nvptx_compiler_test.cc │ │ │ ├── outfeed_manager.cc │ │ │ ├── outfeed_manager.h │ │ │ ├── outfeed_thunk.cc │ │ │ ├── outfeed_thunk.h │ │ │ ├── parallel_loop_emitter.cc │ │ │ ├── parallel_loop_emitter.h │ │ │ ├── prepare_hlo_for_ir_emitting_pipeline.cc │ │ │ ├── prepare_hlo_for_ir_emitting_pipeline.h │ │ │ ├── priority_fusion.cc │ │ │ ├── priority_fusion.h │ │ │ ├── priority_fusion_test.cc │ │ │ ├── reduction_degenerate_dim_remover.cc │ │ │ ├── reduction_degenerate_dim_remover.h │ │ │ ├── reduction_dimension_grouper.cc │ │ │ ├── reduction_dimension_grouper.h │ │ │ ├── reduction_layout_normalizer.cc │ │ │ ├── reduction_layout_normalizer.h │ │ │ ├── reduction_splitter.cc │ │ │ ├── reduction_splitter.h │ │ │ ├── reduction_splitter_test.cc │ │ │ ├── reduction_utils.cc │ │ │ ├── reduction_utils.h │ │ │ ├── replica_id_thunk.cc │ │ │ ├── replica_id_thunk.h │ │ │ ├── runtime │ │ │ │ ├── BUILD │ │ │ │ ├── annotation.cc │ │ │ │ ├── annotation.h │ │ │ │ ├── cholesky.cc │ │ │ │ ├── cholesky.h │ │ │ │ ├── collectives.cc │ │ │ │ ├── collectives.h │ │ │ │ ├── concurrent_region.cc │ │ │ │ ├── concurrent_region.h │ │ │ │ ├── conv.cc │ │ │ │ ├── conv.h │ │ │ │ ├── conv_reorder.cc │ │ │ │ ├── conv_reorder.h │ │ │ │ ├── cub_sort.cc │ │ │ │ ├── cub_sort.h │ │ │ │ ├── custom_call.cc │ │ │ │ ├── custom_call.h │ │ │ │ ├── custom_call_registry.cc │ │ │ │ ├── custom_call_registry.h │ │ │ │ ├── executable.cc │ │ │ │ ├── executable.h │ │ │ │ ├── fft.cc │ │ │ │ ├── fft.h │ │ │ │ ├── fused_attention.cc │ │ │ │ ├── fused_attention.h │ │ │ │ ├── gemm.cc │ │ │ │ ├── gemm.h │ │ │ │ ├── gpu_kernel_helper.h │ │ │ │ ├── gpublas_lt_matmul.cc │ │ │ │ ├── gpublas_lt_matmul.h │ │ │ │ ├── graph_launch.cc │ │ │ │ ├── graph_launch.h │ │ │ │ ├── io_feed.cc │ │ │ │ ├── io_feed.h │ │ │ │ ├── kernel_launch.cc │ │ │ │ ├── kernel_launch.h │ │ │ │ ├── memcpy.cc │ │ │ │ ├── memcpy.h │ │ │ │ ├── memset.cc │ │ │ │ ├── memset.h │ │ │ │ ├── norm.cc │ │ │ │ ├── norm.h │ │ │ │ ├── send_recv.cc │ │ │ │ ├── send_recv.h │ │ │ │ ├── sleep_kernel.cu.cc │ │ │ │ ├── sleep_kernel.h │ │ │ │ ├── stream_synchronization.cc │ │ │ │ ├── stream_synchronization.h │ │ │ │ ├── support.cc │ │ │ │ ├── support.h │ │ │ │ ├── topk.cc │ │ │ │ ├── topk.h │ │ │ │ ├── topk_kernel.cc │ │ │ │ ├── topk_kernel.cu.h │ │ │ │ ├── topk_kernel.h │ │ │ │ ├── topk_kernel_bfloat16.cu.cc │ │ │ │ ├── topk_kernel_common.h │ │ │ │ ├── topk_kernel_float.cu.cc │ │ │ │ ├── topk_kernel_test.cc │ │ │ │ ├── topk_test.cc │ │ │ │ ├── tracing.cc │ │ │ │ ├── tracing.h │ │ │ │ ├── triangular_solve.cc │ │ │ │ └── triangular_solve.h │ │ │ ├── runtime3 │ │ │ │ ├── BUILD │ │ │ │ ├── README.md │ │ │ │ ├── cholesky_thunk.cc │ │ │ │ ├── cholesky_thunk.h │ │ │ │ ├── command_buffer_allocations.cc │ │ │ │ ├── command_buffer_allocations.h │ │ │ │ ├── command_buffer_cmd.cc │ │ │ │ ├── command_buffer_cmd.h │ │ │ │ ├── command_buffer_cmd_emitter.cc │ │ │ │ ├── command_buffer_cmd_emitter.h │ │ │ │ ├── command_buffer_cmd_test.cc │ │ │ │ ├── command_buffer_thunk.cc │ │ │ │ ├── command_buffer_thunk.h │ │ │ │ ├── command_buffer_thunk_test.cc │ │ │ │ ├── custom_call_thunk.cc │ │ │ │ ├── custom_call_thunk.h │ │ │ │ ├── fft_thunk.cc │ │ │ │ ├── fft_thunk.h │ │ │ │ ├── triangular_solve_thunk.cc │ │ │ │ └── triangular_solve_thunk.h │ │ │ ├── runtime_intrinsics.cc │ │ │ ├── runtime_intrinsics.h │ │ │ ├── runtime_intrinsics_test.cc │ │ │ ├── scatter_slice_simplifier.cc │ │ │ ├── scatter_slice_simplifier.h │ │ │ ├── scatter_slice_simplifier_test.cc │ │ │ ├── sequential_thunk.cc │ │ │ ├── sequential_thunk.h │ │ │ ├── softmax_fusion.h │ │ │ ├── softmax_rewriter_triton.cc │ │ │ ├── softmax_rewriter_triton.h │ │ │ ├── softmax_rewriter_triton_test.cc │ │ │ ├── split_k_gemm_rewriter.cc │ │ │ ├── split_k_gemm_rewriter.h │ │ │ ├── split_k_gemm_rewriter_test.cc │ │ │ ├── stream_executor_util.cc │ │ │ ├── stream_executor_util.h │ │ │ ├── target_constants.h │ │ │ ├── target_util.cc │ │ │ ├── target_util.h │ │ │ ├── target_util_test.cc │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── add_preds.hlo │ │ │ │ ├── calling_convention.hlo │ │ │ │ ├── concat.hlo │ │ │ │ ├── constant.hlo │ │ │ │ ├── copy.hlo │ │ │ │ ├── copy_nested.hlo │ │ │ │ ├── dynamic_shared_memory_test.cc │ │ │ │ ├── dynamic_update_slice_inplace.hlo │ │ │ │ ├── element_wise_row_vectorization.hlo │ │ │ │ ├── element_wise_row_vectorization_test.cc │ │ │ │ ├── fused_scatter.hlo │ │ │ │ ├── fused_slice.hlo │ │ │ │ ├── fused_slice_different_operands.hlo │ │ │ │ ├── fusion.hlo │ │ │ │ ├── gemm_broadcast_folding_rewrite_test.cc │ │ │ │ ├── gemm_rewrite_test.cc │ │ │ │ ├── gpu_alignment_test.cc │ │ │ │ ├── gpu_all_gather_optimizer_test.cc │ │ │ │ ├── gpu_atomic_test.cc │ │ │ │ ├── gpu_codegen_test.cc │ │ │ │ ├── gpu_codegen_test.h │ │ │ │ ├── gpu_compilation_parallelism_test.cc │ │ │ │ ├── gpu_convolution_regression_test.cc │ │ │ │ ├── gpu_copy_alone_test.cc │ │ │ │ ├── gpu_copy_test.cc │ │ │ │ ├── gpu_dyn_shape_test.cc │ │ │ │ ├── gpu_ftz_test.cc │ │ │ │ ├── gpu_fused_mha_test.cc │ │ │ │ ├── gpu_fusion_pipeline_test.cc │ │ │ │ ├── gpu_fusion_test.cc │ │ │ │ ├── gpu_index_test.cc │ │ │ │ ├── gpu_input_fusible_slice_test.cc │ │ │ │ ├── gpu_int4_test.cc │ │ │ │ ├── gpu_kernel_tiling_test.cc │ │ │ │ ├── gpu_ldg_test.cc │ │ │ │ ├── gpu_noalias_test.cc │ │ │ │ ├── gpu_reduce_scatter_creator_test.cc │ │ │ │ ├── gpu_spmd_e2e_compile_test.cc │ │ │ │ ├── gpu_too_many_blocks_test.cc │ │ │ │ ├── gpu_unrolling_test.cc │ │ │ │ ├── hlo_to_llvm_ir.cc │ │ │ │ ├── in_place_op_test.cc │ │ │ │ ├── infeed_test.cc │ │ │ │ ├── kernel_launch_test.cc │ │ │ │ ├── kernel_reuse.hlo │ │ │ │ ├── launch_dimensions.hlo │ │ │ │ ├── mock_custom_call_test.cc │ │ │ │ ├── pad_to_static.hlo │ │ │ │ ├── parallel_reduction_test.cc │ │ │ │ ├── pred_arithmetic_test.cc │ │ │ │ ├── reduce_unnested.hlo │ │ │ │ ├── reduction_degenerate_dim_remover_test.cc │ │ │ │ ├── reduction_dimension_grouper_test.cc │ │ │ │ ├── reduction_emitter_test.cc │ │ │ │ ├── reduction_layout_normalizer_test.cc │ │ │ │ ├── reduction_vectorization_sm_all.hlo │ │ │ │ ├── reduction_vectorization_test.cc │ │ │ │ ├── rng_get_and_update_state.hlo │ │ │ │ ├── scatter.hlo │ │ │ │ ├── select_and_scatter.hlo │ │ │ │ ├── select_and_scatter_test.cc │ │ │ │ ├── simple_optimization_test.cc │ │ │ │ ├── single_instruction.hlo │ │ │ │ ├── slice_to_dynamic.hlo │ │ │ │ ├── sorting.hlo │ │ │ │ ├── sorting_test.cc │ │ │ │ ├── swap_conv_operands_test.cc │ │ │ │ ├── tensor_float_32_global_var_test.cc │ │ │ │ ├── test_autotune_cache.textproto │ │ │ │ ├── transpose_emitter_test.cc │ │ │ │ ├── tree_reduction_rewriter_test.cc │ │ │ │ └── triton_naming.hlo │ │ │ ├── thunk.cc │ │ │ ├── thunk.h │ │ │ ├── topk_specializer.cc │ │ │ ├── topk_specializer.h │ │ │ ├── topk_splitter.cc │ │ │ ├── topk_splitter.h │ │ │ ├── topk_splitter_test.cc │ │ │ ├── tree_reduction_rewriter.cc │ │ │ ├── tree_reduction_rewriter.h │ │ │ ├── triangular_solve_rewriter.cc │ │ │ ├── triangular_solve_rewriter.h │ │ │ ├── triton_autotuner.cc │ │ │ ├── triton_autotuner.h │ │ │ ├── triton_autotuner_test.cc │ │ │ ├── triton_fusion_analysis.cc │ │ │ ├── triton_fusion_analysis.h │ │ │ ├── triton_fusion_analysis_test.cc │ │ │ ├── triton_support.cc │ │ │ ├── triton_support.h │ │ │ ├── triton_tiling_propagation.cc │ │ │ ├── triton_tiling_propagation.h │ │ │ ├── variadic_op_splitter.cc │ │ │ ├── variadic_op_splitter.h │ │ │ ├── variadic_op_splitter_test.cc │ │ │ ├── while_thunk.cc │ │ │ ├── while_thunk.h │ │ │ ├── while_transformer_test.cc │ │ │ ├── xfeed_queue.h │ │ │ └── xla_executor_state.h │ │ ├── gpu_compilation_environment.cc │ │ ├── gpu_compilation_environment.h │ │ ├── gpu_compilation_environment_test.cc │ │ ├── graphcycles │ │ │ ├── BUILD │ │ │ ├── graphcycles.cc │ │ │ ├── graphcycles.h │ │ │ ├── graphcycles_test.cc │ │ │ ├── ordered_set.h │ │ │ └── ordered_set_test.cc │ │ ├── heap_simulator.cc │ │ ├── heap_simulator.h │ │ ├── heap_simulator_test.cc │ │ ├── hlo.proto │ │ ├── hlo_alias_analysis.cc │ │ ├── hlo_alias_analysis.h │ │ ├── hlo_alias_analysis_test.cc │ │ ├── hlo_buffer.cc │ │ ├── hlo_buffer.h │ │ ├── hlo_casting_utils_test.cc │ │ ├── hlo_computation_deduplicator.cc │ │ ├── hlo_computation_deduplicator.h │ │ ├── hlo_computation_deduplicator_test.cc │ │ ├── hlo_computation_test.cc │ │ ├── hlo_constant_folding.cc │ │ ├── hlo_constant_folding.h │ │ ├── hlo_constant_folding_test.cc │ │ ├── hlo_cost_analysis.cc │ │ ├── hlo_cost_analysis.h │ │ ├── hlo_cost_analysis_test.cc │ │ ├── hlo_creation_utils.cc │ │ ├── hlo_creation_utils.h │ │ ├── hlo_creation_utils_test.cc │ │ ├── hlo_cse.cc │ │ ├── hlo_cse.h │ │ ├── hlo_cse_test.cc │ │ ├── hlo_dataflow_analysis.cc │ │ ├── hlo_dataflow_analysis.h │ │ ├── hlo_dataflow_analysis_test.cc │ │ ├── hlo_dce.cc │ │ ├── hlo_dce.h │ │ ├── hlo_dce_test.cc │ │ ├── hlo_domain_isolator.cc │ │ ├── hlo_domain_isolator.h │ │ ├── hlo_domain_map.cc │ │ ├── hlo_domain_map.h │ │ ├── hlo_domain_remover.cc │ │ ├── hlo_domain_remover.h │ │ ├── hlo_domain_test.cc │ │ ├── hlo_domain_verifier.cc │ │ ├── hlo_domain_verifier.h │ │ ├── hlo_element_type_converter.cc │ │ ├── hlo_element_type_converter.h │ │ ├── hlo_element_type_converter_test.cc │ │ ├── hlo_execution_profile.cc │ │ ├── hlo_execution_profile.h │ │ ├── hlo_execution_profile_data.proto │ │ ├── hlo_execution_profile_test.cc │ │ ├── hlo_graph_dumper.cc │ │ ├── hlo_graph_dumper.h │ │ ├── hlo_graph_dumper_test.cc │ │ ├── hlo_input_output_alias_config_test.cc │ │ ├── hlo_instruction_test.cc │ │ ├── hlo_lexer.cc │ │ ├── hlo_lexer.h │ │ ├── hlo_liveness_analysis.cc │ │ ├── hlo_liveness_analysis.h │ │ ├── hlo_liveness_analysis_test.cc │ │ ├── hlo_memory_scheduler.cc │ │ ├── hlo_memory_scheduler.h │ │ ├── hlo_memory_scheduler_test.cc │ │ ├── hlo_module_config.cc │ │ ├── hlo_module_config.h │ │ ├── hlo_module_config_test.cc │ │ ├── hlo_module_dce.cc │ │ ├── hlo_module_dce.h │ │ ├── hlo_module_dce_test.cc │ │ ├── hlo_module_group_metadata.cc │ │ ├── hlo_module_group_metadata.h │ │ ├── hlo_module_group_test.cc │ │ ├── hlo_module_group_util.cc │ │ ├── hlo_module_group_util.h │ │ ├── hlo_module_metadata_test.cc │ │ ├── hlo_module_test.cc │ │ ├── hlo_module_util.cc │ │ ├── hlo_module_util.h │ │ ├── hlo_opcode_test.cc │ │ ├── hlo_ordering.cc │ │ ├── hlo_ordering.h │ │ ├── hlo_ordering_test.cc │ │ ├── hlo_parser.cc │ │ ├── hlo_parser.h │ │ ├── hlo_parser_test.cc │ │ ├── hlo_pass_fix.h │ │ ├── hlo_pass_interface.h │ │ ├── hlo_pass_pipeline.cc │ │ ├── hlo_pass_pipeline.h │ │ ├── hlo_pass_pipeline_test.cc │ │ ├── hlo_phi_graph.cc │ │ ├── hlo_phi_graph.h │ │ ├── hlo_phi_graph_test.cc │ │ ├── hlo_profile_printer.cc │ │ ├── hlo_profile_printer.h │ │ ├── hlo_profile_printer_data.proto │ │ ├── hlo_proto_util.cc │ │ ├── hlo_proto_util.h │ │ ├── hlo_proto_util_test.cc │ │ ├── hlo_reachability_test.cc │ │ ├── hlo_rematerialization.cc │ │ ├── hlo_rematerialization.h │ │ ├── hlo_rematerialization_test.cc │ │ ├── hlo_rematerialization_test_utils.h │ │ ├── hlo_rematerialization_test_utils_test.cc │ │ ├── hlo_replication_analysis.cc │ │ ├── hlo_replication_analysis.h │ │ ├── hlo_replication_analysis_test.cc │ │ ├── hlo_runner.cc │ │ ├── hlo_runner.h │ │ ├── hlo_runner_interface.cc │ │ ├── hlo_runner_interface.h │ │ ├── hlo_runner_pjrt.cc │ │ ├── hlo_runner_pjrt.h │ │ ├── hlo_schedule_test.cc │ │ ├── hlo_sharding_test.cc │ │ ├── hlo_value.cc │ │ ├── hlo_value.h │ │ ├── hlo_value_semantics_analysis.cc │ │ ├── hlo_value_semantics_analysis.h │ │ ├── hlo_value_semantics_analysis_test.cc │ │ ├── hlo_verifier.cc │ │ ├── hlo_verifier.h │ │ ├── hlo_verifier_test.cc │ │ ├── human_readable_profile_builder.cc │ │ ├── human_readable_profile_builder.h │ │ ├── indexed_array_analysis.cc │ │ ├── indexed_array_analysis.h │ │ ├── indexed_array_analysis_test.cc │ │ ├── instruction_fusion.cc │ │ ├── instruction_fusion.h │ │ ├── instruction_fusion_test.cc │ │ ├── instruction_hoister.cc │ │ ├── instruction_hoister.h │ │ ├── latency_hiding_scheduler.cc │ │ ├── latency_hiding_scheduler.h │ │ ├── latency_hiding_scheduler_test.cc │ │ ├── layout_assignment.cc │ │ ├── layout_assignment.h │ │ ├── layout_assignment_test.cc │ │ ├── layout_normalization.cc │ │ ├── layout_normalization.h │ │ ├── layout_normalization_test.cc │ │ ├── llvm_compiler.cc │ │ ├── llvm_compiler.h │ │ ├── llvm_ir │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── alias_analysis.cc │ │ │ ├── alias_analysis.h │ │ │ ├── alias_analysis_test.cc │ │ │ ├── buffer_assignment_util.cc │ │ │ ├── buffer_assignment_util.h │ │ │ ├── dynamic_update_slice_util.cc │ │ │ ├── dynamic_update_slice_util.h │ │ │ ├── fused_ir_emitter.cc │ │ │ ├── fused_ir_emitter.h │ │ │ ├── ir_array.cc │ │ │ ├── ir_array.h │ │ │ ├── ir_array_test.cc │ │ │ ├── ir_builder_mixin.h │ │ │ ├── kernel_support_library.cc │ │ │ ├── kernel_support_library.h │ │ │ ├── llvm_command_line_options.h │ │ │ ├── llvm_loop.cc │ │ │ ├── llvm_loop.h │ │ │ ├── llvm_type_conversion_util.h │ │ │ ├── llvm_util.cc │ │ │ ├── llvm_util.h │ │ │ ├── loop_emitter.cc │ │ │ ├── loop_emitter.h │ │ │ ├── math_ops.cc │ │ │ ├── math_ops.h │ │ │ ├── sort_util.cc │ │ │ ├── sort_util.h │ │ │ ├── tuple_ops.cc │ │ │ └── tuple_ops.h │ │ ├── local_service.cc │ │ ├── local_service.h │ │ ├── local_service_utils.cc │ │ ├── local_service_utils.h │ │ ├── logical_buffer.cc │ │ ├── logical_buffer.h │ │ ├── logical_buffer_analysis.cc │ │ ├── logical_buffer_analysis.h │ │ ├── logistic_expander.cc │ │ ├── logistic_expander.h │ │ ├── logistic_expander_test.cc │ │ ├── loop_schedule_linearizer.cc │ │ ├── loop_schedule_linearizer.h │ │ ├── loop_schedule_linearizer_test.cc │ │ ├── map_inliner.cc │ │ ├── map_inliner.h │ │ ├── map_inliner_test.cc │ │ ├── mapped_ptr_container_sorter.h │ │ ├── mapped_ptr_container_sorter_test.cc │ │ ├── maybe_owning_device_memory.cc │ │ ├── maybe_owning_device_memory.h │ │ ├── memory_space_assignment │ │ │ ├── BUILD │ │ │ ├── best_fit_repacker.cc │ │ │ ├── best_fit_repacker.h │ │ │ ├── best_fit_repacker_test.cc │ │ │ ├── memory_space_assignment.cc │ │ │ ├── memory_space_assignment.h │ │ │ ├── memory_space_assignment.proto │ │ │ ├── memory_space_assignment_test.cc │ │ │ ├── repacking.h │ │ │ ├── tuning_utils.cc │ │ │ ├── tuning_utils.h │ │ │ ├── utils.cc │ │ │ └── utils.h │ │ ├── memory_space_propagation.cc │ │ ├── memory_space_propagation.h │ │ ├── memory_space_propagation_test.cc │ │ ├── metrics.proto │ │ ├── metrics_hook_interface.h │ │ ├── multi_output_fusion.cc │ │ ├── multi_output_fusion.h │ │ ├── name_uniquer.cc │ │ ├── name_uniquer.h │ │ ├── name_uniquer_test.cc │ │ ├── op_expander_pass.cc │ │ ├── op_expander_pass.h │ │ ├── operand_upcaster.cc │ │ ├── operand_upcaster.h │ │ ├── operand_upcaster_test.cc │ │ ├── optimization_barrier_expander.cc │ │ ├── optimization_barrier_expander.h │ │ ├── optimize_input_output_buffer_alias.cc │ │ ├── optimize_input_output_buffer_alias.h │ │ ├── optimize_input_output_buffer_alias_test.cc │ │ ├── p2p_schedule_preparation.cc │ │ ├── p2p_schedule_preparation.h │ │ ├── p2p_schedule_preparation_test.cc │ │ ├── pattern_matcher.h │ │ ├── pattern_matcher_gmock.h │ │ ├── pattern_matcher_gmock_test.cc │ │ ├── pattern_matcher_test.cc │ │ ├── platform_util.cc │ │ ├── platform_util.h │ │ ├── profile_guided_latency_estimator.cc │ │ ├── profile_guided_latency_estimator.h │ │ ├── profile_guided_latency_estimator_test.cc │ │ ├── qr_expander.cc │ │ ├── qr_expander.h │ │ ├── real_imag_expander.cc │ │ ├── real_imag_expander.h │ │ ├── real_imag_expander_test.cc │ │ ├── reduce_decomposer.cc │ │ ├── reduce_decomposer.h │ │ ├── reduce_decomposer_test.cc │ │ ├── reduce_scatter_combiner.cc │ │ ├── reduce_scatter_combiner.h │ │ ├── reduce_scatter_combiner_test.cc │ │ ├── reduce_scatter_decomposer.cc │ │ ├── reduce_scatter_decomposer.h │ │ ├── reduce_scatter_decomposer_test.cc │ │ ├── reduce_scatter_reassociate.cc │ │ ├── reduce_scatter_reassociate.h │ │ ├── reduce_scatter_reassociate_test.cc │ │ ├── rendezvous.cc │ │ ├── rendezvous.h │ │ ├── reshape_decomposer.cc │ │ ├── reshape_decomposer.h │ │ ├── reshape_decomposer_test.cc │ │ ├── reshape_mover.cc │ │ ├── reshape_mover.h │ │ ├── reshape_mover_test.cc │ │ ├── result_caster.cc │ │ ├── result_caster.h │ │ ├── result_caster_test.cc │ │ ├── rng_bit_generator_expander.cc │ │ ├── rng_bit_generator_expander.h │ │ ├── rng_expander.cc │ │ ├── rng_expander.h │ │ ├── root_instruction_sinker.cc │ │ ├── root_instruction_sinker.h │ │ ├── root_instruction_sinker_test.cc │ │ ├── scatter_expander.cc │ │ ├── scatter_expander.h │ │ ├── scatter_expander_test.cc │ │ ├── scatter_simplifier.cc │ │ ├── scatter_simplifier.h │ │ ├── scatter_simplifier_test.cc │ │ ├── select_and_scatter_expander.cc │ │ ├── select_and_scatter_expander.h │ │ ├── select_and_scatter_expander_test.cc │ │ ├── service.cc │ │ ├── service.h │ │ ├── service_executable_run_options.h │ │ ├── shape_inference.cc │ │ ├── shape_inference.h │ │ ├── shape_inference_test.cc │ │ ├── shaped_buffer.cc │ │ ├── shaped_buffer.h │ │ ├── shaped_buffer_test.cc │ │ ├── sharding_format_picker.cc │ │ ├── sharding_format_picker.h │ │ ├── sharding_propagation.cc │ │ ├── sharding_propagation.h │ │ ├── sharding_propagation_test.cc │ │ ├── sharding_remover.cc │ │ ├── sharding_remover.h │ │ ├── sharding_remover_test.cc │ │ ├── simplify_fp_conversions.cc │ │ ├── simplify_fp_conversions.h │ │ ├── simplify_fp_conversions_test.cc │ │ ├── slice_sinker.cc │ │ ├── slice_sinker.h │ │ ├── slice_sinker_test.cc │ │ ├── slow_operation_alarm.cc │ │ ├── slow_operation_alarm.h │ │ ├── sort_simplifier.cc │ │ ├── sort_simplifier.h │ │ ├── sort_simplifier_test.cc │ │ ├── source_map_util.h │ │ ├── space_to_batch_converter.cc │ │ ├── space_to_batch_converter.h │ │ ├── space_to_batch_converter_test.cc │ │ ├── sparse_util.cc │ │ ├── sparse_util.h │ │ ├── spmd │ │ │ ├── BUILD │ │ │ ├── canonicalize_all_gather_for_cse.cc │ │ │ ├── canonicalize_all_gather_for_cse.h │ │ │ ├── canonicalize_all_gather_for_cse_test.cc │ │ │ ├── collective_permute_motion.cc │ │ │ ├── collective_permute_motion.h │ │ │ ├── collective_permute_motion_test.cc │ │ │ ├── convolution_handler.cc │ │ │ ├── convolution_handler.h │ │ │ ├── custom_call_handler.cc │ │ │ ├── custom_call_handler.h │ │ │ ├── dot_handler.cc │ │ │ ├── fft_handler.cc │ │ │ ├── gather_scatter_handler.cc │ │ │ ├── partition_assignment.cc │ │ │ ├── partition_assignment.h │ │ │ ├── partition_assignment_test.cc │ │ │ ├── schedule_aware_collective_ops_cse.cc │ │ │ ├── schedule_aware_collective_ops_cse.h │ │ │ ├── schedule_aware_collective_ops_cse_test.cc │ │ │ ├── spmd_partitioner.cc │ │ │ ├── spmd_partitioner.h │ │ │ ├── spmd_partitioner_test.cc │ │ │ ├── spmd_partitioner_util.cc │ │ │ ├── spmd_partitioner_util.h │ │ │ ├── spmd_prepare.cc │ │ │ ├── spmd_prepare.h │ │ │ ├── spmd_prepare_test.cc │ │ │ ├── stateful_rng_spmd_partitioner.cc │ │ │ ├── stateful_rng_spmd_partitioner.h │ │ │ ├── stateful_rng_spmd_partitioner_test.cc │ │ │ ├── whole_graph_manual_pass.cc │ │ │ ├── whole_graph_manual_pass.h │ │ │ └── whole_graph_manual_pass_test.cc │ │ ├── stable_sort_expander.cc │ │ ├── stable_sort_expander.h │ │ ├── stable_sort_expander_test.cc │ │ ├── stochastic_convert_decomposer.cc │ │ ├── stochastic_convert_decomposer.h │ │ ├── stochastic_convert_decomposer_test.cc │ │ ├── stream_pool.cc │ │ ├── stream_pool.h │ │ ├── stream_pool_test.cc │ │ ├── sub_byte_normalization.cc │ │ ├── sub_byte_normalization.h │ │ ├── symbol_repository.h │ │ ├── test_compilation_environment.proto │ │ ├── time_utils.cc │ │ ├── time_utils.h │ │ ├── topk_rewriter.cc │ │ ├── topk_rewriter.h │ │ ├── topk_rewriter_test.cc │ │ ├── tpu_computation_placer.cc │ │ ├── tpu_computation_placer.h │ │ ├── transfer_manager.cc │ │ ├── transfer_manager.h │ │ ├── transpose_folding.cc │ │ ├── transpose_folding.h │ │ ├── transpose_folding_test.cc │ │ ├── tree_reduction_rewriter.cc │ │ ├── tree_reduction_rewriter.h │ │ ├── triangular_solve_expander.cc │ │ ├── triangular_solve_expander.h │ │ ├── triangular_solve_expander_test.cc │ │ ├── tuple_points_to_analysis.cc │ │ ├── tuple_points_to_analysis.h │ │ ├── tuple_points_to_analysis_test.cc │ │ ├── tuple_simplifier.cc │ │ ├── tuple_simplifier.h │ │ ├── tuple_simplifier_test.cc │ │ ├── tuple_util.cc │ │ ├── tuple_util.h │ │ ├── tuple_util_test.cc │ │ ├── value_range.cc │ │ ├── value_range.h │ │ ├── value_range_test.cc │ │ ├── while_loop_all_reduce_code_motion.cc │ │ ├── while_loop_all_reduce_code_motion.h │ │ ├── while_loop_all_reduce_code_motion_test.cc │ │ ├── while_loop_analysis.cc │ │ ├── while_loop_analysis.h │ │ ├── while_loop_analysis_test.cc │ │ ├── while_loop_concat_code_motion.cc │ │ ├── while_loop_concat_code_motion.h │ │ ├── while_loop_concat_code_motion_test.cc │ │ ├── while_loop_constant_sinking.cc │ │ ├── while_loop_constant_sinking.h │ │ ├── while_loop_constant_sinking_test.cc │ │ ├── while_loop_expensive_invariant_code_motion.cc │ │ ├── while_loop_expensive_invariant_code_motion.h │ │ ├── while_loop_expensive_invariant_code_motion_test.cc │ │ ├── while_loop_fusible_sinking.cc │ │ ├── while_loop_fusible_sinking.h │ │ ├── while_loop_fusible_sinking_test.cc │ │ ├── while_loop_invariant_code_motion.cc │ │ ├── while_loop_invariant_code_motion.h │ │ ├── while_loop_invariant_code_motion_test.cc │ │ ├── while_loop_simplifier.cc │ │ ├── while_loop_simplifier.h │ │ ├── while_loop_simplifier_test.cc │ │ ├── while_loop_trip_count_annotator.cc │ │ ├── while_loop_trip_count_annotator.h │ │ ├── while_loop_trip_count_annotator_test.cc │ │ ├── while_loop_unroller.cc │ │ ├── while_loop_unroller.h │ │ ├── while_loop_unroller_test.cc │ │ ├── while_util.cc │ │ ├── while_util.h │ │ ├── while_util_test.cc │ │ ├── xla_aot_compile_cpu_test.cc │ │ ├── xla_aot_compile_gpu_test.cc │ │ ├── xla_aot_compile_stablehlo_cpu_test.cc │ │ ├── xla_aot_compile_stablehlo_test.mlir │ │ ├── xla_aot_compile_test.hlo │ │ ├── xla_aot_compile_test.mlir │ │ ├── xla_aot_compile_test_autotune_results.prototxt │ │ ├── xla_aot_compile_test_constant.mlir │ │ ├── xla_aot_compile_test_convolution.mlir │ │ ├── xla_aot_compile_test_gemm.mlir │ │ ├── xla_aot_compile_test_gpu_target_config.prototxt │ │ ├── xla_compile.bzl │ │ ├── xla_compile_main.cc │ │ ├── xla_compile_result.proto │ │ ├── xla_debug_info_manager.cc │ │ ├── xla_debug_info_manager.h │ │ ├── xla_debug_info_manager_test.cc │ │ ├── zero_sized_hlo_elimination.cc │ │ ├── zero_sized_hlo_elimination.h │ │ └── zero_sized_hlo_elimination_test.cc │ │ ├── service_interface.h │ │ ├── shape.cc │ │ ├── shape.h │ │ ├── shape_layout.cc │ │ ├── shape_layout.h │ │ ├── shape_test.cc │ │ ├── shape_tree.cc │ │ ├── shape_tree.h │ │ ├── shape_tree_test.cc │ │ ├── shape_util.cc │ │ ├── shape_util.h │ │ ├── shape_util_test.cc │ │ ├── sharding_op_util.cc │ │ ├── sharding_op_util.h │ │ ├── side_effect_util.cc │ │ ├── side_effect_util.h │ │ ├── status.h │ │ ├── status_macros.cc │ │ ├── status_macros.h │ │ ├── status_macros_test.cc │ │ ├── statusor.h │ │ ├── stream_executor │ │ ├── BUILD │ │ ├── allocator_stats.cc │ │ ├── allocator_stats.h │ │ ├── blas.cc │ │ ├── blas.h │ │ ├── build_defs.bzl │ │ ├── command_buffer.cc │ │ ├── command_buffer.h │ │ ├── cuda │ │ │ ├── BUILD │ │ │ ├── cuda_activation.h │ │ │ ├── cuda_asm_compiler.cc │ │ │ ├── cuda_blas.cc │ │ │ ├── cuda_blas.h │ │ │ ├── cuda_blas_lt.cc │ │ │ ├── cuda_blas_lt.h │ │ │ ├── cuda_blas_utils.cc │ │ │ ├── cuda_blas_utils.h │ │ │ ├── cuda_command_buffer_test.cc │ │ │ ├── cuda_conditional_kernels.cc │ │ │ ├── cuda_conditional_kernels.cu.cc │ │ │ ├── cuda_diagnostics.cc │ │ │ ├── cuda_diagnostics.h │ │ │ ├── cuda_dnn.cc │ │ │ ├── cuda_dnn.h │ │ │ ├── cuda_driver.cc │ │ │ ├── cuda_driver.h │ │ │ ├── cuda_driver_test.cc │ │ │ ├── cuda_event.cc │ │ │ ├── cuda_event.h │ │ │ ├── cuda_executor.cc │ │ │ ├── cuda_fft.cc │ │ │ ├── cuda_fft.h │ │ │ ├── cuda_helpers.h │ │ │ ├── cuda_kernel.cc │ │ │ ├── cuda_kernel.h │ │ │ ├── cuda_kernel_test.cc │ │ │ ├── cuda_platform.cc │ │ │ ├── cuda_platform.h │ │ │ ├── cuda_platform_id.cc │ │ │ ├── cuda_platform_id.h │ │ │ ├── cuda_runtime.cc │ │ │ ├── cuda_stream.h │ │ │ ├── cuda_test_kernels.cu.cc │ │ │ ├── cuda_test_kernels.h │ │ │ ├── memcpy_test.cc │ │ │ ├── redzone_allocator_test.cc │ │ │ └── stream_search_test.cc │ │ ├── data_type.h │ │ ├── device_description.cc │ │ ├── device_description.h │ │ ├── device_description.proto │ │ ├── device_id_utils.h │ │ ├── device_memory.h │ │ ├── device_memory_allocator.h │ │ ├── device_options.h │ │ ├── dnn.cc │ │ ├── dnn.h │ │ ├── dnn_test.cc │ │ ├── event.cc │ │ ├── event.h │ │ ├── executor_cache.cc │ │ ├── executor_cache.h │ │ ├── fft.h │ │ ├── gpu │ │ │ ├── BUILD │ │ │ ├── asm_compiler.cc │ │ │ ├── asm_compiler.h │ │ │ ├── gpu_activation.cc │ │ │ ├── gpu_activation.h │ │ │ ├── gpu_asm_opts.h │ │ │ ├── gpu_blas_lt.cc │ │ │ ├── gpu_blas_lt.h │ │ │ ├── gpu_command_buffer.cc │ │ │ ├── gpu_command_buffer.h │ │ │ ├── gpu_cudamallocasync_allocator.cc │ │ │ ├── gpu_cudamallocasync_allocator.h │ │ │ ├── gpu_diagnostics.h │ │ │ ├── gpu_driver.h │ │ │ ├── gpu_event.cc │ │ │ ├── gpu_event.h │ │ │ ├── gpu_executor.h │ │ │ ├── gpu_graph.cc │ │ │ ├── gpu_graph.h │ │ │ ├── gpu_helpers.h │ │ │ ├── gpu_init.cc │ │ │ ├── gpu_init.h │ │ │ ├── gpu_kernel.h │ │ │ ├── gpu_runtime.h │ │ │ ├── gpu_stream.cc │ │ │ ├── gpu_stream.h │ │ │ ├── gpu_timer.cc │ │ │ ├── gpu_timer.h │ │ │ ├── gpu_types.h │ │ │ ├── redzone_allocator.cc │ │ │ └── redzone_allocator.h │ │ ├── host │ │ │ ├── BUILD │ │ │ ├── host_gpu_executor.cc │ │ │ ├── host_gpu_executor.h │ │ │ ├── host_platform.cc │ │ │ ├── host_platform.h │ │ │ ├── host_platform_id.cc │ │ │ ├── host_platform_id.h │ │ │ ├── host_stream.cc │ │ │ ├── host_stream.h │ │ │ └── host_stream_test.cc │ │ ├── host_or_device_scalar.h │ │ ├── integrations │ │ │ ├── BUILD │ │ │ ├── device_host_allocator.h │ │ │ ├── device_mem_allocator.h │ │ │ ├── tf_allocator_adapter.cc │ │ │ └── tf_allocator_adapter.h │ │ ├── kernel.cc │ │ ├── kernel.h │ │ ├── kernel_spec.cc │ │ ├── kernel_spec.h │ │ ├── kernel_test.cc │ │ ├── launch_dim.h │ │ ├── lazy_op_runner.h │ │ ├── module_spec.h │ │ ├── multi_platform_manager.cc │ │ ├── multi_platform_manager.h │ │ ├── numeric_options.h │ │ ├── platform.cc │ │ ├── platform.h │ │ ├── platform │ │ │ ├── BUILD │ │ │ ├── default │ │ │ │ ├── BUILD │ │ │ │ ├── dso_loader.h │ │ │ │ └── initialize.h │ │ │ ├── dso_loader.h │ │ │ ├── initialize.h │ │ │ ├── platform.h │ │ │ └── port.h │ │ ├── plugin_registry.cc │ │ ├── plugin_registry.h │ │ ├── rocm │ │ │ ├── BUILD │ │ │ ├── hip_blas_lt.cc │ │ │ ├── hip_blas_lt.h │ │ │ ├── hip_blas_utils.cc │ │ │ ├── hip_blas_utils.h │ │ │ ├── hip_conditional_kernels.cu.cc │ │ │ ├── hipblaslt_wrapper.h │ │ │ ├── hipsolver_wrapper.h │ │ │ ├── hipsparse_wrapper.h │ │ │ ├── rocblas_wrapper.h │ │ │ ├── rocm_activation.h │ │ │ ├── rocm_blas.cc │ │ │ ├── rocm_blas.h │ │ │ ├── rocm_diagnostics.cc │ │ │ ├── rocm_diagnostics.h │ │ │ ├── rocm_dnn.cc │ │ │ ├── rocm_dnn.h │ │ │ ├── rocm_driver.cc │ │ │ ├── rocm_driver.h │ │ │ ├── rocm_driver_wrapper.h │ │ │ ├── rocm_event.cc │ │ │ ├── rocm_fft.cc │ │ │ ├── rocm_fft.h │ │ │ ├── rocm_gpu_executor.cc │ │ │ ├── rocm_helpers.cu.cc │ │ │ ├── rocm_kernel.cc │ │ │ ├── rocm_platform.cc │ │ │ ├── rocm_platform.h │ │ │ ├── rocm_platform_id.cc │ │ │ ├── rocm_platform_id.h │ │ │ ├── rocsolver_wrapper.h │ │ │ └── roctracer_wrapper.h │ │ ├── scratch_allocator.cc │ │ ├── scratch_allocator.h │ │ ├── stream.cc │ │ ├── stream.h │ │ ├── stream_executor.h │ │ ├── stream_executor_internal.h │ │ ├── stream_executor_pimpl.cc │ │ ├── stream_executor_pimpl.h │ │ ├── stream_test.cc │ │ ├── temporary_device_memory.cc │ │ ├── temporary_device_memory.h │ │ ├── temporary_memory_manager.cc │ │ ├── temporary_memory_manager.h │ │ ├── tpu │ │ │ ├── BUILD │ │ │ ├── c_api_conversions.cc │ │ │ ├── c_api_conversions.h │ │ │ ├── c_api_conversions_test.cc │ │ │ ├── c_api_decl.h │ │ │ ├── c_api_defn.h │ │ │ ├── libtftpu.h │ │ │ ├── noncopyable_buffer.h │ │ │ ├── proto_helper.cc │ │ │ ├── proto_helper.h │ │ │ ├── status_helper.h │ │ │ ├── tpu_api.cc │ │ │ ├── tpu_api.h │ │ │ ├── tpu_api_dlsym_set_fn.h │ │ │ ├── tpu_event.h │ │ │ ├── tpu_executable.cc │ │ │ ├── tpu_executable.h │ │ │ ├── tpu_executable_interface.cc │ │ │ ├── tpu_executable_interface.h │ │ │ ├── tpu_executor.cc │ │ │ ├── tpu_executor.h │ │ │ ├── tpu_executor_api.cc │ │ │ ├── tpu_executor_api.h │ │ │ ├── tpu_executor_c_api.h │ │ │ ├── tpu_executor_init_fns.inc │ │ │ ├── tpu_executor_interface.h │ │ │ ├── tpu_initialize_util.cc │ │ │ ├── tpu_initialize_util.h │ │ │ ├── tpu_library_init_fns.inc │ │ │ ├── tpu_node_context.cc │ │ │ ├── tpu_node_context.h │ │ │ ├── tpu_on_demand_compiler.cc │ │ │ ├── tpu_op_executable.cc │ │ │ ├── tpu_op_executable.h │ │ │ ├── tpu_ops_c_api.h │ │ │ ├── tpu_platform.cc │ │ │ ├── tpu_platform.h │ │ │ ├── tpu_platform_id.cc │ │ │ ├── tpu_platform_id.h │ │ │ ├── tpu_platform_interface.cc │ │ │ ├── tpu_platform_interface.h │ │ │ ├── tpu_platform_registration.cc │ │ │ ├── tpu_profiler_c_api.h │ │ │ ├── tpu_profiler_init_fns.inc │ │ │ ├── tpu_stream.h │ │ │ ├── tpu_stream_interface.h │ │ │ ├── tpu_topology.cc │ │ │ ├── tpu_topology.h │ │ │ ├── tpu_transfer_manager.cc │ │ │ ├── tpu_transfer_manager.h │ │ │ ├── tpu_transfer_manager_interface.cc │ │ │ ├── tpu_transfer_manager_interface.h │ │ │ ├── tpu_transfer_manager_registration.cc │ │ │ └── tsl_status_helper.h │ │ └── trace_listener.h │ │ ├── strict.default.bzl │ │ ├── test.h │ │ ├── test_helpers.h │ │ ├── tests │ │ ├── BUILD │ │ ├── all_reduce_test.cc │ │ ├── array_elementwise_ops_test.cc │ │ ├── axpy_simple_test.cc │ │ ├── bad_rng_shape_validation_test.cc │ │ ├── batch_normalization_test.cc │ │ ├── bfloat16_test.cc │ │ ├── binop_scaling_test.cc │ │ ├── bitcast_convert_test.cc │ │ ├── broadcast_simple_test.cc │ │ ├── broadcast_test.cc │ │ ├── buffer_donation_test.cc │ │ ├── build_defs.bzl │ │ ├── call_test.cc │ │ ├── check_execution_arity_test.cc │ │ ├── cholesky_test.cc │ │ ├── client_library_test_base.cc │ │ ├── client_library_test_base.h │ │ ├── client_test.cc │ │ ├── codegen_test_base.cc │ │ ├── codegen_test_base.h │ │ ├── collective_ops_test.cc │ │ ├── collective_ops_test_e2e.cc │ │ ├── collective_pipeliner_execution_test.cc │ │ ├── compilation_cache_test.cc │ │ ├── compute_constant_test.cc │ │ ├── concat_test.cc │ │ ├── conditional_test.cc │ │ ├── constant_reduction_function_test.cc │ │ ├── constants_test.cc │ │ ├── conv_depthwise_backprop_filter_test.cc │ │ ├── conv_depthwise_common.cc │ │ ├── conv_depthwise_common.h │ │ ├── conv_depthwise_test.cc │ │ ├── convert_test.cc │ │ ├── convolution_cudnn_test.cc │ │ ├── convolution_dimension_numbers_test.cc │ │ ├── convolution_test.cc │ │ ├── convolution_test_1d.cc │ │ ├── convolution_variants_test.cc │ │ ├── copy_test.cc │ │ ├── cpu_gpu_fusion_test.cc │ │ ├── custom_call_test.cc │ │ ├── deallocation_test.cc │ │ ├── deconstruct_tuple_test.cc │ │ ├── deep_graph_test.cc │ │ ├── dot_operation_test.cc │ │ ├── dynamic_ops_test.cc │ │ ├── exhaustive │ │ │ ├── BUILD │ │ │ ├── exhaustive_binary_16_bit_test.cc │ │ │ ├── exhaustive_binary_test_f32_f64.cc │ │ │ ├── exhaustive_op_test_utils.cc │ │ │ ├── exhaustive_op_test_utils.h │ │ │ ├── exhaustive_unary_test_complex.cc │ │ │ ├── exhaustive_unary_test_f32_or_smaller.cc │ │ │ └── exhaustive_unary_test_f64.cc │ │ ├── fft_test.cc │ │ ├── filecheck.cc │ │ ├── filecheck.h │ │ ├── float8_test.cc │ │ ├── floor_ceil_test.cc │ │ ├── fmax_fmin_test.cc │ │ ├── fuzz │ │ │ ├── BUILD │ │ │ ├── build_defs.bzl │ │ │ ├── hlo_test_lib.cc │ │ │ ├── rand_000000.hlo │ │ │ ├── rand_000001.hlo │ │ │ ├── rand_000003.hlo │ │ │ ├── rand_000004.hlo │ │ │ ├── rand_000005.hlo │ │ │ ├── rand_000006.hlo │ │ │ ├── rand_000007.hlo │ │ │ ├── rand_000008.hlo │ │ │ ├── rand_000009.hlo │ │ │ ├── rand_000013.hlo │ │ │ ├── rand_000015.hlo │ │ │ ├── rand_000016.hlo │ │ │ ├── rand_000017.hlo │ │ │ ├── rand_000018.hlo │ │ │ ├── rand_000019.hlo │ │ │ ├── rand_000020.hlo │ │ │ ├── rand_000022.hlo │ │ │ ├── rand_000024.hlo │ │ │ ├── rand_000025.hlo │ │ │ ├── rand_000026.hlo │ │ │ ├── rand_000030.hlo │ │ │ ├── rand_000031.hlo │ │ │ ├── rand_000032.hlo │ │ │ ├── rand_000033.hlo │ │ │ ├── rand_000034.hlo │ │ │ ├── rand_000035.hlo │ │ │ ├── rand_000036.hlo │ │ │ ├── rand_000039.hlo │ │ │ ├── rand_000040.hlo │ │ │ ├── rand_000041.hlo │ │ │ ├── rand_000043.hlo │ │ │ ├── rand_000049.hlo │ │ │ ├── rand_000053.hlo │ │ │ ├── rand_000056.hlo │ │ │ ├── rand_000059.hlo │ │ │ ├── rand_000060.hlo │ │ │ ├── rand_000061.hlo │ │ │ ├── rand_000062.hlo │ │ │ ├── rand_000064.hlo │ │ │ ├── rand_000066.hlo │ │ │ ├── rand_000067.hlo │ │ │ ├── rand_000069.hlo │ │ │ ├── rand_000071.hlo │ │ │ ├── rand_000072.hlo │ │ │ ├── rand_000077.hlo │ │ │ ├── rand_000078.hlo │ │ │ ├── rand_000079.hlo │ │ │ ├── rand_000081.hlo │ │ │ ├── rand_000084.hlo │ │ │ ├── rand_000085.hlo │ │ │ ├── rand_000086.hlo │ │ │ ├── rand_000088.hlo │ │ │ ├── rand_000089.hlo │ │ │ ├── rand_000090.hlo │ │ │ ├── rand_000092.hlo │ │ │ ├── rand_000094.hlo │ │ │ └── rand_000095.hlo │ │ ├── gather_operation_test.cc │ │ ├── get_dimension_size_test.cc │ │ ├── grouped_convolution_test.cc │ │ ├── half_test.cc │ │ ├── hlo_metadata_test.cc │ │ ├── hlo_test_base.cc │ │ ├── hlo_test_base.h │ │ ├── int4_test.cc │ │ ├── iota_test.cc │ │ ├── isolated_convolution.hlo │ │ ├── literal_test_util.cc │ │ ├── literal_test_util.h │ │ ├── literal_test_util_test.cc │ │ ├── llvm_compiler_test.cc │ │ ├── llvm_irgen_test_base.cc │ │ ├── llvm_irgen_test_base.h │ │ ├── local_client_allocation_test.cc │ │ ├── local_client_aot_test.cc │ │ ├── local_client_aot_test_helper.cc │ │ ├── local_client_execute_test.cc │ │ ├── local_client_test_base.cc │ │ ├── local_client_test_base.h │ │ ├── log_test.cc │ │ ├── manifest_checking_test.cc │ │ ├── manifest_checking_test.h │ │ ├── map_test.cc │ │ ├── matmul_test.cc │ │ ├── matrix_ops_simple_test.cc │ │ ├── multidimensional_slice_test.cc │ │ ├── multioutput_fusion_test.cc │ │ ├── multiple_devices_on_host_test.cc │ │ ├── multithreaded_compilation_test.cc │ │ ├── onednn_matmul_test.cc │ │ ├── outfeed_in_nested_computation_test.cc │ │ ├── pad_test.cc │ │ ├── params_test.cc │ │ ├── pjrt_client_registry.cc │ │ ├── pjrt_client_registry.h │ │ ├── pjrt_cpu_client_registry.cc │ │ ├── pjrt_gpu_client_registry.cc │ │ ├── plugin.bzl │ │ ├── pred_test.cc │ │ ├── prng_test.cc │ │ ├── ptxas_bug_120501638.cc │ │ ├── query_inferred_shape_test.cc │ │ ├── reduce_hlo_test.cc │ │ ├── reduce_precision_test.cc │ │ ├── reduce_test.cc │ │ ├── reduce_window_test.cc │ │ ├── replay_test.cc │ │ ├── replicated_io_feed_test.cc │ │ ├── reshape_motion_test.cc │ │ ├── reshape_test.cc │ │ ├── reverse_test.cc │ │ ├── round_trip_packed_literal_test.cc │ │ ├── round_trip_transfer_test.cc │ │ ├── sample_file_test.cc │ │ ├── sample_text_test.cc │ │ ├── scalar_computations_test.cc │ │ ├── scatter_test.cc │ │ ├── select_and_scatter_test.cc │ │ ├── select_test.cc │ │ ├── slice_test.cc │ │ ├── test_macros.cc │ │ ├── test_macros.h │ │ ├── test_utils.cc │ │ ├── test_utils.h │ │ ├── test_utils_test.cc │ │ ├── tile_assignment_test.cc │ │ ├── token_hlo_test.cc │ │ ├── topk_test.cc │ │ ├── transfer_manager_test.cc │ │ ├── transpose_test.cc │ │ ├── triangular_solve_test.cc │ │ ├── tuple_test.cc │ │ ├── unary_op_test.cc │ │ ├── value_inference_test.cc │ │ ├── vector_ops_reduce_test.cc │ │ ├── vector_ops_simple_test.cc │ │ ├── verified_hlo_module.cc │ │ ├── verified_hlo_module.h │ │ ├── while_test.cc │ │ ├── xla_hlo_profile_test.cc │ │ └── xla_internal_test_main.cc │ │ ├── text_literal_reader.cc │ │ ├── text_literal_reader.h │ │ ├── text_literal_reader_test.cc │ │ ├── text_literal_writer.cc │ │ ├── text_literal_writer.h │ │ ├── text_literal_writer_test.cc │ │ ├── tools │ │ ├── BUILD │ │ ├── compute_cost.cc │ │ ├── convert_computation.cc │ │ ├── data │ │ │ ├── add.hlo │ │ │ ├── benchmarking │ │ │ │ └── mobilenet_v2.hlo │ │ │ ├── must_alias.hlo │ │ │ └── must_alias_with_sharding.hlo │ │ ├── driver.cc │ │ ├── dumped_computation_to_operation_list.cc │ │ ├── dumped_computation_to_text.cc │ │ ├── hex_floats_to_packed_literal.cc │ │ ├── hlo_bisect │ │ │ ├── BUILD │ │ │ ├── hlo_bisect.cc │ │ │ ├── hlo_bisect_state.cc │ │ │ ├── hlo_bisect_state.h │ │ │ ├── hlo_bisect_state_test.cc │ │ │ ├── hlo_bisect_utils.cc │ │ │ └── hlo_bisect_utils.h │ │ ├── hlo_control_flow_flattening.cc │ │ ├── hlo_control_flow_flattening.h │ │ ├── hlo_control_flow_flattening_test.cc │ │ ├── hlo_expand.cc │ │ ├── hlo_expand.h │ │ ├── hlo_expand_main.cc │ │ ├── hlo_extractor.cc │ │ ├── hlo_extractor.h │ │ ├── hlo_extractor_test.cc │ │ ├── hlo_module_loader.cc │ │ ├── hlo_module_loader.h │ │ ├── hlo_module_loader_test.cc │ │ ├── hlo_opt │ │ │ ├── BUILD │ │ │ ├── cpu_hlo.hlo │ │ │ ├── cpu_opt.cc │ │ │ ├── gpu_hlo.hlo │ │ │ ├── gpu_hlo_buffers.hlo │ │ │ ├── gpu_hlo_llvm.hlo │ │ │ ├── gpu_hlo_ptx.hlo │ │ │ ├── gpu_opt.cc │ │ │ ├── gpu_specs │ │ │ │ └── a100.txtpb │ │ │ ├── opt_lib.cc │ │ │ ├── opt_lib.h │ │ │ └── opt_main.cc │ │ ├── hlo_proto_to_json.cc │ │ ├── hlo_slicer.cc │ │ ├── hlo_slicer.h │ │ ├── hlo_slicer_test.cc │ │ ├── interactive_graphviz.cc │ │ ├── interactive_graphviz_bin_test.cc │ │ ├── multihost_hlo_runner │ │ │ ├── BUILD │ │ │ ├── README.md │ │ │ ├── data │ │ │ │ ├── sharded_16_devices.hlo │ │ │ │ ├── sharded_2_devices.hlo │ │ │ │ ├── single_device.hlo │ │ │ │ └── single_device_tupled.hlo │ │ │ ├── functional_hlo_runner.cc │ │ │ ├── functional_hlo_runner.h │ │ │ ├── functional_hlo_runner_test.cc │ │ │ ├── hlo_runner_flags.cc │ │ │ ├── hlo_runner_flags.h │ │ │ └── hlo_runner_main.cc │ │ ├── prepare_reference_module.cc │ │ ├── prepare_reference_module.h │ │ ├── run_hlo_module.cc │ │ ├── run_hlo_module.h │ │ ├── run_hlo_module.proto │ │ ├── run_hlo_module_bin_test.cc │ │ ├── run_hlo_module_main.cc │ │ ├── show_literal.cc │ │ ├── show_signature.cc │ │ ├── show_text_literal.cc │ │ ├── tests │ │ │ ├── cholesky.hlo │ │ │ ├── hlo_expand_test.cc │ │ │ └── invalid_concat.hlo │ │ ├── xla_compile_lib.cc │ │ ├── xla_compile_lib.h │ │ └── xla_compile_lib_test.cc │ │ ├── translate │ │ ├── BUILD │ │ ├── hlo_to_mhlo │ │ │ ├── BUILD │ │ │ ├── attribute_importer.cc │ │ │ ├── attribute_importer.h │ │ │ ├── hlo_function_importer.cc │ │ │ ├── hlo_function_importer.h │ │ │ ├── hlo_module_importer.cc │ │ │ ├── hlo_module_importer.h │ │ │ ├── hlo_to_mlir_hlo.cc │ │ │ ├── hlo_to_mlir_hlo.h │ │ │ ├── hlo_utils.cc │ │ │ ├── hlo_utils.h │ │ │ ├── hlo_utils_test.cc │ │ │ ├── location_importer.cc │ │ │ ├── location_importer.h │ │ │ ├── stack_location_utils.cc │ │ │ ├── stack_location_utils.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── bool_compare.hlotxt │ │ │ │ ├── case_conditional.hlotxt │ │ │ │ ├── dynamic_param.hlo │ │ │ │ ├── entry_computation_layout.hlotxt │ │ │ │ ├── frontend_attributes.hlotxt │ │ │ │ ├── fully_connected_reference_model.hlotxt │ │ │ │ ├── fusion.hlotxt │ │ │ │ ├── if_conditional.hlotxt │ │ │ │ ├── import.hlotxt │ │ │ │ ├── import_async.hlotxt │ │ │ │ ├── layouts_and_names.hlotxt │ │ │ │ ├── location.hlotxt │ │ │ │ ├── module_attributes.hlo │ │ │ │ ├── send_recv.hlotxt │ │ │ │ ├── simple.hlo │ │ │ │ ├── spmd_module_sharding.hlo │ │ │ │ ├── stacktrace_to_location.hlo │ │ │ │ ├── types.hlotxt │ │ │ │ └── while.hlotxt │ │ │ ├── translate.cc │ │ │ ├── translate.h │ │ │ └── translate_registration.cc │ │ ├── mhlo_to_hlo │ │ │ ├── BUILD │ │ │ ├── attribute_exporter.cc │ │ │ ├── attribute_exporter.h │ │ │ ├── layout_util.cc │ │ │ ├── layout_util.h │ │ │ ├── location_exporter.cc │ │ │ ├── location_exporter.h │ │ │ ├── mlir_hlo_to_hlo.cc │ │ │ ├── mlir_hlo_to_hlo.h │ │ │ ├── operator_writer_gen.cc │ │ │ ├── stack_frame_index_builder.cc │ │ │ ├── stack_frame_index_builder.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── add.mlir │ │ │ │ ├── case.mlir │ │ │ │ ├── dynamic.mlir │ │ │ │ ├── export-with-layouts.mlir │ │ │ │ ├── export.mlir │ │ │ │ ├── export_and_check_layouts.mlir │ │ │ │ ├── export_large_constants.mlir │ │ │ │ ├── export_replicas.mlir │ │ │ │ ├── frontend_attributes.mlir │ │ │ │ ├── fusion.mlir │ │ │ │ ├── if.mlir │ │ │ │ ├── input_output_aliasing.mlir │ │ │ │ ├── int4.mlir │ │ │ │ ├── layouts_and_names.mlir │ │ │ │ ├── location_to_op_metadata.mlir │ │ │ │ ├── location_to_stacktrace.mlir │ │ │ │ ├── missing_main.mlir │ │ │ │ ├── module_attributes.mlir │ │ │ │ ├── multiple_return_tuple.mlir │ │ │ │ ├── opaque_elements_attr.mlir │ │ │ │ ├── rng_get_and_update_state.mlir │ │ │ │ ├── sharding.mlir │ │ │ │ ├── simple.mlir │ │ │ │ ├── unsupported_type.mlir │ │ │ │ └── while.mlir │ │ │ ├── translate.cc │ │ │ ├── translate.h │ │ │ ├── translate_registration.cc │ │ │ ├── type_to_shape.cc │ │ │ ├── type_to_shape.h │ │ │ └── type_to_shape_test.cc │ │ ├── mhlo_to_lhlo_with_xla │ │ │ ├── BUILD │ │ │ ├── mhlo_to_lhlo_with_xla.cc │ │ │ ├── mhlo_to_lhlo_with_xla.h │ │ │ ├── tests │ │ │ │ ├── BUILD │ │ │ │ ├── hlo_text_to_lhlo_no_opt.hlotxt │ │ │ │ ├── no_opt_ops.hlotxt │ │ │ │ └── non_identity_layouts.hlotxt │ │ │ ├── translate_registration.cc │ │ │ └── xla_translate_opt_main.cc │ │ └── xla_translate_main.cc │ │ ├── types.h │ │ ├── types_test.cc │ │ ├── union_find.h │ │ ├── util.cc │ │ ├── util.h │ │ ├── util_test.cc │ │ ├── window_util.cc │ │ ├── window_util.h │ │ ├── window_util_test.cc │ │ ├── xla.bzl │ │ ├── xla.proto │ │ └── xla_data.proto └── zlib.BUILD └── tools └── tf_env_collect.sh /.bazelignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.bazelignore -------------------------------------------------------------------------------- /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.bazelrc -------------------------------------------------------------------------------- /.bazelversion: -------------------------------------------------------------------------------- 1 | 6.4.0 2 | # NOTE: Update Bazel version in tensorflow/tools/ci_build/release/common.sh.oss -------------------------------------------------------------------------------- /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.clang-format -------------------------------------------------------------------------------- /.github/bot_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.github/bot_config.yml -------------------------------------------------------------------------------- /.github/workflows/arm-cd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.github/workflows/arm-cd.yml -------------------------------------------------------------------------------- /.github/workflows/arm-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.github/workflows/arm-ci.yml -------------------------------------------------------------------------------- /.github/workflows/cffconvert.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.github/workflows/cffconvert.yml -------------------------------------------------------------------------------- /.github/workflows/create_issue.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.github/workflows/create_issue.js -------------------------------------------------------------------------------- /.github/workflows/update-rbe.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.github/workflows/update-rbe.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- 1 | tensorflow/tools/ci_build/pylintrc -------------------------------------------------------------------------------- /.zenodo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/.zenodo.json -------------------------------------------------------------------------------- /AUTHORS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/AUTHORS -------------------------------------------------------------------------------- /BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/BUILD -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /ISSUES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ISSUES.md -------------------------------------------------------------------------------- /ISSUE_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ISSUE_TEMPLATE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/README.md -------------------------------------------------------------------------------- /RELEASE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/RELEASE.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/SECURITY.md -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/WORKSPACE -------------------------------------------------------------------------------- /arm_compiler.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/arm_compiler.BUILD -------------------------------------------------------------------------------- /ci/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/README.md -------------------------------------------------------------------------------- /ci/devinfra/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/devinfra/README.md -------------------------------------------------------------------------------- /ci/official/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/README.md -------------------------------------------------------------------------------- /ci/official/any.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/any.sh -------------------------------------------------------------------------------- /ci/official/bisect.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/bisect.sh -------------------------------------------------------------------------------- /ci/official/code_check_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/code_check_full.sh -------------------------------------------------------------------------------- /ci/official/containers/linux_arm64/.gitignore: -------------------------------------------------------------------------------- 1 | .*.swp 2 | -------------------------------------------------------------------------------- /ci/official/debug_tfci.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/debug_tfci.sh -------------------------------------------------------------------------------- /ci/official/envs/ci_default: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/envs/ci_default -------------------------------------------------------------------------------- /ci/official/envs/sample: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/envs/sample -------------------------------------------------------------------------------- /ci/official/libtensorflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/libtensorflow.sh -------------------------------------------------------------------------------- /ci/official/pycpp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/pycpp.sh -------------------------------------------------------------------------------- /ci/official/utilities/docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/utilities/docker.sh -------------------------------------------------------------------------------- /ci/official/utilities/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/utilities/setup.sh -------------------------------------------------------------------------------- /ci/official/wheel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/wheel.sh -------------------------------------------------------------------------------- /ci/official/wheel_test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/wheel_test/BUILD -------------------------------------------------------------------------------- /ci/official/wheel_test/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/wheel_test/README.md -------------------------------------------------------------------------------- /ci/official/wheel_test/WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/ci/official/wheel_test/WORKSPACE -------------------------------------------------------------------------------- /configure: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/configure -------------------------------------------------------------------------------- /configure.cmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/configure.cmd -------------------------------------------------------------------------------- /configure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/configure.py -------------------------------------------------------------------------------- /models.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/models.BUILD -------------------------------------------------------------------------------- /requirements_lock_3_10.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/requirements_lock_3_10.txt -------------------------------------------------------------------------------- /requirements_lock_3_11.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/requirements_lock_3_11.txt -------------------------------------------------------------------------------- /requirements_lock_3_12.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/requirements_lock_3_12.txt -------------------------------------------------------------------------------- /requirements_lock_3_9.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/requirements_lock_3_9.txt -------------------------------------------------------------------------------- /tensorflow/.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/.clang-format -------------------------------------------------------------------------------- /tensorflow/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/BUILD -------------------------------------------------------------------------------- /tensorflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/__init__.py -------------------------------------------------------------------------------- /tensorflow/c/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/BUILD -------------------------------------------------------------------------------- /tensorflow/c/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/README.md -------------------------------------------------------------------------------- /tensorflow/c/c_api.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api.cc -------------------------------------------------------------------------------- /tensorflow/c/c_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api.h -------------------------------------------------------------------------------- /tensorflow/c/c_api_experimental.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api_experimental.h -------------------------------------------------------------------------------- /tensorflow/c/c_api_function.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api_function.cc -------------------------------------------------------------------------------- /tensorflow/c/c_api_internal.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api_internal.h -------------------------------------------------------------------------------- /tensorflow/c/c_api_macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api_macros.h -------------------------------------------------------------------------------- /tensorflow/c/c_api_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_api_test.cc -------------------------------------------------------------------------------- /tensorflow/c/c_op_requires.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_op_requires.h -------------------------------------------------------------------------------- /tensorflow/c/c_test.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_test.c -------------------------------------------------------------------------------- /tensorflow/c/c_test_util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_test_util.cc -------------------------------------------------------------------------------- /tensorflow/c/c_test_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/c_test_util.h -------------------------------------------------------------------------------- /tensorflow/c/checkpoint_reader.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/checkpoint_reader.cc -------------------------------------------------------------------------------- /tensorflow/c/checkpoint_reader.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/checkpoint_reader.h -------------------------------------------------------------------------------- /tensorflow/c/conversion_macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/conversion_macros.h -------------------------------------------------------------------------------- /tensorflow/c/eager/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/BUILD -------------------------------------------------------------------------------- /tensorflow/c/eager/c_api.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/c_api.cc -------------------------------------------------------------------------------- /tensorflow/c/eager/c_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/c_api.h -------------------------------------------------------------------------------- /tensorflow/c/eager/c_api_debug.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/c_api_debug.cc -------------------------------------------------------------------------------- /tensorflow/c/eager/c_api_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/c_api_test.cc -------------------------------------------------------------------------------- /tensorflow/c/eager/dlpack.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/dlpack.cc -------------------------------------------------------------------------------- /tensorflow/c/eager/dlpack.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/dlpack.h -------------------------------------------------------------------------------- /tensorflow/c/eager/dlpack_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/dlpack_test.cc -------------------------------------------------------------------------------- /tensorflow/c/eager/gradients.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/gradients.cc -------------------------------------------------------------------------------- /tensorflow/c/eager/gradients.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/gradients.h -------------------------------------------------------------------------------- /tensorflow/c/eager/tape.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/eager/tape.h -------------------------------------------------------------------------------- /tensorflow/c/env.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/env.cc -------------------------------------------------------------------------------- /tensorflow/c/env.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/env.h -------------------------------------------------------------------------------- /tensorflow/c/env_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/env_test.cc -------------------------------------------------------------------------------- /tensorflow/c/exported_symbols.lds: -------------------------------------------------------------------------------- 1 | _TF_* 2 | _TFE_* 3 | -------------------------------------------------------------------------------- /tensorflow/c/generate-pc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/generate-pc.sh -------------------------------------------------------------------------------- /tensorflow/c/kernels.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/kernels.cc -------------------------------------------------------------------------------- /tensorflow/c/kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/kernels.h -------------------------------------------------------------------------------- /tensorflow/c/kernels/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/kernels/BUILD -------------------------------------------------------------------------------- /tensorflow/c/kernels_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/kernels_test.cc -------------------------------------------------------------------------------- /tensorflow/c/logging.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/logging.cc -------------------------------------------------------------------------------- /tensorflow/c/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/logging.h -------------------------------------------------------------------------------- /tensorflow/c/ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/ops.cc -------------------------------------------------------------------------------- /tensorflow/c/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/ops.h -------------------------------------------------------------------------------- /tensorflow/c/ops_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/ops_test.cc -------------------------------------------------------------------------------- /tensorflow/c/python_api.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/python_api.cc -------------------------------------------------------------------------------- /tensorflow/c/python_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/python_api.h -------------------------------------------------------------------------------- /tensorflow/c/safe_ptr.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/safe_ptr.cc -------------------------------------------------------------------------------- /tensorflow/c/safe_ptr.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/safe_ptr.h -------------------------------------------------------------------------------- /tensorflow/c/tensor_interface.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tensor_interface.h -------------------------------------------------------------------------------- /tensorflow/c/test_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/test_op.cc -------------------------------------------------------------------------------- /tensorflow/c/test_op1.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/test_op1.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_attrtype.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_attrtype.h -------------------------------------------------------------------------------- /tensorflow/c/tf_buffer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_buffer.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_buffer.h -------------------------------------------------------------------------------- /tensorflow/c/tf_datatype.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_datatype.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_datatype.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_datatype.h -------------------------------------------------------------------------------- /tensorflow/c/tf_shape.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_shape.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_shape.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_shape.h -------------------------------------------------------------------------------- /tensorflow/c/tf_status.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_status.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_status.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_status.h -------------------------------------------------------------------------------- /tensorflow/c/tf_tensor.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_tensor.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_tensor.h -------------------------------------------------------------------------------- /tensorflow/c/tf_tstring.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_tstring.cc -------------------------------------------------------------------------------- /tensorflow/c/tf_tstring.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/c/tf_tstring.h -------------------------------------------------------------------------------- /tensorflow/cc/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/BUILD -------------------------------------------------------------------------------- /tensorflow/cc/framework/ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/framework/ops.cc -------------------------------------------------------------------------------- /tensorflow/cc/framework/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/framework/ops.h -------------------------------------------------------------------------------- /tensorflow/cc/ops/const_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/ops/const_op.cc -------------------------------------------------------------------------------- /tensorflow/cc/ops/const_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/ops/const_op.h -------------------------------------------------------------------------------- /tensorflow/cc/ops/while_loop.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/ops/while_loop.h -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/README.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/AssetModule/assets/test_asset.txt: -------------------------------------------------------------------------------- 1 | TEST ASSET FILE CONTENTS 2 | -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/fuzz_generated/bad_node_attr/assets/empty: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/fuzz_generated/const_with_no_value/assets/empty: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/fuzz_generated/negative_shape/assets/empty: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two/00000123/assets/foo.txt: -------------------------------------------------------------------------------- 1 | asset-file-contents -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two/00000123/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- 1 | ?@@@ -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two_main_op/00000123/assets/foo.txt: -------------------------------------------------------------------------------- 1 | asset-file-contents -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two_main_op/00000123/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- 1 | ?@@@ -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/assets/foo.txt: -------------------------------------------------------------------------------- 1 | asset-file-contents -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- 1 | ?@@@ -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two_v2/00000123/assets/foo.txt: -------------------------------------------------------------------------------- 1 | asset-file-contents -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/half_plus_two_v2/00000123/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- 1 | ?@@@ -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/static_hashtable_asset.txt: -------------------------------------------------------------------------------- 1 | foo 2 | bar 3 | baz 4 | wombat 5 | -------------------------------------------------------------------------------- /tensorflow/cc/saved_model/testdata/test_asset.txt: -------------------------------------------------------------------------------- 1 | TEST ASSET FILE CONTENTS 2 | -------------------------------------------------------------------------------- /tensorflow/cc/tools/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/cc/tools/BUILD -------------------------------------------------------------------------------- /tensorflow/compiler/aot/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/compiler/aot/BUILD -------------------------------------------------------------------------------- /tensorflow/compiler/jit/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/compiler/jit/BUILD -------------------------------------------------------------------------------- /tensorflow/compiler/jit/defs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/compiler/jit/defs.h -------------------------------------------------------------------------------- /tensorflow/compiler/mlir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/compiler/mlir/BUILD -------------------------------------------------------------------------------- /tensorflow/compiler/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/core/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/BUILD -------------------------------------------------------------------------------- /tensorflow/core/api_def/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/api_def/BUILD -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchCholesky.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchCholesky" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchFFT.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchFFT" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchFFT2D.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchFFT2D" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchFFT3D.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchFFT3D" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchIFFT.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchIFFT" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchIFFT2D.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchIFFT2D" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchIFFT3D.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchIFFT3D" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchMatrixDiag.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchMatrixDiag" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchMatrixSolve.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchMatrixSolve" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_BatchSvd.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchSvd" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_CopyToMesh.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "CopyToMesh" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_CopyToMeshGrad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "CopyToMeshGrad" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_DTensorRestoreV2.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "DTensorRestoreV2" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_Erfinv.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Erfinv" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_Ndtri.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Ndtri" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_Relayout.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Relayout" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_RelayoutLike.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RelayoutLike" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_ResourceGatherNd.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ResourceGatherNd" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_SelectV2.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "SelectV2" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_Softplus.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Softplus" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArray.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArray" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArrayClose.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArrayClose" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArrayGrad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArrayGrad" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArrayPack.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArrayPack" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArrayRead.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArrayRead" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArraySize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArraySize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArraySplit.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArraySplit" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/base_api/api_def_TensorArrayWrite.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArrayWrite" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Abort.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Abort" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_All.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "All" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Any.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Any" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Assign.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Assign" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_AssignAdd.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "AssignAdd" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_AssignSub.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "AssignSub" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_AssignVariableOp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "AssignVariableOp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Barrier.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Barrier" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BarrierClose.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BarrierClose" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BarrierReadySize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BarrierReadySize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BarrierTakeMany.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BarrierTakeMany" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Batch.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Batch" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BatchFunction.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchFunction" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BatchToSpace.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BatchToSpace" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Bitcast.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Bitcast" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BroadcastArgs.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BroadcastArgs" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_BroadcastTo.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "BroadcastTo" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Bucketize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Bucketize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ClipByValue.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ClipByValue" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ConcatOffset.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ConcatOffset" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ConsumeMutexLock.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ConsumeMutexLock" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ControlTrigger.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ControlTrigger" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_CountUpTo.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "CountUpTo" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_DeepCopy.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "DeepCopy" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_DynamicPartition.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "DynamicPartition" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_DynamicStitch.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "DynamicStitch" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_EditDistance.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "EditDistance" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Empty.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Empty" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_EmptyTensorList.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "EmptyTensorList" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_EncodeProto.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "EncodeProto" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_EnsureShape.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "EnsureShape" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Enter.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Enter" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Exit.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Exit" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ExpandDims.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ExpandDims" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_FakeParam.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "FakeParam" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Fill.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Fill" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_For.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "For" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_GetSessionTensor.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "GetSessionTensor" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_GuaranteeConst.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "GuaranteeConst" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_HostConst.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "HostConst" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Identity.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Identity" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_IdentityN.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "IdentityN" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_If.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "If" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ImmutableConst.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ImmutableConst" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_InplaceAdd.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "InplaceAdd" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_InplaceSub.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "InplaceSub" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_InplaceUpdate.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "InplaceUpdate" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_LoopCond.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "LoopCond" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_LowerBound.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "LowerBound" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapClear.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapClear" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapDefun.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapDefun" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapPeek.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapPeek" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapSize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapSize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapStage.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapStage" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapUnstage.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapUnstage" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MapUnstageNoKey.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MapUnstageNoKey" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Max.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Max" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Merge.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Merge" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Min.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Min" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MirrorPad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MirrorPad" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MirrorPadGrad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MirrorPadGrad" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_MutexLock.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "MutexLock" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_NcclAllReduce.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "NcclAllReduce" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_NcclBroadcast.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "NcclBroadcast" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_NcclReduce.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "NcclReduce" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_NextIteration.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "NextIteration" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_NoOp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "NoOp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_OneHot.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "OneHot" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_OnesLike.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "OnesLike" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_OrderedMapClear.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "OrderedMapClear" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_OrderedMapPeek.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "OrderedMapPeek" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_OrderedMapSize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "OrderedMapSize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_OrderedMapStage.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "OrderedMapStage" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Pack.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Pack" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Pad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Pad" 3 | visibility: SKIP 4 | } 5 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ParallelConcat.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ParallelConcat" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_PartitionedCall.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "PartitionedCall" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Placeholder.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Placeholder" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Prod.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Prod" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_QuantizedConcat.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "QuantizedConcat" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_QuantizedReshape.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "QuantizedReshape" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RaggedGather.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RaggedGather" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RaggedRange.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RaggedRange" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Range.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Range" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Rank.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Rank" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ReadVariableOp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ReadVariableOp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ReduceDataset.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ReduceDataset" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefEnter.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefEnter" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefExit.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefExit" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefIdentity.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefIdentity" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefMerge.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefMerge" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefNextIteration.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefNextIteration" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefSelect.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefSelect" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RefSwitch.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RefSwitch" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_RemoteCall.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "RemoteCall" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ResourceGather.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ResourceGather" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ReverseSequence.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ReverseSequence" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Roll.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Roll" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Save.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Save" 3 | visibility: SKIP 4 | } 5 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterAdd.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterAdd" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterDiv.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterDiv" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterMax.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterMax" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterMin.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterMin" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterMul.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterMul" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterNdAdd.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterNdAdd" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterNdSub.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterNdSub" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterNdUpdate.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterNdUpdate" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterSub.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterSub" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ScatterUpdate.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ScatterUpdate" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Select.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Select" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_SetSize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "SetSize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Shape.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Shape" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ShapeN.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ShapeN" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Size.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Size" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Skipgram.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Skipgram" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Slice.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Slice" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Snapshot.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Snapshot" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Split.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Split" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_SplitV.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "SplitV" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Squeeze.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Squeeze" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Stage.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Stage" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StageClear.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StageClear" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StagePeek.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StagePeek" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StageSize.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StageSize" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StatelessIf.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StatelessIf" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StatelessWhile.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StatelessWhile" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StopGradient.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StopGradient" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StridedSlice.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StridedSlice" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_StridedSliceGrad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "StridedSliceGrad" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Sum.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Sum" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_TensorArrayPack.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorArrayPack" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_TensorListGather.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorListGather" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_TensorListLength.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorListLength" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_TensorListStack.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TensorListStack" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Timestamp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Timestamp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_TopK.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "TopK" 3 | visibility: SKIP 4 | } 5 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Unbatch.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Unbatch" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_UnbatchGrad.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "UnbatchGrad" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Unpack.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Unpack" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_UnravelIndex.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "UnravelIndex" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Unstage.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Unstage" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_UpperBound.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "UpperBound" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_VarHandleOp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "VarHandleOp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_VariableShape.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "VariableShape" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_Where.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "Where" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_While.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "While" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/java_api/api_def_ZerosLike.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | graph_op_name: "ZerosLike" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/python_api/api_def_For.pbtxt: -------------------------------------------------------------------------------- 1 | op { graph_op_name: "For" visibility: HIDDEN } 2 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/python_api/api_def_If.pbtxt: -------------------------------------------------------------------------------- 1 | op { graph_op_name: "If" visibility: HIDDEN } 2 | -------------------------------------------------------------------------------- /tensorflow/core/api_def/python_api/api_def_While.pbtxt: -------------------------------------------------------------------------------- 1 | op { graph_op_name: "While" visibility: HIDDEN } 2 | -------------------------------------------------------------------------------- /tensorflow/core/build_defs.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/build_defs.bzl -------------------------------------------------------------------------------- /tensorflow/core/config/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/config/BUILD -------------------------------------------------------------------------------- /tensorflow/core/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/core/config/flags.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/config/flags.h -------------------------------------------------------------------------------- /tensorflow/core/data/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/data/BUILD -------------------------------------------------------------------------------- /tensorflow/core/data/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/data/utils.cc -------------------------------------------------------------------------------- /tensorflow/core/data/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/data/utils.h -------------------------------------------------------------------------------- /tensorflow/core/debug/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/debug/BUILD -------------------------------------------------------------------------------- /tensorflow/core/debug/debug.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/debug/debug.cc -------------------------------------------------------------------------------- /tensorflow/core/example/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/example/BUILD -------------------------------------------------------------------------------- /tensorflow/core/framework/op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/framework/op.h -------------------------------------------------------------------------------- /tensorflow/core/graph/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/graph/BUILD -------------------------------------------------------------------------------- /tensorflow/core/graph/colors.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/graph/colors.h -------------------------------------------------------------------------------- /tensorflow/core/graph/graph.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/graph/graph.cc -------------------------------------------------------------------------------- /tensorflow/core/graph/graph.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/graph/graph.h -------------------------------------------------------------------------------- /tensorflow/core/graph/types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/graph/types.h -------------------------------------------------------------------------------- /tensorflow/core/grappler/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/grappler/BUILD -------------------------------------------------------------------------------- /tensorflow/core/grappler/inputs/testdata/test_file.txt: -------------------------------------------------------------------------------- 1 | Hello World 2 | -------------------------------------------------------------------------------- /tensorflow/core/ir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/BUILD -------------------------------------------------------------------------------- /tensorflow/core/ir/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/README.md -------------------------------------------------------------------------------- /tensorflow/core/ir/dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/dialect.h -------------------------------------------------------------------------------- /tensorflow/core/ir/dialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/dialect.td -------------------------------------------------------------------------------- /tensorflow/core/ir/importexport/tests/mlir_to_graphdef/empty.mlir: -------------------------------------------------------------------------------- 1 | // RUN: tfg-translate -mlir-to-graphdef %s 2 | 3 | -------------------------------------------------------------------------------- /tensorflow/core/ir/ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/ops.cc -------------------------------------------------------------------------------- /tensorflow/core/ir/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/ops.h -------------------------------------------------------------------------------- /tensorflow/core/ir/ops.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/ops.td -------------------------------------------------------------------------------- /tensorflow/core/ir/ops_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/ops_test.cc -------------------------------------------------------------------------------- /tensorflow/core/ir/tests/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/tests/BUILD -------------------------------------------------------------------------------- /tensorflow/core/ir/types/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/types/BUILD -------------------------------------------------------------------------------- /tensorflow/core/ir/utility.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/utility.cc -------------------------------------------------------------------------------- /tensorflow/core/ir/utility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ir/utility.h -------------------------------------------------------------------------------- /tensorflow/core/kernels/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/kernels/BUILD -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/decode_csv/61317d2434a1af0da4e8ae7cdbd213b23572f9a5: -------------------------------------------------------------------------------- 1 | a,,b,,, 2 | a&x",d, 3 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/scatter_nd/5b0e5f8d2990c3cac80fa792ba141c43: -------------------------------------------------------------------------------- 1 |  2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/scatter_nd/5b61fa3a30dd267828f12d9ea2b2a191: -------------------------------------------------------------------------------- 1 |  2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/scatter_nd/8bc8b7d8beb3483c48158739791e56b0: -------------------------------------------------------------------------------- 1 |  2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/scatter_nd/d2ef31d47578e9de8323bb0e4806f1be: -------------------------------------------------------------------------------- 1 |  2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/scatter_nd/e2791edcf2c8d9f4af3678a75d43a3e4: -------------------------------------------------------------------------------- 1 |  2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/2db83ea58639b6d7d585fa12e3947a82: -------------------------------------------------------------------------------- 1 | 6.023e+23 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/36b4a931886b941dc41180050d12ca94: -------------------------------------------------------------------------------- 1 | 6.023e-23 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/50a2fabfdd276f573ff97ace8b11c5f4: -------------------------------------------------------------------------------- 1 | 42 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/62edb2a1eee34b001652cd86584becf2: -------------------------------------------------------------------------------- 1 | 0xabcdef 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/90013d1ec28c46a5c00574e60c70b6fc: -------------------------------------------------------------------------------- 1 | 3.14159265359 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/94f3e3cee6957ce5815326d6788c85f4: -------------------------------------------------------------------------------- 1 | 0.69314718056 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/96f547bc04bb913da0bc08915238ebd8: -------------------------------------------------------------------------------- 1 | 6.023e23 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/d3a903d18fc11e1f35c572ad4da690ed: -------------------------------------------------------------------------------- 1 | 1.61803 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/e3b629c92af44260c189deb32d6f06f3: -------------------------------------------------------------------------------- 1 | -42 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/fuzzing/corpus/string_to_number/f03eecf3bcfe4967a1888156a3115c8d: -------------------------------------------------------------------------------- 1 | 6.023E+23 2 | -------------------------------------------------------------------------------- /tensorflow/core/kernels/loss.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/kernels/loss.h -------------------------------------------------------------------------------- /tensorflow/core/kernels/mfcc.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/kernels/mfcc.h -------------------------------------------------------------------------------- /tensorflow/core/lib/bmp/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/bmp/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/core/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/core/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/db/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/db/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/gif/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/gif/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/gtl/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/gtl/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/hash/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/hash/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/io/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/io/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/io/block.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/io/block.h -------------------------------------------------------------------------------- /tensorflow/core/lib/io/cache.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/io/cache.h -------------------------------------------------------------------------------- /tensorflow/core/lib/io/path.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/io/path.h -------------------------------------------------------------------------------- /tensorflow/core/lib/io/table.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/io/table.h -------------------------------------------------------------------------------- /tensorflow/core/lib/jpeg/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/jpeg/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/math/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/math/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/png/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/png/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/psnr/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/psnr/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/ssim/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/ssim/BUILD -------------------------------------------------------------------------------- /tensorflow/core/lib/wav/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/lib/wav/BUILD -------------------------------------------------------------------------------- /tensorflow/core/nccl/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/nccl/BUILD -------------------------------------------------------------------------------- /tensorflow/core/ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/BUILD -------------------------------------------------------------------------------- /tensorflow/core/ops/compat/ops_history_v1/ControlTrigger.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | name: "ControlTrigger" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/ops/compat/ops_history_v1/NoOp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | name: "NoOp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/ops/compat/ops_history_v2/ControlTrigger.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | name: "ControlTrigger" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/ops/compat/ops_history_v2/NoOp.pbtxt: -------------------------------------------------------------------------------- 1 | op { 2 | name: "NoOp" 3 | } 4 | -------------------------------------------------------------------------------- /tensorflow/core/ops/ctc_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/ctc_ops.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/io_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/io_ops.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/map_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/map_ops.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/nn_grad.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/nn_grad.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/nn_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/nn_ops.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/no_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/no_op.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/ops.pbtxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/ops.pbtxt -------------------------------------------------------------------------------- /tensorflow/core/ops/rnn_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/rnn_ops.cc -------------------------------------------------------------------------------- /tensorflow/core/ops/set_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/ops/set_ops.cc -------------------------------------------------------------------------------- /tensorflow/core/platform/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/BUILD -------------------------------------------------------------------------------- /tensorflow/core/platform/abi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/abi.h -------------------------------------------------------------------------------- /tensorflow/core/platform/env.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/env.h -------------------------------------------------------------------------------- /tensorflow/core/platform/gif.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/gif.h -------------------------------------------------------------------------------- /tensorflow/core/platform/mem.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/mem.h -------------------------------------------------------------------------------- /tensorflow/core/platform/net.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/net.h -------------------------------------------------------------------------------- /tensorflow/core/platform/png.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/platform/png.h -------------------------------------------------------------------------------- /tensorflow/core/profiler/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/profiler/BUILD -------------------------------------------------------------------------------- /tensorflow/core/protobuf/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/protobuf/BUILD -------------------------------------------------------------------------------- /tensorflow/core/public/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/public/BUILD -------------------------------------------------------------------------------- /tensorflow/core/summary/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/summary/BUILD -------------------------------------------------------------------------------- /tensorflow/core/tfrt/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/tfrt/BUILD -------------------------------------------------------------------------------- /tensorflow/core/tfrt/mla/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/tfrt/mla/BUILD -------------------------------------------------------------------------------- /tensorflow/core/tfrt/ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/tfrt/ops/BUILD -------------------------------------------------------------------------------- /tensorflow/core/tpu/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/tpu/BUILD -------------------------------------------------------------------------------- /tensorflow/core/tpu/ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/tpu/ops/BUILD -------------------------------------------------------------------------------- /tensorflow/core/tpu/tpu_defs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/tpu/tpu_defs.h -------------------------------------------------------------------------------- /tensorflow/core/user_ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/user_ops/BUILD -------------------------------------------------------------------------------- /tensorflow/core/util/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/BUILD -------------------------------------------------------------------------------- /tensorflow/core/util/bcast.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/bcast.cc -------------------------------------------------------------------------------- /tensorflow/core/util/bcast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/bcast.h -------------------------------------------------------------------------------- /tensorflow/core/util/ctc/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/ctc/BUILD -------------------------------------------------------------------------------- /tensorflow/core/util/env_var.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/env_var.h -------------------------------------------------------------------------------- /tensorflow/core/util/padding.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/padding.h -------------------------------------------------------------------------------- /tensorflow/core/util/port.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/port.cc -------------------------------------------------------------------------------- /tensorflow/core/util/port.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/port.h -------------------------------------------------------------------------------- /tensorflow/core/util/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/util.cc -------------------------------------------------------------------------------- /tensorflow/core/util/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/core/util/util.h -------------------------------------------------------------------------------- /tensorflow/docs_src/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/docs_src/README.md -------------------------------------------------------------------------------- /tensorflow/dtensor/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/dtensor/BUILD -------------------------------------------------------------------------------- /tensorflow/dtensor/cc/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/dtensor/cc/BUILD -------------------------------------------------------------------------------- /tensorflow/dtensor/mlir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/dtensor/mlir/BUILD -------------------------------------------------------------------------------- /tensorflow/dtensor/mlir/dce.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/dtensor/mlir/dce.cc -------------------------------------------------------------------------------- /tensorflow/dtensor/proto/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/dtensor/proto/BUILD -------------------------------------------------------------------------------- /tensorflow/dtensor/tests/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/dtensor/tests/BUILD -------------------------------------------------------------------------------- /tensorflow/examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/examples/README.md -------------------------------------------------------------------------------- /tensorflow/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/examples/adding_an_op/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/go/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/BUILD -------------------------------------------------------------------------------- /tensorflow/go/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/README.md -------------------------------------------------------------------------------- /tensorflow/go/android.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/android.go -------------------------------------------------------------------------------- /tensorflow/go/attrs.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/attrs.go -------------------------------------------------------------------------------- /tensorflow/go/attrs_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/attrs_test.go -------------------------------------------------------------------------------- /tensorflow/go/context.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/context.go -------------------------------------------------------------------------------- /tensorflow/go/context_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/context_test.go -------------------------------------------------------------------------------- /tensorflow/go/doc.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/doc.go -------------------------------------------------------------------------------- /tensorflow/go/genop/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/genop/.gitignore -------------------------------------------------------------------------------- /tensorflow/go/genop/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/genop/BUILD -------------------------------------------------------------------------------- /tensorflow/go/genop/main.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/genop/main.go -------------------------------------------------------------------------------- /tensorflow/go/graph.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/graph.go -------------------------------------------------------------------------------- /tensorflow/go/graph_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/graph_test.go -------------------------------------------------------------------------------- /tensorflow/go/lib.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/lib.go -------------------------------------------------------------------------------- /tensorflow/go/op/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/BUILD -------------------------------------------------------------------------------- /tensorflow/go/op/generate.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/generate.go -------------------------------------------------------------------------------- /tensorflow/go/op/gradients.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/gradients.go -------------------------------------------------------------------------------- /tensorflow/go/op/op.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/op.go -------------------------------------------------------------------------------- /tensorflow/go/op/op_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/op_test.go -------------------------------------------------------------------------------- /tensorflow/go/op/scope.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/scope.go -------------------------------------------------------------------------------- /tensorflow/go/op/scope_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/scope_test.go -------------------------------------------------------------------------------- /tensorflow/go/op/wrappers.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/op/wrappers.go -------------------------------------------------------------------------------- /tensorflow/go/operation.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/operation.go -------------------------------------------------------------------------------- /tensorflow/go/saved_model.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/saved_model.go -------------------------------------------------------------------------------- /tensorflow/go/session.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/session.go -------------------------------------------------------------------------------- /tensorflow/go/session_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/session_test.go -------------------------------------------------------------------------------- /tensorflow/go/shape.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/shape.go -------------------------------------------------------------------------------- /tensorflow/go/shape_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/shape_test.go -------------------------------------------------------------------------------- /tensorflow/go/signature.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/signature.go -------------------------------------------------------------------------------- /tensorflow/go/status.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/status.go -------------------------------------------------------------------------------- /tensorflow/go/tensor.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/tensor.go -------------------------------------------------------------------------------- /tensorflow/go/tensor_handle.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/tensor_handle.go -------------------------------------------------------------------------------- /tensorflow/go/tensor_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/tensor_test.go -------------------------------------------------------------------------------- /tensorflow/go/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/test.sh -------------------------------------------------------------------------------- /tensorflow/go/testdata/saved_model/half_plus_two/00000123/assets/foo.txt: -------------------------------------------------------------------------------- 1 | asset-file-contents -------------------------------------------------------------------------------- /tensorflow/go/testdata/saved_model/half_plus_two/00000123/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- 1 | ?@@@ -------------------------------------------------------------------------------- /tensorflow/go/util_test.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/util_test.go -------------------------------------------------------------------------------- /tensorflow/go/version.go: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/go/version.go -------------------------------------------------------------------------------- /tensorflow/java/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/java/BUILD -------------------------------------------------------------------------------- /tensorflow/java/LEGACY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/java/LEGACY.md -------------------------------------------------------------------------------- /tensorflow/java/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/java/README.md -------------------------------------------------------------------------------- /tensorflow/java/build_defs.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/java/build_defs.bzl -------------------------------------------------------------------------------- /tensorflow/java/src/gen/resources/test.java.snippet: -------------------------------------------------------------------------------- 1 | // Here is a little snippet 2 | System.out.println("Hello!"); 3 | -------------------------------------------------------------------------------- /tensorflow/js/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/js/BUILD -------------------------------------------------------------------------------- /tensorflow/js/ops/ts_op_gen.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/js/ops/ts_op_gen.cc -------------------------------------------------------------------------------- /tensorflow/js/ops/ts_op_gen.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/js/ops/ts_op_gen.h -------------------------------------------------------------------------------- /tensorflow/lite/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/CMakeLists.txt -------------------------------------------------------------------------------- /tensorflow/lite/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/README.md -------------------------------------------------------------------------------- /tensorflow/lite/allocation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/allocation.cc -------------------------------------------------------------------------------- /tensorflow/lite/allocation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/allocation.h -------------------------------------------------------------------------------- /tensorflow/lite/array.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/array.cc -------------------------------------------------------------------------------- /tensorflow/lite/array.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/array.h -------------------------------------------------------------------------------- /tensorflow/lite/array_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/array_test.cc -------------------------------------------------------------------------------- /tensorflow/lite/async/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/async/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/async/c/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/async/c/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/async/c/task.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/async/c/task.h -------------------------------------------------------------------------------- /tensorflow/lite/build_def.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/build_def.bzl -------------------------------------------------------------------------------- /tensorflow/lite/builtin_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/builtin_ops.h -------------------------------------------------------------------------------- /tensorflow/lite/c/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/c/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/README.md -------------------------------------------------------------------------------- /tensorflow/lite/c/c_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/c_api.h -------------------------------------------------------------------------------- /tensorflow/lite/c/c_test.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/c_test.c -------------------------------------------------------------------------------- /tensorflow/lite/c/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/common.h -------------------------------------------------------------------------------- /tensorflow/lite/c/exported_symbols.lds: -------------------------------------------------------------------------------- 1 | _TfLite* 2 | -------------------------------------------------------------------------------- /tensorflow/lite/c/jni/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/jni/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/c/test_util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/test_util.cc -------------------------------------------------------------------------------- /tensorflow/lite/c/test_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/c/test_util.h -------------------------------------------------------------------------------- /tensorflow/lite/context.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/context.h -------------------------------------------------------------------------------- /tensorflow/lite/context_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/context_util.h -------------------------------------------------------------------------------- /tensorflow/lite/core/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/core/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/README.md -------------------------------------------------------------------------------- /tensorflow/lite/core/api/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/api/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/core/c/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/c/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/core/c/c_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/c/c_api.h -------------------------------------------------------------------------------- /tensorflow/lite/core/macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/macros.h -------------------------------------------------------------------------------- /tensorflow/lite/core/model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/core/model.h -------------------------------------------------------------------------------- /tensorflow/lite/delegates/flex/exported_symbols.lds: -------------------------------------------------------------------------------- 1 | *AcquireFlexDelegate* -------------------------------------------------------------------------------- /tensorflow/lite/examples/ios/camera/data/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/lite/generate-pc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/generate-pc.sh -------------------------------------------------------------------------------- /tensorflow/lite/graph_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/graph_info.cc -------------------------------------------------------------------------------- /tensorflow/lite/graph_info.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/graph_info.h -------------------------------------------------------------------------------- /tensorflow/lite/internal/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/internal/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/interpreter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/interpreter.h -------------------------------------------------------------------------------- /tensorflow/lite/ios/allowlist_TensorFlowLiteC.txt: -------------------------------------------------------------------------------- 1 | _TfLite* 2 | *AcquireFlexDelegate* 3 | -------------------------------------------------------------------------------- /tensorflow/lite/ios/ios.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/ios/ios.bzl -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/duplicate_names/foo.o: -------------------------------------------------------------------------------- 1 | first foo.o -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/duplicate_names/foo_1.o: -------------------------------------------------------------------------------- 1 | second foo.o -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/duplicate_names/foo_2.o: -------------------------------------------------------------------------------- 1 | third foo.o -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/extended_filename/long_file_name_with_extended_format.o: -------------------------------------------------------------------------------- 1 | long file name -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/extended_filename/short.o: -------------------------------------------------------------------------------- 1 | short file content -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/odd_bytes/even.o: -------------------------------------------------------------------------------- 1 | even bytes -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/odd_bytes/odd.o: -------------------------------------------------------------------------------- 1 | odd bytes -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/simple/bar.o: -------------------------------------------------------------------------------- 1 | bar file content -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/simple/foo.o: -------------------------------------------------------------------------------- 1 | foo file content -------------------------------------------------------------------------------- /tensorflow/lite/ios/testdata/skip_same_file/foo.o: -------------------------------------------------------------------------------- 1 | foo file content -------------------------------------------------------------------------------- /tensorflow/lite/java/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/java/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/java/demo/settings.gradle: -------------------------------------------------------------------------------- 1 | include ':app' 2 | -------------------------------------------------------------------------------- /tensorflow/lite/java/jni/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/java/jni/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/java/ovic/demo/settings.gradle: -------------------------------------------------------------------------------- 1 | include ':app' 2 | -------------------------------------------------------------------------------- /tensorflow/lite/java/src/testdata/invalid_model.bin: -------------------------------------------------------------------------------- 1 | This is an invalid model. -------------------------------------------------------------------------------- /tensorflow/lite/kernels/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/kernels/add.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/add.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/div.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/div.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/exp.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/exp.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/if.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/if.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/mul.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/mul.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/neg.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/neg.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/pad.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/pad.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/pow.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/pow.cc -------------------------------------------------------------------------------- /tensorflow/lite/kernels/sub.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/kernels/sub.cc -------------------------------------------------------------------------------- /tensorflow/lite/logger.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/logger.cc -------------------------------------------------------------------------------- /tensorflow/lite/logger.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/logger.h -------------------------------------------------------------------------------- /tensorflow/lite/model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/model.h -------------------------------------------------------------------------------- /tensorflow/lite/namespace.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/namespace.h -------------------------------------------------------------------------------- /tensorflow/lite/nnapi/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/nnapi/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/nnapi/sl/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/nnapi/sl/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/objc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/objc/README.md -------------------------------------------------------------------------------- /tensorflow/lite/op_resolver.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/op_resolver.h -------------------------------------------------------------------------------- /tensorflow/lite/python/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/python/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/python/lite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/python/lite.py -------------------------------------------------------------------------------- /tensorflow/lite/python/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/python/util.py -------------------------------------------------------------------------------- /tensorflow/lite/schema/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/schema/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/string_type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/string_type.h -------------------------------------------------------------------------------- /tensorflow/lite/string_util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/string_util.cc -------------------------------------------------------------------------------- /tensorflow/lite/string_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/string_util.h -------------------------------------------------------------------------------- /tensorflow/lite/test_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/test_util.h -------------------------------------------------------------------------------- /tensorflow/lite/testing/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/testing/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/testing/join.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/testing/join.h -------------------------------------------------------------------------------- /tensorflow/lite/testing/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/testing/util.h -------------------------------------------------------------------------------- /tensorflow/lite/toco/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/toco/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/toco/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/toco/README.md -------------------------------------------------------------------------------- /tensorflow/lite/toco/args.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/toco/args.cc -------------------------------------------------------------------------------- /tensorflow/lite/toco/args.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/toco/args.h -------------------------------------------------------------------------------- /tensorflow/lite/toco/logging/testdata/toco_tf_graph.dot: -------------------------------------------------------------------------------- 1 | digraph {a -> b} 2 | -------------------------------------------------------------------------------- /tensorflow/lite/toco/logging/testdata/toco_tflite_graph.dot: -------------------------------------------------------------------------------- 1 | digraph {a -> b} 2 | -------------------------------------------------------------------------------- /tensorflow/lite/toco/model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/toco/model.h -------------------------------------------------------------------------------- /tensorflow/lite/toco/toco.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/toco/toco.cc -------------------------------------------------------------------------------- /tensorflow/lite/tools/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/tools/BUILD -------------------------------------------------------------------------------- /tensorflow/lite/tools/benchmark/ios/TFLiteBenchmark/TFLiteBenchmark/Frameworks/.gitignore: -------------------------------------------------------------------------------- 1 | *.framework/ 2 | -------------------------------------------------------------------------------- /tensorflow/lite/tools/evaluation/testdata/empty.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/lite/tools/pip_package/MANIFEST.in: -------------------------------------------------------------------------------- 1 | recursive-include * *.py 2 | -------------------------------------------------------------------------------- /tensorflow/lite/tools/pip_package/debian/compat: -------------------------------------------------------------------------------- 1 | 9 2 | -------------------------------------------------------------------------------- /tensorflow/lite/tools/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/tools/utils.cc -------------------------------------------------------------------------------- /tensorflow/lite/tools/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/tools/utils.h -------------------------------------------------------------------------------- /tensorflow/lite/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/util.cc -------------------------------------------------------------------------------- /tensorflow/lite/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/util.h -------------------------------------------------------------------------------- /tensorflow/lite/util_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/util_test.cc -------------------------------------------------------------------------------- /tensorflow/lite/version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/lite/version.h -------------------------------------------------------------------------------- /tensorflow/py.default.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/py.default.bzl -------------------------------------------------------------------------------- /tensorflow/python/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/BUILD -------------------------------------------------------------------------------- /tensorflow/python/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/__init__.py -------------------------------------------------------------------------------- /tensorflow/python/autograph/pyct/common_transformers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/client/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/client/BUILD -------------------------------------------------------------------------------- /tensorflow/python/client/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/compat/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/compat/BUILD -------------------------------------------------------------------------------- /tensorflow/python/compiler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/data/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/data/BUILD -------------------------------------------------------------------------------- /tensorflow/python/debug/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/debug/BUILD -------------------------------------------------------------------------------- /tensorflow/python/debug/cli/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/debug/lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/dlpack/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/dlpack/BUILD -------------------------------------------------------------------------------- /tensorflow/python/eager/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/eager/BUILD -------------------------------------------------------------------------------- /tensorflow/python/feature_column/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/feature_column/testdata/warriors_vocabulary.txt: -------------------------------------------------------------------------------- 1 | 30 2 | 35 3 | 11 4 | 23 5 | 22 6 | -------------------------------------------------------------------------------- /tensorflow/python/feature_column/testdata/wire_vocabulary.txt: -------------------------------------------------------------------------------- 1 | omar 2 | stringer 3 | marlo 4 | -------------------------------------------------------------------------------- /tensorflow/python/framework/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/keras/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/keras/BUILD -------------------------------------------------------------------------------- /tensorflow/python/keras/engine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/keras/legacy_tf_layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/keras/saving/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/keras/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/kernel_tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/layers/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/layers/BUILD -------------------------------------------------------------------------------- /tensorflow/python/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/lib/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/lib/BUILD -------------------------------------------------------------------------------- /tensorflow/python/lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/lib/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/lib/io/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/lib/io/BUILD -------------------------------------------------------------------------------- /tensorflow/python/lib/io/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/module/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/module/BUILD -------------------------------------------------------------------------------- /tensorflow/python/ops/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/ops/BUILD -------------------------------------------------------------------------------- /tensorflow/python/ops/cond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/ops/cond.py -------------------------------------------------------------------------------- /tensorflow/python/ops/linalg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/ops/linalg/sparse/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/ops/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/ops/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/ops/nn.py -------------------------------------------------------------------------------- /tensorflow/python/ops/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/ops/rnn.py -------------------------------------------------------------------------------- /tensorflow/python/ops/sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/ops/sets.py -------------------------------------------------------------------------------- /tensorflow/python/ops/structured/__init__.py: -------------------------------------------------------------------------------- 1 | """Structured Tensors.""" 2 | 3 | -------------------------------------------------------------------------------- /tensorflow/python/platform/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/profiler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/summary/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/tf2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/tf2.py -------------------------------------------------------------------------------- /tensorflow/python/tools/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/tools/BUILD -------------------------------------------------------------------------------- /tensorflow/python/tpu/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/tpu/BUILD -------------------------------------------------------------------------------- /tensorflow/python/tpu/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/tpu/api.py -------------------------------------------------------------------------------- /tensorflow/python/tpu/tpu.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/tpu/tpu.bzl -------------------------------------------------------------------------------- /tensorflow/python/tpu/tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/tpu/tpu.py -------------------------------------------------------------------------------- /tensorflow/python/trackable/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/types/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/types/BUILD -------------------------------------------------------------------------------- /tensorflow/python/user_ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/util/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/util/BUILD -------------------------------------------------------------------------------- /tensorflow/python/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/util/nest.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/util/nest.cc -------------------------------------------------------------------------------- /tensorflow/python/util/nest.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/util/nest.h -------------------------------------------------------------------------------- /tensorflow/python/util/nest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/util/nest.py -------------------------------------------------------------------------------- /tensorflow/python/util/protobuf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/python/util/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/util/util.cc -------------------------------------------------------------------------------- /tensorflow/python/util/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/python/util/util.h -------------------------------------------------------------------------------- /tensorflow/pytype.default.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/pytype.default.bzl -------------------------------------------------------------------------------- /tensorflow/security/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/security/README.md -------------------------------------------------------------------------------- /tensorflow/strict.default.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/strict.default.bzl -------------------------------------------------------------------------------- /tensorflow/tensorflow.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tensorflow.bzl -------------------------------------------------------------------------------- /tensorflow/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/android/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/android/test/jni/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/android/test/settings.gradle: -------------------------------------------------------------------------------- 1 | rootProject.name = "TensorFlow sample" 2 | -------------------------------------------------------------------------------- /tensorflow/tools/api/lib/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/api/lib/BUILD -------------------------------------------------------------------------------- /tensorflow/tools/ci_build/pep8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/ci_build/pep8 -------------------------------------------------------------------------------- /tensorflow/tools/common/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/common/BUILD -------------------------------------------------------------------------------- /tensorflow/tools/docs/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/docs/BUILD -------------------------------------------------------------------------------- /tensorflow/tools/git/.gitignore: -------------------------------------------------------------------------------- 1 | gen 2 | -------------------------------------------------------------------------------- /tensorflow/tools/git/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/git/BUILD -------------------------------------------------------------------------------- /tensorflow/tools/mlpbtxt/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/mlpbtxt/BUILD -------------------------------------------------------------------------------- /tensorflow/tools/pip_package/README: -------------------------------------------------------------------------------- 1 | TensorFlow 2 | -------------------------------------------------------------------------------- /tensorflow/tools/proto_text/placeholder.txt: -------------------------------------------------------------------------------- 1 | Contents are unused. See gen_proto_functions.cc for details. 2 | -------------------------------------------------------------------------------- /tensorflow/tools/tensorflow_builder/config_detector/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/tools/test/BUILD -------------------------------------------------------------------------------- /tensorflow/tools/tf_sig_build_dockerfiles/devel.usertools/nvidia.bazelrc: -------------------------------------------------------------------------------- 1 | gpu.bazelrc -------------------------------------------------------------------------------- /tensorflow/tools/toolchains/cpus/aarch64/crosstool/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/toolchains/embedded/arm-linux/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/toolchains/python/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/toolchains/remote/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/toolchains/remote/BUILD.tpl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/tools/toolchains/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tensorflow/workspace0.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/workspace0.bzl -------------------------------------------------------------------------------- /tensorflow/workspace1.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/workspace1.bzl -------------------------------------------------------------------------------- /tensorflow/workspace2.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/workspace2.bzl -------------------------------------------------------------------------------- /tensorflow/workspace3.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tensorflow/workspace3.bzl -------------------------------------------------------------------------------- /third_party/BUILD: -------------------------------------------------------------------------------- 1 | licenses(["notice"]) # Apache 2.0 2 | -------------------------------------------------------------------------------- /third_party/FP16/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/FP16/BUILD -------------------------------------------------------------------------------- /third_party/FP16/FP16.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/FP16/FP16.BUILD -------------------------------------------------------------------------------- /third_party/FP16/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/FP16/workspace.bzl -------------------------------------------------------------------------------- /third_party/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/absl/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/absl/BUILD -------------------------------------------------------------------------------- /third_party/absl/system.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/absl/system.BUILD -------------------------------------------------------------------------------- /third_party/absl/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/absl/workspace.bzl -------------------------------------------------------------------------------- /third_party/android/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/android/android_configure.BUILD.tpl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/benchmark/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/benchmark/BUILD -------------------------------------------------------------------------------- /third_party/boringssl/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/clang_toolchain/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/codegen.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/codegen.BUILD -------------------------------------------------------------------------------- /third_party/coremltools.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/coremltools.BUILD -------------------------------------------------------------------------------- /third_party/cub.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/cub.BUILD -------------------------------------------------------------------------------- /third_party/curl.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/curl.BUILD -------------------------------------------------------------------------------- /third_party/cython.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/cython.BUILD -------------------------------------------------------------------------------- /third_party/dlpack/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/dlpack/BUILD -------------------------------------------------------------------------------- /third_party/ducc/BUILD: -------------------------------------------------------------------------------- 1 | # DUCC FFT library (https://gitlab.mpcdf.mpg.de/mtr/ducc). 2 | -------------------------------------------------------------------------------- /third_party/ducc/ducc.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ducc/ducc.BUILD -------------------------------------------------------------------------------- /third_party/ducc/fft.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ducc/fft.cc -------------------------------------------------------------------------------- /third_party/ducc/fft.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ducc/fft.h -------------------------------------------------------------------------------- /third_party/ducc/threading.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ducc/threading.cc -------------------------------------------------------------------------------- /third_party/ducc/threading.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ducc/threading.h -------------------------------------------------------------------------------- /third_party/ducc/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ducc/workspace.bzl -------------------------------------------------------------------------------- /third_party/eigen3/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/eigen3/BUILD -------------------------------------------------------------------------------- /third_party/eigen3/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/eigen3/LICENSE -------------------------------------------------------------------------------- /third_party/farmhash/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/farmhash/BUILD -------------------------------------------------------------------------------- /third_party/fft2d/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/fft2d/BUILD -------------------------------------------------------------------------------- /third_party/fft2d/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/fft2d/LICENSE -------------------------------------------------------------------------------- /third_party/fft2d/fft.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/fft2d/fft.h -------------------------------------------------------------------------------- /third_party/fft2d/fft2d.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/fft2d/fft2d.BUILD -------------------------------------------------------------------------------- /third_party/fft2d/fft2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/fft2d/fft2d.h -------------------------------------------------------------------------------- /third_party/flatbuffers/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/flatbuffers/BUILD -------------------------------------------------------------------------------- /third_party/gemmlowp/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/gemmlowp/BUILD -------------------------------------------------------------------------------- /third_party/gif.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/gif.BUILD -------------------------------------------------------------------------------- /third_party/git/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/git/BUILD.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/git/BUILD.tpl -------------------------------------------------------------------------------- /third_party/googleapis/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/googleapis/BUILD -------------------------------------------------------------------------------- /third_party/gpus/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/gpus/crosstool/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/gpus/cuda/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/gpus/cuda/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/gpus/cuda/LICENSE -------------------------------------------------------------------------------- /third_party/gpus/rocm/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/grpc/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/hexagon/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/hexagon/BUILD -------------------------------------------------------------------------------- /third_party/highwayhash/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/highwayhash/BUILD -------------------------------------------------------------------------------- /third_party/hwloc/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/hwloc/BUILD -------------------------------------------------------------------------------- /third_party/hwloc/BUILD.system: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/hwloc/BUILD.system -------------------------------------------------------------------------------- /third_party/hwloc/hwloc.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/hwloc/hwloc.BUILD -------------------------------------------------------------------------------- /third_party/icu/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/icu/BUILD -------------------------------------------------------------------------------- /third_party/icu/BUILD.system: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/icu/BUILD.system -------------------------------------------------------------------------------- /third_party/icu/data/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/icu/data/LICENSE -------------------------------------------------------------------------------- /third_party/icu/icu.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/icu/icu.BUILD -------------------------------------------------------------------------------- /third_party/icu/udata.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/icu/udata.patch -------------------------------------------------------------------------------- /third_party/icu/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/icu/workspace.bzl -------------------------------------------------------------------------------- /third_party/implib_so/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/implib_so/BUILD -------------------------------------------------------------------------------- /third_party/jpeg/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/jpeg/BUILD -------------------------------------------------------------------------------- /third_party/jpeg/BUILD.system: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/jpeg/BUILD.system -------------------------------------------------------------------------------- /third_party/jpeg/jpeg.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/jpeg/jpeg.BUILD -------------------------------------------------------------------------------- /third_party/jpeg/jpeg_helpers.BUILD.bazel: -------------------------------------------------------------------------------- 1 | licenses(["notice"]) 2 | -------------------------------------------------------------------------------- /third_party/jpeg/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/jpeg/workspace.bzl -------------------------------------------------------------------------------- /third_party/kissfft/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/kissfft/BUILD -------------------------------------------------------------------------------- /third_party/linenoise.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/linenoise.BUILD -------------------------------------------------------------------------------- /third_party/llvm/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm/BUILD -------------------------------------------------------------------------------- /third_party/llvm/build.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm/build.patch -------------------------------------------------------------------------------- /third_party/llvm/llvm.BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/llvm/run_lit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm/run_lit.sh -------------------------------------------------------------------------------- /third_party/llvm/setup.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm/setup.bzl -------------------------------------------------------------------------------- /third_party/llvm/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm/workspace.bzl -------------------------------------------------------------------------------- /third_party/llvm/zstd.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm/zstd.patch -------------------------------------------------------------------------------- /third_party/llvm_openmp/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/llvm_openmp/BUILD -------------------------------------------------------------------------------- /third_party/mkl/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/mkl/BUILD -------------------------------------------------------------------------------- /third_party/mkl/build_defs.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/mkl/build_defs.bzl -------------------------------------------------------------------------------- /third_party/mkl_dnn/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/mkl_dnn/BUILD -------------------------------------------------------------------------------- /third_party/mkl_dnn/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/mkl_dnn/LICENSE -------------------------------------------------------------------------------- /third_party/mpi/.gitignore: -------------------------------------------------------------------------------- 1 | *.h 2 | *.dylib 3 | *.so 4 | -------------------------------------------------------------------------------- /third_party/nasm/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nasm/BUILD -------------------------------------------------------------------------------- /third_party/nasm/BUILD.system: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nasm/BUILD.system -------------------------------------------------------------------------------- /third_party/nasm/config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nasm/config.h -------------------------------------------------------------------------------- /third_party/nasm/nasm.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nasm/nasm.BUILD -------------------------------------------------------------------------------- /third_party/nasm/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nasm/workspace.bzl -------------------------------------------------------------------------------- /third_party/nccl/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/nccl/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nccl/LICENSE -------------------------------------------------------------------------------- /third_party/nccl/archive.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nccl/archive.BUILD -------------------------------------------------------------------------------- /third_party/nccl/archive.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nccl/archive.patch -------------------------------------------------------------------------------- /third_party/net_zstd.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/net_zstd.BUILD -------------------------------------------------------------------------------- /third_party/nsync.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nsync.patch -------------------------------------------------------------------------------- /third_party/nvtx.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/nvtx.BUILD -------------------------------------------------------------------------------- /third_party/ortools/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/ortools/glpk.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ortools/glpk.BUILD -------------------------------------------------------------------------------- /third_party/ortools/scip.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ortools/scip.BUILD -------------------------------------------------------------------------------- /third_party/ortools/scip.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ortools/scip.patch -------------------------------------------------------------------------------- /third_party/pasta/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/pasta/BUILD -------------------------------------------------------------------------------- /third_party/pasta/BUILD.system: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/pasta/BUILD.system -------------------------------------------------------------------------------- /third_party/pasta/pasta.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/pasta/pasta.BUILD -------------------------------------------------------------------------------- /third_party/png.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/png.BUILD -------------------------------------------------------------------------------- /third_party/png_fix_rpi.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/png_fix_rpi.patch -------------------------------------------------------------------------------- /third_party/pprof.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/pprof.BUILD -------------------------------------------------------------------------------- /third_party/protobuf/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/BUILD.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/py/BUILD.tpl -------------------------------------------------------------------------------- /third_party/py/ml_dtypes/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/non_hermetic/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/non_hermetic/ml_dtypes/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/numpy/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/py/numpy/BUILD -------------------------------------------------------------------------------- /third_party/py/numpy/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/py/numpy/LICENSE -------------------------------------------------------------------------------- /third_party/py/numpy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/py/numpy/README.md -------------------------------------------------------------------------------- /third_party/pybind11.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/pybind11.BUILD -------------------------------------------------------------------------------- /third_party/pybind11_bazel/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/repo.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/repo.bzl -------------------------------------------------------------------------------- /third_party/riegeli_fix.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/riegeli_fix.patch -------------------------------------------------------------------------------- /third_party/ruy/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ruy/BUILD -------------------------------------------------------------------------------- /third_party/ruy/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/ruy/workspace.bzl -------------------------------------------------------------------------------- /third_party/six.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/six.BUILD -------------------------------------------------------------------------------- /third_party/snappy.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/snappy.BUILD -------------------------------------------------------------------------------- /third_party/sobol_data/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/sobol_data/BUILD -------------------------------------------------------------------------------- /third_party/sqlite.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/sqlite.BUILD -------------------------------------------------------------------------------- /third_party/stablehlo/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/systemlibs/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/systemlibs/BUILD.tpl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/systemlibs/protobuf_deps.bzl: -------------------------------------------------------------------------------- 1 | def protobuf_deps(): 2 | pass 3 | -------------------------------------------------------------------------------- /third_party/tensorrt/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/tensorrt/BUILD.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/tensorrt/BUILD.tpl -------------------------------------------------------------------------------- /third_party/tensorrt/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/tensorrt/LICENSE -------------------------------------------------------------------------------- /third_party/tf_runtime/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/triton/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/triton/BUILD -------------------------------------------------------------------------------- /third_party/xla/.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/.bazelrc -------------------------------------------------------------------------------- /third_party/xla/.bazelversion: -------------------------------------------------------------------------------- 1 | 6.4.0 2 | # NOTE: Update Bazel version in tensorflow/tools/ci_build/release/common.sh.oss -------------------------------------------------------------------------------- /third_party/xla/.clang-tidy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/.clang-tidy -------------------------------------------------------------------------------- /third_party/xla/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/.gitignore -------------------------------------------------------------------------------- /third_party/xla/.kokoro/windows/build.bat: -------------------------------------------------------------------------------- 1 | bash -l %0/../windows_build.sh %* 2 | exit /b %ERRORLEVEL% 3 | -------------------------------------------------------------------------------- /third_party/xla/BUILD.bazel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/BUILD.bazel -------------------------------------------------------------------------------- /third_party/xla/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/LICENSE -------------------------------------------------------------------------------- /third_party/xla/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/README.md -------------------------------------------------------------------------------- /third_party/xla/WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/WORKSPACE -------------------------------------------------------------------------------- /third_party/xla/build_tools/docker/context/.bazelversion: -------------------------------------------------------------------------------- 1 | 5.3.0 -------------------------------------------------------------------------------- /third_party/xla/build_tools/lint/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/configure: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/configure -------------------------------------------------------------------------------- /third_party/xla/configure.cmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/configure.cmd -------------------------------------------------------------------------------- /third_party/xla/configure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/configure.py -------------------------------------------------------------------------------- /third_party/xla/docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/docs/index.md -------------------------------------------------------------------------------- /third_party/xla/docs/shapes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/docs/shapes.md -------------------------------------------------------------------------------- /third_party/xla/third_party/BUILD: -------------------------------------------------------------------------------- 1 | licenses(["notice"]) # Apache 2.0 2 | -------------------------------------------------------------------------------- /third_party/xla/third_party/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/ortools/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/py/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/py/ml_dtypes/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/py/non_hermetic/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/py/non_hermetic/ml_dtypes/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/stablehlo/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/.kokoro/windows/cpu/build_cpu_py39.cfg: -------------------------------------------------------------------------------- 1 | build_file: "tsl/.kokoro/windows/build.bat" 2 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/BUILD: -------------------------------------------------------------------------------- 1 | licenses(["notice"]) # Apache 2.0 2 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/clang_toolchain/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/ducc/BUILD: -------------------------------------------------------------------------------- 1 | # DUCC FFT library (https://gitlab.mpcdf.mpg.de/mtr/ducc). 2 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/git/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/gpus/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/gpus/crosstool/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/gpus/cuda/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/gpus/rocm/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/grpc/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/jpeg/jpeg_helpers.BUILD.bazel: -------------------------------------------------------------------------------- 1 | licenses(["notice"]) 2 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/nccl/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/protobuf/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/py/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/py/ml_dtypes/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/py/non_hermetic/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/py/non_hermetic/ml_dtypes/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/pybind11_bazel/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/systemlibs/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/systemlibs/BUILD.tpl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/systemlibs/protobuf_deps.bzl: -------------------------------------------------------------------------------- 1 | def protobuf_deps(): 2 | pass 3 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/tensorrt/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/third_party/tf_runtime/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/tools/toolchains/cpus/aarch64/crosstool/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/tools/toolchains/embedded/arm-linux/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/tools/toolchains/python/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/tools/toolchains/remote/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/tools/toolchains/remote/BUILD.tpl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/third_party/tsl/tools/toolchains/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/tools/toolchains/cpus/aarch64/crosstool/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/tools/toolchains/embedded/arm-linux/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/tools/toolchains/python/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/tools/toolchains/remote/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/tools/toolchains/remote/BUILD.tpl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/tools/toolchains/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/workspace0.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/workspace0.bzl -------------------------------------------------------------------------------- /third_party/xla/workspace1.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/workspace1.bzl -------------------------------------------------------------------------------- /third_party/xla/workspace2.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/workspace2.bzl -------------------------------------------------------------------------------- /third_party/xla/workspace3.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/workspace3.bzl -------------------------------------------------------------------------------- /third_party/xla/workspace4.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/workspace4.bzl -------------------------------------------------------------------------------- /third_party/xla/xla/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/BUILD -------------------------------------------------------------------------------- /third_party/xla/xla/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/README.md -------------------------------------------------------------------------------- /third_party/xla/xla/array.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/array.cc -------------------------------------------------------------------------------- /third_party/xla/xla/array.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/array.h -------------------------------------------------------------------------------- /third_party/xla/xla/array2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/array2d.h -------------------------------------------------------------------------------- /third_party/xla/xla/array3d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/array3d.h -------------------------------------------------------------------------------- /third_party/xla/xla/array4d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/array4d.h -------------------------------------------------------------------------------- /third_party/xla/xla/bit_cast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/bit_cast.h -------------------------------------------------------------------------------- /third_party/xla/xla/c/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/c/BUILD -------------------------------------------------------------------------------- /third_party/xla/xla/experiments/triton_autotuning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/xla/ffi/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/ffi/BUILD -------------------------------------------------------------------------------- /third_party/xla/xla/ffi/ffi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/ffi/ffi.h -------------------------------------------------------------------------------- /third_party/xla/xla/layout.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/layout.cc -------------------------------------------------------------------------------- /third_party/xla/xla/layout.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/layout.h -------------------------------------------------------------------------------- /third_party/xla/xla/lazy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/lazy.h -------------------------------------------------------------------------------- /third_party/xla/xla/literal.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/literal.cc -------------------------------------------------------------------------------- /third_party/xla/xla/literal.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/literal.h -------------------------------------------------------------------------------- /third_party/xla/xla/map_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/map_util.h -------------------------------------------------------------------------------- /third_party/xla/xla/mlir_hlo/.clang-format: -------------------------------------------------------------------------------- 1 | BasedOnStyle: LLVM 2 | AlwaysBreakTemplateDeclarations: Yes 3 | -------------------------------------------------------------------------------- /third_party/xla/xla/mlir_hlo/build_tools/llvm_version.txt: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /third_party/xla/xla/pjrt/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/pjrt/BUILD -------------------------------------------------------------------------------- /third_party/xla/xla/printer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/printer.cc -------------------------------------------------------------------------------- /third_party/xla/xla/printer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/printer.h -------------------------------------------------------------------------------- /third_party/xla/xla/python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xla/xla/shape.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/shape.cc -------------------------------------------------------------------------------- /third_party/xla/xla/shape.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/shape.h -------------------------------------------------------------------------------- /third_party/xla/xla/status.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/status.h -------------------------------------------------------------------------------- /third_party/xla/xla/statusor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/statusor.h -------------------------------------------------------------------------------- /third_party/xla/xla/test.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/test.h -------------------------------------------------------------------------------- /third_party/xla/xla/types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/types.h -------------------------------------------------------------------------------- /third_party/xla/xla/util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/util.cc -------------------------------------------------------------------------------- /third_party/xla/xla/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/util.h -------------------------------------------------------------------------------- /third_party/xla/xla/xla.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/xla.bzl -------------------------------------------------------------------------------- /third_party/xla/xla/xla.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/xla/xla/xla.proto -------------------------------------------------------------------------------- /third_party/zlib.BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/third_party/zlib.BUILD -------------------------------------------------------------------------------- /tools/tf_env_collect.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tanveerpot/tensorflow/HEAD/tools/tf_env_collect.sh --------------------------------------------------------------------------------