tree: e26cae721011e66726e97e4d2056272513cbbcc1 [path history] [tgz]
  1. registration/
  2. allocation_description.proto
  3. allocator.cc
  4. allocator.h
  5. allocator_registry.cc
  6. allocator_registry.h
  7. allocator_test.cc
  8. api_def.proto
  9. attr_value.proto
  10. attr_value_util.cc
  11. attr_value_util.h
  12. attr_value_util_test.cc
  13. batch_util_test.cc
  14. bfloat16.cc
  15. bfloat16.h
  16. bfloat16_test.cc
  17. bounds_check.h
  18. BUILD
  19. cancellation.cc
  20. cancellation.h
  21. cancellation_test.cc
  22. collective.cc
  23. collective.h
  24. common_shape_fns.cc
  25. common_shape_fns.h
  26. common_shape_fns_test.cc
  27. control_flow.h
  28. cost_graph.proto
  29. cpu_allocator_impl.cc
  30. dataset.cc
  31. dataset.h
  32. dataset_metadata.proto
  33. dataset_options.proto
  34. dataset_stateful_op_allowlist.h
  35. dataset_test.cc
  36. device.cc
  37. device.h
  38. device_attributes.proto
  39. device_base.cc
  40. device_base.h
  41. device_base_test.cc
  42. device_factory.cc
  43. device_factory.h
  44. disable_jit_test.cc
  45. fake_input.cc
  46. fake_input.h
  47. full_type.proto
  48. full_type_inference_util.cc
  49. full_type_inference_util.h
  50. full_type_inference_util_test.cc
  51. full_type_util.cc
  52. full_type_util.h
  53. full_type_util_test.cc
  54. function.cc
  55. function.h
  56. function.proto
  57. function_handle_cache.cc
  58. function_handle_cache.h
  59. function_test.cc
  60. function_testlib.cc
  61. function_testlib.h
  62. graph.proto
  63. graph_def_util.cc
  64. graph_def_util.h
  65. graph_def_util_test.cc
  66. graph_to_functiondef.cc
  67. graph_to_functiondef.h
  68. graph_to_functiondef_test.cc
  69. graph_transfer_info.proto
  70. kernel_def.proto
  71. kernel_def_builder.cc
  72. kernel_def_builder.h
  73. kernel_def_builder_test.cc
  74. kernel_def_util.cc
  75. kernel_def_util.h
  76. kernel_def_util_test.cc
  77. kernel_shape_util.cc
  78. kernel_shape_util.h
  79. load_library.cc
  80. local_rendezvous.cc
  81. local_rendezvous.h
  82. log_memory.cc
  83. log_memory.h
  84. log_memory.proto
  85. logging.cc
  86. logging.h
  87. lookup_interface.cc
  88. lookup_interface.h
  89. memory_types.cc
  90. memory_types.h
  91. memory_types_test.cc
  92. metrics.cc
  93. metrics.h
  94. model.cc
  95. model.h
  96. model.proto
  97. model_test.cc
  98. node_def.proto
  99. node_def_builder.cc
  100. node_def_builder.h
  101. node_def_builder_test.cc
  102. node_def_util.cc
  103. node_def_util.h
  104. node_def_util_test.cc
  105. node_properties.cc
  106. node_properties.h
  107. node_properties_test.cc
  108. numeric_op.h
  109. numeric_types.h
  110. op.cc
  111. op.h
  112. op_compatibility_test.cc
  113. op_def.proto
  114. op_def_builder.cc
  115. op_def_builder.h
  116. op_def_builder_test.cc
  117. op_def_util.cc
  118. op_def_util.h
  119. op_def_util_test.cc
  120. op_gen_lib.cc
  121. op_gen_lib.h
  122. op_gen_lib_test.cc
  123. op_kernel.cc
  124. op_kernel.h
  125. op_kernel_test.cc
  126. op_kernel_test_base.h
  127. op_registration_test.cc
  128. op_requires.h
  129. op_segment.cc
  130. op_segment.h
  131. op_segment_test.cc
  132. ops_util.cc
  133. ops_util.h
  134. partial_tensor_shape.h
  135. partial_tensor_shape_test.cc
  136. queue_interface.h
  137. reader_base.cc
  138. reader_base.h
  139. reader_base.proto
  140. reader_interface.h
  141. reader_op_kernel.h
  142. register_types.h
  143. register_types_traits.h
  144. rendezvous.cc
  145. rendezvous.h
  146. rendezvous_test.cc
  147. resource_base.h
  148. resource_handle.cc
  149. resource_handle.h
  150. resource_handle.proto
  151. resource_handle_test.cc
  152. resource_mgr.cc
  153. resource_mgr.h
  154. resource_mgr_test.cc
  155. resource_op_kernel.h
  156. resource_op_kernel_test.cc
  157. resource_var.cc
  158. resource_var.h
  159. resource_var_test.cc
  160. rng_alg.h
  161. run_handler.cc
  162. run_handler.h
  163. run_handler_test.cc
  164. run_handler_util.cc
  165. run_handler_util.h
  166. run_handler_util_test.cc
  167. session_state.h
  168. shape_inference.cc
  169. shape_inference.h
  170. shape_inference_test.cc
  171. shape_inference_testutil.cc
  172. shape_inference_testutil.h
  173. shape_inference_testutil_test.cc
  174. shared_ptr_variant.h
  175. stats_aggregator.h
  176. step_stats.proto
  177. summary.proto
  178. tensor.cc
  179. tensor.h
  180. tensor.proto
  181. tensor_description.proto
  182. tensor_key.h
  183. tensor_reference.h
  184. tensor_shape.cc
  185. tensor_shape.h
  186. tensor_shape.proto
  187. tensor_shape_test.cc
  188. tensor_slice.cc
  189. tensor_slice.h
  190. tensor_slice.proto
  191. tensor_slice_test.cc
  192. tensor_test.cc
  193. tensor_testutil.cc
  194. tensor_testutil.h
  195. tensor_testutil_test.cc
  196. tensor_types.h
  197. tensor_util.cc
  198. tensor_util.h
  199. tensor_util_test.cc
  200. thread_factory.h
  201. tracking_allocator.cc
  202. tracking_allocator.h
  203. tracking_allocator_test.cc
  204. type_index.h
  205. type_traits.h
  206. typed_allocator.cc
  207. typed_allocator.h
  208. types.cc
  209. types.h
  210. types.proto
  211. types_test.cc
  212. variable.proto
  213. variant.cc
  214. variant.h
  215. variant_encode_decode.h
  216. variant_op_copy_test.cc
  217. variant_op_registry.cc
  218. variant_op_registry.h
  219. variant_op_registry_test.cc
  220. variant_tensor_data.cc
  221. variant_tensor_data.h
  222. variant_test.cc
  223. versions.cc
  224. versions.h
  225. versions.proto