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