STT-tensorflow/tensorflow/compiler/jit
Ruoxin Sang 02ad000479 Support dynamic outputs for XLA on demand ops.
PiperOrigin-RevId: 317902879
Change-Id: I6b6dfa54855d5996ac15d4b5c48a5db5dc230025
2020-07-01 11:11:47 -07:00
..
graphcycles During contraction of an edge from "a" to "b" there is a choice which node should be used to represent the union of the nodes. Using the node with the largest degree minimizes the number of operation that should be performed. 2020-03-16 18:10:55 -07:00
kernels [TF2XLA] [NFC] Refactor datastructures for resource variables to not require snapshotting for compilation 2020-06-22 12:55:49 -07:00
ops
tests [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
BUILD Moving TPU ops components as TPU kernels library. 2020-05-20 16:36:27 -07:00
build_xla_ops_pass.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
build_xla_ops_pass.h [NFC] Clarify documentation of JIT passes 2019-12-19 11:20:28 -08:00
build_xla_ops_pass_test.cc
clone_constants_for_better_clustering.cc
clone_constants_for_better_clustering.h
clone_constants_for_better_clustering_test.cc
cluster_scoping_pass.cc
cluster_scoping_pass.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
compilability_check_util.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
compilability_check_util.h [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
compilability_check_util_test.cc [Build cleanup] Split "core_cpu_impl" into fine-grained targets (4/n). 2020-04-28 09:53:59 -07:00
compilation_passes_test_main.cc
deadness_analysis.cc Code review changes 2020-01-14 23:58:14 -05:00
deadness_analysis.h
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
defs.cc Move must-compilation from experimental_compile=True to a separate attribute 2019-12-20 20:31:28 -08:00
defs.h Move must-compilation from experimental_compile=True to a separate attribute 2019-12-20 20:31:28 -08:00
device_util.cc
device_util.h
device_util_test.cc
encapsulate_subgraphs_pass.cc [TF/XLA] Minor clean up 2020-02-09 17:23:44 -08:00
encapsulate_subgraphs_pass.h [NFC] Clarify documentation of JIT passes 2019-12-19 11:20:28 -08:00
encapsulate_subgraphs_pass_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
encapsulate_util.cc
encapsulate_util.h
encapsulate_util_test.cc
encapsulate_xla_computations_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
encapsulate_xla_computations_pass.h [NFC] Clarify documentation of JIT passes 2019-12-19 11:20:28 -08:00
encapsulate_xla_computations_pass_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
extract_outside_compilation_pass.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
extract_outside_compilation_pass.h
extract_outside_compilation_pass_test.cc
flags.cc Provides an environment variable via TF_XLA_FLAGS to turn on the MLIR bridge. 2020-06-12 09:26:57 -07:00
flags.h Provides an environment variable via TF_XLA_FLAGS to turn on the MLIR bridge. 2020-06-12 09:26:57 -07:00
force_xla_constants_on_host_pass.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
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
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
increase_dynamism_for_auto_jit_pass.cc
increase_dynamism_for_auto_jit_pass.h
increase_dynamism_for_auto_jit_pass_test.cc
introduce_floating_point_jitter_pass.cc
introduce_floating_point_jitter_pass.h
introduce_floating_point_jitter_pass_internal.h
introduce_floating_point_jitter_pass_test.cc
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.cc Add DeviceIndex xla op. 2020-06-22 15:06:38 -07:00
mark_for_compilation_pass.h [NFC] Clarify documentation of JIT passes 2019-12-19 11:20:28 -08:00
mark_for_compilation_pass_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
mark_for_compilation_pass_test_helper.cc
mark_for_compilation_pass_test_helper.h
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
node_matchers_test.cc
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
partially_decluster_pass_test.cc [Build cleanup] Update #includes of moved header "graph/graph_constructor.h". 2020-04-29 09:20:48 -07:00
rearrange_function_argument_pass_test.cc
report_clustering_info_pass.cc
report_clustering_info_pass.h
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
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
shape_inference.cc Fixes for outside compilation. 2020-01-03 14:27:55 -08:00
shape_inference.h
shape_inference_helpers.cc
shape_inference_helpers.h
shape_inference_test.cc
test_util.cc
test_util.h
union_find.h
xla_activity.proto
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_listener_test.cc
xla_activity_logging_listener.cc
xla_cluster_util.cc
xla_cluster_util.h
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_compilation_cache.cc Support XlaCompiler arguments of type kConstant in XlaCompilationCache 2020-05-05 16:12:29 -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_compilation_cache_test.cc
xla_compile_on_demand_op.cc [TF2XLA] [NFC] Refactor datastructures for resource variables to not require snapshotting for compilation 2020-06-22 12:55:49 -07:00
xla_compile_on_demand_op.h [TF2XLA] [NFC] Refactor datastructures for resource variables to not require snapshotting for compilation 2020-06-22 12:55:49 -07:00
xla_cpu_device.cc [TF/XLA] Only enable XLA_ devices if TF_XLA_FLAGS=--tf_xla_enable_xla_devices is set. For now, set the flag to "true" by default. 2020-01-09 11:46:30 -08:00
xla_device.cc Use INFO logging for device deprecation, for easier filtering. 2020-05-27 10:30:49 -07:00
xla_device.h Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
xla_device_context.cc Export memory fragmentation in memstats. 2020-04-14 22:18:56 -07:00
xla_device_context.h Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
xla_device_ops.cc
xla_device_ops.h Resubmitting the change to make iterator deleter not on host memory 2020-05-08 11:11:31 -07:00
xla_gpu_device.cc [ROCm] Fix to enable XLA_GPU device registration for ROCm platform 2020-01-15 14:50:10 +00:00
xla_interpreter_device.cc
xla_kernel_creator.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
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_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_util.cc [TF/XLA] On compilation failure, do not overflow the max size of the bad status by a huge list of function inputs 2020-05-27 11:59:06 -07:00
xla_kernel_creator_util.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
xla_launch_util.cc Support dynamic outputs for XLA on demand ops. 2020-07-01 11:11:47 -07:00
xla_launch_util.h [TF2XLA] [NFC] Refactor datastructures for resource variables to not require snapshotting for compilation 2020-06-22 12:55:49 -07:00
xla_tensor.cc [TF/XLA] Remove the wrapping function XlaTensor::RefCountIsOne 2020-06-15 11:44:06 -07:00
xla_tensor.h [XLA] Do not needlessly store wrapped Tensor and ScopedShapedBuffer inside XlaTensor on a heap 2020-06-15 12:52:08 -07:00