experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla
History
David Majnemer 4fcfcbe2c6 [XLA] Refactor dot_operation_test's parameterization 
PiperOrigin-RevId: 352483336
Change-Id: Ib5d3adf1e91a495fa7adb18d47ad99b94f99f9b4
2021-01-18 20:08:51 -08:00
..
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
Merge pull request #31745 from take-cheeze:patch-1
2019-08-19 15:45:46 -07:00
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

README.md

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