| .. |
|
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
|
Define helper function for effective MlirBridgeRollout enum
|
2021-03-02 05:29:23 -08:00 |
|
build_xla_ops_pass_test.cc
|
Use StatefulPartitionedCall instead of PartitionedCall in auto-clustering
|
2021-02-16 12:00:23 -08:00 |
|
build_xla_ops_pass.cc
|
Use StatefulPartitionedCall instead of PartitionedCall in auto-clustering
|
2021-02-16 12:00:23 -08: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
|
|
|
|
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] Remove redundant compilability checking in xla_kernel_creator
|
2021-02-22 14:58:08 -08:00 |
|
compilability_check_util.h
|
[TF2XLA] Remove redundant compilability checking in xla_kernel_creator
|
2021-02-22 14:58:08 -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
|
Use TryGetNodeAttr instead of GetNodeAttr in a few places
|
2021-02-15 14:26:06 -08: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
|
Define helper function for effective MlirBridgeRollout enum
|
2021-03-02 05:29:23 -08:00 |
|
flags.h
|
Define helper function for effective MlirBridgeRollout enum
|
2021-03-02 05:29:23 -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
|
Extend experimental_get_compiler_ir to get serialized HLO modules.
|
2021-01-26 23:30:30 -08:00 |
|
get_compiler_ir.h
|
Extend experimental_get_compiler_ir to get serialized HLO modules.
|
2021-01-26 23:30:30 -08: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
|
|
|
|
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
|
Add a new unittest in mark_for_compilation_pass_test.
|
2019-08-16 13:37:19 -07:00 |
|
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
|
Introduce Xla{Dot,Conv}V2
|
2021-02-22 11:25:16 -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
|
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
|
|
|
|
rearrange_function_argument_pass_test.cc
|
Update RearrangeFunctionArgumentForFunctionTest to use correct index when creating Retval nodes.
|
2019-09-27 14:25:28 -07:00 |
|
report_clustering_info_pass.cc
|
Roll forward "Add an XLA "activity listener" mechanism."
|
2019-06-14 17:14:19 -07:00 |
|
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
|
Prefixing TensorFlow thread annotation macros with TF_.
|
2020-03-05 08:42:01 -08:00 |
|
xla_activity_listener.h
|
[TF:XLA] Logging of known cases of performance deficiencies using XLA with Tensorflow.
|
2019-08-07 04:04:01 -07:00 |
|
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
|
Lookup functions in optional FunctionLibraryDefinition when checking for supported features for MLIR bridge.
|
2021-03-05 09:34:14 -08:00 |
|
xla_compilation_cache.h
|
Add MlirXlaOpKernel, which is used to implement XlaOpKernels using MLIR legalization.
|
2021-03-03 07:27:30 -08: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
|
Register VariableShapeOp for TPU.
|
2021-02-08 15:50:16 -08: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
|
Finish migrating {Tf|Platform}GpuId to {Tf|Platform}DeviceId
|
2021-03-05 17:15:52 -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] Remove redundant compilability checking in xla_kernel_creator
|
2021-02-22 14:58:08 -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
|
Propagate the frame where the resource var was created to resource var handles
|
2021-01-25 14:25:35 -08:00 |
|
xla_launch_util.h
|
Implement allocation tracking support for XLA allocated tensors.
|
2021-03-05 02:35:46 -08:00 |
|
xla_ops_on_regular_devices.cc
|
Introduce Xla{Dot,Conv}V2
|
2021-02-22 11:25:16 -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 |
|
xla_tpu_device.cc
|
[TPU] Move TPU node and system device initializers to compiler/jit
|
2021-01-19 22:43:21 -08:00 |
|
xla_tpu_device.h
|
[TPU] Move TPU node and system device initializers to compiler/jit
|
2021-01-19 22:43:21 -08:00 |
