STT-tensorflow/tensorflow/python/framework
TensorFlower Gardener 7c38468051 Merge pull request from nluehr:TF32-v2
PiperOrigin-RevId: 317757557
Change-Id: I0a0f0cc9025db7d1fbc7975b07d7d934c6fa8c2f
2020-06-22 17:23:41 -07:00
..
testdata Updating metrics_export_meta_graph.pb for fixing //tensorflow/python:framework_meta_graph_test on big endian 2019-11-08 00:19:30 -08:00
__init__.py
auto_control_deps_test.py Allow functions with raw Send & Recv to be inlined 2020-04-30 15:47:16 -07:00
auto_control_deps_utils.py Support ordering of certain collectives. 2020-03-03 16:24:21 -08:00
auto_control_deps.py Wrap save/restore logic in tf.function when in eager mode. This allows parallel saving and restoring when using multiple devices. 2020-06-22 13:23:14 -07:00
c_api_util_test.py
c_api_util.py Move the _TFBufferWrapper helper to a common place instead of in the tpu codebase. 2020-02-19 10:18:12 -08:00
combinations.py Remove default context for EagerGraphCombination. 2020-03-11 14:42:47 -07:00
common_shapes_test.py
common_shapes.py
composite_tensor_test.py Small adjustments on import spacing. 2019-12-18 20:32:12 -08:00
composite_tensor_utils_test.py
composite_tensor_utils.py
composite_tensor.py
config_test.py Add get_device_details API. 2020-06-11 11:42:48 -07:00
config.py Merge pull request from nluehr:TF32-v2 2020-06-22 17:23:41 -07:00
constant_op_test.py Fix segfault when attempting to convert string to float16. 2019-12-20 15:44:32 -08:00
constant_op.py Move 'enabled' from _pywrap_traceme to trace 2020-06-12 10:57:27 -07:00
convert_to_constants_test.py Add _SwitchN op to the handled op lists in convert_variables_to_constants_v2. 2020-06-22 12:59:11 -07:00
convert_to_constants.py Add _SwitchN op to the handled op lists in convert_variables_to_constants_v2. 2020-06-22 12:59:11 -07:00
cpp_shape_inference.proto
device_spec_test.py Implement several optimizations to reduce graph construction time. 2019-04-10 21:08:37 -07:00
device_spec.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
device_test.py
device.py
dtypes_test.py Always return the interned version from as_dtype, so that dtypes created dynamically using DType(value) retain the exact value and type of their original counterparts. This allows more brittle code performing exact type checks to continue to work as expected once lands. 2020-06-22 06:41:19 -07:00
dtypes.cc systemlibs: unbundle pybind11 2020-04-08 03:38:38 +08:00
dtypes.py Always return the interned version from as_dtype, so that dtypes created dynamically using DType(value) retain the exact value and type of their original counterparts. This allows more brittle code performing exact type checks to continue to work as expected once lands. 2020-06-22 06:41:19 -07:00
error_interpolation_test.py Make error interpolation and debug info tolerant of ops without a traceback. 2019-11-13 11:52:04 -08:00
error_interpolation.py Make error interpolation and debug info tolerant of ops without a traceback. 2019-11-13 11:52:04 -08:00
errors_impl.py Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2020-01-29 18:29:05 -08:00
errors_test.py Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2020-01-29 18:29:05 -08:00
errors.py
fast_tensor_util.pyx
file_system_test.py Automated rollback of commit 2af37fc088 2020-02-10 11:54:00 -08:00
framework_lib.py
func_graph.py Support packed variable for tf data captured function. 2020-06-22 16:24:37 -07:00
function_def_to_graph_test.py
function_def_to_graph.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
function_test.py
function.py Let Defun helper methods able to recognize TF's new FuncGraph. 2020-04-29 20:01:34 -07:00
gpu_util.py [ROCm] Fix for the broken ROCm CSB - 191223 2019-12-23 16:22:25 +00:00
graph_building_benchmark.py Add some graph building benchmarks. 2020-01-21 18:43:47 -08:00
graph_io.py Make exported MetaGraphDefs deterministic 2020-03-18 22:07:42 -07:00
graph_to_function_def.py Replace list comprehension with generator expressions. 2019-12-16 12:26:12 -08:00
graph_util_impl.py Unify convert to constants logic. 2020-06-09 09:32:59 -07:00
graph_util_test.py Unify convert to constants logic. 2020-06-09 09:32:59 -07:00
graph_util.py
importer_test.py Remove old exceptions as post removal date 2020-03-30 08:38:40 -07:00
importer.py Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2020-01-29 18:29:05 -08:00
indexed_slices.py Fixed error from printing out of IndexedSlices objects. 2020-06-15 12:47:57 -07:00
is_mlir_bridge_test_true.py Add an option to enable MLIR bridge for tf_xla_py_test rule 2020-04-13 12:40:48 -07:00
is_tfrt_test_true.py Creates a tfrt_enabled flag for TF python tests which if set will run the test with TFRT enabled. It creates a new BUILD target for tests when this flag is set which ensures that the EagerContext for that test has tfrt_enabled. 2020-04-01 16:03:33 -07:00
is_xla_test_true.py
kernels_test.py
kernels.py Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2020-01-29 18:29:05 -08:00
load_library.py Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2020-01-29 18:29:05 -08:00
memory_checker_test_helper.cc systemlibs: unbundle pybind11 2020-04-08 03:38:38 +08:00
memory_checker_test.py Move keras related memory_checker_test to keras/tests. 2020-04-11 20:41:49 -07:00
memory_checker.py Introduce a memory leak detection utility. 2020-02-12 16:10:29 -08:00
meta_graph_test.py
meta_graph.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
op_callbacks_test.py [tfdbg2] Ensure Const ops in graphs are captured by op_callbacks 2020-04-21 18:42:31 -07:00
op_callbacks.py Minor spelling and grammar fixes in op_callbacks.py 2020-04-05 16:27:12 -07:00
op_def_library_test.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
op_def_library.py [tf.function] In graph mode, preserve nested attributes for func-valued attrs. 2020-01-06 14:58:21 -08:00
op_def_registry.cc systemlibs: unbundle pybind11 2020-04-08 03:38:38 +08:00
op_def_registry.py
ops_enable_eager_test.py
ops_test.py Enable more TFRT tests. 2020-05-19 00:24:26 -07:00
ops.py Remove a defensive check that prevented adding Generic as superclass of Tensor. This is required to unblock . 2020-06-09 07:44:58 -07:00
proto_test.py
python_memory_checker_helper.cc systemlibs: unbundle pybind11 2020-04-08 03:38:38 +08:00
python_memory_checker.py Introduce a memory leak detection utility. 2020-02-12 16:10:29 -08:00
python_op_gen_internal.cc Fix invalid syntax error when import carla is present 2019-12-05 03:40:35 +00:00
python_op_gen_internal.h
python_op_gen_main.cc
python_op_gen_test.cc Create an empty test for python_op_gen. 2020-06-18 12:44:13 -07:00
python_op_gen_wrapper.cc systemlibs: unbundle pybind11 2020-04-08 03:38:38 +08:00
python_op_gen.cc Create an empty test for python_op_gen. 2020-06-18 12:44:13 -07:00
python_op_gen.h Create an empty test for python_op_gen. 2020-06-18 12:44:13 -07:00
random_seed_test.py
random_seed.py Merge pull request from kiszk:spelling_tweaks_python 2020-02-12 13:41:49 -08:00
registry_test.py
registry.py
smart_cond_test.py
smart_cond.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
sparse_tensor_test.py Simplify shape code. 2020-04-22 08:25:09 -07:00
sparse_tensor.py Move TensorLike under types.internal and rename it NativeObject to distinguish from types which actually behave like tensors. 2020-04-22 13:02:41 -07:00
subscribe_test.py
subscribe.py Prefer generator expressions over list comprehensions 2020-05-26 22:15:07 +01:00
tensor_conversion_registry.py
tensor_shape_div_test.py
tensor_shape_test.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
tensor_shape.py Reduce TensorShape.__init__ overhead by 50%. 2020-06-17 11:08:15 -07:00
tensor_spec_test.py
tensor_spec.py as_numpy_dtype is a property (so this effectively returns 0) 2019-12-16 15:11:29 -08:00
tensor_util_test.py Update make_tensor_proto to support nested nparray values when dtype is specified. 2020-05-06 13:08:53 -07:00
tensor_util.py Replace deprecated numpy.ndarray.tostring() with tobytes() 2020-06-22 15:44:17 +02:00
test_combinations_test.py
test_combinations.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
test_file_system.cc
test_ops_2.cc
test_ops.cc Make the fastpath correctly respect default types. 2019-11-22 18:15:05 -08:00
test_util_test.py Add skip_if_error context manager in test_util.py to conveniently skip errors that are not related to what is being tested. 2020-05-08 10:40:38 -07:00
test_util.py Add an option to enable MLIR bridge for tpu_py_test rule 2020-06-18 13:58:47 -07:00
tf2_test.py
traceable_stack_test.py
traceable_stack.py
type_spec_test.py
type_spec.py Adds type_spec_from_value, which can be used to get the TypeSpec for a Tensor or CompositeTensor value. 2020-06-01 14:08:15 -07:00
versions_test.py
versions.py Export the TF Session classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2020-01-29 18:29:05 -08:00