| .. |
|
kernels
|
[TF2XLA] Support must-be-constant resource variables for compilation
|
2020-11-09 15:04:58 -08:00 |
|
ops
|
PY2 removal cleanup
|
2021-01-15 16:48:57 -08: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
|
|
|
|
clone_constants_for_better_clustering_test.cc
|
|
|
|
clone_constants_for_better_clustering.cc
|
|
|
|
clone_constants_for_better_clustering.h
|
|
|
|
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
|
|
|
|
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] Bumping recursion depth limit for conversion
|
2021-01-14 12:21:40 -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 #41734 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
|
|
|
|
encapsulate_util_test.cc
|
|
|
|
encapsulate_util.cc
|
c, compiler, jit resolutions
|
2020-07-26 20:42:00 +00:00 |
|
encapsulate_util.h
|
|
|
|
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
|
Fix outside compilation to add new function instead of inplace function update.
|
2021-01-08 17:16:29 -08: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
|
[TF/XLA] Force all tensors which need to be constant during the XLA compilation to be located on the host
|
2020-04-01 11:34:32 -07:00 |
|
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
|
|
|
|
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_floating_point_jitter_pass_internal.h
|
|
|
|
introduce_floating_point_jitter_pass_test.cc
|
|
|
|
introduce_floating_point_jitter_pass.cc
|
|
|
|
introduce_floating_point_jitter_pass.h
|
|
|
|
jit_compilation_pass_registration.cc
|
[TF/XLA] Force all tensors which need to be constant during the XLA compilation to be located on the host
|
2020-04-01 11:34:32 -07:00 |
|
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
|
|
|
|
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
|
|
|
|
partially_decluster_pass.h
|
|
|
|
rearrange_function_argument_pass_test.cc
|
|
|
|
report_clustering_info_pass.cc
|
|
|
|
report_clustering_info_pass.h
|
|
|
|
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
|
Storing kRead kinds doesn't have any effect, but might take a lot of RAM.
|
2020-03-17 09:54:28 -07:00 |
|
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
|
Roll forward change to run use MLIR based TensorFlow compiler in XLA on demand compiler
|
2020-03-23 14:50:24 -07:00 |
|
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 #43134 from ROCmSoftwarePlatform:google-upstream-thenexecute-deadlock
|
2020-09-18 05:20:22 -07:00 |
|
xla_device_context.h
|
Merge pull request #43134 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
|
|
|
|
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
|
|
|
|
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
|
Cleans up a newly added XlaVariadicSort op to take Input instead of Attr.
|
2021-01-06 06:42:01 -08: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 |
