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