tree: 682ef3fe30437e3f07b88dc67dd779a46fe17ea4 [path history] [tgz]
  1. cpu/
  2. g3doc/
  3. gpu/
  4. interpreter/
  5. llvm_ir/
  6. mlir_gpu/
  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_decomposer.cc
  13. all_gather_decomposer.h
  14. all_gather_decomposer_test.cc
  15. all_reduce_combiner.cc
  16. all_reduce_combiner.h
  17. all_reduce_combiner_test.cc
  18. all_reduce_simplifier.cc
  19. all_reduce_simplifier.h
  20. all_reduce_simplifier_test.cc
  21. allocation_tracker.cc
  22. allocation_tracker.h
  23. ar_crs_combiner.cc
  24. ar_crs_combiner.h
  25. ar_crs_combiner_test.cc
  26. backend.cc
  27. backend.h
  28. batch_dot_simplification.cc
  29. batch_dot_simplification.h
  30. batch_dot_simplification_test.cc
  31. batchnorm_expander.cc
  32. batchnorm_expander.h
  33. batchnorm_expander_test.cc
  34. bfloat16_conversion_folding.cc
  35. bfloat16_conversion_folding.h
  36. bfloat16_conversion_folding_test.cc
  37. bfloat16_normalization.cc
  38. bfloat16_normalization.h
  39. bfloat16_normalization_test.cc
  40. bfloat16_propagation.cc
  41. bfloat16_propagation.h
  42. bfloat16_propagation_test.cc
  43. bfloat16_support.cc
  44. bfloat16_support.h
  45. buffer_assignment.cc
  46. buffer_assignment.h
  47. buffer_assignment_test.cc
  48. buffer_value.cc
  49. buffer_value.h
  50. buffer_value_containers.h
  51. BUILD
  52. call_graph.cc
  53. call_graph.h
  54. call_graph_test.cc
  55. call_inliner.cc
  56. call_inliner.h
  57. call_inliner_test.cc
  58. channel_tracker.cc
  59. channel_tracker.h
  60. cholesky_expander.cc
  61. cholesky_expander.h
  62. collective_ops_utils.cc
  63. collective_ops_utils.h
  64. comparison_expander.cc
  65. comparison_expander.h
  66. compilation_cache.cc
  67. compilation_cache.h
  68. compilation_stats.cc
  69. compilation_stats.h
  70. compile_only_service.cc
  71. compile_only_service.h
  72. compiler.cc
  73. compiler.h
  74. computation_layout.cc
  75. computation_layout.h
  76. computation_placer.cc
  77. computation_placer.h
  78. conditional_canonicalizer.cc
  79. conditional_canonicalizer.h
  80. conditional_canonicalizer_test.cc
  81. conditional_code_motion.cc
  82. conditional_code_motion.h
  83. conditional_code_motion_test.cc
  84. conditional_simplifier.cc
  85. conditional_simplifier.h
  86. conditional_simplifier_test.cc
  87. conditional_to_select.cc
  88. conditional_to_select.h
  89. conditional_to_select_test.cc
  90. convolution_4d_expander.cc
  91. convolution_4d_expander.h
  92. convolution_4d_expander_test.cc
  93. convolution_group_converter.cc
  94. convolution_group_converter.h
  95. convolution_group_converter_test.cc
  96. copy_insertion.cc
  97. copy_insertion.h
  98. copy_insertion_test.cc
  99. custom_call_target_registry.cc
  100. custom_call_target_registry.h
  101. defuser.cc
  102. defuser.h
  103. defuser_test.cc
  104. despecializer.cc
  105. despecializer.h
  106. dfs_hlo_visitor.cc
  107. dfs_hlo_visitor.h
  108. dfs_hlo_visitor_with_default.h
  109. dfs_hlo_visitor_with_default_test.cc
  110. dot_as_convolution_util.cc
  111. dot_as_convolution_util.h
  112. dot_decomposer.cc
  113. dot_decomposer.h
  114. dot_decomposer_test.cc
  115. dump.cc
  116. dump.h
  117. dynamic_dimension_inference.cc
  118. dynamic_dimension_inference.h
  119. dynamic_dimension_inference_test.cc
  120. dynamic_index_splitter.cc
  121. dynamic_index_splitter.h
  122. dynamic_index_splitter_test.cc
  123. dynamic_padder.cc
  124. dynamic_padder.h
  125. dynamic_padder_test.cc
  126. dynamic_parameter_binding.cc
  127. dynamic_parameter_binding.h
  128. dynamic_parameter_binding_test.cc
  129. dynamic_update_slice_test.cc
  130. elemental_ir_emitter.cc
  131. elemental_ir_emitter.h
  132. elemental_ir_emitter_test.cc
  133. executable.cc
  134. executable.h
  135. execution_tracker.cc
  136. execution_tracker.h
  137. flatten_call_graph.cc
  138. flatten_call_graph.h
  139. flatten_call_graph_test.cc
  140. fusion_node_indexing_evaluation.cc
  141. fusion_node_indexing_evaluation.h
  142. fusion_node_indexing_evaluation_test.cc
  143. fusion_queue.h
  144. gather_expander.cc
  145. gather_expander.h
  146. gather_expander_test.cc
  147. generic_transfer_manager.cc
  148. generic_transfer_manager.h
  149. heap_simulator.cc
  150. heap_simulator.h
  151. heap_simulator_test.cc
  152. hlo.proto
  153. hlo_alias_analysis.cc
  154. hlo_alias_analysis.h
  155. hlo_alias_analysis_test.cc
  156. hlo_buffer.cc
  157. hlo_buffer.h
  158. hlo_casting_utils.h
  159. hlo_casting_utils_test.cc
  160. hlo_clone_context.h
  161. hlo_computation.cc
  162. hlo_computation.h
  163. hlo_computation_test.cc
  164. hlo_constant_folding.cc
  165. hlo_constant_folding.h
  166. hlo_constant_folding_test.cc
  167. hlo_cost_analysis.cc
  168. hlo_cost_analysis.h
  169. hlo_cost_analysis_test.cc
  170. hlo_creation_utils.cc
  171. hlo_creation_utils.h
  172. hlo_creation_utils_test.cc
  173. hlo_cse.cc
  174. hlo_cse.h
  175. hlo_cse_test.cc
  176. hlo_dataflow_analysis.cc
  177. hlo_dataflow_analysis.h
  178. hlo_dataflow_analysis_test.cc
  179. hlo_dce.cc
  180. hlo_dce.h
  181. hlo_dce_test.cc
  182. hlo_domain_isolator.cc
  183. hlo_domain_isolator.h
  184. hlo_domain_map.cc
  185. hlo_domain_map.h
  186. hlo_domain_metadata.h
  187. hlo_domain_remover.cc
  188. hlo_domain_remover.h
  189. hlo_domain_test.cc
  190. hlo_domain_verifier.cc
  191. hlo_domain_verifier.h
  192. hlo_element_type_converter.cc
  193. hlo_element_type_converter.h
  194. hlo_element_type_converter_test.cc
  195. hlo_evaluator.cc
  196. hlo_evaluator.h
  197. hlo_evaluator_test.cc
  198. hlo_evaluator_typed_visitor.h
  199. hlo_evaluator_typed_visitor_bfloat16.cc
  200. hlo_evaluator_typed_visitor_bool.cc
  201. hlo_evaluator_typed_visitor_complex128.cc
  202. hlo_evaluator_typed_visitor_complex64.cc
  203. hlo_evaluator_typed_visitor_double.cc
  204. hlo_evaluator_typed_visitor_float.cc
  205. hlo_evaluator_typed_visitor_half.cc
  206. hlo_evaluator_typed_visitor_int16.cc
  207. hlo_evaluator_typed_visitor_int32.cc
  208. hlo_evaluator_typed_visitor_int64.cc
  209. hlo_evaluator_typed_visitor_int8.cc
  210. hlo_evaluator_typed_visitor_uint16.cc
  211. hlo_evaluator_typed_visitor_uint32.cc
  212. hlo_evaluator_typed_visitor_uint64.cc
  213. hlo_evaluator_typed_visitor_uint8.cc
  214. hlo_execution_profile.cc
  215. hlo_execution_profile.h
  216. hlo_execution_profile_data.proto
  217. hlo_execution_profile_test.cc
  218. hlo_get_dimension_size_rewriter.h
  219. hlo_graph_dumper.cc
  220. hlo_graph_dumper.h
  221. hlo_graph_dumper_test.cc
  222. hlo_input_output_alias_config.cc
  223. hlo_input_output_alias_config.h
  224. hlo_input_output_alias_config_test.cc
  225. hlo_instruction.cc
  226. hlo_instruction.h
  227. hlo_instruction_test.cc
  228. hlo_instructions.cc
  229. hlo_instructions.h
  230. hlo_lexer.cc
  231. hlo_lexer.h
  232. hlo_live_range.cc
  233. hlo_live_range.h
  234. hlo_live_range_test.cc
  235. hlo_liveness_analysis.cc
  236. hlo_liveness_analysis.h
  237. hlo_liveness_analysis_test.cc
  238. hlo_matchers.cc
  239. hlo_matchers.h
  240. hlo_matchers_test.cc
  241. hlo_memory_scheduler.cc
  242. hlo_memory_scheduler.h
  243. hlo_memory_scheduler_test.cc
  244. hlo_module.cc
  245. hlo_module.h
  246. hlo_module_config.cc
  247. hlo_module_config.h
  248. hlo_module_dce.cc
  249. hlo_module_dce.h
  250. hlo_module_dce_test.cc
  251. hlo_module_group.cc
  252. hlo_module_group.h
  253. hlo_module_group_metadata.cc
  254. hlo_module_group_metadata.h
  255. hlo_module_group_test.cc
  256. hlo_module_group_util.cc
  257. hlo_module_group_util.h
  258. hlo_module_test.cc
  259. hlo_opcode.cc
  260. hlo_opcode.h
  261. hlo_opcode_test.cc
  262. hlo_ordering.cc
  263. hlo_ordering.h
  264. hlo_ordering_test.cc
  265. hlo_parser.cc
  266. hlo_parser.h
  267. hlo_parser_test.cc
  268. hlo_pass_fix.h
  269. hlo_pass_interface.h
  270. hlo_pass_pipeline.cc
  271. hlo_pass_pipeline.h
  272. hlo_pass_pipeline_test.cc
  273. hlo_phi_graph.cc
  274. hlo_phi_graph.h
  275. hlo_phi_graph_test.cc
  276. hlo_profile_printer.cc
  277. hlo_profile_printer.h
  278. hlo_profile_printer_data.proto
  279. hlo_proto_util.cc
  280. hlo_proto_util.h
  281. hlo_proto_util_test.cc
  282. hlo_query.cc
  283. hlo_query.h
  284. hlo_reachability.cc
  285. hlo_reachability.h
  286. hlo_reachability_test.cc
  287. hlo_rematerialization.cc
  288. hlo_rematerialization.h
  289. hlo_rematerialization_test.cc
  290. hlo_rematerialization_test_utils.h
  291. hlo_rematerialization_test_utils_test.cc
  292. hlo_replication_analysis.cc
  293. hlo_replication_analysis.h
  294. hlo_replication_analysis_test.cc
  295. hlo_runner.cc
  296. hlo_runner.h
  297. hlo_schedule.cc
  298. hlo_schedule.h
  299. hlo_schedule_test.cc
  300. hlo_sharding.cc
  301. hlo_sharding.h
  302. hlo_sharding_metadata.cc
  303. hlo_sharding_metadata.h
  304. hlo_sharding_test.cc
  305. hlo_sharding_util.cc
  306. hlo_sharding_util.h
  307. hlo_sharding_util_test.cc
  308. hlo_subcomputation_unification.cc
  309. hlo_subcomputation_unification.h
  310. hlo_subcomputation_unification_test.cc
  311. hlo_value.cc
  312. hlo_value.h
  313. hlo_verifier.cc
  314. hlo_verifier.h
  315. hlo_verifier_test.cc
  316. human_readable_profile_builder.cc
  317. human_readable_profile_builder.h
  318. indexed_array_analysis.cc
  319. indexed_array_analysis.h
  320. indexed_array_analysis_test.cc
  321. instruction_fusion.cc
  322. instruction_fusion.h
  323. instruction_fusion_test.cc
  324. layout_assignment.cc
  325. layout_assignment.h
  326. layout_assignment_test.cc
  327. llvm_compiler.cc
  328. llvm_compiler.h
  329. local_service.cc
  330. local_service.h
  331. logical_buffer.cc
  332. logical_buffer.h
  333. logical_buffer_analysis.cc
  334. logical_buffer_analysis.h
  335. logistic_expander.cc
  336. logistic_expander.h
  337. logistic_expander_test.cc
  338. map_inliner.cc
  339. map_inliner.h
  340. map_inliner_test.cc
  341. maybe_owning_device_memory.cc
  342. maybe_owning_device_memory.h
  343. memory_space_assignment.cc
  344. memory_space_assignment.h
  345. memory_space_assignment_best_fit_repacker.cc
  346. memory_space_assignment_best_fit_repacker.h
  347. memory_space_assignment_best_fit_repacker_test.cc
  348. memory_space_assignment_repacking.h
  349. memory_space_assignment_test.cc
  350. memory_space_assignment_utils.cc
  351. memory_space_assignment_utils.h
  352. memory_space_propagation.cc
  353. memory_space_propagation.h
  354. memory_space_propagation_test.cc
  355. multi_output_fusion.cc
  356. multi_output_fusion.h
  357. name_uniquer.cc
  358. name_uniquer.h
  359. name_uniquer_test.cc
  360. op_expander_pass.cc
  361. op_expander_pass.h
  362. optimize_input_output_buffer_alias.cc
  363. optimize_input_output_buffer_alias.h
  364. optimize_input_output_buffer_alias_test.cc
  365. pattern_matcher.h
  366. pattern_matcher_gmock.h
  367. pattern_matcher_gmock_test.cc
  368. pattern_matcher_test.cc
  369. platform_util.cc
  370. platform_util.h
  371. reshape_mover.cc
  372. reshape_mover.h
  373. reshape_mover_test.cc
  374. rng_bit_generator_expander.cc
  375. rng_bit_generator_expander.h
  376. rng_expander.cc
  377. rng_expander.h
  378. root_instruction_sinker.cc
  379. root_instruction_sinker.h
  380. root_instruction_sinker_test.cc
  381. scatter_expander.cc
  382. scatter_expander.h
  383. scatter_expander_test.cc
  384. service.cc
  385. service.h
  386. service_executable_run_options.h
  387. shape_inference.cc
  388. shape_inference.h
  389. shape_inference_test.cc
  390. shaped_buffer.cc
  391. shaped_buffer.h
  392. shaped_buffer_test.cc
  393. sharding_propagation.cc
  394. sharding_propagation.h
  395. sharding_propagation_test.cc
  396. slice_sinker.cc
  397. slice_sinker.h
  398. slice_sinker_test.cc
  399. slow_operation_alarm.cc
  400. slow_operation_alarm.h
  401. sort_simplifier.cc
  402. sort_simplifier.h
  403. sort_simplifier_test.cc
  404. source_map_util.h
  405. stable_sort_expander.cc
  406. stable_sort_expander.h
  407. stable_sort_expander_test.cc
  408. stream_pool.cc
  409. stream_pool.h
  410. stream_pool_test.cc
  411. topk_rewriter.cc
  412. topk_rewriter.h
  413. topk_rewriter_test.cc
  414. transfer_manager.cc
  415. transfer_manager.h
  416. transpose_folding.cc
  417. transpose_folding.h
  418. transpose_folding_test.cc
  419. tree_reduction_rewriter.cc
  420. tree_reduction_rewriter.h
  421. triangular_solve_expander.cc
  422. triangular_solve_expander.h
  423. triangular_solve_expander_test.cc
  424. tuple_points_to_analysis.cc
  425. tuple_points_to_analysis.h
  426. tuple_points_to_analysis_test.cc
  427. tuple_simplifier.cc
  428. tuple_simplifier.h
  429. tuple_simplifier_test.cc
  430. tuple_util.cc
  431. tuple_util.h
  432. tuple_util_test.cc
  433. while_loop_analysis.cc
  434. while_loop_analysis.h
  435. while_loop_analysis_test.cc
  436. while_loop_constant_sinking.cc
  437. while_loop_constant_sinking.h
  438. while_loop_constant_sinking_test.cc
  439. while_loop_invariant_code_motion.cc
  440. while_loop_invariant_code_motion.h
  441. while_loop_invariant_code_motion_test.cc
  442. while_loop_simplifier.cc
  443. while_loop_simplifier.h
  444. while_loop_simplifier_test.cc
  445. while_loop_trip_count_annotator.cc
  446. while_loop_trip_count_annotator.h
  447. while_loop_trip_count_annotator_test.cc
  448. while_util.cc
  449. while_util.h
  450. while_util_test.cc
  451. zero_sized_hlo_elimination.cc
  452. zero_sized_hlo_elimination.h
  453. zero_sized_hlo_elimination_test.cc