tree: 3c99f7379de87f7322e879eaed352cfd3230745f [path history] [tgz]
  1. experimental/
  2. testdata/
  3. __init__.py
  4. auto_control_deps.py
  5. auto_control_deps_test.py
  6. auto_control_deps_utils.py
  7. BUILD
  8. byte_swap_tensor.py
  9. c_api_util.py
  10. c_api_util_test.py
  11. combinations.py
  12. common_shapes.py
  13. common_shapes_test.py
  14. composite_tensor.py
  15. composite_tensor_gradient.py
  16. composite_tensor_test.py
  17. config.py
  18. config_test.py
  19. config_vgpu_test.py
  20. constant_op.py
  21. constant_op_test.py
  22. convert_to_constants.py
  23. convert_to_constants_test.py
  24. cpp_shape_inference.proto
  25. device.py
  26. device_spec.py
  27. device_spec_test.py
  28. device_test.py
  29. dtypes.cc
  30. dtypes.py
  31. dtypes_test.py
  32. error_interpolation.py
  33. error_interpolation_test.py
  34. errors.py
  35. errors_impl.py
  36. errors_test.py
  37. errors_test_helper.cc
  38. extension_type.py
  39. extension_type_field.py
  40. extension_type_field_test.py
  41. extension_type_test.py
  42. fast_tensor_util.pyx
  43. file_system_test.py
  44. flexible_dtypes.py
  45. flexible_dtypes_test.py
  46. framework_lib.py
  47. func_graph.py
  48. function.py
  49. function_def_to_graph.py
  50. function_def_to_graph_test.py
  51. function_test.py
  52. gpu_util.py
  53. graph_building_benchmark.py
  54. graph_io.py
  55. graph_to_function_def.py
  56. graph_util.py
  57. graph_util_impl.py
  58. graph_util_test.py
  59. immutable_dict.py
  60. immutable_dict_test.py
  61. importer.py
  62. importer_test.py
  63. indexed_slices.py
  64. indexed_slices_test.py
  65. is_mlir_bridge_test_false.py
  66. is_mlir_bridge_test_true.py
  67. is_xla_test_true.py
  68. kernels.py
  69. kernels_test.py
  70. kythe_metadata.proto
  71. load_library.py
  72. memory_checker.py
  73. memory_checker_test.py
  74. memory_checker_test_helper.cc
  75. meta_graph.py
  76. meta_graph_test.py
  77. node_file_writer_test.py
  78. offset_counter.cc
  79. offset_counter_helper.cc
  80. offset_counter_helper.h
  81. offset_counter_helper_test.cc
  82. op_allowlist_namespace_test.py
  83. op_callbacks.py
  84. op_callbacks_test.py
  85. op_def_library.py
  86. op_def_library_pybind.cc
  87. op_def_library_pybind.py
  88. op_def_library_test.py
  89. op_def_registry.cc
  90. op_def_registry.py
  91. op_def_util.cc
  92. op_def_util.h
  93. op_def_util_pybind.cc
  94. op_def_util_test.py
  95. op_reg_offset.proto
  96. ops.py
  97. ops_enable_eager_test.py
  98. ops_test.py
  99. proto_comparators.cc
  100. proto_test.py
  101. py_context_manager.cc
  102. py_context_manager.h
  103. py_context_manager_pybind.cc
  104. py_context_manager_test.py
  105. python_api_dispatcher.cc
  106. python_api_dispatcher.h
  107. python_api_dispatcher_test.py
  108. python_api_dispatcher_wrapper.cc
  109. python_api_info.cc
  110. python_api_info.h
  111. python_api_info_test.py
  112. python_api_info_wrapper.cc
  113. python_api_parameter_converter.cc
  114. python_api_parameter_converter.h
  115. python_api_parameter_converter_test.py
  116. python_api_parameter_converter_wrapper.cc
  117. python_memory_checker.py
  118. python_memory_checker_helper.cc
  119. python_op_gen.cc
  120. python_op_gen.h
  121. python_op_gen_annotation_test.py
  122. python_op_gen_annotator.cc
  123. python_op_gen_annotator.h
  124. python_op_gen_annotator_test.cc
  125. python_op_gen_main.cc
  126. python_op_gen_test.cc
  127. python_op_gen_wrapper.cc
  128. python_tensor_converter.cc
  129. python_tensor_converter.h
  130. python_tensor_converter_test.py
  131. python_tensor_converter_wrapper.cc
  132. random_seed.py
  133. random_seed_test.py
  134. registry.py
  135. registry_test.py
  136. smart_cond.py
  137. smart_cond_test.py
  138. sparse_tensor.py
  139. sparse_tensor_test.py
  140. stack.py
  141. strict_mode.py
  142. subscribe.py
  143. subscribe_test.py
  144. summary_test_util.py
  145. tensor.py
  146. tensor_conversion.py
  147. tensor_conversion_registry.py
  148. tensor_shape.py
  149. tensor_shape_casters.h
  150. tensor_shape_test.py
  151. tensor_spec.py
  152. tensor_test.py
  153. tensor_util.py
  154. tensor_util_test.py
  155. test_combinations.py
  156. test_combinations_test.py
  157. test_file_system.cc
  158. test_metrics_util.cc
  159. test_ops.cc
  160. test_ops.cu.cc
  161. test_ops.h
  162. test_util.py
  163. test_util_test.py
  164. tf2_test.py
  165. tfrt_utils.py
  166. traceable_stack.py
  167. traceable_stack_test.py
  168. type_spec.py
  169. type_spec_registry.py
  170. type_spec_test.py
  171. type_utils.py
  172. versions.py
  173. versions_test.py
  174. weak_tensor.py
  175. weak_tensor_test.py