experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/tf2xla
History
Roman Dzhabarov 10e5d5cdb1 [mlir] Add more logging on MLIR bridge execution. 
In the default case, we do not want to flood log with the "Running MLIR ..." messages. Log if bridge was executed once. User can enable v=1 option for more detailed logging.

PiperOrigin-RevId: 358199172
Change-Id: I03267b05a9ee4e05e7b0f926809a0002d6f60b42
2021-02-18 09:51:33 -08:00
..
cc
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
g3doc
…
kernels
Propagate value bounds into control flow.
2021-02-17 15:42:19 -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 an explicit sharding attribute to XlaSharding op.
2021-01-26 19:56:27 -08:00
python
Add an explicit sharding attribute to XlaSharding op.
2021-01-26 19:56:27 -08:00
BUILD
Use proper header for absl::optional
2021-02-17 22:38:26 -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
[TF2XLA] [NFC] Remove the comment which was already done
2021-02-03 16:46:44 -08: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
Automatically set up user aliasing in tf2xla when a resource update is presented.
2020-02-13 13:22:20 -08:00
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
[mlir] Add more logging on MLIR bridge execution.
2021-02-18 09:51:33 -08:00
mlir_bridge_pass.h
Let MlirFunctionOptimizationPass query individual registered passes whether they need to run in a shadow mode.
2021-01-26 12:30:16 -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] Look for stack traces in global FunctionLibraryDefinition if they are not found in local FunctionLibraryDefinition
2021-01-26 19:51:52 -08:00
rearrange_function_argument.h
[TF2XLA] Look for stack traces in global FunctionLibraryDefinition if they are not found in local FunctionLibraryDefinition
2021-01-26 19:51:52 -08: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
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
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 to populate OpMetadata in OpSharding.
2021-01-19 10:26:20 -08:00
sharding_util.cc
Update DistributedTPURewritePass to populate OpMetadata in OpSharding.
2021-01-19 10:26:20 -08:00
sharding_util.h
Update DistributedTPURewritePass to populate OpMetadata in OpSharding.
2021-01-19 10:26:20 -08:00
side_effect_util.cc
[XLA] Break dependencies of XLA on TF.
2021-01-19 14:07:56 -08:00
side_effect_util.h
[XLA] Break dependencies of XLA on TF.
2021-01-19 14:07:56 -08:00
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
Update DistributedTPURewritePass to populate OpMetadata in OpSharding.
2021-01-19 10:26:20 -08:00
tf2xla_util.cc
[TF2XLA] Propagate stack traces properly for more functions
2021-01-25 16:21:46 -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
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
Add uninitialized resource args to unsupported features for MLIR bridge
2021-02-16 19:12:45 -08:00
xla_argument.h
Use proper header for absl::optional
2021-02-17 22:38:26 -08:00
xla_compilation_device.cc
Update DistributedTPURewritePass to populate OpMetadata in OpSharding.
2021-01-19 10:26:20 -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
Propagate value bounds into control flow.
2021-02-17 15:42:19 -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
Propagate value bounds into control flow.
2021-02-17 15:42:19 -08:00
xla_expression.h
Propagate value bounds into control flow.
2021-02-17 15:42:19 -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] Attach Python stack trace to all INVALID_ARGUMENT Status objects created from the bridge
2021-02-08 18:28:58 -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
Use TryGetNodeAttr instead of GetNodeAttr in a few places
2021-02-15 14:26:06 -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
xla_tpu_backend.cc
[TPU] Colocate tpu_compilation_device registration with other registration in compiler/tf2xla
2021-01-22 16:51:35 -08:00