tree: 6661418cd36c13c6d7ded0047539a6faa4571863 [path history] [tgz]
  1. device/
  2. eager/
  3. gpu/
  4. next_pluggable_device/
  5. pluggable_device/
  6. accumulate_n_optimizer.cc
  7. all_to_all.cc
  8. all_to_all.h
  9. all_to_all_test.cc
  10. allocator_retry.h
  11. arg_ret_placement.cc
  12. arg_ret_placement.h
  13. arg_ret_placement_test.cc
  14. base_collective_executor.cc
  15. base_collective_executor.h
  16. bfc_allocator.h
  17. buf_rendezvous.cc
  18. buf_rendezvous.h
  19. buf_rendezvous_test.cc
  20. BUILD
  21. build_graph_options.cc
  22. build_graph_options.h
  23. collective_executor_mgr.cc
  24. collective_executor_mgr.h
  25. collective_executor_mgr_test.cc
  26. collective_param_resolver_local.cc
  27. collective_param_resolver_local.h
  28. collective_param_resolver_local_test.cc
  29. collective_rma_local.cc
  30. collective_rma_local.h
  31. collective_rma_local_test.cc
  32. collective_test_util.cc
  33. collective_test_util.h
  34. collective_util.cc
  35. collective_util.h
  36. colocation_graph.cc
  37. colocation_graph.h
  38. composite_device.cc
  39. composite_device.h
  40. composite_device_test.cc
  41. constant_folding.cc
  42. constant_folding.h
  43. constant_folding_test.cc
  44. copy_tensor.cc
  45. copy_tensor.h
  46. cost_constants.h
  47. cost_measurement.h
  48. cost_measurement_registry.cc
  49. cost_measurement_registry.h
  50. cost_measurement_registry_test.cc
  51. cost_util.cc
  52. cost_util.h
  53. cost_util_test.cc
  54. costmodel_manager.cc
  55. costmodel_manager.h
  56. debugger_state_interface.cc
  57. debugger_state_interface.h
  58. device.h
  59. device_factory.h
  60. device_mgr.cc
  61. device_mgr.h
  62. device_mgr_test.cc
  63. device_propagation.cc
  64. device_propagation.h
  65. device_propagation_test.cc
  66. device_resolver_local.cc
  67. device_resolver_local.h
  68. device_resolver_local_test.cc
  69. device_set.cc
  70. device_set.h
  71. device_set_test.cc
  72. direct_session.cc
  73. direct_session.h
  74. direct_session_test.cc
  75. direct_session_with_tracking_alloc_test.cc
  76. dma_helper.h
  77. dynamic_device_mgr.cc
  78. dynamic_device_mgr_test.cc
  79. entry.h
  80. eval_const_tensor.cc
  81. eval_const_tensor.h
  82. eval_const_tensor_test.cc
  83. executor.cc
  84. executor.h
  85. executor_factory.cc
  86. executor_factory.h
  87. executor_test.cc
  88. function.cc
  89. function.h
  90. function_body.cc
  91. function_body.h
  92. function_def_utils.cc
  93. function_def_utils.h
  94. function_optimization_registration_test.cc
  95. function_optimization_registry.cc
  96. function_optimization_registry.h
  97. function_optimization_registry_no_pass_test.cc
  98. function_optimization_registry_pass_failure_test.cc
  99. function_optimization_registry_test.cc
  100. function_test.cc
  101. function_testlib.cc
  102. function_testlib.h
  103. function_threadpool_test.cc
  104. function_utils.cc
  105. function_utils.h
  106. gpu_device_context.h
  107. gradients.cc
  108. gradients.h
  109. graph_constructor.cc
  110. graph_constructor.h
  111. graph_constructor_fuzz.cc
  112. graph_constructor_test.cc
  113. graph_def_builder_util.cc
  114. graph_def_builder_util.h
  115. graph_execution_state.cc
  116. graph_execution_state.h
  117. graph_optimizer.cc
  118. graph_optimizer.h
  119. graph_runner.cc
  120. graph_runner.h
  121. graph_runner_test.cc
  122. graph_view.cc
  123. graph_view.h
  124. hierarchical_tree_broadcaster.cc
  125. hierarchical_tree_broadcaster.h
  126. hierarchical_tree_broadcaster_test.cc
  127. immutable_executor_state.cc
  128. immutable_executor_state.h
  129. inline_function_utils.cc
  130. inline_function_utils.h
  131. inline_function_utils_test.cc
  132. input_colocation_exemption_registry.cc
  133. input_colocation_exemption_registry.h
  134. input_colocation_exemption_registry_test.cc
  135. inspecting_placer.cc
  136. inspecting_placer.h
  137. int32_fulltype.cc
  138. int32_fulltype.h
  139. int32_fulltype_test.cc
  140. isolate_placer_inspection_required_ops_pass.cc
  141. isolate_placer_inspection_required_ops_pass.h
  142. isolate_placer_inspection_required_ops_pass_test.cc
  143. kernel_benchmark_testlib.cc
  144. kernel_benchmark_testlib.h
  145. layout_pass_util.cc
  146. layout_pass_util.h
  147. local_device.cc
  148. local_device.h
  149. local_executor_params.h
  150. local_session_selection.cc
  151. local_session_selection.h
  152. lower_case_op.cc
  153. lower_case_op.h
  154. lower_case_op_test.cc
  155. lower_function_call_inline_policy.cc
  156. lower_function_call_inline_policy.h
  157. lower_function_call_op.cc
  158. lower_function_call_op.h
  159. lower_function_call_op_test.cc
  160. lower_functional_ops.cc
  161. lower_functional_ops.h
  162. lower_functional_ops_test.cc
  163. lower_if_op.cc
  164. lower_if_op.h
  165. lower_if_op_test.cc
  166. lower_while_op.cc
  167. lower_while_op.h
  168. lower_while_op_test.cc
  169. memory_types.cc
  170. memory_types.h
  171. memory_types_test.cc
  172. mkl_cpu_allocator.cc
  173. mkl_cpu_allocator.h
  174. mkl_cpu_allocator_test.cc
  175. mkl_layout_pass.cc
  176. mkl_layout_pass.h
  177. mkl_threadpool_device_test.cc
  178. no_op_cost_measurement.cc
  179. no_op_cost_measurement.h
  180. no_op_cost_measurement_test.cc
  181. node_file_writer.cc
  182. node_file_writer.h
  183. null_request_cost_accessor.cc
  184. null_request_cost_accessor.h
  185. null_request_cost_accessor_test.cc
  186. optimization_registry.cc
  187. optimization_registry.h
  188. optimization_registry_test.cc
  189. optimize_cross_host_control_deps.cc
  190. optimize_cross_host_control_deps.h
  191. optimize_cross_host_control_deps_test.cc
  192. optimize_function_graph_utils.cc
  193. optimize_function_graph_utils.h
  194. optimize_function_graph_utils_test.cc
  195. optimized_function_graph_info.cc
  196. optimized_function_graph_info.h
  197. optimized_function_graph_info_test.cc
  198. parallel_concat_optimizer.cc
  199. partitioning_utils.cc
  200. partitioning_utils.h
  201. partitioning_utils_test.cc
  202. pending_counts.h
  203. pending_counts_test.cc
  204. permuter.cc
  205. permuter.h
  206. permuter_test.cc
  207. placer.cc
  208. placer.h
  209. placer_inspection_required_ops_utils.cc
  210. placer_inspection_required_ops_utils.h
  211. placer_inspection_required_ops_utils_test.cc
  212. placer_test.cc
  213. pool_allocator.cc
  214. pool_allocator.h
  215. process_function_library_runtime.cc
  216. process_function_library_runtime.h
  217. process_function_library_runtime_test.cc
  218. process_state.cc
  219. process_state.h
  220. process_util.cc
  221. process_util.h
  222. process_util_test.cc
  223. profile_handler.h
  224. propagator_debug_utils.cc
  225. propagator_debug_utils.h
  226. propagator_state.cc
  227. propagator_state.h
  228. quantize_training.cc
  229. quantize_training.h
  230. quantize_training_test.cc
  231. renamed_device.cc
  232. renamed_device.h
  233. rendezvous_mgr.cc
  234. rendezvous_mgr.h
  235. rendezvous_util.cc
  236. rendezvous_util.h
  237. rendezvous_util_test.cc
  238. replicate_per_replica_nodes.cc
  239. replicate_per_replica_nodes.h
  240. replicate_per_replica_nodes_test.cc
  241. request_cost.cc
  242. request_cost.h
  243. request_cost_accessor.h
  244. request_cost_accessor_registry.cc
  245. request_cost_accessor_registry.h
  246. request_cost_accessor_registry_test.cc
  247. request_cost_test.cc
  248. ring_alg.cc
  249. ring_alg.h
  250. ring_gatherer.cc
  251. ring_gatherer.h
  252. ring_gatherer_test.cc
  253. ring_reducer.cc
  254. ring_reducer.h
  255. ring_reducer_test.cc
  256. scoped_allocator.cc
  257. scoped_allocator.h
  258. scoped_allocator_mgr.cc
  259. scoped_allocator_mgr.h
  260. scoped_allocator_mgr_test.cc
  261. session.cc
  262. session_factory.cc
  263. session_factory.h
  264. session_options.cc
  265. session_state.cc
  266. session_test.cc
  267. shape_refiner.cc
  268. shape_refiner.h
  269. shape_refiner_test.cc
  270. shared_counter.h
  271. simple_propagator_state.cc
  272. simple_propagator_state.h
  273. single_threaded_cpu_device.cc
  274. single_threaded_cpu_device.h
  275. single_threaded_executor.cc
  276. single_threaded_executor.h
  277. single_threaded_executor_test.cc
  278. stats_publisher_interface.cc
  279. stats_publisher_interface.h
  280. step_stats_collector.cc
  281. step_stats_collector.h
  282. test_collective_executor_mgr.h
  283. testlib_ops.cc
  284. threadpool_device.cc
  285. threadpool_device.h
  286. threadpool_device_factory.cc
  287. threadpool_device_test.cc
  288. type_inference.cc
  289. type_inference.h
  290. type_inference_test.cc
  291. zen_layout_pass.cc