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