| .. |
|
cc
|
Add target environment constraints to a subset of TensorFlow packages and targets.
|
2020-09-21 10:57:01 -07:00 |
|
g3doc
|
|
|
|
kernels
|
Remove dead helper function TransposeInputForGroupConvolutionBackpropFilter
|
2021-01-15 14:04:26 -08:00 |
|
lib
|
Add target environment constraints to a subset of TensorFlow packages and targets.
|
2020-09-21 10:57:01 -07:00 |
|
ops
|
Add auto-generated op for TF XlaVariadicReduce op
|
2021-01-14 03:17:33 -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
|
Do not round trip to MLIR in the MLIR bridge if no TPUs are found
|
2021-01-13 14:31:09 -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
|
|
|
|
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
|
Replace calls to deprecated googletest macros *TEST_CASE() with *TEST_SUITE()
|
2019-01-16 11:08:17 -08:00 |
|
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
|
Do not round trip to MLIR in the MLIR bridge if no TPUs are found
|
2021-01-13 14:31:09 -08:00 |
|
mlir_bridge_pass.h
|
Do not round trip to MLIR in the MLIR bridge if no TPUs are found
|
2021-01-13 14:31:09 -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
|
[TF2XLA] Do not lose stack traces while doing tf2xla function transformations
|
2020-12-22 20:09:55 -08: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
|
|
|
|
resource_util.cc
|
|
|
|
resource_util.h
|
|
|
|
shape_util.cc
|
Reuse unknown sentinel when not static
|
2021-01-12 10:40:05 -08:00 |
|
shape_util.h
|
[TF2XLA] Support unknown ranked tensor in fake param.
|
2020-04-14 11:15:54 -07:00 |
|
sharding_util_test.cc
|
Update DistributedTPURewritePass and sharding util to populate OpMetadata in OpSharding.
|
2021-01-14 13:56:17 -08:00 |
|
sharding_util.cc
|
Update DistributedTPURewritePass and sharding util to populate OpMetadata in OpSharding.
|
2021-01-14 13:56:17 -08:00 |
|
sharding_util.h
|
Update DistributedTPURewritePass and sharding util to populate OpMetadata in OpSharding.
|
2021-01-14 13:56:17 -08: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] Do not lose stack traces while doing tf2xla function transformations
|
2020-12-22 20:09:55 -08: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
|
|
|
|
tf2xla.proto
|
|
|
|
type_util.cc
|
|
|
|
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
|
Update DistributedTPURewritePass and sharding util to populate OpMetadata in OpSharding.
|
2021-01-14 13:56:17 -08: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
|
When resolving dynamism fails, conservatively assume value is dynamic.
|
2021-01-08 15:05:10 -08:00 |
|
xla_op_kernel.h
|
[TF2XLA] Show stack traces of op definitions for some common tf2xla error messages
|
2020-12-22 16:01:50 -08: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 |
