| .. |
|
client
|
Internal change
|
2021-03-05 13:51:17 -08:00 |
|
experimental/xla_sharding
|
Adds get_sharding_tile_shape() function.
|
2021-03-05 16:20:35 -08:00 |
|
g3doc
|
[XLA:GPU Docs] Clarify how to make XLA:GPU deterministic
|
2021-03-05 17:09:56 -08:00 |
|
pjrt
|
[PJRT] Expose TPU runtime version via new PjRtClient::platform_version string.
|
2021-03-05 13:45:22 -08:00 |
|
python
|
[PJRT] Expose TPU runtime version via new PjRtClient::platform_version string.
|
2021-03-05 13:45:22 -08:00 |
|
python_api
|
Add public visibility to target as nightly build is broken otherwise
|
2021-03-04 18:59:00 -08:00 |
|
rpc
|
Comment out a number of google-internal targets when copybara-exporting instead of removing them.
|
2021-01-26 05:47:31 -08:00 |
|
service
|
[XLA:GPU] [NFC] Clarify types in AllReduceCombiner
|
2021-03-05 15:16:41 -08:00 |
|
tests
|
Ensure that trivial types are passed by value rather than reference
|
2021-03-04 22:09:46 -08:00 |
|
tools
|
[XLA] Small ergonomic improvements to replay_computation.
|
2021-02-16 10:04:28 -08:00 |
|
.clang-format
|
|
|
|
array2d_test.cc
|
Enable half precision convolution for the CPU and GPU backends.
|
2018-02-15 10:42:55 -08:00 |
|
array2d.h
|
|
|
|
array3d_test.cc
|
|
|
|
array3d.h
|
|
|
|
array4d_test.cc
|
|
|
|
array4d.h
|
|
|
|
array_test.cc
|
|
|
|
array.cc
|
|
|
|
array.h
|
[XLA] Rework Array constructors a bit to allow creating a Array of unique_ptr<>
|
2021-02-22 16:23:41 -08:00 |
|
bit_cast_test.cc
|
|
|
|
bit_cast.h
|
|
|
|
BUILD
|
Internal change to build files.
|
2021-03-03 19:42:45 -08:00 |
|
comparison_util.cc
|
Fail if ToString of comparisons is called during initialization.
|
2021-01-22 22:24:44 -08:00 |
|
comparison_util.h
|
|
|
|
cpu_function_runtime.cc
|
[XLA] Move tensorflow/compiler/tf2xla/cpu_function_runtime.h to tensorflow/compiler/xla/cpu_function_runtime.h
|
2019-04-01 09:27:18 -07:00 |
|
cpu_function_runtime.h
|
[XLA:CPU] Reconsider alignment heuristics
|
2020-04-24 02:51:55 -07:00 |
|
debug_options_flags.cc
|
Merge pull request #47121 from nouiz:upstream-llvm_file
|
2021-02-26 02:44:59 -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
|
|
|
|
layout.h
|
[XLA] Break dependencies of XLA on TF.
|
2021-01-19 14:07:56 -08:00 |
|
literal_comparison.cc
|
fix typos in compiler dir
|
2020-10-29 01:08:30 +03:00 |
|
literal_comparison.h
|
|
|
|
literal_test.cc
|
|
|
|
literal_util.cc
|
|
|
|
literal_util.h
|
|
|
|
literal.cc
|
Add a literal field to custom call instruction.
|
2021-02-17 12:28:55 -08:00 |
|
literal.h
|
Add a method to update a literal using dynamic bounds if necessary.
|
2021-02-18 13:49:53 -08:00 |
|
map_util.h
|
|
|
|
metric_table_report.cc
|
|
|
|
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
|
|
|
|
permutation_util_test.cc
|
[XLA] Change xla::IsPermutation() not to check the size, and make its implementation O(n).
|
2021-02-09 19:13:04 -08:00 |
|
permutation_util.cc
|
[XLA] Change xla::IsPermutation() not to check the size, and make its implementation O(n).
|
2021-02-09 19:13:04 -08:00 |
|
permutation_util.h
|
[XLA] Invert the meaning of xla::Permute, add an inverse xla::PermuteInverse().
|
2021-02-10 12:21:04 -08:00 |
|
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
|
[XLA] Uniquify sharding metadata when propagated.
|
2021-01-27 10:21:28 -08:00 |
|
protobuf_util.h
|
[XLA] Uniquify sharding metadata when propagated.
|
2021-01-27 10:21:28 -08:00 |
|
README.md
|
|
|
|
refcounting_hash_map_test.cc
|
Ensure that trivial types are passed by value rather than reference
|
2021-03-04 22:09:46 -08:00 |
|
refcounting_hash_map.h
|
Store NcclCliques in new NcclCliqueMap.
|
2021-01-29 04:16:02 -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
|
|
|
|
shape_test.cc
|
Some shape benchmarks.
|
2021-02-02 10:46:30 -08:00 |
|
shape_tree_test.cc
|
Internal tests cleanup.
|
2020-10-28 07:40:36 -07:00 |
|
shape_tree.h
|
|
|
|
shape_util_test.cc
|
[XLA] Invert the permutation convention for xla::ShapeUtil::PermuteDimensions.
|
2021-02-10 13:20:44 -08:00 |
|
shape_util.cc
|
[XLA] Invert the permutation convention for xla::ShapeUtil::PermuteDimensions.
|
2021-02-10 13:20:44 -08:00 |
|
shape_util.h
|
[XLA] Invert the permutation convention for xla::ShapeUtil::PermuteDimensions.
|
2021-02-10 13:20:44 -08:00 |
|
shape.cc
|
Considering leaf rank in shape comparison
|
2021-01-12 11:22:46 +00:00 |
|
shape.h
|
relaxing requirements for clonewithnewoperands to preserve sharding information
|
2020-11-30 16:57:32 +00:00 |
|
side_effect_util.cc
|
[XLA] Break dependencies of XLA on TF.
|
2021-01-19 14:07:56 -08:00 |
|
side_effect_util.h
|
[XLA] Break dependencies of XLA on TF.
|
2021-01-19 14:07:56 -08:00 |
|
status_macros_test.cc
|
|
|
|
status_macros.cc
|
|
|
|
status_macros.h
|
|
|
|
status.h
|
[XLA] s/tensorflow::Status/Status/.
|
2018-05-11 16:07:35 -07:00 |
|
statusor.h
|
|
|
|
test_helpers.h
|
|
|
|
test.h
|
|
|
|
text_literal_reader_test.cc
|
|
|
|
text_literal_reader.cc
|
Change Shape parsing from regexp matcher to parser.
|
2018-12-14 08:37:21 -08:00 |
|
text_literal_reader.h
|
|
|
|
text_literal_writer_test.cc
|
|
|
|
text_literal_writer.cc
|
Remove unused inclusions and add missing inclusions
|
2019-02-05 14:06:22 -08:00 |
|
text_literal_writer.h
|
|
|
|
types.h
|
|
|
|
union_find.h
|
|
|
|
util_test.cc
|
Fix oob access in CommonFactors.
|
2021-03-04 21:33:38 -08:00 |
|
util.cc
|
Fix oob access in CommonFactors.
|
2021-03-04 21:33:38 -08:00 |
|
util.h
|
[XLA] Split permutation utilities from xla/util.* into a new xla/permutation_util.*
|
2021-02-09 17:01:49 -08:00 |
|
window_util_test.cc
|
|
|
|
window_util.cc
|
|
|
|
window_util.h
|
|
|
|
xla_data.proto
|
[XLA] Move HLO instruction OpMetadata dummy names to a new field.
|
2021-02-08 17:19:21 -08:00 |
|
xla.bzl
|
|
|
|
xla.proto
|
Updates weight broadcast option to go through different proto.
|
2021-02-27 01:55:04 -08:00 |
