..
cc
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
g3doc
kernels
Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument,
2020-12-21 22:35:25 -08:00
lib
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
ops
Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument,
2020-12-21 22:35:25 -08:00
python
Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument,
2020-12-21 22:35:25 -08:00
BUILD
[TF2XLA] Remove temporary #ifdef statement
2020-12-17 18:40:37 -08:00
const_analysis_test.cc
Rollback of rollback of disabling XLA:CPU/GPU devices by default.
2020-08-11 13:06:59 -07:00
const_analysis.cc
generate stateless_case op if all ops in all branches are stateless
2020-07-30 16:23:47 -07:00
const_analysis.h
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_test.cc
Add capability to restrict functionalization via node filter
2020-06-29 10:56:59 -07:00
functionalize_cond.cc
XLA Parallel reduce.
2020-08-27 15:21:33 -07:00
functionalize_cond.h
Add capability to restrict functionalization via node filter
2020-06-29 10:56:59 -07:00
functionalize_control_flow_pass_registration.cc
functionalize_control_flow_test.cc
Also functionalize control flow in functions for UpgradeLegacyGraph
2020-08-14 17:04:03 -07:00
functionalize_control_flow_util.cc
Add capability to restrict functionalization via node filter
2020-06-29 10:56:59 -07:00
functionalize_control_flow_util.h
Add capability to restrict functionalization via node filter
2020-06-29 10:56:59 -07:00
functionalize_control_flow.cc
XLA Parallel reduce.
2020-08-27 15:21:33 -07:00
functionalize_control_flow.h
Also functionalize control flow in functions for UpgradeLegacyGraph
2020-08-14 17:04:03 -07:00
functionalize_while.cc
XLA Parallel reduce.
2020-08-27 15:21:33 -07:00
functionalize_while.h
Add capability to restrict functionalization via node filter
2020-06-29 10:56:59 -07:00
fused_batchnorm_reserve_space_test.cc
Rollback of rollback of disabling XLA:CPU/GPU devices by default.
2020-08-11 13:06:59 -07:00
graph_compiler_util.cc
[-Wsign-compare] warning fixes batch 9
2020-07-07 01:37:54 +00:00
graph_compiler_util.h
graph_compiler.cc
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
graph_compiler.h
host_compute_metadata.proto
literal_util_test.cc
literal_util.cc
tf2xla directory resolutions
2020-07-26 22:06:02 +00:00
literal_util.h
Add a HostToBorrowingLiteral function that takes an xla::Shape
2020-02-21 14:44:42 -08:00
mlir_bridge_pass_registration.cc
[TF:MLIR] Add generic MLIR pass registration mechanism.
2020-03-10 16:04:53 -07:00
mlir_bridge_pass.cc
Remove reference variable tracking from import phase.
2020-11-18 17:44:31 -08:00
mlir_bridge_pass.h
MlirBridgeV1CompatPass does not support shadow runs, disable it.
2020-11-22 19:44:37 -08:00
mlir_tf2xla.cc
Propagate unknown rank of tensor shape information to MLIR
2020-11-17 00:38:39 -08:00
rearrange_function_argument.cc
recommonded updates
2020-07-29 16:27:57 +00:00
rearrange_function_argument.h
Move function argument rearrangement from a graph pass to XlaCompiler.
2019-05-15 16:58:38 -07:00
resource_operation_table_test.cc
resource_operation_table.cc
Internal change
2020-09-17 21:37:45 -07:00
resource_operation_table.h
resource_util_test.cc
Pipe ConfigProto through FLR so that it can be accessed by Ops like PartitionedCallOp.
2019-10-03 16:20:55 -07:00
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
shape_util.cc
[-Wsign-compare] warning fixes batch 9
2020-07-07 01:37:54 +00:00
shape_util.h
[TF2XLA] Support unknown ranked tensor in fake param.
2020-04-14 11:15:54 -07:00
sharding_util_test.cc
sharding_util.cc
Adds ParseShardingFromEdgeSource().
2020-08-03 15:44:43 -07:00
sharding_util.h
Adds ParseShardingFromEdgeSource().
2020-08-03 15:44:43 -07:00
side_effect_util.cc
side_effect_util.h
test_util.cc
test_util.h
tf2xla_supported_ops_main.cc
tf2xla_supported_ops.cc
tf2xla_supported_ops.h
tf2xla_test.cc
Fix TF2XLA's InitGraph for unused feeds.
2020-05-15 10:23:18 -07:00
tf2xla_util_test.cc
tf2xla_util.cc
tf2xla directory resolutions
2020-07-26 22:06:02 +00:00
tf2xla_util.h
Restrict propagation of internal attributes in functionalization
2020-07-01 13:41:52 -07:00
tf2xla.cc
Remove an obsolete TODO since allowed_device has been deprecated.
2020-08-20 15:54:51 -07:00
tf2xla.h
Internal change
2020-03-05 14:01:37 -08:00
tf2xla.proto
type_util.cc
[TF/XLA] Fix several layout issues.
2020-02-18 13:47:52 -08:00
type_util.h
xla_argument.cc
[TF2XLA] [NFC] Break apart the [TF2XLA/MLIR] -> xla_compiler dependency edge
2020-07-28 13:36:06 -07:00
xla_argument.h
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_compilation_device.cc
[TF2XLA] Display Python location information in XLA metadata
2020-12-10 16:53:44 -08:00
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_test.cc
Fix build and reenable xla_compiler_test.
2020-08-24 10:03:56 -07:00
xla_compiler.cc
[TF2XLA] Enable MLIR bridge usage in XlaCompiler::CompileGraph entry point
2020-12-21 12:59:53 -08:00
xla_compiler.h
Fix enable MLIR bridge logic in xla_compiler
2020-11-30 15:25:39 -08:00
xla_context.cc
[TF2XLA] Display Python location information in XLA metadata
2020-12-10 16:53:44 -08:00
xla_context.h
[TF2XLA] Display Python location information in XLA metadata
2020-12-10 16:53:44 -08:00
xla_cpu_backend.cc
xla_expression_test.cc
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_expression.cc
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_expression.h
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_gpu_backend.cc
xla_helpers.cc
[XLA:SPMD] Minor fixes and utils for manual sharding
2020-12-01 13:37:36 -08:00
xla_helpers.h
[TF2XLA] [NFC] Break apart the [TF2XLA/MLIR] -> xla_compiler dependency edge
2020-07-28 13:36:06 -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_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_op_kernel.cc
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_op_kernel.h
Support dynamic leading dimension for tensorlist.
2020-08-11 15:58:45 -07:00
xla_op_registry_test.cc
xla_op_registry.cc
[TF2XLA] [NFC] Log which ops require which constant inputs
2020-11-05 15:45:28 -08:00
xla_op_registry.h
[TF2XLA] Remove the serialization of CustomKernelCreator, since there is only one, and we won't add new ones
2020-09-22 10:59:59 -07:00
xla_resource.cc
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_resource.h
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
682d130cf6