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