..
cc
…
g3doc
…
kernels
Add DeviceIndex xla op.
2020-06-22 15:06:38 -07:00
lib
…
ops
[XLA:SPMD] Add API for the experimental mechanism to mix auto and manual partitioning
2020-05-04 16:57:03 -07:00
python
Add Bessel functions to the public api:
2020-06-17 21:06:09 -07:00
BUILD
Use Bazel target names aligned with upstream library names
2020-06-10 14:51:07 -07:00
const_analysis.cc
Support interprocedural constant meta-information propagation for compilation
2020-03-30 17:51:05 -07:00
const_analysis.h
Support interprocedural constant meta-information propagation for compilation
2020-03-30 17:51:05 -07:00
const_analysis_test.cc
Support interprocedural constant meta-information propagation for compilation
2020-03-30 17:51:05 -07:00
cpu_function_runtime_test.cc
[XLA:CPU] Reconsider alignment heuristics
2020-04-24 02:51:55 -07:00
frontend_attributes_util.cc
…
frontend_attributes_util.h
…
functionalize_cond.cc
…
functionalize_cond.h
…
functionalize_cond_test.cc
…
functionalize_control_flow.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
functionalize_control_flow.h
…
functionalize_control_flow_pass_registration.cc
…
functionalize_control_flow_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
functionalize_control_flow_util.cc
…
functionalize_control_flow_util.h
…
functionalize_while.cc
…
functionalize_while.h
…
fused_batchnorm_reserve_space_test.cc
Support computing exponential running mean and variance in fused_batch_norm.
2020-02-21 17:55:52 -08:00
graph_compiler.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
graph_compiler.h
…
graph_compiler_util.cc
Fix TF2XLA's InitGraph for unused feeds.
2020-05-15 10:23:18 -07:00
graph_compiler_util.h
…
host_compute_metadata.proto
…
literal_util.cc
Add a HostToBorrowingLiteral function that takes an xla::Shape
2020-02-21 14:44:42 -08:00
literal_util.h
Add a HostToBorrowingLiteral function that takes an xla::Shape
2020-02-21 14:44:42 -08:00
literal_util_test.cc
…
mlir_bridge_pass.cc
Instrument the number of times the MLIR-based TF Bridge is enabled.
2020-05-14 17:04:52 -07:00
mlir_bridge_pass.h
[TF:MLIR] Add generic MLIR pass registration mechanism.
2020-03-10 16:04:53 -07:00
mlir_bridge_pass_registration.cc
[TF:MLIR] Add generic MLIR pass registration mechanism.
2020-03-10 16:04:53 -07:00
mlir_tf2xla.cc
- Eliminate all uses of passes that mark function visibility since the visibility
2020-06-18 13:12:22 -07:00
rearrange_function_argument.cc
…
rearrange_function_argument.h
…
resource_operation_table.cc
…
resource_operation_table.h
…
resource_operation_table_test.cc
…
resource_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-07 14:38:02 -07:00
resource_util.h
…
resource_util_test.cc
…
shape_util.cc
[TF2XLA] Support unknown ranked tensor in fake param.
2020-04-14 11:15:54 -07:00
shape_util.h
[TF2XLA] Support unknown ranked tensor in fake param.
2020-04-14 11:15:54 -07:00
sharding_util.cc
Refactor xla_sharding to be more useful.
2020-02-18 15:39:55 -08:00
sharding_util.h
Refactor xla_sharding to be more useful.
2020-02-18 15:39:55 -08:00
sharding_util_test.cc
…
side_effect_util.cc
…
side_effect_util.h
…
test_util.cc
…
test_util.h
…
tf2xla.cc
Roll forward https://github.com/tensorflow/tensorflow/commit/ed0e46c219efa62559901 ...
2020-03-09 16:40:34 -07:00
tf2xla.h
Internal change
2020-03-05 14:01:37 -08:00
tf2xla.proto
…
tf2xla_supported_ops.cc
…
tf2xla_supported_ops.h
…
tf2xla_supported_ops_main.cc
…
tf2xla_test.cc
Fix TF2XLA's InitGraph for unused feeds.
2020-05-15 10:23:18 -07:00
tf2xla_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-07 14:38:02 -07:00
tf2xla_util.h
…
tf2xla_util_test.cc
…
type_util.cc
[TF/XLA] Fix several layout issues.
2020-02-18 13:47:52 -08:00
type_util.h
…
xla_compilation_device.cc
…
xla_compilation_device.h
…
xla_compiled_cpu_function.cc
[xla] add better support for variables(num_variables, index lookup by name) in xla aot/jit.
2020-03-10 21:11:29 -07:00
xla_compiled_cpu_function.h
[XLA:CPU] Reconsider alignment heuristics
2020-04-24 02:51:55 -07:00
xla_compiler.cc
[TF2XLA] Set up aliasing for resource variables even when not returning a tuple
2020-06-19 18:20:41 -07:00
xla_compiler.h
Add support for updating argument/result shapes and layouts with associated shardings of entry function.
2020-03-26 13:32:04 -07:00
xla_compiler_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
xla_context.cc
…
xla_context.h
…
xla_cpu_backend.cc
…
xla_expression.cc
Handle constant inputs in xla-legalize-tf-with-tf2xla pass
2020-04-30 15:04:26 -07:00
xla_expression.h
…
xla_expression_test.cc
…
xla_gpu_backend.cc
…
xla_helpers.cc
…
xla_helpers.h
…
xla_jit_compiled_cpu_function.cc
[xla] add better support for variables(num_variables, index lookup by name) in xla aot/jit.
2020-03-10 21:11:29 -07:00
xla_jit_compiled_cpu_function.h
[xla] add better support for variables(num_variables, index lookup by name) in xla aot/jit.
2020-03-10 21:11:29 -07:00
xla_jit_compiled_cpu_function_test.cc
[xla] add better support for variables(num_variables, index lookup by name) in xla aot/jit.
2020-03-10 21:11:29 -07:00
xla_op_kernel.cc
[TF:XLA] Update an error message to include the word "uninitialized" because many tests
2020-06-09 19:20:57 -07:00
xla_op_kernel.h
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -07:00
xla_op_registry.cc
…
xla_op_registry.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
xla_op_registry_test.cc
…
xla_resource.cc
…
xla_resource.h
…
