experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/jit
History
Lucy Fox db14325178 Update naming of MLIR bridge flag and add some documentation. 
PiperOrigin-RevId: 338152017
Change-Id: Ib0573db86327077245d60c7eb57d8e10ec051b6e
2020-10-20 15:51:11 -07:00
..
graphcycles
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
kernels
Remove ThenExecuteFunction references from eager XLA op dispatch.
2020-10-14 11:25:54 -07: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
Clean up _pywrap_tfe deps. pywrap_tfe is a separate shared library and should not include tensorflow sources that are included in the main shared library.
2020-10-16 11:42:44 -07: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
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.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 #31594 from trentlo:github-cluster-scoping
2019-08-21 20:15:13 -07:00
cluster_scoping_pass.h
Merge pull request #31594 from trentlo:github-cluster-scoping
2019-08-21 20:15:13 -07:00
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] Check 'case' op compilation from autoclustering codepath, but not from tf.function
2020-10-07 12:44:29 -07:00
compilability_check_util.h
[TF2XLA] Check 'case' op compilation from autoclustering codepath, but not from tf.function
2020-10-07 12:44:29 -07: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
Code review changes
2020-01-14 23:58:14 -05:00
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
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.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 #41734 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
Update naming of MLIR bridge flag and add some documentation.
2020-10-20 15:51:11 -07:00
flags.h
Update tf_mlir_enable_mlir_bridge support unspecified
2020-10-16 09:55:12 -07: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] Explicitly copy must-be-constants to host in get_compiler_ir
2020-10-01 13:04:39 -07: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
Add a debug-only pass that introduces a small error to a designated TF node
2019-03-28 12:25:47 -07:00
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
Add a new unittest in mark_for_compilation_pass_test.
2019-08-16 13:37:19 -07:00
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
Exposes variadic reduce to TF python via xla.py
2020-10-13 13:16:03 -07: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
minor spelling tweaks
2019-12-09 18:21:12 +09:00
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
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
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
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
[TF:XLA] Update resource_operation_safety_analysis file comment.
2019-07-16 18:03:24 -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
Move shape inference utility to open source.
2018-10-15 21:23:29 -07:00
test_util.cc
…
test_util.h
[TF/XLA Bridge] [NFC] Reduce the amount of boilerplate required to create GraphOptimizationPassOptions
2019-09-30 20:48:18 -07:00
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
…
xla_activity_logging_listener.cc
[TF:XLA] Logging of known cases of performance deficiencies using XLA with Tensorflow.
2019-08-07 04:04:01 -07:00
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
Utility function to figure out whether a cluster has reference variables
2019-09-19 17:07:52 -07:00
xla_compilation_cache_test.cc
[TF2XLA] Add a kill switch to disable XLA compilation
2020-08-27 12:20:59 -07:00
xla_compilation_cache.cc
Disable using MLIR for VarIsInitializedOp via XlaCompilationCache::CompileSingleOp.
2020-10-16 15:31:05 -07: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] [NFC] Simplify GetVariableInfosFromInputs and BuildXlaCompilerArguments
2020-09-02 21:43:21 -07:00
xla_compile_on_demand_op.h
[TF2XLA] [NFC] Simplify XlaPlatformInfoFromDevice
2020-08-28 11:20:02 -07:00
xla_cpu_device.cc
[TF2XLA] [NFC] Lower the verbosity for not creating XLA:CPU/GPU devices log
2020-08-28 00:01:44 -07: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
[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_device.h
[TF2XLA] [NFC] Simplify XlaPlatformInfoFromDevice
2020-08-28 11:20:02 -07:00
xla_gpu_device.cc
[TF2XLA] [NFC] Lower the verbosity for not creating XLA:CPU/GPU devices log
2020-08-28 00:01:44 -07: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
Update tf_mlir_enable_mlir_bridge support unspecified
2020-10-16 09:55:12 -07: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
[TF2XLA] Correct misleading error message about var placement
2020-09-21 16:53:52 -07:00
xla_launch_util.h
[TF2XLA] [NFC] Simplify GetVariableInfosFromInputs and BuildXlaCompilerArguments
2020-09-02 21:43:21 -07: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