tree: 3ec2f19af3f342712df4d2a5fe9d525d41874ef7 [path history] [tgz]
  1. cpu/
  2. g3doc/
  3. gpu/
  4. graphcycles/
  5. interpreter/
  6. llvm_ir/
  7. spmd/
  8. algebraic_simplifier.cc
  9. algebraic_simplifier.h
  10. algebraic_simplifier_proof_distributive_property.py
  11. algebraic_simplifier_test.cc
  12. all_gather_broadcast_reorder.cc
  13. all_gather_broadcast_reorder.h
  14. all_gather_broadcast_reorder_test.cc
  15. all_gather_combiner.cc
  16. all_gather_combiner.h
  17. all_gather_combiner_test.cc
  18. all_gather_decomposer.cc
  19. all_gather_decomposer.h
  20. all_gather_decomposer_test.cc
  21. all_reduce_combiner.cc
  22. all_reduce_combiner.h
  23. all_reduce_combiner_test.cc
  24. all_reduce_contiguous.cc
  25. all_reduce_contiguous.h
  26. all_reduce_contiguous_test.cc
  27. all_reduce_folder.cc
  28. all_reduce_folder.h
  29. all_reduce_folder_test.cc
  30. all_reduce_key.cc
  31. all_reduce_key.h
  32. all_reduce_reassociate.cc
  33. all_reduce_reassociate.h
  34. all_reduce_reassociate_test.cc
  35. all_reduce_simplifier.cc
  36. all_reduce_simplifier.h
  37. all_reduce_simplifier_test.cc
  38. all_to_all_decomposer.cc
  39. all_to_all_decomposer.h
  40. allocation_tracker.cc
  41. allocation_tracker.h
  42. ar_crs_combiner.cc
  43. ar_crs_combiner.h
  44. ar_crs_combiner_test.cc
  45. async_collective_creator.cc
  46. async_collective_creator.h
  47. async_collective_creator_test.cc
  48. async_op_canonicalizer.cc
  49. async_op_canonicalizer.h
  50. async_op_canonicalizer_test.cc
  51. backend.cc
  52. backend.h
  53. batch_dot_simplification.cc
  54. batch_dot_simplification.h
  55. batch_dot_simplification_test.cc
  56. batchnorm_expander.cc
  57. batchnorm_expander.h
  58. batchnorm_expander_test.cc
  59. bfloat16_conversion_folding.cc
  60. bfloat16_conversion_folding.h
  61. bfloat16_conversion_folding_test.cc
  62. bfloat16_normalization.cc
  63. bfloat16_normalization.h
  64. bfloat16_normalization_test.cc
  65. bfloat16_propagation.cc
  66. bfloat16_propagation.h
  67. bfloat16_propagation_test.cc
  68. bfloat16_support.cc
  69. bfloat16_support.h
  70. bitcast_decomposer.cc
  71. bitcast_decomposer.h
  72. bitcast_decomposer_test.cc
  73. bitcast_dtypes_expander.cc
  74. bitcast_dtypes_expander.h
  75. bitcast_dtypes_expander_test.cc
  76. broadcast_canonicalizer.cc
  77. broadcast_canonicalizer.h
  78. broadcast_canonicalizer_test.cc
  79. buffer_assignment.cc
  80. buffer_assignment.h
  81. buffer_assignment_test.cc
  82. buffer_value.cc
  83. buffer_value.h
  84. buffer_value_containers.h
  85. BUILD
  86. call_graph.cc
  87. call_graph.h
  88. call_graph_test.cc
  89. call_inliner.cc
  90. call_inliner.h
  91. call_inliner_test.cc
  92. change_op_data_type.cc
  93. change_op_data_type.h
  94. change_op_data_type_test.cc
  95. channel_tracker.cc
  96. channel_tracker.h
  97. cholesky_expander.cc
  98. cholesky_expander.h
  99. collective_combiner_utils.h
  100. collective_decomposer_utils.cc
  101. collective_decomposer_utils.h
  102. collective_ops_utils.cc
  103. collective_ops_utils.h
  104. collective_ops_utils_test.cc
  105. collectives_schedule_linearizer.cc
  106. collectives_schedule_linearizer.h
  107. collectives_schedule_linearizer_test.cc
  108. comparison_expander.cc
  109. comparison_expander.h
  110. compilation_cache.cc
  111. compilation_cache.h
  112. compilation_environments.cc
  113. compilation_environments.h
  114. compilation_environments_test.cc
  115. compilation_stats.cc
  116. compilation_stats.h
  117. compile_only_service.cc
  118. compile_only_service.h
  119. compile_time_cap.h
  120. compiler.cc
  121. compiler.h
  122. computation_layout.cc
  123. computation_layout.h
  124. computation_placer.cc
  125. computation_placer.h
  126. conditional_canonicalizer.cc
  127. conditional_canonicalizer.h
  128. conditional_canonicalizer_test.cc
  129. conditional_code_motion.cc
  130. conditional_code_motion.h
  131. conditional_code_motion_test.cc
  132. conditional_simplifier.cc
  133. conditional_simplifier.h
  134. conditional_simplifier_test.cc
  135. conditional_to_select.cc
  136. conditional_to_select.h
  137. conditional_to_select_test.cc
  138. convert_mover.cc
  139. convert_mover.h
  140. convert_mover_test.cc
  141. convert_operand_folding.cc
  142. convert_operand_folding.h
  143. convert_operand_folding_test.cc
  144. convolution_4d_expander.cc
  145. convolution_4d_expander.h
  146. convolution_4d_expander_test.cc
  147. convolution_group_converter.cc
  148. convolution_group_converter.h
  149. convolution_group_converter_test.cc
  150. convolution_pred_expander.cc
  151. convolution_pred_expander.h
  152. convolution_pred_expander_test.cc
  153. copy_insertion.cc
  154. copy_insertion.h
  155. copy_insertion_test.cc
  156. custom_call_sharding_helper.cc
  157. custom_call_sharding_helper.h
  158. custom_call_status.cc
  159. custom_call_status.h
  160. custom_call_status_internal.h
  161. custom_call_status_test.cc
  162. custom_call_status_test_c_caller.c
  163. custom_call_status_test_c_caller.h
  164. custom_call_target_registry.cc
  165. custom_call_target_registry.h
  166. defuser.cc
  167. defuser.h
  168. defuser_test.cc
  169. despecializer.cc
  170. despecializer.h
  171. dfs_hlo_visitor.cc
  172. dfs_hlo_visitor.h
  173. dfs_hlo_visitor_with_default.h
  174. dfs_hlo_visitor_with_default_test.cc
  175. dot_as_convolution_util.cc
  176. dot_as_convolution_util.h
  177. dot_decomposer.cc
  178. dot_decomposer.h
  179. dot_decomposer_test.cc
  180. dot_merger.cc
  181. dot_merger.h
  182. dot_merger_test.cc
  183. dump.cc
  184. dump.h
  185. dynamic_dimension_inference.cc
  186. dynamic_dimension_inference.h
  187. dynamic_dimension_inference_test.cc
  188. dynamic_dimension_simplifier.cc
  189. dynamic_dimension_simplifier.h
  190. dynamic_dimension_simplifier_test.cc
  191. dynamic_index_splitter.cc
  192. dynamic_index_splitter.h
  193. dynamic_index_splitter_test.cc
  194. dynamic_padder.cc
  195. dynamic_padder.h
  196. dynamic_padder_test.cc
  197. dynamic_parameter_binding.cc
  198. dynamic_parameter_binding.h
  199. dynamic_parameter_binding_test.cc
  200. dynamic_update_slice_test.cc
  201. dynamic_window_utils.cc
  202. dynamic_window_utils.h
  203. eigh_expander.cc
  204. eigh_expander.h
  205. elemental_ir_emitter.cc
  206. elemental_ir_emitter.h
  207. elemental_ir_emitter_test.cc
  208. executable.cc
  209. executable.h
  210. execution_tracker.cc
  211. execution_tracker.h
  212. flatten_call_graph.cc
  213. flatten_call_graph.h
  214. flatten_call_graph_test.cc
  215. fusion_node_indexing_evaluation.cc
  216. fusion_node_indexing_evaluation.h
  217. fusion_node_indexing_evaluation_test.cc
  218. fusion_queue.h
  219. gather_expander.cc
  220. gather_expander.h
  221. gather_expander_test.cc
  222. gather_scatter_utils.cc
  223. gather_scatter_utils.h
  224. gather_simplifier.cc
  225. gather_simplifier.h
  226. gather_simplifier_test.cc
  227. generate_test_hlo_checks.py
  228. generate_test_hlo_checks_test.py
  229. generic_transfer_manager.cc
  230. generic_transfer_manager.h
  231. global_device_id.cc
  232. global_device_id.h
  233. heap_simulator.cc
  234. heap_simulator.h
  235. heap_simulator_test.cc
  236. hlo.proto
  237. hlo_alias_analysis.cc
  238. hlo_alias_analysis.h
  239. hlo_alias_analysis_test.cc
  240. hlo_buffer.cc
  241. hlo_buffer.h
  242. hlo_casting_utils.h
  243. hlo_casting_utils_test.cc
  244. hlo_clone_context.h
  245. hlo_computation.cc
  246. hlo_computation.h
  247. hlo_computation_test.cc
  248. hlo_constant_folding.cc
  249. hlo_constant_folding.h
  250. hlo_constant_folding_test.cc
  251. hlo_cost_analysis.cc
  252. hlo_cost_analysis.h
  253. hlo_cost_analysis_test.cc
  254. hlo_creation_utils.cc
  255. hlo_creation_utils.h
  256. hlo_creation_utils_test.cc
  257. hlo_cse.cc
  258. hlo_cse.h
  259. hlo_cse_test.cc
  260. hlo_dataflow_analysis.cc
  261. hlo_dataflow_analysis.h
  262. hlo_dataflow_analysis_test.cc
  263. hlo_dce.cc
  264. hlo_dce.h
  265. hlo_dce_test.cc
  266. hlo_domain_isolator.cc
  267. hlo_domain_isolator.h
  268. hlo_domain_map.cc
  269. hlo_domain_map.h
  270. hlo_domain_metadata.h
  271. hlo_domain_remover.cc
  272. hlo_domain_remover.h
  273. hlo_domain_test.cc
  274. hlo_domain_verifier.cc
  275. hlo_domain_verifier.h
  276. hlo_element_type_converter.cc
  277. hlo_element_type_converter.h
  278. hlo_element_type_converter_test.cc
  279. hlo_evaluator.cc
  280. hlo_evaluator.h
  281. hlo_evaluator_test.cc
  282. hlo_evaluator_typed_visitor.h
  283. hlo_evaluator_typed_visitor_bfloat16.cc
  284. hlo_evaluator_typed_visitor_bool.cc
  285. hlo_evaluator_typed_visitor_complex128.cc
  286. hlo_evaluator_typed_visitor_complex64.cc
  287. hlo_evaluator_typed_visitor_double.cc
  288. hlo_evaluator_typed_visitor_float.cc
  289. hlo_evaluator_typed_visitor_half.cc
  290. hlo_evaluator_typed_visitor_int16.cc
  291. hlo_evaluator_typed_visitor_int32.cc
  292. hlo_evaluator_typed_visitor_int64.cc
  293. hlo_evaluator_typed_visitor_int8.cc
  294. hlo_evaluator_typed_visitor_uint16.cc
  295. hlo_evaluator_typed_visitor_uint32.cc
  296. hlo_evaluator_typed_visitor_uint64.cc
  297. hlo_evaluator_typed_visitor_uint8.cc
  298. hlo_execution_profile.cc
  299. hlo_execution_profile.h
  300. hlo_execution_profile_data.proto
  301. hlo_execution_profile_test.cc
  302. hlo_graph_dumper.cc
  303. hlo_graph_dumper.h
  304. hlo_graph_dumper_test.cc
  305. hlo_input_output_alias_config.cc
  306. hlo_input_output_alias_config.h
  307. hlo_input_output_alias_config_test.cc
  308. hlo_instruction.cc
  309. hlo_instruction.h
  310. hlo_instruction_test.cc
  311. hlo_instructions.cc
  312. hlo_instructions.h
  313. hlo_lexer.cc
  314. hlo_lexer.h
  315. hlo_live_range.cc
  316. hlo_live_range.h
  317. hlo_live_range_test.cc
  318. hlo_liveness_analysis.cc
  319. hlo_liveness_analysis.h
  320. hlo_liveness_analysis_test.cc
  321. hlo_matchers.cc
  322. hlo_matchers.h
  323. hlo_matchers_test.cc
  324. hlo_memory_scheduler.cc
  325. hlo_memory_scheduler.h
  326. hlo_memory_scheduler_test.cc
  327. hlo_module.cc
  328. hlo_module.h
  329. hlo_module_config.cc
  330. hlo_module_config.h
  331. hlo_module_dce.cc
  332. hlo_module_dce.h
  333. hlo_module_dce_test.cc
  334. hlo_module_group.cc
  335. hlo_module_group.h
  336. hlo_module_group_metadata.cc
  337. hlo_module_group_metadata.h
  338. hlo_module_group_test.cc
  339. hlo_module_group_util.cc
  340. hlo_module_group_util.h
  341. hlo_module_metadata.cc
  342. hlo_module_metadata.h
  343. hlo_module_metadata_test.cc
  344. hlo_module_test.cc
  345. hlo_module_util.cc
  346. hlo_module_util.h
  347. hlo_op_metadata.cc
  348. hlo_op_metadata.h
  349. hlo_opcode.cc
  350. hlo_opcode.h
  351. hlo_opcode_test.cc
  352. hlo_ordering.cc
  353. hlo_ordering.h
  354. hlo_ordering_test.cc
  355. hlo_parser.cc
  356. hlo_parser.h
  357. hlo_parser_test.cc
  358. hlo_pass_fix.h
  359. hlo_pass_interface.h
  360. hlo_pass_pipeline.cc
  361. hlo_pass_pipeline.h
  362. hlo_pass_pipeline_test.cc
  363. hlo_phi_graph.cc
  364. hlo_phi_graph.h
  365. hlo_phi_graph_test.cc
  366. hlo_profile_printer.cc
  367. hlo_profile_printer.h
  368. hlo_profile_printer_data.proto
  369. hlo_proto_util.cc
  370. hlo_proto_util.h
  371. hlo_proto_util_test.cc
  372. hlo_query.cc
  373. hlo_query.h
  374. hlo_reachability.cc
  375. hlo_reachability.h
  376. hlo_reachability_test.cc
  377. hlo_rematerialization.cc
  378. hlo_rematerialization.h
  379. hlo_rematerialization_test.cc
  380. hlo_rematerialization_test_utils.h
  381. hlo_rematerialization_test_utils_test.cc
  382. hlo_replication_analysis.cc
  383. hlo_replication_analysis.h
  384. hlo_replication_analysis_test.cc
  385. hlo_runner.cc
  386. hlo_runner.h
  387. hlo_runner_interface.cc
  388. hlo_runner_interface.h
  389. hlo_schedule.cc
  390. hlo_schedule.h
  391. hlo_schedule_test.cc
  392. hlo_sharding.cc
  393. hlo_sharding.h
  394. hlo_sharding_metadata.cc
  395. hlo_sharding_metadata.h
  396. hlo_sharding_test.cc
  397. hlo_sharding_util.cc
  398. hlo_sharding_util.h
  399. hlo_sharding_util_test.cc
  400. hlo_subcomputation_unification.cc
  401. hlo_subcomputation_unification.h
  402. hlo_subcomputation_unification_test.cc
  403. hlo_value.cc
  404. hlo_value.h
  405. hlo_verifier.cc
  406. hlo_verifier.h
  407. hlo_verifier_test.cc
  408. human_readable_profile_builder.cc
  409. human_readable_profile_builder.h
  410. indexed_array_analysis.cc
  411. indexed_array_analysis.h
  412. indexed_array_analysis_test.cc
  413. instruction_fusion.cc
  414. instruction_fusion.h
  415. instruction_fusion_test.cc
  416. instruction_hoister.cc
  417. instruction_hoister.h
  418. layout_assignment.cc
  419. layout_assignment.h
  420. layout_assignment_test.cc
  421. layout_normalization.cc
  422. layout_normalization.h
  423. layout_normalization_test.cc
  424. llvm_compiler.cc
  425. llvm_compiler.h
  426. local_service.cc
  427. local_service.h
  428. logical_buffer.cc
  429. logical_buffer.h
  430. logical_buffer_analysis.cc
  431. logical_buffer_analysis.h
  432. logistic_expander.cc
  433. logistic_expander.h
  434. logistic_expander_test.cc
  435. loop_schedule_linearizer.cc
  436. loop_schedule_linearizer.h
  437. loop_schedule_linearizer_test.cc
  438. map_inliner.cc
  439. map_inliner.h
  440. map_inliner_test.cc
  441. mapped_ptr_container_sorter.h
  442. mapped_ptr_container_sorter_test.cc
  443. maybe_owning_device_memory.cc
  444. maybe_owning_device_memory.h
  445. memory_space_assignment.cc
  446. memory_space_assignment.h
  447. memory_space_assignment_best_fit_repacker.cc
  448. memory_space_assignment_best_fit_repacker.h
  449. memory_space_assignment_best_fit_repacker_test.cc
  450. memory_space_assignment_repacking.h
  451. memory_space_assignment_test.cc
  452. memory_space_assignment_tuning_utils.cc
  453. memory_space_assignment_tuning_utils.h
  454. memory_space_assignment_utils.cc
  455. memory_space_assignment_utils.h
  456. memory_space_propagation.cc
  457. memory_space_propagation.h
  458. memory_space_propagation_test.cc
  459. multi_output_fusion.cc
  460. multi_output_fusion.h
  461. name_uniquer.cc
  462. name_uniquer.h
  463. name_uniquer_test.cc
  464. op_expander_pass.cc
  465. op_expander_pass.h
  466. operand_upcaster.cc
  467. operand_upcaster.h
  468. operand_upcaster_test.cc
  469. optimization_barrier_expander.cc
  470. optimization_barrier_expander.h
  471. optimize_input_output_buffer_alias.cc
  472. optimize_input_output_buffer_alias.h
  473. optimize_input_output_buffer_alias_test.cc
  474. pattern_matcher.h
  475. pattern_matcher_gmock.h
  476. pattern_matcher_gmock_test.cc
  477. pattern_matcher_test.cc
  478. platform_util.cc
  479. platform_util.h
  480. qr_expander.cc
  481. qr_expander.h
  482. real_imag_expander.cc
  483. real_imag_expander.h
  484. real_imag_expander_test.cc
  485. reduce_decomposer.cc
  486. reduce_decomposer.h
  487. reduce_decomposer_test.cc
  488. reduce_scatter_combiner.cc
  489. reduce_scatter_combiner.h
  490. reduce_scatter_combiner_test.cc
  491. reduce_scatter_decomposer.cc
  492. reduce_scatter_decomposer.h
  493. reduce_scatter_decomposer_test.cc
  494. reduce_scatter_utils.cc
  495. reduce_scatter_utils.h
  496. rendezvous.cc
  497. rendezvous.h
  498. reshape_decomposer.cc
  499. reshape_decomposer.h
  500. reshape_decomposer_test.cc
  501. reshape_mover.cc
  502. reshape_mover.h
  503. reshape_mover_test.cc
  504. result_caster.cc
  505. result_caster.h
  506. result_caster_test.cc
  507. rng_bit_generator_expander.cc
  508. rng_bit_generator_expander.h
  509. rng_expander.cc
  510. rng_expander.h
  511. root_instruction_sinker.cc
  512. root_instruction_sinker.h
  513. root_instruction_sinker_test.cc
  514. scatter_expander.cc
  515. scatter_expander.h
  516. scatter_expander_test.cc
  517. scatter_simplifier.cc
  518. scatter_simplifier.h
  519. scatter_simplifier_test.cc
  520. select_and_scatter_expander.cc
  521. select_and_scatter_expander.h
  522. select_and_scatter_expander_test.cc
  523. service.cc
  524. service.h
  525. service_executable_run_options.h
  526. shape_inference.cc
  527. shape_inference.h
  528. shape_inference_test.cc
  529. shaped_buffer.cc
  530. shaped_buffer.h
  531. shaped_buffer_test.cc
  532. sharding_propagation.cc
  533. sharding_propagation.h
  534. sharding_propagation_test.cc
  535. sharding_remover.cc
  536. sharding_remover.h
  537. sharding_remover_test.cc
  538. simplify_fp_conversions.cc
  539. simplify_fp_conversions.h
  540. simplify_fp_conversions_test.cc
  541. slice_sinker.cc
  542. slice_sinker.h
  543. slice_sinker_test.cc
  544. slow_operation_alarm.cc
  545. slow_operation_alarm.h
  546. sort_simplifier.cc
  547. sort_simplifier.h
  548. sort_simplifier_test.cc
  549. source_map_util.h
  550. space_to_batch_converter.cc
  551. space_to_batch_converter.h
  552. space_to_batch_converter_test.cc
  553. stable_sort_expander.cc
  554. stable_sort_expander.h
  555. stable_sort_expander_test.cc
  556. stream_pool.cc
  557. stream_pool.h
  558. stream_pool_test.cc
  559. test_compilation_environment.proto
  560. topk_rewriter.cc
  561. topk_rewriter.h
  562. topk_rewriter_test.cc
  563. tpu_computation_placer.cc
  564. tpu_computation_placer.h
  565. transfer_manager.cc
  566. transfer_manager.h
  567. transpose_folding.cc
  568. transpose_folding.h
  569. transpose_folding_test.cc
  570. tree_reduction_rewriter.cc
  571. tree_reduction_rewriter.h
  572. triangular_solve_expander.cc
  573. triangular_solve_expander.h
  574. triangular_solve_expander_test.cc
  575. tuple_points_to_analysis.cc
  576. tuple_points_to_analysis.h
  577. tuple_points_to_analysis_test.cc
  578. tuple_simplifier.cc
  579. tuple_simplifier.h
  580. tuple_simplifier_test.cc
  581. tuple_util.cc
  582. tuple_util.h
  583. tuple_util_test.cc
  584. while_loop_all_reduce_code_motion.cc
  585. while_loop_all_reduce_code_motion.h
  586. while_loop_all_reduce_code_motion_test.cc
  587. while_loop_analysis.cc
  588. while_loop_analysis.h
  589. while_loop_analysis_test.cc
  590. while_loop_concat_code_motion.cc
  591. while_loop_concat_code_motion.h
  592. while_loop_concat_code_motion_test.cc
  593. while_loop_constant_sinking.cc
  594. while_loop_constant_sinking.h
  595. while_loop_constant_sinking_test.cc
  596. while_loop_expensive_invariant_code_motion.cc
  597. while_loop_expensive_invariant_code_motion.h
  598. while_loop_expensive_invariant_code_motion_test.cc
  599. while_loop_invariant_code_motion.cc
  600. while_loop_invariant_code_motion.h
  601. while_loop_invariant_code_motion_test.cc
  602. while_loop_simplifier.cc
  603. while_loop_simplifier.h
  604. while_loop_simplifier_test.cc
  605. while_loop_trip_count_annotator.cc
  606. while_loop_trip_count_annotator.h
  607. while_loop_trip_count_annotator_test.cc
  608. while_util.cc
  609. while_util.h
  610. while_util_test.cc
  611. xla_debug_info_manager.cc
  612. xla_debug_info_manager.h
  613. xla_debug_info_manager_test.cc
  614. zero_sized_hlo_elimination.cc
  615. zero_sized_hlo_elimination.h
  616. zero_sized_hlo_elimination_test.cc