experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/tf2xla
History
George Karpenkov 2c596172d2 [TF2XLA] Attach Python stack trace to all INVALID_ARGUMENT Status objects created from the bridge 
PiperOrigin-RevId: 356400325
Change-Id: I1b8fc2a771bcb6492ccde5fbeb14844154b38791
2021-02-08 18:28:58 -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
[TF2XLA] Attach Python stack trace to all INVALID_ARGUMENT Status objects created from the bridge
2021-02-08 18:28:58 -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
[TPU] Colocate tpu_compilation_device registration with other registration in compiler/tf2xla
2021-01-22 16:51:35 -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
Merge pull request #31129 from AnthonyBarbier:frontend_attributes
2019-08-27 16:48:01 -07:00
frontend_attributes_util.h
- Fix: Save frontend attributes in while loop
2019-08-19 11:41:38 +01:00
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
Move MLIR bridge pass to before all other passes.
2019-11-25 12:30:15 -08:00
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
Support function calls through PartitionedCall in tf2xla
2019-03-08 11:32:38 -08:00
host_compute_metadata.proto
Add metadata for gathering information about host compute transfers while compiling XLA.
2018-03-06 16:53:12 -08:00
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
Introduce MlirOptimizationPassState::Fallback state to the MlirOptimizationPass.
2021-02-08 11:53:44 -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
[XLA] Migrate from gtl::FlatMap to absl::flat_hash_map
2018-10-01 13:53:07 -07:00
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
Add function call support in resource op analysis
2019-07-08 20:00:50 -07:00
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
Add a utility function to build node name to node index.
2018-10-12 06:39:26 -07:00
test_util.h
Add a utility function to build node name to node index.
2018-10-12 06:39:26 -07:00
tf2xla_supported_ops_main.cc
Add the tf2xla_supported_ops tool, which dumps ops supported by tf2xla.
2017-12-05 11:46:00 -08:00
tf2xla_supported_ops.cc
Use absl functions instead of str_util within tf2xla.
2018-08-23 19:21:32 -07:00
tf2xla_supported_ops.h
Add the tf2xla_supported_ops tool, which dumps ops supported by tf2xla.
2017-12-05 11:46:00 -08:00
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
minor spelling tweaks
2019-12-09 18:21:12 +09:00
type_util.cc
[TF/XLA] Fix several layout issues.
2020-02-18 13:47:52 -08:00
type_util.h
[TF:XLA] Add EncodePrimitiveTypeAsDataType into tf2xla/type_util.{h,cc}.
2019-07-02 19:15:28 -07:00
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
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
[TF2XLA] Look for stack traces in global FunctionLibraryDefinition if they are not found in local FunctionLibraryDefinition
2021-01-26 19:51:52 -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
[TF:XLA] Fix typo in function name.
2018-10-17 14:01:30 -07:00
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
[TF:XLA] Fix typo in function name.
2018-10-17 14:01:30 -07:00
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
[TF:XLA] Don't register kernels with infeasible type constraints.
2018-06-26 09:57:45 -07:00
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
xla_tpu_backend.cc
[TPU] Colocate tpu_compilation_device registration with other registration in compiler/tf2xla
2021-01-22 16:51:35 -08:00