| .. |
|
client
|
Enables per-host dummy args for TPUExecute (TF1) and adds XLA options.
|
2021-01-14 17:03:51 -08:00 |
|
experimental/xla_sharding
|
PY2 removal cleanup
|
2021-01-15 16:48:57 -08:00 |
|
g3doc
|
[XLA] Add AllGather documentation
|
2021-01-18 14:04:56 -08:00 |
|
pjrt
|
Internal changes only.
|
2021-01-15 18:51:23 -08:00 |
|
python
|
Move jax_jit functions to the jax namespace.
|
2021-01-16 15:12:36 -08:00 |
|
python_api
|
PY2 removal cleanup
|
2021-01-15 16:48:57 -08:00 |
|
rpc
|
Internal change
|
2020-10-02 20:33:42 -07:00 |
|
service
|
Support 0-sized shape with dynamic reshapes.
|
2021-01-16 10:23:25 -08:00 |
|
tests
|
[XLA] Refactor dot_operation_test's parameterization
|
2021-01-18 20:08:51 -08:00 |
|
tools
|
Internal changes only.
|
2021-01-14 19:09:44 -08:00 |
|
.clang-format
|
|
|
|
array2d_test.cc
|
|
|
|
array2d.h
|
|
|
|
array3d_test.cc
|
|
|
|
array3d.h
|
|
|
|
array4d_test.cc
|
|
|
|
array4d.h
|
|
|
|
array_test.cc
|
|
|
|
array.cc
|
|
|
|
array.h
|
Fix C++14 build.
|
2020-08-08 13:12:15 -07:00 |
|
bit_cast_test.cc
|
|
|
|
bit_cast.h
|
Move bfloat16 header to tensorflow/core/platform.
|
2020-08-14 13:44:57 -07:00 |
|
BUILD
|
Move llvm dependency from xla/util.h to xla/service/gpu/ir_emission_utils.h
|
2020-12-14 16:51:06 -08:00 |
|
comparison_util.cc
|
Adding total-order comparison support in proto and HloInstruction.
|
2020-08-10 09:40:29 -07:00 |
|
comparison_util.h
|
Adding total-order comparison support in proto and HloInstruction.
|
2020-08-10 09:40:29 -07:00 |
|
cpu_function_runtime.cc
|
|
|
|
cpu_function_runtime.h
|
[XLA:CPU] Reconsider alignment heuristics
|
2020-04-24 02:51:55 -07:00 |
|
debug_options_flags.cc
|
[XLA:GPU] Add --xla_gpu_deterministic_ops to force run-to-run determinism.
|
2020-12-15 12:51:36 -08:00 |
|
debug_options_flags.h
|
|
|
|
debug_options_parsers_test.cc
|
|
|
|
debug_options_parsers.h
|
|
|
|
device_util.h
|
|
|
|
error_spec.h
|
|
|
|
executable_run_options.cc
|
[NFC] Moved GlobalDeviceId into its own file as it is also used by the CPU runtime.
|
2020-11-24 04:21:38 -08:00 |
|
executable_run_options.h
|
[NFC] Moved GlobalDeviceId into its own file as it is also used by the CPU runtime.
|
2020-11-24 04:21:38 -08:00 |
|
execution_options_util.cc
|
|
|
|
execution_options_util.h
|
|
|
|
index_util_test.cc
|
|
|
|
index_util.cc
|
xla directory resolutions
|
2020-07-26 22:14:33 +00:00 |
|
index_util.h
|
|
|
|
iterator_util_test.cc
|
|
|
|
iterator_util.h
|
|
|
|
layout_test.cc
|
|
|
|
layout_util_test.cc
|
Add MakeAscendingLayout().
|
2021-01-08 08:21:45 -08:00 |
|
layout_util.cc
|
Add MakeAscendingLayout().
|
2021-01-08 08:21:45 -08:00 |
|
layout_util.h
|
Add MakeAscendingLayout().
|
2021-01-08 08:21:45 -08:00 |
|
layout.cc
|
This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor.
|
2020-04-06 17:55:23 -07:00 |
|
layout.h
|
|
|
|
literal_comparison.cc
|
fix typos in compiler dir
|
2020-10-29 01:08:30 +03:00 |
|
literal_comparison.h
|
|
|
|
literal_test.cc
|
[Resubmit] Dynamic literal support
|
2020-07-17 18:19:30 -07:00 |
|
literal_util.cc
|
xla directory resolutions
|
2020-07-26 22:14:33 +00:00 |
|
literal_util.h
|
|
|
|
literal.cc
|
Add several fixes for s390x architecture and other Big-Endian systems
|
2020-10-28 11:14:47 -04:00 |
|
literal.h
|
[Resubmit] Dynamic literal support
|
2020-07-17 18:19:30 -07:00 |
|
map_util.h
|
|
|
|
metric_table_report.cc
|
Merge pull request #41752 from tg-at-google:wsign-compare-semi-final-xla
|
2020-07-29 19:10:26 -07:00 |
|
metric_table_report.h
|
|
|
|
overflow_util.h
|
|
|
|
packed_literal_reader.cc
|
|
|
|
packed_literal_reader.h
|
|
|
|
parse_flags_from_env_test.cc
|
Crash if XLA flags are to be read from a file but we could not open the file
|
2020-12-30 14:23:16 -08:00 |
|
parse_flags_from_env.cc
|
Crash if XLA flags are to be read from a file but we could not open the file
|
2020-12-30 14:23:16 -08:00 |
|
parse_flags_from_env.h
|
|
|
|
primitive_util_test.cc
|
[XLA] Be more consistent about inferring mixed operand shapes
|
2021-01-12 13:29:04 -08:00 |
|
primitive_util.cc
|
[XLA] Be more consistent about inferring mixed operand shapes
|
2021-01-12 13:29:04 -08:00 |
|
primitive_util.h
|
[XLA] Be more consistent about inferring mixed operand shapes
|
2021-01-12 13:29:04 -08:00 |
|
protobuf_util.cc
|
|
|
|
protobuf_util.h
|
|
|
|
README.md
|
|
|
|
refcounting_hash_map_test.cc
|
Added RefCountingHashMap::GetOrTryCreateIfAbsent to allow for factory methods that may fail.
|
2020-11-26 06:10:04 -08:00 |
|
refcounting_hash_map.h
|
[XLA] Fix race condition in RefcountingHashMap that could lead to a use-after-free segfault.
|
2020-12-15 12:11:48 -08:00 |
|
reference_util_test.cc
|
|
|
|
reference_util.cc
|
[XLA] Add an optional preferred_element_type argument to Dot/Conv builder methods to enable generating HLOs that has wider accumulation type than default shape inference result.
|
2020-11-20 13:16:56 -08:00 |
|
reference_util.h
|
|
|
|
service_interface.h
|
|
|
|
shape_layout.cc
|
|
|
|
shape_layout.h
|
Introduce dynamic reshape op.
|
2020-08-14 15:48:19 -07:00 |
|
shape_test.cc
|
|
|
|
shape_tree_test.cc
|
Internal tests cleanup.
|
2020-10-28 07:40:36 -07:00 |
|
shape_tree.h
|
Split ShapeTreeIterator into two classes: ShapeTreeIterator and ShapeTreeLeafIterator
|
2020-09-15 14:29:58 -07:00 |
|
shape_util_test.cc
|
Speed up Shape creation by avoiding unnecessary validations.
|
2020-11-22 20:40:07 -08:00 |
|
shape_util.cc
|
Log error before failing
|
2021-01-11 17:59:46 -08:00 |
|
shape_util.h
|
[XLA] Be more consistent about inferring mixed operand shapes
|
2021-01-12 13:29:04 -08:00 |
|
shape.cc
|
|
|
|
shape.h
|
[libtpu] Reimplement XLA_Shape in C API for performance.
|
2020-10-19 14:17:22 -07:00 |
|
status_macros_test.cc
|
|
|
|
status_macros.cc
|
|
|
|
status_macros.h
|
|
|
|
status.h
|
|
|
|
statusor.h
|
|
|
|
test_helpers.h
|
|
|
|
test.h
|
|
|
|
text_literal_reader_test.cc
|
|
|
|
text_literal_reader.cc
|
|
|
|
text_literal_reader.h
|
|
|
|
text_literal_writer_test.cc
|
|
|
|
text_literal_writer.cc
|
|
|
|
text_literal_writer.h
|
|
|
|
types.h
|
|
|
|
union_find.h
|
XLA Parallel reduce.
|
2020-08-27 15:21:33 -07:00 |
|
util_test.cc
|
Support 0-sized shape with dynamic reshapes.
|
2021-01-16 10:23:25 -08:00 |
|
util.cc
|
Support 0-sized shape with dynamic reshapes.
|
2021-01-16 10:23:25 -08:00 |
|
util.h
|
Support 0-sized shape with dynamic reshapes.
|
2021-01-16 10:23:25 -08:00 |
|
window_util_test.cc
|
|
|
|
window_util.cc
|
|
|
|
window_util.h
|
|
|
|
xla_data.proto
|
Add OpMetadata list to OpShardingProto/HloSharding.
|
2021-01-12 12:56:24 -08:00 |
|
xla.bzl
|
|
|
|
xla.proto
|
Enables per-host dummy args for TPUExecute (TF1) and adds XLA options.
|
2021-01-14 17:03:51 -08:00 |
