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