experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/xla/service
History
George Karpenkov de5bfc8bb6 [XLA:GPU] [NFC] Clarify types in AllReduceCombiner 
PiperOrigin-RevId: 361234109
Change-Id: I5a89b41f600d6e03c75366fb41e1b5fdbefe9f1a
2021-03-05 15:16:41 -08:00
..
cpu
[XLA] Have XLA:CPU depend on AArch64CodeGen in all ARM configurations
2021-03-04 17:34:20 -08:00
g3doc
…
gpu
[XLA:GPU] Enforce collectives ordering to be their appearence in the module
2021-03-04 21:47:17 -08:00
graphcycles
…
interpreter
[XLA] Change eigh client library to produce a CustomCall that is expanded by an expander pass.
2021-03-01 08:12:43 -08:00
llvm_ir
Ensure that trivial types are passed by value rather than reference
2021-03-04 22:09:46 -08:00
mlir_gpu
[XLA:GPU] Fix build failure
2021-02-26 14:24:18 -08:00
spmd
[XLA] Make the canonicalize all-gather pass add a new/exclusive all-gather dimension.
2021-03-03 01:06:58 -08:00
algebraic_simplifier_proof_distributive_property.py
…
algebraic_simplifier_test.cc
Internal clean up
2021-02-22 08:46:38 -08:00
algebraic_simplifier.cc
[XLA] Simplify unsigned comparisons against 0
2021-02-12 23:10:45 -08:00
algebraic_simplifier.h
…
all_gather_decomposer_test.cc
…
all_gather_decomposer.cc
[XLA] Fix some all-gather issues.
2020-05-19 17:55:08 -07:00
all_gather_decomposer.h
…
all_reduce_combiner_test.cc
…
all_reduce_combiner.cc
[XLA:GPU] [NFC] Clarify types in AllReduceCombiner
2021-03-05 15:16:41 -08:00
all_reduce_combiner.h
…
all_reduce_simplifier_test.cc
[XLA] replace all-reduces with a singleton replica_group_size with the operand.
2020-09-11 08:58:29 -07:00
all_reduce_simplifier.cc
…
all_reduce_simplifier.h
…
all_to_all_decomposer.cc
[XLA] Use array all to all in the client frontend
2021-01-29 20:50:10 -08:00
all_to_all_decomposer.h
[XLA] Use array all to all in the client frontend
2021-01-29 20:50:10 -08:00
allocation_tracker.cc
…
allocation_tracker.h
…
ar_crs_combiner_test.cc
[XLA] Use CollectiveOpGroup mode to drive some of the HLO verification
2021-02-23 10:15:41 -08:00
ar_crs_combiner.cc
…
ar_crs_combiner.h
…
backend.cc
…
backend.h
…
batch_dot_simplification_test.cc
…
batch_dot_simplification.cc
…
batch_dot_simplification.h
…
batchnorm_expander_test.cc
…
batchnorm_expander.cc
…
batchnorm_expander.h
…
bfloat16_conversion_folding_test.cc
…
bfloat16_conversion_folding.cc
…
bfloat16_conversion_folding.h
Add interface for HLO passes which run on HloModuleGroup.
2018-09-19 08:16:42 -07:00
bfloat16_normalization_test.cc
…
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
…
bfloat16_propagation.cc
TPU layout method cleanup by removing one parameter.
2021-03-01 15:38:15 -08:00
bfloat16_propagation.h
TPU layout method cleanup by removing one parameter.
2021-03-01 15:38:15 -08:00
bfloat16_support.cc
…
bfloat16_support.h
…
buffer_assignment_test.cc
…
buffer_assignment.cc
[XLA] Print live ranges by every moment.
2021-02-12 17:48:57 -08:00
buffer_assignment.h
…
buffer_value_containers.h
…
buffer_value.cc
…
buffer_value.h
…
BUILD
[XLA:GPU] Enforce collectives ordering to be their appearence in the module
2021-03-04 21:47:17 -08:00
call_graph_test.cc
…
call_graph.cc
…
call_graph.h
…
call_inliner_test.cc
…
call_inliner.cc
…
call_inliner.h
…
channel_tracker.cc
…
channel_tracker.h
…
cholesky_expander.cc
…
cholesky_expander.h
…
collective_ops_utils_test.cc
[XLA:GPU] Fix collective communication ops to work correctly in SPMD mode.
2021-02-22 12:42:21 -08:00
collective_ops_utils.cc
[XLA] Use CollectiveOpGroup mode to drive some of the HLO verification
2021-02-23 10:15:41 -08:00
collective_ops_utils.h
[XLA:GPU][NFC] Add some additional logging for collective thunks.
2021-02-22 14:47:14 -08:00
collectives_schedule_linearizer_test.cc
[XLA:GPU] [NFC] CollectiveScheduleLinearizer: test for dependent collectives
2021-03-05 15:01:35 -08:00
collectives_schedule_linearizer.cc
[XLA:GPU] [NFC] CollectiveScheduleLinearizer: test for dependent collectives
2021-03-05 15:01:35 -08:00
collectives_schedule_linearizer.h
[XLA:GPU] Enforce collectives ordering to be their appearence in the module
2021-03-04 21:47:17 -08:00
comparison_expander.cc
…
comparison_expander.h
…
compilation_cache.cc
…
compilation_cache.h
…
compilation_stats.cc
[TF:XLA] Collect compilation statistics about the HLO passes.
2019-04-11 23:57:47 -07:00
compilation_stats.h
…
compile_only_service.cc
Deprecate broadcast-replicated-params option in xla::ExecutionOptions proto.
2021-02-18 12:50:44 -08:00
compile_only_service.h
…
compiler.cc
…
compiler.h
Updates weight broadcast option to go through different proto.
2021-02-27 01:55:04 -08:00
computation_layout.cc
…
computation_layout.h
…
computation_placer.cc
[XLA][NFC] Add a struct to describe the LogicalID of a device.
2021-02-22 19:40:16 -08:00
computation_placer.h
[XLA][NFC] Add a struct to describe the LogicalID of a device.
2021-02-22 19:40:16 -08:00
conditional_canonicalizer_test.cc
…
conditional_canonicalizer.cc
…
conditional_canonicalizer.h
…
conditional_code_motion_test.cc
[XLA] Add auto-tuning support for conditional code motion.
2021-02-22 14:41:21 -08:00
conditional_code_motion.cc
[XLA] Add auto-tuning support for conditional code motion.
2021-02-22 14:41:21 -08:00
conditional_code_motion.h
[XLA] Add auto-tuning support for conditional code motion.
2021-02-22 14:41:21 -08:00
conditional_simplifier_test.cc
…
conditional_simplifier.cc
…
conditional_simplifier.h
…
conditional_to_select_test.cc
…
conditional_to_select.cc
…
conditional_to_select.h
…
convert_operand_folding_test.cc
…
convert_operand_folding.cc
…
convert_operand_folding.h
…
convolution_4d_expander_test.cc
…
convolution_4d_expander.cc
…
convolution_4d_expander.h
…
convolution_group_converter_test.cc
…
convolution_group_converter.cc
…
convolution_group_converter.h
…
copy_insertion_test.cc
[XLA/GPU] Re-enable h-loop-fusion to share operands with users.
2021-02-24 15:34:18 -08:00
copy_insertion.cc
[XLA] Fix TryElideCopy in copy insertion to traverse all related sources/destinations when considering removing a copy. Also modify the constructor of CopyRemover to check the ordering of aliased live ranges only when using the default dependence-based HloOrdering.
2021-02-25 10:10:41 -08:00
copy_insertion.h
[XLA] Fix TryElideCopy in copy insertion to traverse all related sources/destinations when considering removing a copy. Also modify the constructor of CopyRemover to check the ordering of aliased live ranges only when using the default dependence-based HloOrdering.
2021-02-25 10:10:41 -08: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
…
dfs_hlo_visitor.cc
…
dfs_hlo_visitor.h
…
dot_as_convolution_util.cc
…
dot_as_convolution_util.h
…
dot_decomposer_test.cc
…
dot_decomposer.cc
[XLA] Invert the permutation convention for xla::ShapeUtil::PermuteDimensions.
2021-02-10 13:20:44 -08:00
dot_decomposer.h
…
dump.cc
Minor changes on dumping utility functions.
2021-03-04 21:54:25 -08:00
dump.h
Minor changes on dumping utility functions.
2021-03-04 21:54:25 -08:00
dynamic_dimension_inference_test.cc
…
dynamic_dimension_inference.cc
Support more cases for strided_slice gradient.
2021-02-17 12:18:39 -08:00
dynamic_dimension_inference.h
…
dynamic_index_splitter_test.cc
…
dynamic_index_splitter.cc
…
dynamic_index_splitter.h
…
dynamic_padder_test.cc
Add a field in hlo proto to record if graph has dynamic shapes.
2021-02-22 21:37:05 -08:00
dynamic_padder.cc
Add a field in hlo proto to record if graph has dynamic shapes.
2021-02-22 21:37:05 -08:00
dynamic_padder.h
…
dynamic_parameter_binding_test.cc
…
dynamic_parameter_binding.cc
…
dynamic_parameter_binding.h
…
dynamic_update_slice_test.cc
…
dynamic_window_utils.cc
…
dynamic_window_utils.h
…
eigh_expander.cc
[XLA] Add support for Hermitian eigendecompositions in eigh expander pass.
2021-03-02 06:58:20 -08:00
eigh_expander.h
[XLA] Change eigh client library to produce a CustomCall that is expanded by an expander pass.
2021-03-01 08:12:43 -08:00
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
[XLA:CPU] More accurate expm1 when x is small, take two
2021-02-22 11:29:40 -08:00
elemental_ir_emitter.h
Merge pull request #47164 from nouiz:upstream-llvm_var_name
2021-02-17 06:12:19 -08:00
executable.cc
[XLA] Use DumpToFileInDir for hlo_execution_profile_data
2021-02-05 15:13:22 -08:00
executable.h
…
execution_tracker.cc
…
execution_tracker.h
…
flatten_call_graph_test.cc
…
flatten_call_graph.cc
…
flatten_call_graph.h
…
fusion_node_indexing_evaluation_test.cc
…
fusion_node_indexing_evaluation.cc
…
fusion_node_indexing_evaluation.h
…
fusion_queue.h
* Add virtual function NotFusingInstruction to FusionQueue to get necessary information for some subclasses.
2021-02-16 10:38:49 -08:00
gather_expander_test.cc
…
gather_expander.cc
…
gather_expander.h
…
generic_transfer_manager.cc
…
generic_transfer_manager.h
…
global_device_id.cc
…
global_device_id.h
…
heap_simulator_test.cc
…
heap_simulator.cc
…
heap_simulator.h
…
hlo_alias_analysis_test.cc
…
hlo_alias_analysis.cc
…
hlo_alias_analysis.h
Internal change
2020-09-12 16:03:35 -07:00
hlo_buffer.cc
…
hlo_buffer.h
…
hlo_casting_utils_test.cc
…
hlo_casting_utils.h
Add an error message when HLO casting fails.
2021-02-18 12:30:57 -08:00
hlo_clone_context.h
…
hlo_computation_test.cc
…
hlo_computation.cc
[XLA] Move HLO instruction OpMetadata dummy names to a new field.
2021-02-08 17:19:21 -08:00
hlo_computation.h
…
hlo_constant_folding_test.cc
[XLA] Invert the meaning of xla::Permute, add an inverse xla::PermuteInverse().
2021-02-10 12:21:04 -08:00
hlo_constant_folding.cc
…
hlo_constant_folding.h
…
hlo_cost_analysis_test.cc
[XLA] Make HloCostAnalysis account for nested shapes to calculate bytes accessed in fusion, infeed and outfeed
2021-02-10 11:44:40 -08:00
hlo_cost_analysis.cc
[XLA] Make HloCostAnalysis account for nested shapes to calculate bytes accessed in fusion, infeed and outfeed
2021-02-10 11:44:40 -08:00
hlo_cost_analysis.h
…
hlo_creation_utils_test.cc
…
hlo_creation_utils.cc
…
hlo_creation_utils.h
…
hlo_cse_test.cc
…
hlo_cse.cc
…
hlo_cse.h
…
hlo_dataflow_analysis_test.cc
[XLA/GPU] Re-enable h-loop-fusion to share operands with users.
2021-02-24 15:34:18 -08:00
hlo_dataflow_analysis.cc
[XLA/GPU] Fix a bug in CanShareOperandBufferWithUser().
2021-02-24 15:34:29 -08:00
hlo_dataflow_analysis.h
…
hlo_dce_test.cc
…
hlo_dce.cc
…
hlo_dce.h
…
hlo_domain_isolator.cc
…
hlo_domain_isolator.h
…
hlo_domain_map.cc
…
hlo_domain_map.h
…
hlo_domain_metadata.h
…
hlo_domain_remover.cc
…
hlo_domain_remover.h
Add interface for HLO passes which run on HloModuleGroup.
2018-09-19 08:16:42 -07:00
hlo_domain_test.cc
…
hlo_domain_verifier.cc
…
hlo_domain_verifier.h
…
hlo_element_type_converter_test.cc
…
hlo_element_type_converter.cc
Make the comparison computation for Sort required.
2019-02-08 02:31:58 -08:00
hlo_element_type_converter.h
…
hlo_evaluator_test.cc
[XLA] Invert the meaning of xla::Permute, add an inverse xla::PermuteInverse().
2021-02-10 12:21:04 -08: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
Support S16 and U16 in the HLO evaluator
2019-01-11 06:06:32 -08:00
hlo_evaluator_typed_visitor_uint32.cc
…
hlo_evaluator_typed_visitor_uint64.cc
…
hlo_evaluator_typed_visitor.h
…
hlo_evaluator.cc
…
hlo_evaluator.h
…
hlo_execution_profile_data.proto
…
hlo_execution_profile_test.cc
…
hlo_execution_profile.cc
…
hlo_execution_profile.h
…
hlo_get_dimension_size_rewriter.h
…
hlo_graph_dumper_test.cc
…
hlo_graph_dumper.cc
fix a typo. this shows up in GraphViewer's tooltip
2021-02-11 18:09:25 -08:00
hlo_graph_dumper.h
…
hlo_input_output_alias_config_test.cc
…
hlo_input_output_alias_config.cc
…
hlo_input_output_alias_config.h
…
hlo_instruction_test.cc
Merge pull request #45276 from Alfie-Edwards:relaxing-requirements-for-clonewithnewoperands-to-preserve-sharding-information
2021-02-02 04:18:59 -08:00
hlo_instruction.cc
Add a literal field to custom call instruction.
2021-02-17 12:28:55 -08:00
hlo_instruction.h
XLA value inference.
2021-03-03 14:17:48 -08:00
hlo_instructions.cc
Add a literal field to custom call instruction.
2021-02-17 12:28:55 -08:00
hlo_instructions.h
Add a literal field to custom call instruction.
2021-02-17 12:28:55 -08:00
hlo_lexer.cc
…
hlo_lexer.h
…
hlo_live_range_test.cc
…
hlo_live_range.cc
[XLA] Print live ranges by every moment.
2021-02-12 17:48:57 -08:00
hlo_live_range.h
…
hlo_liveness_analysis_test.cc
…
hlo_liveness_analysis.cc
…
hlo_liveness_analysis.h
…
hlo_matchers_test.cc
…
hlo_matchers.cc
…
hlo_matchers.h
[XLA] Add a variant of WhileLoopInvariantCodeMotion pass to hoist groups of loop expensive invariant HLOs out of while if they are not size-inflating as a whole.
2021-02-08 13:13:32 -08:00
hlo_memory_scheduler_test.cc
[XLA] Use templates in heap simulator to allow opaque type to be different than HloValue (NFC)
2020-08-14 15:58:46 -07:00
hlo_memory_scheduler.cc
…
hlo_memory_scheduler.h
[XLA] Use module scheduler to reduce compile time.
2019-07-16 14:16:07 -07:00
hlo_module_config.cc
Rename GenerateCubinForTfCode to GenerateGpuBinaryForTfCode and add ROCm support for it
2020-08-26 13:50:01 +00:00
hlo_module_config.h
Updates weight broadcast option to go through different proto.
2021-02-27 01:55:04 -08:00
hlo_module_dce_test.cc
…
hlo_module_dce.cc
…
hlo_module_dce.h
…
hlo_module_group_metadata.cc
…
hlo_module_group_metadata.h
…
hlo_module_group_test.cc
…
hlo_module_group_util.cc
…
hlo_module_group_util.h
…
hlo_module_group.cc
…
hlo_module_group.h
…
hlo_module_metadata_test.cc
…
hlo_module_metadata.cc
…
hlo_module_metadata.h
…
hlo_module_test.cc
…
hlo_module_util.cc
Deprecate broadcast-replicated-params option in xla::ExecutionOptions proto.
2021-02-18 12:50:44 -08:00
hlo_module_util.h
Move CreateModuleConfig to a new hlo_module_util header.
2020-12-14 15:14:29 -08:00
hlo_module.cc
Add a field in hlo proto to record if graph has dynamic shapes.
2021-02-22 21:37:05 -08:00
hlo_module.h
Add a field in hlo proto to record if graph has dynamic shapes.
2021-02-22 21:37:05 -08:00
hlo_op_metadata.cc
…
hlo_op_metadata.h
…
hlo_opcode_test.cc
…
hlo_opcode.cc
…
hlo_opcode.h
…
hlo_ordering_test.cc
[XLA] Automated g4 rollback of changelist 352846603.
2021-01-26 14:45:12 -08:00
hlo_ordering.cc
…
hlo_ordering.h
…
hlo_parser_test.cc
Add a literal field to custom call instruction.
2021-02-17 12:28:55 -08:00
hlo_parser.cc
Add a literal field to custom call instruction.
2021-02-17 12:28:55 -08:00
hlo_parser.h
…
hlo_pass_fix.h
…
hlo_pass_interface.h
…
hlo_pass_pipeline_test.cc
…
hlo_pass_pipeline.cc
[XLA] Add a space in a log statement
2021-02-12 12:40:41 -08:00
hlo_pass_pipeline.h
…
hlo_phi_graph_test.cc
…
hlo_phi_graph.cc
…
hlo_phi_graph.h
a bit more
2020-10-29 18:16:19 +03: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
With other internal changes, extend the TF XLA interface to support running post-optimizations HLO modules. Also fix a few bugs in XLA related to saving and loading HLO from protobuf.
2020-11-04 12:25:32 -08:00
hlo_query.cc
…
hlo_query.h
…
hlo_reachability_test.cc
…
hlo_reachability.cc
…
hlo_reachability.h
…
hlo_rematerialization_test_utils_test.cc
…
hlo_rematerialization_test_utils.h
…
hlo_rematerialization_test.cc
…
hlo_rematerialization.cc
…
hlo_rematerialization.h
[XLA] Initialize fields of RematerializationSizes by default
2021-02-05 16:48:25 -08:00
hlo_replication_analysis_test.cc
[XLA] Use CollectiveOpGroup mode to drive some of the HLO verification
2021-02-23 10:15:41 -08:00
hlo_replication_analysis.cc
…
hlo_replication_analysis.h
…
hlo_runner_interface.cc
…
hlo_runner_interface.h
Add new data transfer support.
2021-03-04 14:21:05 -08:00
hlo_runner.cc
…
hlo_runner.h
Add SPMD alternative implementations.
2021-01-29 18:34:22 -08:00
hlo_schedule_test.cc
…
hlo_schedule.cc
…
hlo_schedule.h
…
hlo_sharding_metadata.cc
…
hlo_sharding_metadata.h
…
hlo_sharding_test.cc
…
hlo_sharding_util_test.cc
…
hlo_sharding_util.cc
…
hlo_sharding_util.h
…
hlo_sharding.cc
…
hlo_sharding.h
…
hlo_subcomputation_unification_test.cc
…
hlo_subcomputation_unification.cc
…
hlo_subcomputation_unification.h
…
hlo_value.cc
…
hlo_value.h
…
hlo_verifier_test.cc
[XLA] Extend CollectivePermute verification to check if source/target IDs are in range.
2021-02-26 12:19:15 -08:00
hlo_verifier.cc
[XLA] Extend CollectivePermute verification to check if source/target IDs are in range.
2021-02-26 12:19:15 -08:00
hlo_verifier.h
[XLA] Fix dot/conv in ShapeVerifier for !allow_mixed_precision
2021-02-09 15:48:16 -08:00
hlo.proto
Add a field in hlo proto to record if graph has dynamic shapes.
2021-02-22 21:37:05 -08:00
human_readable_profile_builder.cc
…
human_readable_profile_builder.h
…
indexed_array_analysis_test.cc
…
indexed_array_analysis.cc
…
indexed_array_analysis.h
…
instruction_fusion_test.cc
…
instruction_fusion.cc
* Add virtual function NotFusingInstruction to FusionQueue to get necessary information for some subclasses.
2021-02-16 10:38:49 -08:00
instruction_fusion.h
* Add virtual function NotFusingInstruction to FusionQueue to get necessary information for some subclasses.
2021-02-16 10:38:49 -08:00
layout_assignment_test.cc
[XLA] Add a test for dynamic shapes vs ComputationLayout
2021-02-16 13:11:38 -08:00
layout_assignment.cc
[XLA] Rollback layout assignment change
2021-02-25 19:40:25 -08:00
layout_assignment.h
…
llvm_compiler.cc
- Fix PjRt GPU Client to allow num_partitions > 1
2021-02-01 12:33:28 -08:00
llvm_compiler.h
…
local_service.cc
…
local_service.h
…
logical_buffer_analysis.cc
…
logical_buffer_analysis.h
…
logical_buffer.cc
…
logical_buffer.h
…
logistic_expander_test.cc
…
logistic_expander.cc
[XLA] Add expander pass for kLogistic.
2020-06-24 22:32:37 -07:00
logistic_expander.h
…
loop_schedule_linearizer_test.cc
…
loop_schedule_linearizer.cc
…
loop_schedule_linearizer.h
…
map_inliner_test.cc
…
map_inliner.cc
Rename "Inliner" to "MapInliner".
2018-10-04 14:29:35 -07:00
map_inliner.h
…
maybe_owning_device_memory.cc
…
maybe_owning_device_memory.h
…
memory_space_assignment_best_fit_repacker_test.cc
…
memory_space_assignment_best_fit_repacker.cc
…
memory_space_assignment_best_fit_repacker.h
…
memory_space_assignment_repacking.h
[XLA] Use templates in heap simulator to allow opaque type to be different than HloValue (NFC)
2020-08-14 15:58:46 -07:00
memory_space_assignment_test.cc
…
memory_space_assignment_utils.cc
Internal change
2020-09-24 10:22:25 -07:00
memory_space_assignment_utils.h
…
memory_space_assignment.cc
Delete dead code in memory space assignment
2021-02-12 12:02:58 -08:00
memory_space_assignment.h
Delete dead code in memory space assignment
2021-02-12 12:02:58 -08:00
memory_space_propagation_test.cc
[XLA] Bitcasts should define values when propagating memory spaces.
2021-02-23 15:27:16 -08:00
memory_space_propagation.cc
[XLA] Bitcasts should define values when propagating memory spaces.
2021-02-23 15:27:16 -08:00
memory_space_propagation.h
…
multi_output_fusion.cc
…
multi_output_fusion.h
…
name_uniquer_test.cc
…
name_uniquer.cc
…
name_uniquer.h
…
op_expander_pass.cc
…
op_expander_pass.h
…
operand_upcaster_test.cc
…
operand_upcaster.cc
…
operand_upcaster.h
…
optimize_input_output_buffer_alias_test.cc
…
optimize_input_output_buffer_alias.cc
…
optimize_input_output_buffer_alias.h
…
pattern_matcher_gmock_test.cc
…
pattern_matcher_gmock.h
…
pattern_matcher_test.cc
…
pattern_matcher.h
Virtualize the AddCopiesOnConditional function on copy insertion.
2021-02-02 10:01:15 -08:00
platform_util.cc
…
platform_util.h
…
qr_expander.cc
[XLA] Fix underflow in norm calculation in Householder reflection for complex inputs.
2021-02-23 18:24:36 -08:00
qr_expander.h
[XLA] Split QR decomposition along the same lines as LAPACK xGEQRf/xORGQR.
2021-02-23 14:13:25 -08:00
reshape_mover_test.cc
…
reshape_mover.cc
[XLA] Split permutation utilities from xla/util.* into a new xla/permutation_util.*
2021-02-09 17:01:49 -08:00
reshape_mover.h
…
rng_bit_generator_expander.cc
…
rng_bit_generator_expander.h
…
rng_expander.cc
…
rng_expander.h
…
root_instruction_sinker_test.cc
…
root_instruction_sinker.cc
…
root_instruction_sinker.h
…
scatter_expander_test.cc
…
scatter_expander.cc
…
scatter_expander.h
…
service_executable_run_options.h
…
service.cc
…
service.h
…
shape_inference_test.cc
…
shape_inference.cc
Support more cases for strided_slice gradient.
2021-02-17 12:18:39 -08:00
shape_inference.h
…
shaped_buffer_test.cc
…
shaped_buffer.cc
…
shaped_buffer.h
…
sharding_propagation_test.cc
Allow more sharding propagation from broadcast operands
2021-02-08 16:13:46 -08:00
sharding_propagation.cc
Allow more sharding propagation from broadcast operands
2021-02-08 16:13:46 -08:00
sharding_propagation.h
…
slice_sinker_test.cc
…
slice_sinker.cc
…
slice_sinker.h
…
slow_operation_alarm.cc
…
slow_operation_alarm.h
…
sort_simplifier_test.cc
…
sort_simplifier.cc
…
sort_simplifier.h
…
source_map_util.h
…
space_to_batch_converter_test.cc
Skip space-to-batch optimization on convs that are used by a different rank reduce-window or select-and-scatter.
2021-03-04 19:12:55 -08:00
space_to_batch_converter.cc
Skip space-to-batch optimization on convs that are used by a different rank reduce-window or select-and-scatter.
2021-03-04 19:12:55 -08:00
space_to_batch_converter.h
…
stable_sort_expander_test.cc
…
stable_sort_expander.cc
…
stable_sort_expander.h
Add an HLO Pass to make Sort stable.
2019-02-14 01:28:16 -08:00
stream_pool_test.cc
…
stream_pool.cc
…
stream_pool.h
…
topk_rewriter_test.cc
[XLA] Rewrite 1d sort to TopK
2020-08-05 02:40:46 -07:00
topk_rewriter.cc
…
topk_rewriter.h
…
tpu_computation_placer.cc
…
tpu_computation_placer.h
…
transfer_manager.cc
…
transfer_manager.h
…
transpose_folding_test.cc
…
transpose_folding.cc
…
transpose_folding.h
…
tree_reduction_rewriter.cc
…
tree_reduction_rewriter.h
…
triangular_solve_expander_test.cc
Merge pull request #39690 from whoozle:triangular-solve-expander-block-size
2020-07-15 06:57:41 -07:00
triangular_solve_expander.cc
…
triangular_solve_expander.h
…
tuple_points_to_analysis_test.cc
…
tuple_points_to_analysis.cc
…
tuple_points_to_analysis.h
…
tuple_simplifier_test.cc
…
tuple_simplifier.cc
…
tuple_simplifier.h
…
tuple_util_test.cc
…
tuple_util.cc
…
tuple_util.h
…
while_loop_all_reduce_code_motion_test.cc
…
while_loop_all_reduce_code_motion.cc
[XLA] Rollback layout assignment change
2021-02-25 19:40:25 -08:00
while_loop_all_reduce_code_motion.h
…
while_loop_analysis_test.cc
…
while_loop_analysis.cc
[XLA] Rollback layout assignment change
2021-02-25 19:40:25 -08:00
while_loop_analysis.h
…
while_loop_constant_sinking_test.cc
…
while_loop_constant_sinking.cc
minor spelling tweaks
2019-12-09 18:21:12 +09:00
while_loop_constant_sinking.h
…
while_loop_expensive_invariant_code_motion_test.cc
[XLA] Add a variant of WhileLoopInvariantCodeMotion pass to hoist groups of loop expensive invariant HLOs out of while if they are not size-inflating as a whole.
2021-02-08 13:13:32 -08:00
while_loop_expensive_invariant_code_motion.cc
[XLA] Add a variant of WhileLoopInvariantCodeMotion pass to hoist groups of loop expensive invariant HLOs out of while if they are not size-inflating as a whole.
2021-02-08 13:13:32 -08:00
while_loop_expensive_invariant_code_motion.h
[XLA] Add a variant of WhileLoopInvariantCodeMotion pass to hoist groups of loop expensive invariant HLOs out of while if they are not size-inflating as a whole.
2021-02-08 13:13:32 -08:00
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
…
while_loop_simplifier_test.cc
[XLA] Remove tuple elements in while loop if unused both inside or outside loop.
2021-02-22 19:53:24 -08:00
while_loop_simplifier.cc
[XLA] Remove tuple elements in while loop if unused both inside or outside loop.
2021-02-22 19:53:24 -08: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
…
while_util.cc
…
while_util.h
…
xla_debug_info_manager_test.cc
move singleton xla::gpu::GPUDebugInfoManager to xla::XlaDebugInfoManager. in hope that it can be reused by XLA/CPU.
2021-02-09 10:28:57 -08:00
xla_debug_info_manager.cc
move singleton xla::gpu::GPUDebugInfoManager to xla::XlaDebugInfoManager. in hope that it can be reused by XLA/CPU.
2021-02-09 10:28:57 -08:00
xla_debug_info_manager.h
move singleton xla::gpu::GPUDebugInfoManager to xla::XlaDebugInfoManager. in hope that it can be reused by XLA/CPU.
2021-02-09 10:28:57 -08:00
zero_sized_hlo_elimination_test.cc
…
zero_sized_hlo_elimination.cc
…
zero_sized_hlo_elimination.h
…