experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/service
History
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
..
cpu
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
2020-06-22 12:32:43 -07:00
g3doc
[XLA] Remove unsupported sparse layout
2020-01-02 18:04:52 -08:00
gpu
Make gpu_ftz_test ready for CUDA 11.
2020-06-22 12:24:29 -07:00
interpreter
Allocate input/output buffers at the correct memory space.
2020-06-22 17:31:14 -07:00
llvm_ir
Enable XLA:CPU fast math for min/max by default to be similar to TF's behavior.
2020-06-17 01:27:08 -07:00
mlir_gpu
[NFC] Eliminate use of .getBlocks() when not needed
2020-06-22 10:34:01 -07:00
spmd
[XLA:SPMD] Limit halo exchange to be smaller than replication.
2020-06-08 17:37:39 -07:00
algebraic_simplifier_proof_distributive_property.py
…
algebraic_simplifier_test.cc
[XLA] Turn variadic reduces with a single input and output into normal reduces
2020-06-10 22:44:39 -07:00
algebraic_simplifier.cc
Simplify broadcast plus compare
2020-06-17 11:32:30 -07:00
algebraic_simplifier.h
[XLA] Swap convolution operands if a naive convolution implementation would do less flops.
2020-04-15 09:23:25 -07:00
all_gather_decomposer_test.cc
[XLA] Fix some all-gather issues.
2020-05-19 17:55:08 -07:00
all_gather_decomposer.cc
[XLA] Fix some all-gather issues.
2020-05-19 17:55:08 -07:00
all_gather_decomposer.h
[XLA] Fix some all-gather issues.
2020-05-19 17:55:08 -07:00
all_reduce_combiner_test.cc
Verify replica count from AllReduce replica group config
2020-04-27 08:53:44 -07:00
all_reduce_combiner.cc
Enable using global ids for replica groups in AllReduce
2020-02-21 19:01:52 -08:00
all_reduce_combiner.h
[XLA:GPU] Add an AllReduceCombiner pass, that merges AllReduce operations.
2020-02-19 22:07:07 -08:00
all_reduce_simplifier_test.cc
Verify replica count from AllReduce replica group config
2020-04-27 08:53:44 -07:00
all_reduce_simplifier.cc
Combine cross-replica / cross-partition AllReduce after SPMD partition
2019-12-03 13:32:47 -08:00
all_reduce_simplifier.h
…
allocation_tracker.cc
…
allocation_tracker.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
ar_crs_combiner_test.cc
Verify replica count from AllReduce replica group config
2020-04-27 08:53:44 -07:00
ar_crs_combiner.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-06 17:55:23 -07:00
ar_crs_combiner.h
Combine cross-replica / cross-partition AllReduce after SPMD partition
2019-12-03 13:32:47 -08:00
backend.cc
Replace NumSchedulableCPUs() with MaxParallelism()
2019-06-28 15:10:56 -07:00
backend.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
batch_dot_simplification_test.cc
…
batch_dot_simplification.cc
…
batch_dot_simplification.h
…
batchnorm_expander_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
batchnorm_expander.cc
[XLA] Reduce the amount of compute performed on the operand shape in
2019-09-25 13:29:46 -07:00
batchnorm_expander.h
[XLA:GPU] Mark constructor explicit to comply with C++ style.
2019-08-08 07:38:33 -07:00
bfloat16_conversion_folding_test.cc
Enable using global ids for replica groups in AllReduce
2020-02-21 19:01:52 -08:00
bfloat16_conversion_folding.cc
[XLA] Extract a common visitor for rewriting instructions.
2019-07-09 16:44:10 -07:00
bfloat16_conversion_folding.h
…
bfloat16_normalization_test.cc
Add optional layout constraint to AllToAll
2020-04-27 18:37:21 -07:00
bfloat16_normalization.cc
[XLA] Make BF16 normalization better handle tuple-shaped inputs/outputs.
2019-10-17 23:06:11 -07:00
bfloat16_normalization.h
…
bfloat16_propagation_test.cc
[XLA] Support conditional in bf16 propagation
2020-04-05 16:46:27 -07:00
bfloat16_propagation.cc
[XLA] Support conditional in bf16 propagation
2020-04-05 16:46:27 -07:00
bfloat16_propagation.h
[XLA] Support conditional in bf16 propagation
2020-04-05 16:46:27 -07:00
bfloat16_support.cc
[XLA] Introducing AllGather HLO and a decomposition pass
2020-05-09 12:22:17 -07:00
bfloat16_support.h
…
buffer_assignment_test.cc
[XLA] Better support for while loops in memory space assignment.
2020-04-13 10:37:30 -07:00
buffer_assignment.cc
[XLA] BufferValue::Color now type aliases int64.
2020-05-27 09:35:14 -07:00
buffer_assignment.h
[XLA] BufferValue::Color now type aliases int64.
2020-05-27 09:35:14 -07:00
buffer_value_containers.h
…
buffer_value.cc
[XLA] BufferValue::Color now type aliases int64.
2020-05-27 09:35:14 -07:00
buffer_value.h
[XLA] BufferValue::Color now type aliases int64.
2020-05-27 09:35:14 -07:00
BUILD
Support dynamic outputs for XLA on demand ops.
2020-07-01 11:11:47 -07:00
call_graph_test.cc
…
call_graph.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-06 17:55:23 -07:00
call_graph.h
…
call_inliner_test.cc
Convert absl string_view constants to const, rather than constexpr.
2020-02-05 11:15:53 -08:00
call_inliner.cc
[XLA:CPU] Restrict CallInliner to functions with a single call site
2020-01-23 09:19:45 -08:00
call_inliner.h
[XLA:CPU] Restrict CallInliner to functions with a single call site
2020-01-23 09:19:45 -08:00
channel_tracker.cc
…
channel_tracker.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
cholesky_expander.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
cholesky_expander.h
…
collective_ops_utils.cc
[XLA:CPU/GPU] Implement all-reduce and/or of pred
2020-03-30 11:10:37 -07:00
collective_ops_utils.h
[XLA:CPU] CollectivePermute support on CPU
2020-04-06 22:07:56 -07:00
compilation_cache.cc
…
compilation_cache.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
compilation_stats.cc
…
compilation_stats.h
…
compile_only_service.cc
[XLA] Expose SPMD/MPMD as an option
2020-05-11 16:07:03 -07:00
compile_only_service.h
…
compiler.cc
Add a pass that converts MLIR HLO to MLIR LHLO by using the XLA pipeline
2020-04-22 21:26:39 -07:00
compiler.h
[XLA] Expose SPMD/MPMD as an option
2020-05-11 16:07:03 -07:00
computation_layout.cc
* Add logic to compare computation layout.
2019-08-22 17:04:34 -07:00
computation_layout.h
* Add logic to compare computation layout.
2019-08-22 17:04:34 -07:00
computation_placer.cc
Register ROCm platform on XLA computation placer
2019-07-03 22:17:29 +00:00
computation_placer.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
conditional_code_motion_test.cc
Hoisting unconditional converts from conditional branch computations.
2020-06-18 20:58:13 -07:00
conditional_code_motion.cc
Hoisting unconditional converts from conditional branch computations.
2020-06-18 20:58:13 -07:00
conditional_code_motion.h
Hoisting unconditional converts from conditional branch computations.
2020-06-18 20:58:13 -07:00
conditional_simplifier_test.cc
[XLA] Add token support to conditional_simplifier
2020-06-02 11:00:01 -07:00
conditional_simplifier.cc
[XLA] Add token support to conditional_simplifier
2020-06-02 11:00:01 -07:00
conditional_simplifier.h
…
conditional_to_select_test.cc
…
conditional_to_select.cc
…
conditional_to_select.h
…
convolution_4d_expander_test.cc
Convert 4D convolutions with trivial dimensions to lower-dimensional convolutions.
2020-01-28 02:40:48 -08:00
convolution_4d_expander.cc
Convert 4D convolutions with trivial dimensions to lower-dimensional convolutions.
2020-01-28 02:40:48 -08:00
convolution_4d_expander.h
Convert 4D convolutions with trivial dimensions to lower-dimensional convolutions.
2020-01-28 02:40:48 -08:00
convolution_group_converter_test.cc
[XLA] Fix the condition for rewriting batch group convolutions to include when the input batch is not equal to the batch group count.
2020-04-08 16:50:13 -07:00
convolution_group_converter.cc
Handle batch group conv - backward filter grouped conv
2020-05-11 17:21:23 -07:00
convolution_group_converter.h
Fix confusing way of specifying filter_expansion parameter.
2020-01-08 00:43:03 -08:00
copy_insertion_test.cc
Adding additional tests that exercise CopyInsertion when a while feeds another while. The structure of the added tests is as follows:
2020-06-10 22:31:55 -07:00
copy_insertion.cc
Iterate through snapshotted instructions in copy insertion.
2020-03-10 11:35:18 -07:00
copy_insertion.h
Remove the use of buffer liveness.
2019-07-10 11:18:56 -07:00
custom_call_target_registry.cc
…
custom_call_target_registry.h
…
defuser_test.cc
…
defuser.cc
…
defuser.h
…
despecializer.cc
…
despecializer.h
…
dfs_hlo_visitor_with_default_test.cc
…
dfs_hlo_visitor_with_default.h
[XLA] Introduce asynchronous collective-permute (CollectivePermuteStart and CollectivePermuteDone) HLO opcodes.
2020-05-21 21:48:39 -07:00
dfs_hlo_visitor.cc
…
dfs_hlo_visitor.h
[XLA] Introduce asynchronous collective-permute (CollectivePermuteStart and CollectivePermuteDone) HLO opcodes.
2020-05-21 21:48:39 -07:00
dot_as_convolution_util.cc
[XLA] Move sharding propagation to third party
2020-05-22 18:18:30 -07:00
dot_as_convolution_util.h
[XLA] Move sharding propagation to third party
2020-05-22 18:18:30 -07:00
dot_decomposer_test.cc
Improve DotDecomposer to not add unnecessary non-contracting dimensions.
2020-05-26 05:49:19 -07:00
dot_decomposer.cc
Polish some comments in dot decomposer.
2020-06-15 14:41:41 -07:00
dot_decomposer.h
…
dump.cc
lsc. no new code.
2020-04-16 04:23:49 -07:00
dump.h
[XLA] Add a configurable limit to the number of HLO dumps
2020-02-03 22:32:17 -08:00
dynamic_dimension_inference_test.cc
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -07:00
dynamic_dimension_inference.cc
[XLA]Add HLO infrastructure to support dynamic op lowering.
2020-04-30 17:00:49 -07:00
dynamic_dimension_inference.h
[XLA]Add HLO infrastructure to support dynamic op lowering.
2020-04-30 17:00:49 -07:00
dynamic_index_splitter_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
dynamic_index_splitter.cc
…
dynamic_index_splitter.h
…
dynamic_padder_test.cc
[XLA]Add HLO infrastructure to support dynamic op lowering.
2020-04-30 17:00:49 -07:00
dynamic_padder.cc
[XLA]Add HLO infrastructure to support dynamic op lowering.
2020-04-30 17:00:49 -07:00
dynamic_padder.h
[XLA]Add HLO infrastructure to support dynamic op lowering.
2020-04-30 17:00:49 -07:00
dynamic_parameter_binding_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
dynamic_parameter_binding.cc
…
dynamic_parameter_binding.h
…
dynamic_update_slice_test.cc
…
elemental_ir_emitter_test.cc
[XLA] Fix type conversion from and to bf16 and improve test coverage.
2019-11-26 10:31:34 -08:00
elemental_ir_emitter.cc
Enable XLA:CPU fast math for min/max by default to be similar to TF's behavior.
2020-06-17 01:27:08 -07:00
elemental_ir_emitter.h
Enable XLA:CPU fast math for min/max by default to be similar to TF's behavior.
2020-06-17 01:27:08 -07:00
executable.cc
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
2020-06-22 12:32:43 -07:00
executable.h
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
2020-06-22 12:32:43 -07:00
execution_tracker.cc
…
execution_tracker.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
flatten_call_graph_test.cc
…
flatten_call_graph.cc
…
flatten_call_graph.h
…
fusion_node_indexing_evaluation_test.cc
Replace IsFusedIrEmitterInefficient with FusionNodeIndexingEvaluation class.
2020-02-03 06:08:01 -08:00
fusion_node_indexing_evaluation.cc
Don't replace Transposes with Bitcasts on the GPU backend.
2020-04-02 02:07:13 -07:00
fusion_node_indexing_evaluation.h
Replace IsFusedIrEmitterInefficient with FusionNodeIndexingEvaluation class.
2020-02-03 06:08:01 -08:00
fusion_queue.h
[XLA] Minor clean-up of fusion_queue.h.
2019-08-19 06:33:00 -07:00
gather_expander_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
gather_expander.cc
…
gather_expander.h
…
generic_transfer_manager.cc
[XLA:GPU] Don't defer host callbacks until BlockHostUntilDone().
2020-02-12 10:21:01 -08:00
generic_transfer_manager.h
[XLA] Implement dynamic input and output in DynamicPadder.
2019-12-12 10:31:20 -08:00
heap_simulator_test.cc
[XLA] Add a remove method in heap simulator's interval tree.
2020-02-27 11:31:52 -08:00
heap_simulator.cc
[XLA] Add a remove method in heap simulator's interval tree.
2020-02-27 11:31:52 -08:00
heap_simulator.h
[XLA] Implement an allocation retry mechanism.
2020-06-04 16:37:06 -07:00
hlo_alias_analysis_test.cc
Make use of the CreateBitcast method.
2019-07-19 17:07:30 -07:00
hlo_alias_analysis.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_alias_analysis.h
…
hlo_buffer.cc
…
hlo_buffer.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_casting_utils_test.cc
…
hlo_casting_utils.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_clone_context.h
…
hlo_computation_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
hlo_computation.cc
[XLA] Turn variadic reduces with a single input and output into normal reduces
2020-06-10 22:44:39 -07:00
hlo_computation.h
[XLA] Turn variadic reduces with a single input and output into normal reduces
2020-06-10 22:44:39 -07:00
hlo_constant_folding_test.cc
Replace absl::Span where we use initializer lists in tests with std::vector.
2020-02-28 17:37:42 -08:00
hlo_constant_folding.cc
…
hlo_constant_folding.h
…
hlo_cost_analysis_test.cc
[XLA] Separate bytes accessed in cost analysis by operands and outputs.
2019-10-23 16:02:21 -07:00
hlo_cost_analysis.cc
[XLA:TPU] Move per-memory-space bytes read/written code to HloCostAnalysis.
2020-05-26 16:04:09 -07:00
hlo_cost_analysis.h
[XLA:TPU] Move per-memory-space bytes read/written code to HloCostAnalysis.
2020-05-26 16:04:09 -07:00
hlo_creation_utils_test.cc
Merge pull request #32264 from nouiz:utils_pr_hlo_creation
2019-09-09 16:28:58 -07:00
hlo_creation_utils.cc
Implement Mul(Convert(Pred), operand) => select(pred, operand, 0) optimization.
2020-06-16 13:00:07 -07:00
hlo_creation_utils.h
Implement Mul(Convert(Pred), operand) => select(pred, operand, 0) optimization.
2020-06-16 13:00:07 -07:00
hlo_cse_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_cse.cc
[XLA] Avoid hash collisions in CseHash.
2020-02-20 00:01:30 -08:00
hlo_cse.h
…
hlo_dataflow_analysis_test.cc
[XLA] Create explicit phi graph optimization during dataflow analysis.
2020-03-17 14:14:22 -07:00
hlo_dataflow_analysis.cc
Properly support nest phi reduction in reverse order.
2020-06-17 18:47:22 -07:00
hlo_dataflow_analysis.h
[XLA] Adds transformation from collective-permute to the asynchronous version, and adds emitter for the asynchronous version.
2020-06-01 18:54:45 -07:00
hlo_dce_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
hlo_dce.cc
[XLA] Introducing AllGather HLO and a decomposition pass
2020-05-09 12:22:17 -07:00
hlo_dce.h
[XLA] Resubmit DCE change
2020-03-03 17:43:00 -08:00
hlo_domain_isolator.cc
…
hlo_domain_isolator.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_domain_map.cc
…
hlo_domain_map.h
…
hlo_domain_metadata.h
…
hlo_domain_remover.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_domain_remover.h
…
hlo_domain_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_domain_verifier.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_domain_verifier.h
…
hlo_element_type_converter_test.cc
…
hlo_element_type_converter.cc
…
hlo_element_type_converter.h
…
hlo_evaluator_test.cc
[XLA] Add support for bf16 map() to HloEvaluator
2020-04-22 05:35:36 -07:00
hlo_evaluator_typed_visitor_bfloat16.cc
…
hlo_evaluator_typed_visitor_bool.cc
…
hlo_evaluator_typed_visitor_complex64.cc
…
hlo_evaluator_typed_visitor_complex128.cc
…
hlo_evaluator_typed_visitor_double.cc
…
hlo_evaluator_typed_visitor_float.cc
…
hlo_evaluator_typed_visitor_half.cc
…
hlo_evaluator_typed_visitor_int8.cc
…
hlo_evaluator_typed_visitor_int16.cc
…
hlo_evaluator_typed_visitor_int32.cc
…
hlo_evaluator_typed_visitor_int64.cc
…
hlo_evaluator_typed_visitor_uint8.cc
…
hlo_evaluator_typed_visitor_uint16.cc
…
hlo_evaluator_typed_visitor_uint32.cc
…
hlo_evaluator_typed_visitor_uint64.cc
…
hlo_evaluator_typed_visitor.h
[XLA] Add Cbrt operation (lowered to pow(1.0/3.0))
2020-05-02 00:28:18 -07:00
hlo_evaluator.cc
Have the HloEvaluator print out intermediate values on VLOG(100)
2020-06-19 22:14:31 -07:00
hlo_evaluator.h
[XLA:CPU] Wire up C64/C128 matmul to Eigen
2020-05-27 09:56:35 -07:00
hlo_execution_profile_data.proto
Adds new protobuf message HloExecutionProfileData, which describes HloExecutionProfile.
2019-09-16 06:08:02 -07:00
hlo_execution_profile_test.cc
Rewrite the pattern we want to match in hlo_execution_profile_test.
2020-01-27 05:22:03 -08:00
hlo_execution_profile.cc
Adds new protobuf message HloExecutionProfileData, which describes HloExecutionProfile.
2019-09-16 06:08:02 -07:00
hlo_execution_profile.h
Adds new protobuf message HloExecutionProfileData, which describes HloExecutionProfile.
2019-09-16 06:08:02 -07:00
hlo_get_dimension_size_rewriter_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
hlo_get_dimension_size_rewriter.cc
Wire dynamic shape support to XRT.
2019-11-18 17:28:41 -08:00
hlo_get_dimension_size_rewriter.h
…
hlo_graph_dumper_test.cc
…
hlo_graph_dumper.cc
Roll forward "Add a show_fusion_subcomputations command to interactive_graphviz" with fix
2020-05-27 11:27:39 -07:00
hlo_graph_dumper.h
Roll forward "Add a show_fusion_subcomputations command to interactive_graphviz" with fix
2020-05-27 11:27:39 -07:00
hlo_input_output_alias_config_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
hlo_input_output_alias_config.cc
[XLA] Serialize aliasing to HloModule textual representation
2020-06-10 16:01:33 -07:00
hlo_input_output_alias_config.h
[XLA] Serialize aliasing to HloModule textual representation
2020-06-10 16:01:33 -07:00
hlo_instruction_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_instruction.cc
Add set_outfeed_config in XLA HloInstruction.
2020-06-18 18:19:13 -07:00
hlo_instruction.h
Add set_outfeed_config in XLA HloInstruction.
2020-06-18 18:19:13 -07:00
hlo_instructions.cc
[XLA] Make IdenticalSlowPath more strict
2020-06-16 15:10:42 -07:00
hlo_instructions.h
Add set_outfeed_config in XLA HloInstruction.
2020-06-18 18:19:13 -07:00
hlo_lexer.cc
[XLA] Unbreak the OSS build
2020-05-20 01:42:35 -07:00
hlo_lexer.h
[XLA] Remove unsupported sparse layout
2020-01-02 18:04:52 -08:00
hlo_live_range_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_live_range.cc
[XLA] Fix live range flattening for while loops.
2019-11-01 15:50:08 -07:00
hlo_live_range.h
[XLA] Implement memory space allocation across sequential calls (e.g. while).
2019-11-20 14:24:43 -08:00
hlo_liveness_analysis_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_liveness_analysis.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_liveness_analysis.h
…
hlo_matchers_test.cc
[XLA] Change CopyStart return type to (target, source, context)
2020-01-28 16:11:12 -08:00
hlo_matchers.cc
[XLA] Improve error message in HLO matchers.
2019-07-18 12:38:55 -07:00
hlo_matchers.h
[XLA] Add AllGather to the HLO matchers.
2020-05-21 22:29:00 -07:00
hlo_memory_scheduler_test.cc
Explicitly tell lambdas to capture existing local variables by copy.
2020-02-05 14:52:20 -08:00
hlo_memory_scheduler.cc
If an outfeed instruction has a config, schedule it early. If an infeed instruction has a config, schedule it late.
2020-06-03 22:15:37 -07:00
hlo_memory_scheduler.h
[XLA] Use module scheduler to reduce compile time.
2019-07-16 14:16:07 -07:00
hlo_module_config.cc
[XLA] Create an alias_passthrough_params config option.
2019-08-29 04:34:28 -07:00
hlo_module_config.h
Modify signature of layout_config().
2020-05-18 15:06:31 -07:00
hlo_module_dce_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_module_dce.cc
…
hlo_module_dce.h
…
hlo_module_group_metadata.cc
Change all_reduce_id to use channel_id
2019-06-26 20:37:58 -07:00
hlo_module_group_metadata.h
Change all_reduce_id to use channel_id
2019-06-26 20:37:58 -07:00
hlo_module_group_test.cc
Use VerifiedHloModule in more tests.
2019-10-18 09:17:33 -07:00
hlo_module_group_util.cc
Change all_reduce_id to use channel_id
2019-06-26 20:37:58 -07:00
hlo_module_group_util.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_module_group.cc
[XLA] Add option to print out hlo pass hash.
2020-02-27 14:54:06 -08:00
hlo_module_group.h
[XLA] Add option to print out hlo pass hash.
2020-02-27 14:54:06 -08:00
hlo_module_test.cc
Use VerifiedHloModule in tests that already have valid HLO.
2019-10-21 05:52:12 -07:00
hlo_module.cc
[XLA] Serialize aliasing to HloModule textual representation
2020-06-10 16:01:33 -07:00
hlo_module.h
Teach MemorySpaceAssignment to find buffers to keep in alternate memory across program scopes.
2020-03-25 14:12:52 -07:00
hlo_opcode_test.cc
…
hlo_opcode.cc
…
hlo_opcode.h
[XLA] Introduce asynchronous collective-permute (CollectivePermuteStart and CollectivePermuteDone) HLO opcodes.
2020-05-21 21:48:39 -07:00
hlo_ordering_test.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_ordering.cc
…
hlo_ordering.h
…
hlo_parser_test.cc
[XLA] Serialize aliasing to HloModule textual representation
2020-06-10 16:01:33 -07:00
hlo_parser.cc
[XLA] Serialize aliasing to HloModule textual representation
2020-06-10 16:01:33 -07:00
hlo_parser.h
Move ParseHloString to VerifiedHloModule.
2019-10-22 06:54:21 -07:00
hlo_pass_fix.h
[XLA] Lower verbosity of "Unexpectedly high number of iterations" warning.
2020-04-29 09:34:46 -07:00
hlo_pass_interface.h
…
hlo_pass_pipeline_test.cc
…
hlo_pass_pipeline.cc
[XLA] Add option to print out hlo pass hash.
2020-02-27 14:54:06 -08:00
hlo_pass_pipeline.h
[XLA] Stop verifying the HLO inside the pass pipeline on optimized builds.
2020-04-02 08:19:57 -07:00
hlo_phi_graph_test.cc
Properly support nest phi reduction in reverse order.
2020-06-17 18:47:22 -07:00
hlo_phi_graph.cc
Properly support nest phi reduction in reverse order.
2020-06-17 18:47:22 -07:00
hlo_phi_graph.h
Properly support nest phi reduction in reverse order.
2020-06-17 18:47:22 -07:00
hlo_profile_printer_data.proto
…
hlo_profile_printer.cc
…
hlo_profile_printer.h
…
hlo_proto_util_test.cc
…
hlo_proto_util.cc
…
hlo_proto_util.h
…
hlo_query.cc
Add some XLA frontend attribute names.
2020-01-16 19:01:38 -08:00
hlo_query.h
Add some XLA frontend attribute names.
2020-01-16 19:01:38 -08:00
hlo_reachability_test.cc
Use VerifiedHloModule in tests that already have valid HLO.
2019-10-21 05:52:12 -07:00
hlo_reachability.cc
Allow split-phase operations for HloReachabilityMap where we can
2020-06-14 14:46:08 -07:00
hlo_reachability.h
Allow split-phase operations for HloReachabilityMap where we can
2020-06-14 14:46:08 -07:00
hlo_rematerialization_test_utils_test.cc
…
hlo_rematerialization_test_utils.h
…
hlo_rematerialization_test.cc
Implements an incremental greedy algorithm for rematerialization. Whenever the memory use at any point in the execution goes over the memory limit, we attempt to rematerialize one instruction at a time to reduce memory usage. If no such candidate is found, we expand the search space to instead rematerialize 2 consecutive instructions at a time, and so on doubling the search space every time. At any point, if we find a block of instructions to rematerialize that reduces the memory usage, we reset the block size to 1 to resume the algorithm.
2020-02-04 15:52:32 -08:00
hlo_rematerialization.cc
Adds a max_rematerialized_block_size field.
2020-03-16 11:49:49 -07:00
hlo_rematerialization.h
Adds a max_rematerialized_block_size field.
2020-03-16 11:49:49 -07:00
hlo_replication_analysis_test.cc
[XLA] Make replication analysis consider AllGather and global device IDs.
2020-06-10 13:32:48 -07:00
hlo_replication_analysis.cc
[XLA] Make replication analysis consider AllGather and global device IDs.
2020-06-10 13:32:48 -07:00
hlo_replication_analysis.h
Combine cross-replica / cross-partition AllReduce after SPMD partition
2019-12-03 13:32:47 -08:00
hlo_runner.cc
[XLA] Support aliasing in HloRunner
2020-06-10 17:17:41 -07:00
hlo_runner.h
[XLA] Support aliasing in HloRunner
2020-06-10 17:17:41 -07:00
hlo_schedule_test.cc
Use VerifiedHloModule in tests that already have valid HLO.
2019-10-21 05:52:12 -07:00
hlo_schedule.cc
…
hlo_schedule.h
Add interfaces to modify an HLO module after already being scheduled.
2019-07-17 13:51:58 -07:00
hlo_sharding_metadata.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
hlo_sharding_metadata.h
…
hlo_sharding_test.cc
Fix uses of private, mangled names for proto enumerators.
2019-07-03 19:17:10 -07:00
hlo_sharding_util_test.cc
[XLA] Move SPMD partitioner to third_party
2020-05-13 11:28:00 -07:00
hlo_sharding_util.cc
[XLA:SPMD] More cases of reverse sharding
2020-05-28 15:44:17 -07:00
hlo_sharding_util.h
[XLA:SPMD] More cases of reverse sharding
2020-05-28 15:44:17 -07:00
hlo_sharding.cc
Fixes a previously unhandled case of a tuple with zero leaf nodes in
2020-04-17 00:30:15 -07:00
hlo_sharding.h
Add a util to get the tile shape on a device
2020-03-26 15:46:16 -07:00
hlo_subcomputation_unification_test.cc
Fix some HLO bugs in tests detected when using VerifiedHloModule.
2019-10-31 08:01:08 -07:00
hlo_subcomputation_unification.cc
…
hlo_subcomputation_unification.h
…
hlo_value.cc
[XLA] BufferValue::Color now type aliases int64.
2020-05-27 09:35:14 -07:00
hlo_value.h
Remove use of tuple points to analysis in buffer assignment.
2019-07-09 18:14:15 -07:00
hlo_verifier_test.cc
[XLA] Adds transformation from collective-permute to the asynchronous version, and adds emitter for the asynchronous version.
2020-06-01 18:54:45 -07:00
hlo_verifier.cc
Check the value shape of set dimension size in shape inference.
2020-06-19 12:07:45 -07:00
hlo_verifier.h
[XLA] Introduce asynchronous collective-permute (CollectivePermuteStart and CollectivePermuteDone) HLO opcodes.
2020-05-21 21:48:39 -07:00
hlo.proto
[XLA] Change backend_config from string to bytes
2020-04-24 17:27:27 -07:00
human_readable_profile_builder.cc
…
human_readable_profile_builder.h
…
indexed_array_analysis_test.cc
Remove 1 + "string" cases and undo -Wno-string-plus-int.
2020-02-05 16:58:37 -08:00
indexed_array_analysis.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
indexed_array_analysis.h
…
instruction_fusion_test.cc
rng loop fusion
2020-06-05 10:34:16 -07:00
instruction_fusion.cc
[XLA] Introduce asynchronous collective-permute (CollectivePermuteStart and CollectivePermuteDone) HLO opcodes.
2020-05-21 21:48:39 -07:00
instruction_fusion.h
Add a method FuseInstruction to InstructionFusion.
2020-01-30 07:24:23 -08:00
layout_assignment_test.cc
Make linear layout more explicit.
2020-06-18 06:55:02 -07:00
layout_assignment.cc
Make linear layout more explicit.
2020-06-18 06:55:02 -07:00
layout_assignment.h
minor spelling tweaks
2020-01-16 14:36:52 +09:00
llvm_compiler.cc
[XLA] Remove unused method RunBackendOnModuleGroup
2019-08-19 10:14:14 -07:00
llvm_compiler.h
Update paths to LLVM.
2019-11-04 13:53:39 -08:00
local_service.cc
[XLA] Expose SPMD/MPMD as an option
2020-05-11 16:07:03 -07:00
local_service.h
Change LocalClient::Compile to support returning multiple executables (one per partition).
2020-01-29 01:03:57 -08:00
logical_buffer_analysis.cc
[XLA] Change CopyStart return type to (target, source, context)
2020-01-28 16:11:12 -08:00
logical_buffer_analysis.h
[XLA] Change CopyStart return type to (target, source, context)
2020-01-28 16:11:12 -08:00
logical_buffer.cc
[XLA] BufferValue::Color now type aliases int64.
2020-05-27 09:35:14 -07:00
logical_buffer.h
…
map_inliner_test.cc
…
map_inliner.cc
…
map_inliner.h
…
maybe_owning_device_memory.cc
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
2020-06-22 12:32:43 -07:00
maybe_owning_device_memory.h
Make XRT CPU/GPU use MaybeOwning buffer interface, so the new copy protection CL won't break aliasing.
2020-06-22 12:32:43 -07:00
memory_space_assignment_test.cc
[XLA] More tweaks and fixes to memory space assignment:
2020-06-16 16:15:14 -07:00
memory_space_assignment_utils.cc
[XLA] Refactor memory_space_assignment.cc.
2020-06-15 15:22:06 -07:00
memory_space_assignment_utils.h
[XLA] Refactor memory_space_assignment.cc.
2020-06-15 15:22:06 -07:00
memory_space_assignment.cc
[XLA] More tweaks and fixes to memory space assignment:
2020-06-16 16:15:14 -07:00
memory_space_assignment.h
[XLA] More tweaks and fixes to memory space assignment:
2020-06-16 16:15:14 -07:00
memory_space_propagation_test.cc
[XLA] Propagate memory spaces recursively inside nested fusions.
2020-06-18 13:49:23 -07:00
memory_space_propagation.cc
[XLA] Propagate memory spaces recursively inside nested fusions.
2020-06-18 13:49:23 -07:00
memory_space_propagation.h
[XLA] Propagate memory spaces recursively inside nested fusions.
2020-06-18 13:49:23 -07:00
multi_output_fusion.cc
Clean up post fusion using DCE.
2020-06-19 11:52:26 -07:00
multi_output_fusion.h
[Resubmit] Fix several issues of multi output fusion.
2020-02-18 20:50:32 -08:00
name_uniquer_test.cc
…
name_uniquer.cc
…
name_uniquer.h
…
op_expander_pass.cc
…
op_expander_pass.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
optimize_input_output_buffer_alias_test.cc
Automated rollback of commit afd980b3ce068c30bd3a43d54929e2590be8bd29
2020-02-12 14:31:05 -08:00
optimize_input_output_buffer_alias.cc
Automated rollback of commit afd980b3ce068c30bd3a43d54929e2590be8bd29
2020-02-12 14:31:05 -08:00
optimize_input_output_buffer_alias.h
Automated rollback of commit afd980b3ce068c30bd3a43d54929e2590be8bd29
2020-02-12 14:31:05 -08:00
pattern_matcher_gmock_test.cc
[XLA] Remove unsupported sparse layout
2020-01-02 18:04:52 -08:00
pattern_matcher_gmock.h
…
pattern_matcher_test.cc
[XLA] Remove unsupported sparse layout
2020-01-02 18:04:52 -08:00
pattern_matcher.h
Standardize how pointer addresses are printed in xla pattern matcher across operating systems.
2020-03-12 22:00:32 -07:00
platform_util.cc
Cleanup PlatformUtil and MultiPlatformManager. NFC.
2019-11-08 20:01:51 -08:00
platform_util.h
Cleanup PlatformUtil and MultiPlatformManager. NFC.
2019-11-08 20:01:51 -08:00
reshape_mover_test.cc
…
reshape_mover.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
reshape_mover.h
…
rng_bit_generator_expander.cc
Rewrite output state to match the input.
2020-04-29 15:51:55 -07:00
rng_bit_generator_expander.h
Introduce new RngBitGenerator HLO
2020-02-06 05:06:44 -08:00
rng_expander.cc
Introduce new RngBitGenerator HLO
2020-02-06 05:06:44 -08:00
rng_expander.h
[XLA] Cache expanded RNG computations
2020-01-23 10:39:48 -08:00
root_instruction_sinker_test.cc
[XLA] Add a RootInstructionSinker pass.
2020-04-22 14:46:15 -07:00
root_instruction_sinker.cc
[XLA] Add a RootInstructionSinker pass.
2020-04-22 14:46:15 -07:00
root_instruction_sinker.h
[XLA] Add a RootInstructionSinker pass.
2020-04-22 14:46:15 -07:00
scatter_expander_test.cc
[XLA] Fix ScatterExpander to handle operands with no layout shapes.
2019-07-02 08:45:05 -07:00
scatter_expander.cc
…
scatter_expander.h
…
service_executable_run_options.h
…
service.cc
[XLA] Expose SPMD/MPMD as an option
2020-05-11 16:07:03 -07:00
service.h
[XLA] Wire through static device assignments from the Python client to the compiler
2020-02-21 17:42:43 -08:00
shape_inference_test.cc
Check the value shape of set dimension size in shape inference.
2020-06-19 12:07:45 -07:00
shape_inference.cc
Check the value shape of set dimension size in shape inference.
2020-06-19 12:07:45 -07:00
shape_inference.h
Check the value shape of set dimension size in shape inference.
2020-06-19 12:07:45 -07:00
shaped_buffer_test.cc
Cleanup PlatformUtil and MultiPlatformManager. NFC.
2019-11-08 20:01:51 -08:00
shaped_buffer.cc
…
shaped_buffer.h
[XLA] Correct ScopedShapedBuffer documentation
2020-05-27 16:24:22 -07:00
sharding_propagation_test.cc
[XLA] Move sharding propagation to third party
2020-05-22 18:18:30 -07:00
sharding_propagation.cc
[XLA] Avoid sharding propagation on trivial dims for dot-general convolution
2020-06-10 23:03:39 -07:00
sharding_propagation.h
[XLA] Move sharding propagation to third party
2020-05-22 18:18:30 -07:00
slice_sinker_test.cc
…
slice_sinker.cc
…
slice_sinker.h
…
slow_operation_alarm.cc
Change std::call_once to absl::call_once
2020-01-29 08:40:19 -08:00
slow_operation_alarm.h
Add absl/base/attributes.h to slow_operation_alarm.h as ABSL_MUST_USE_RESULT is used
2020-02-12 18:22:02 -08:00
sort_simplifier_test.cc
…
sort_simplifier.cc
…
sort_simplifier.h
…
source_map_util.h
…
stable_sort_expander_test.cc
…
stable_sort_expander.cc
…
stable_sort_expander.h
…
stream_pool_test.cc
…
stream_pool.cc
…
stream_pool.h
Prefixing TensorFlow thread annotation macros with TF_.
2020-03-05 08:42:01 -08:00
transfer_manager.cc
Support dynamic outputs for XLA on demand ops.
2020-07-01 11:11:47 -07:00
transfer_manager.h
Support dynamic outputs for XLA on demand ops.
2020-07-01 11:11:47 -07:00
transpose_folding_test.cc
Fix some HLO bugs in tests detected when using VerifiedHloModule.
2019-10-31 08:01:08 -07:00
transpose_folding.cc
…
transpose_folding.h
[XLA] Add the ability to disable the Reduce(Reshape(X)) to Reduce(X) rewrite and have TransposeFolding default to folding all operands of Convolution and Dot
2020-03-07 00:36:17 -08:00
tree_reduction_rewriter.cc
[XLA CPU] Apply a tree reduction rewriter pass on a CPU to reduce the error margin
2019-08-26 11:35:52 -07:00
tree_reduction_rewriter.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
triangular_solve_expander.cc
[XLA] Fix shape error in triangular solve expander for left_side=False where more than one block is present.
2020-03-27 11:49:28 -07:00
triangular_solve_expander.h
…
tuple_points_to_analysis_test.cc
[XLA] Change CopyStart return type to (target, source, context)
2020-01-28 16:11:12 -08:00
tuple_points_to_analysis.cc
[XLA] Change CopyStart return type to (target, source, context)
2020-01-28 16:11:12 -08:00
tuple_points_to_analysis.h
[XLA] Change CopyStart return type to (target, source, context)
2020-01-28 16:11:12 -08:00
tuple_simplifier_test.cc
…
tuple_simplifier.cc
[XLA:HLO] Small refactoring and more comments in tuple_simplifier.
2020-06-08 11:52:07 -07:00
tuple_simplifier.h
[XLA:HLO] Small refactoring and more comments in tuple_simplifier.
2020-06-08 11:52:07 -07:00
tuple_util_test.cc
Move ParseHloString to VerifiedHloModule.
2019-10-22 06:54:21 -07:00
tuple_util.cc
…
tuple_util.h
[XLA] Add a RootInstructionSinker pass.
2020-04-22 14:46:15 -07:00
while_loop_analysis_test.cc
…
while_loop_analysis.cc
…
while_loop_analysis.h
…
while_loop_constant_sinking_test.cc
Use VerifiedHloModule in tests that already have valid HLO.
2019-10-21 05:52:12 -07:00
while_loop_constant_sinking.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
while_loop_constant_sinking.h
…
while_loop_invariant_code_motion_test.cc
…
while_loop_invariant_code_motion.cc
[XLA] Correct WhileLoopInvariantCodeMotion log messages.
2020-05-07 13:35:38 -07:00
while_loop_invariant_code_motion.h
[XLA] Allow custom shape size in loop-invariant code motion
2019-06-27 22:28:08 -07:00
while_loop_simplifier_test.cc
[XLA] Sink side effect check in while loop simplifier only when it's needed.
2020-04-16 18:56:30 -07:00
while_loop_simplifier.cc
[XLA] Sink side effect check in while loop simplifier only when it's needed.
2020-04-16 18:56:30 -07:00
while_loop_simplifier.h
…
while_loop_trip_count_annotator_test.cc
…
while_loop_trip_count_annotator.cc
…
while_loop_trip_count_annotator.h
…
while_util_test.cc
Fix some HLO bugs in tests detected when using VerifiedHloModule.
2019-10-31 08:01:08 -07:00
while_util.cc
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -07:00
while_util.h
[XLA][TF2XLA] Support tensor list with dynamic dimension.
2020-04-22 14:37:18 -07:00
zero_sized_hlo_elimination_test.cc
Fix some HLO bugs in tests detected when using VerifiedHloModule.
2019-10-31 08:01:08 -07:00
zero_sized_hlo_elimination.cc
…
zero_sized_hlo_elimination.h
…