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