experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla
History
George Karpenkov fbd744dfdb [XLA:GPU Docs] Clarify how to make XLA:GPU deterministic 
PiperOrigin-RevId: 361252684
Change-Id: I5ce077dd6e06a76d2230e4ef3f98385796a9620e
2021-03-05 17:09:56 -08:00
..
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
…
array2d.h
…
array3d_test.cc
…
array3d.h
…
array4d_test.cc
Change headers to directly include absl::Span, and clean up the build
2018-08-30 22:52:05 -07:00
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
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
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
…
cpu_function_runtime.h
…
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
Replace std::vector<std::initializer_list> with std::vector<std::vector>.
2018-11-08 03:19:36 -08:00
index_util.cc
…
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
Near comparison now works for non-fp types. Callback uses ShapeIndex parameter.
2019-06-05 16:01:58 -07:00
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
[XLA] Stop including str_util.h.
2018-08-23 21:33:13 -07:00
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
Split ShapeTreeIterator into two classes: ShapeTreeIterator and ShapeTreeLeafIterator
2020-09-15 14:29:58 -07:00
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
minor spelling tweaks
2019-12-09 18:21:12 +09:00
status_macros.cc
…
status_macros.h
…
status.h
…
statusor.h
…
test_helpers.h
…
test.h
…
text_literal_reader_test.cc
Global de-std::unique_ptr cleanup for xla::Literal.
2018-09-10 12:38:19 -07:00
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
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
Support empty window and 0D convolution. For non-GPUs it's already accidentally supported; for GPUs it's not hard to support anyway.
2020-01-13 14:51:47 -08:00
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

README.md

XLA (Accelerated Linear Algebra) is a domain-specific compiler for linear algebra that optimizes TensorFlow computations. See the documentation.