STT-tensorflow/tensorflow/compiler/jit
A. Unique TensorFlower 682d130cf6 Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument,
and XlaKeyValueSort for 2 arguments.)

The new op, XlaVariadicSort, fully exposes the xla::Sort functionality: can
take an arbitrary list of operands, can take dimension and
is_stable arguments, and a comparator function.

PiperOrigin-RevId: 348581487
Change-Id: I82b4ff0c37d433541167584475b004626bc8c82c
2020-12-21 22:35:25 -08:00
..
kernels [TF2XLA] Support must-be-constant resource variables for compilation 2020-11-09 15:04:58 -08:00
ops Add target environment constraints to a subset of TensorFlow packages and targets. 2020-09-21 10:57:01 -07:00
tests Add target environment constraints to a subset of TensorFlow packages and targets. 2020-09-21 10:57:01 -07:00
BUILD [TF2XLA] [NFC] Remove temporary workaround for TPU on GCE 2020-12-21 15:46:41 -08:00
build_xla_ops_pass_test.cc Avoid "unused function" warning 2020-09-01 17:22:45 -07:00
build_xla_ops_pass.cc Update build_xla_ops_pass.cc 2020-07-29 17:27:29 -04:00
build_xla_ops_pass.h [NFC] Clarify documentation of JIT passes 2019-12-19 11:20:28 -08:00
clone_constants_for_better_clustering_test.cc Make constant cloning more aggressive (i.e. make it clone more often) 2019-10-07 12:23:37 -07:00
clone_constants_for_better_clustering.cc
clone_constants_for_better_clustering.h Add an optimization pass that clones Constant nodes to make larger clusters 2019-03-04 15:25:22 -08:00
cluster_scoping_pass_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
cluster_scoping_pass.cc Merge pull request from trentlo:github-cluster-scoping 2019-08-21 20:15:13 -07:00
cluster_scoping_pass.h
compilability_check_util_test.cc [TF2XLA] Check 'case' op compilation from autoclustering codepath, but not from tf.function 2020-10-07 12:44:29 -07:00
compilability_check_util.cc [TF2XLA] Fix a segfault in propagating stack traces in XlaKernelCreator 2020-12-21 12:05:41 -08:00
compilability_check_util.h [TF2XLA] Fix a segfault in propagating stack traces in XlaKernelCreator 2020-12-21 12:05:41 -08:00
compilation_passes_test_main.cc
deadness_analysis_internal.h
deadness_analysis_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
deadness_analysis.cc
deadness_analysis.h
defs.cc [TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation 2020-08-28 17:11:30 -07:00
defs.h [TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation 2020-08-28 17:11:30 -07:00
device_util_test.cc
device_util.cc c, compiler, jit resolutions 2020-07-26 20:42:00 +00:00
device_util.h Merge pull request from tg-at-google:wsign-compare-semi-final-tf-c-compiler-jit 2020-07-30 13:30:14 -07:00
encapsulate_subgraphs_pass_test.cc Adjust TPU build dependencies. 2020-07-27 15:37:03 -07:00
encapsulate_subgraphs_pass.cc [XLA] Insert control edges from write to read instructions for same buffers inside loops 2020-10-26 17:30:57 -07:00
encapsulate_subgraphs_pass.h [NFC] Clarify documentation of JIT passes 2019-12-19 11:20:28 -08:00
encapsulate_util_test.cc Remove unused inclusions and add missing inclusions 2019-02-05 14:06:22 -08:00
encapsulate_util.cc c, compiler, jit resolutions 2020-07-26 20:42:00 +00:00
encapsulate_util.h minor spelling tweaks 2019-12-09 18:21:12 +09:00
encapsulate_xla_computations_pass_test.cc [TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation 2020-08-28 17:11:30 -07:00
encapsulate_xla_computations_pass.cc [TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation 2020-08-28 17:11:30 -07:00
encapsulate_xla_computations_pass.h [TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation 2020-08-28 17:11:30 -07:00
extract_outside_compilation_pass_test.cc Adjust TPU build dependencies. 2020-07-27 15:37:03 -07:00
extract_outside_compilation_pass.cc Merge pull request from tg-at-google:wsign-compare-semi-final-tf-c-compiler-jit 2020-07-30 13:30:14 -07:00
extract_outside_compilation_pass.h
flags.cc Support safe mode in the mlir bridge 2020-12-15 15:27:34 -08:00
flags.h typo 2020-11-12 15:31:54 +08:00
force_xla_constants_on_host_pass_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
force_xla_constants_on_host_pass.cc [TF2XLA] [NFC] Simplify GetBodyAndConstantsAndResources 2020-09-02 22:03:20 -07:00
force_xla_constants_on_host_pass.h
get_compiler_ir.cc [TF2XLA] Support must-be-constant resource variables for compilation 2020-11-09 15:04:58 -08:00
get_compiler_ir.h [TF2XLA] Explicitly copy must-be-constants to host in get_compiler_ir 2020-10-01 13:04:39 -07:00
increase_dynamism_for_auto_jit_pass_test.cc Don't rewrite slices that have a constant begin input 2019-02-06 14:44:47 -08:00
increase_dynamism_for_auto_jit_pass.cc c, compiler, jit resolutions 2020-07-26 20:42:00 +00:00
increase_dynamism_for_auto_jit_pass.h Introduce a pass to increase the amount of dynamism supported by an XLA cluster 2018-10-26 14:01:34 -07:00
introduce_floating_point_jitter_pass_internal.h Add a debug-only pass that introduces a small error to a designated TF node 2019-03-28 12:25:47 -07:00
introduce_floating_point_jitter_pass_test.cc Add a debug-only pass that introduces a small error to a designated TF node 2019-03-28 12:25:47 -07:00
introduce_floating_point_jitter_pass.cc Add a debug-only pass that introduces a small error to a designated TF node 2019-03-28 12:25:47 -07:00
introduce_floating_point_jitter_pass.h
jit_compilation_pass_registration.cc
mark_for_compilation_pass_test_helper.cc
mark_for_compilation_pass_test_helper.h
mark_for_compilation_pass_test.cc Rollback of rollback of disabling XLA:CPU/GPU devices by default. 2020-08-11 13:06:59 -07:00
mark_for_compilation_pass.cc Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument, 2020-12-21 22:35:25 -08:00
mark_for_compilation_pass.h [TF2XLA] [NFC] Make a function only used in a single library fileprivate 2020-09-18 23:30:47 -07:00
node_matchers_test.cc
node_matchers.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
node_matchers.h minor spelling tweaks 2019-12-09 18:21:12 +09:00
partially_decluster_pass_test.cc Rollback of rollback of disabling XLA:CPU/GPU devices by default. 2020-08-11 13:06:59 -07:00
partially_decluster_pass.cc Split out node_def_util, op_def_builder, op_def_util and attr_value_util targets in tensorflow/core/framework/BUILD. Split out 2020-01-06 11:46:01 -08:00
partially_decluster_pass.h Decluster some must-be-constant ops to reduce XLA recompilations 2018-09-07 18:53:18 -07:00
rearrange_function_argument_pass_test.cc
report_clustering_info_pass.cc
report_clustering_info_pass.h Roll forward "Add an XLA "activity listener" mechanism." 2019-06-14 17:14:19 -07:00
resource_operation_safety_analysis_test.cc [Build cleanup] Split "core_cpu_impl" into fine-grained targets (4/n). 2020-04-28 09:53:59 -07:00
resource_operation_safety_analysis.cc
resource_operation_safety_analysis.h [XLA] Insert control edges from write to read instructions for same buffers inside loops 2020-10-26 17:30:57 -07:00
shape_inference_helpers.cc
shape_inference_helpers.h
shape_inference_test.cc
shape_inference.cc c, compiler, jit resolutions 2020-07-26 20:42:00 +00:00
shape_inference.h
test_util.cc
test_util.h
xla_activity_listener_test.cc
xla_activity_listener.cc
xla_activity_listener.h
xla_activity_logging_listener.cc
xla_activity.proto
xla_cluster_util_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
xla_cluster_util.cc c, compiler, jit resolutions 2020-07-26 20:42:00 +00:00
xla_cluster_util.h [XLA] Insert control edges from write to read instructions for same buffers inside loops 2020-10-26 17:30:57 -07:00
xla_compilation_cache_test.cc Internal tests cleanup. 2020-10-27 10:16:12 -07:00
xla_compilation_cache.cc [TF2XLA] [NFC] Remove temporary workaround for TPU on GCE 2020-12-21 15:46:41 -08:00
xla_compilation_cache.h
xla_compile_on_demand_op.cc [TF2XLA] Support must-be-constant resource variables for compilation 2020-11-09 15:04:58 -08:00
xla_compile_on_demand_op.h [TF2XLA] [NFC] Simplify XlaPlatformInfoFromDevice 2020-08-28 11:20:02 -07:00
xla_cpu_device.cc [NFC] [TF2XLA] Reduce log spam: no need to say every time that we are not creating XLA devices 2020-12-09 21:14:55 -08:00
xla_device_context.cc Merge pull request from ROCmSoftwarePlatform:google-upstream-thenexecute-deadlock 2020-09-18 05:20:22 -07:00
xla_device_context.h Merge pull request from ROCmSoftwarePlatform:google-upstream-thenexecute-deadlock 2020-09-18 05:20:22 -07:00
xla_device_ops.cc Rollback of rollback of [TF/XLA] Enable input/output aliasing in the TF2XLA bridge 2020-07-17 15:17:25 -07:00
xla_device_ops.h Resubmitting the change to make iterator deleter not on host memory 2020-05-08 11:11:31 -07:00
xla_device.cc Rationale: 2020-11-03 12:55:59 -08:00
xla_device.h [TF2XLA] [NFC] Simplify XlaPlatformInfoFromDevice 2020-08-28 11:20:02 -07:00
xla_gpu_device.cc [NFC] [TF2XLA] Reduce log spam: no need to say every time that we are not creating XLA devices 2020-12-09 21:14:55 -08:00
xla_kernel_creator_test.cc Replace NodeDef with std::shared_ptr<NodeProperties> in the kernel creation code paths and try to avoid as many copies of NodeDefs as possible. This will in most cases allow sharing the NodeDef between the OpKernel and the graph Node from which it is created. 2020-02-18 13:20:06 -08:00
xla_kernel_creator.cc [TF2XLA] Fix a segfault in propagating stack traces in XlaKernelCreator 2020-12-21 12:05:41 -08:00
xla_kernel_creator.h Replace NodeDef with std::shared_ptr<NodeProperties> in the kernel creation code paths and try to avoid as many copies of NodeDefs as possible. This will in most cases allow sharing the NodeDef between the OpKernel and the graph Node from which it is created. 2020-02-18 13:20:06 -08:00
xla_launch_util.cc Remove the use of host shape in TransferManager::ReadDynamicShapes. 2020-11-17 13:46:04 -08:00
xla_launch_util.h [TF2XLA] Support must-be-constant resource variables for compilation 2020-11-09 15:04:58 -08:00
xla_ops_on_regular_devices.cc [rollback of rollback] [TF2XLA] Do not copy in XLA device implementation; instead, request correct placement from the start. 2020-08-31 17:54:50 -07:00
xla_platform_info.cc [TF2XLA] [NFC] Simplify GenerateCompilerOptions and GetAllocator 2020-08-28 11:34:15 -07:00
xla_platform_info.h [TF2XLA] [NFC] Simplify GenerateCompilerOptions and GetAllocator 2020-08-28 11:34:15 -07:00
xla_tensor.cc [TF/XLA] Remove the wrapping function XlaTensor::RefCountIsOne 2020-06-15 11:44:06 -07:00
xla_tensor.h [TF2XLA] Remove XlaTensor::set_host_tensor. It creates unnecessary complication 2020-09-02 17:02:00 -07:00