| .. |
|
client
|
[XLA] [client] Implement a RunAsync overload which does not need a vector of shapes
|
2020-06-19 17:09:13 -07:00 |
|
experimental/xla_sharding
|
[XLA:SPMD] Add API for the experimental mechanism to mix auto and manual partitioning
|
2020-05-04 16:57:03 -07:00 |
|
g3doc
|
[XLA] Fix formatting of code snippets in the operation_semantics
|
2020-06-02 14:14:05 -07:00 |
|
pjrt
|
Allow a shape to be passed to CopyToHostAsync
|
2020-06-22 01:55:19 -07:00 |
|
python
|
[PJRT] Fix potential misuse of PjRtBuffer::FromHostBuffer.
|
2020-06-16 06:59:42 -07:00 |
|
python_api
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
rpc
|
Use tf_grpc_dependency in tensorflow/compiler/xla
|
2020-04-02 22:42:59 -07:00 |
|
service
|
Support dynamic outputs for XLA on demand ops.
|
2020-07-01 11:11:47 -07:00 |
|
tests
|
Unbreak the MSVC build
|
2020-06-19 21:15:11 -07:00 |
|
tools
|
Roll forward "Add a show_fusion_subcomputations command to interactive_graphviz" with fix
|
2020-05-27 11:27:39 -07:00 |
|
.clang-format
|
|
|
|
array2d_test.cc
|
|
|
|
array2d.h
|
[TF] [XLA] Fix a number of compiler warnings on Mac OS X.
|
2018-11-19 13:19:32 -08:00 |
|
array3d_test.cc
|
|
|
|
array3d.h
|
|
|
|
array4d_test.cc
|
|
|
|
array4d.h
|
Add a default constructor to Array4D
|
2020-02-07 15:23:01 -08:00 |
|
array_test.cc
|
|
|
|
array.cc
|
Add specialization of method Array::FillRandom() for complex64 type.
|
2019-12-13 16:56:06 -08:00 |
|
array.h
|
Add specialization of method Array::FillRandom() for complex64 type.
|
2019-12-13 16:56:06 -08:00 |
|
bit_cast_test.cc
|
Remove absl::bit_casts of non-trivially-copyable Eigen types, and replace them with a new, extensible xla::BitCast call. The latter does everything absl::bit_cast does, but additionally can be specialized to support non-trivial types.
|
2019-06-14 02:44:19 -07:00 |
|
bit_cast.h
|
|
|
|
BUILD
|
[XLA] Basic (R1) support for CPU bounded dynamic shapes.
|
2020-05-13 14:04:06 -07:00 |
|
comparison_util.cc
|
[XLA] Replace individual comparison HLO ops with a single compare op.
|
2019-03-07 14:05:30 -08:00 |
|
comparison_util.h
|
|
|
|
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
|
Enable XLA:CPU fast math for min/max by default to be similar to TF's behavior.
|
2020-06-17 01:27:08 -07:00 |
|
debug_options_flags.h
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
debug_options_parsers_test.cc
|
[ROCm] Fix for compile error in //tensorflow/compiler/xla:debug_options_parsers_test
|
2020-01-19 02:38:56 +00:00 |
|
debug_options_parsers.h
|
Remove ReducePrecisionInsertion pass.
|
2019-11-04 02:09:31 -08:00 |
|
device_util.h
|
|
|
|
error_spec.h
|
|
|
|
executable_run_options.cc
|
Add run_id to TraceMes.
|
2020-03-13 10:42:34 -07:00 |
|
executable_run_options.h
|
Enable XRT cache to be shared among multiple GPU devices.
|
2020-05-07 09:34:58 -07:00 |
|
execution_options_util.cc
|
|
|
|
execution_options_util.h
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
index_util_test.cc
|
|
|
|
index_util.cc
|
|
|
|
index_util.h
|
|
|
|
iterator_util_test.cc
|
|
|
|
iterator_util.h
|
fix C++ header guards.
|
2018-08-21 16:22:05 -07:00 |
|
layout_test.cc
|
[XLA] Remove unsupported sparse layout
|
2020-01-02 18:04:52 -08:00 |
|
layout_util_test.cc
|
[XLA] Remove unsupported sparse layout
|
2020-01-02 18:04:52 -08:00 |
|
layout_util.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_util.h
|
[XLA] Remove unsupported sparse layout
|
2020-01-02 18:04:52 -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
|
[XLA] Remove unsupported sparse layout
|
2020-01-02 18:04:52 -08:00 |
|
literal_comparison.cc
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
literal_comparison.h
|
|
|
|
literal_test.cc
|
Avoid dividing by zero in xla/literal_test.
|
2020-02-10 11:29:21 -08:00 |
|
literal_util.cc
|
Introduce a few new literal conversion functions for various floating point types
|
2020-01-27 07:47:30 -08:00 |
|
literal_util.h
|
Introduce a few new literal conversion functions for various floating point types
|
2020-01-27 07:47:30 -08:00 |
|
literal.cc
|
Internal TF change.
|
2020-05-21 18:49:44 -07:00 |
|
literal.h
|
Internal TF change.
|
2020-05-21 18:49:44 -07: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
|
Handle the cases where stdout may have CRLF suffix on windows in xla parse_flags_from_env_test
|
2020-03-12 14:33:13 -07:00 |
|
parse_flags_from_env.cc
|
[XLA] Use LOG(QFATAL) rather than LOG(FATAL) in XLA_FLAGS argument parsing.
|
2019-11-27 10:27:55 -08:00 |
|
parse_flags_from_env.h
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
primitive_util_test.cc
|
|
|
|
primitive_util.cc
|
[XLA] Rename OPAQUE to OPAQUE_TYPE.
|
2019-04-24 09:41:01 -07:00 |
|
primitive_util.h
|
|
|
|
protobuf_util.cc
|
|
|
|
protobuf_util.h
|
|
|
|
README.md
|
|
|
|
refcounting_hash_map_test.cc
|
[XLA:GPU] Add experimental, lightly tested support for multi-host and multi-process NCCL AllReduce.
|
2020-02-21 14:00:29 -08:00 |
|
refcounting_hash_map.h
|
[XLA:GPU] Add experimental, lightly tested support for multi-host and multi-process NCCL AllReduce.
|
2020-02-21 14:00:29 -08:00 |
|
reference_util_test.cc
|
|
|
|
reference_util.cc
|
[XLA] Do extensive testing of int32 matmuls
|
2019-09-16 08:52:24 -07:00 |
|
reference_util.h
|
[XLA] Do extensive testing of int32 matmuls
|
2019-09-16 08:52:24 -07:00 |
|
service_interface.h
|
[XLA] Split ExecuteGraph into Compile and Execute.
|
2018-11-07 19:20:01 -08:00 |
|
shape_layout.cc
|
[XLA] Implement dynamic input and output in DynamicPadder.
|
2019-12-12 10:31:20 -08:00 |
|
shape_layout.h
|
|
|
|
shape_test.cc
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
shape_tree_test.cc
|
|
|
|
shape_tree.h
|
|
|
|
shape_util_test.cc
|
[XLA] Add a few helper functions around dynamic dimensions.
|
2019-11-25 12:52:48 -08:00 |
|
shape_util.cc
|
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
|
2020-06-22 12:32:43 -07:00 |
|
shape_util.h
|
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
|
2020-06-22 12:32:43 -07:00 |
|
shape.cc
|
[xla::Shape::Shape] Emplace tuple shapes into tuple_shapes_.
|
2020-03-03 15:06:38 -08:00 |
|
shape.h
|
[XLA]Add HLO infrastructure to support dynamic op lowering.
|
2020-04-30 17:00:49 -07:00 |
|
status_macros_test.cc
|
minor spelling tweaks
|
2019-12-09 18:21:12 +09:00 |
|
status_macros.cc
|
|
|
|
status_macros.h
|
Multiple blocks of code in StreamExecutor can be simplified using the
|
2019-05-22 12:00:46 -07:00 |
|
status.h
|
|
|
|
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
|
Use Env::LocalTempFilename for a temp filename.
|
2020-02-19 17:30:20 -08:00 |
|
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
|
|
|
|
util_test.cc
|
[XLA] Make liberal use of inlined vectors to avoid memory allocation inside
|
2019-10-31 14:24:04 -07:00 |
|
util.cc
|
[XLA] Add a utility to transform dimensions from one shape's dimensions to another.
|
2020-03-31 02:34:20 -07:00 |
|
util.h
|
[XLA] Add a utility to transform dimensions from one shape's dimensions to another.
|
2020-03-31 02:34:20 -07:00 |
|
window_util_test.cc
|
Add utiliry fct window_util::MakeWindow(sizes, strides)
|
2019-10-08 12:00:28 -07:00 |
|
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
|
Merge pull request #33257 from nouiz:small2
|
2019-10-14 04:08:20 -07:00 |
|
xla_data.proto
|
Introduce new RngBitGenerator HLO
|
2020-02-06 05:06:44 -08:00 |
|
xla.bzl
|
Migrate from xla_proto_library to tf_proto_library_cc.
|
2019-11-15 01:00:17 -08:00 |
|
xla.proto
|
Update following the comments. Also update the description.
|
2020-06-09 14:22:10 -07:00 |
