├── .gitignore ├── .gitmodules ├── CMakeLists.txt ├── README.md ├── bin ├── relu_model.cc ├── test_xla_client.cc └── tests.cc ├── onnx_xla ├── backend.cc ├── backend.h ├── backend_test.cc ├── backend_test.h ├── conv_pool_helper.cc ├── conv_pool_helper.h ├── onnx_xla_interface.cc ├── onnxifi_helper.cc ├── onnxifi_helper.h ├── operator_registry.cc ├── operator_registry.h ├── translations │ ├── translate_average_pool.cc │ ├── translate_batch_normalization.cc │ ├── translate_concat.cc │ ├── translate_conv.cc │ ├── translate_dropout.cc │ ├── translate_elem_bin_op.cc │ ├── translate_gemm.cc │ ├── translate_global_average_pool.cc │ ├── translate_lrn.cc │ ├── translate_max_pool.cc │ ├── translate_relu.cc │ ├── translate_reshape.cc │ ├── translate_softmax.cc │ ├── translate_sum.cc │ ├── translate_transpose.cc │ ├── translate_undefined.cc │ └── translate_unsqueeze.cc ├── utils.cc ├── utils.h ├── xla_client.cc └── xla_client.h ├── onnx_xla_test.py ├── python_onnxifi ├── data_conversion.cc ├── data_conversion.h ├── onnxifi_backend.py └── python_onnxifi.cc ├── scripts ├── build_xla.sh └── format.sh ├── setup.py ├── tensorflow.patch └── test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/.gitmodules -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/README.md -------------------------------------------------------------------------------- /bin/relu_model.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/bin/relu_model.cc -------------------------------------------------------------------------------- /bin/test_xla_client.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/bin/test_xla_client.cc -------------------------------------------------------------------------------- /bin/tests.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/bin/tests.cc -------------------------------------------------------------------------------- /onnx_xla/backend.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/backend.cc -------------------------------------------------------------------------------- /onnx_xla/backend.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/backend.h -------------------------------------------------------------------------------- /onnx_xla/backend_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/backend_test.cc -------------------------------------------------------------------------------- /onnx_xla/backend_test.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/backend_test.h -------------------------------------------------------------------------------- /onnx_xla/conv_pool_helper.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/conv_pool_helper.cc -------------------------------------------------------------------------------- /onnx_xla/conv_pool_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/conv_pool_helper.h -------------------------------------------------------------------------------- /onnx_xla/onnx_xla_interface.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/onnx_xla_interface.cc -------------------------------------------------------------------------------- /onnx_xla/onnxifi_helper.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/onnxifi_helper.cc -------------------------------------------------------------------------------- /onnx_xla/onnxifi_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/onnxifi_helper.h -------------------------------------------------------------------------------- /onnx_xla/operator_registry.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/operator_registry.cc -------------------------------------------------------------------------------- /onnx_xla/operator_registry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/operator_registry.h -------------------------------------------------------------------------------- /onnx_xla/translations/translate_average_pool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_average_pool.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_batch_normalization.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_batch_normalization.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_concat.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_concat.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_conv.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_conv.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_dropout.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_dropout.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_elem_bin_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_elem_bin_op.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_gemm.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_gemm.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_global_average_pool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_global_average_pool.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_lrn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_lrn.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_max_pool.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_max_pool.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_relu.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_relu.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_reshape.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_reshape.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_softmax.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_softmax.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_sum.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_sum.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_transpose.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_transpose.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_undefined.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_undefined.cc -------------------------------------------------------------------------------- /onnx_xla/translations/translate_unsqueeze.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/translations/translate_unsqueeze.cc -------------------------------------------------------------------------------- /onnx_xla/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/utils.cc -------------------------------------------------------------------------------- /onnx_xla/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/utils.h -------------------------------------------------------------------------------- /onnx_xla/xla_client.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/xla_client.cc -------------------------------------------------------------------------------- /onnx_xla/xla_client.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla/xla_client.h -------------------------------------------------------------------------------- /onnx_xla_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/onnx_xla_test.py -------------------------------------------------------------------------------- /python_onnxifi/data_conversion.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/python_onnxifi/data_conversion.cc -------------------------------------------------------------------------------- /python_onnxifi/data_conversion.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/python_onnxifi/data_conversion.h -------------------------------------------------------------------------------- /python_onnxifi/onnxifi_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/python_onnxifi/onnxifi_backend.py -------------------------------------------------------------------------------- /python_onnxifi/python_onnxifi.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/python_onnxifi/python_onnxifi.cc -------------------------------------------------------------------------------- /scripts/build_xla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/scripts/build_xla.sh -------------------------------------------------------------------------------- /scripts/format.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/scripts/format.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/setup.py -------------------------------------------------------------------------------- /tensorflow.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/tensorflow.patch -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/onnx/onnx-xla/HEAD/test.py --------------------------------------------------------------------------------