STT-tensorflow/tensorflow/python/framework
TensorFlower Gardener 7c38468051 Merge pull request #40624 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 Add a method to list op names in an ApiDefMap. 2018-05-18 19:28:20 -07:00
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 Inlined tensor_shape.{scalar,vector,matrix} 2019-07-23 15:33:44 -07:00
common_shapes.py Removed internal calls to @RegisterShape and related APIs 2019-09-18 15:44:15 -07:00
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 Remove deprecated APIs from CompositeTensor 2019-10-03 14:48:08 -07:00
config_test.py Add get_device_details API. 2020-06-11 11:42:48 -07:00
config.py Merge pull request #40624 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 #40132 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 #40132 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 Changed OpDefLibrary to use op_def_registry 2019-09-27 06:34:31 -07:00
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 Add Python API functions to query kernels 2018-07-24 03:13:49 -07:00
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 op_def_registry.get now delegates to OpRegistry::Global() 2019-09-27 02:24:02 -07:00
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 #40132. 2020-06-09 07:44:58 -07:00
proto_test.py Remove hourglass imports from even more tests 2016-12-16 18:26:25 -08:00
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 Yet another gen_file fix to support '>' namespace separators in op names. (In this case when an op has multiple named outputs) 2019-09-16 18:15:30 -07:00
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 Replace unnecessary () in run_in_graph_and_eager_modes(). 2018-06-22 01:49:29 -07:00
random_seed.py Merge pull request #34985 from kiszk:spelling_tweaks_python 2020-02-12 13:41:49 -08:00
registry_test.py
registry.py Ported tf_stack.extract_stack to C++ 2019-08-16 09:37:31 -07:00
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 Minimized the logic under the mutex in register_tensor_conversion_function 2019-07-08 03:19:27 -07:00
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 Move combinations.py closer to TF's test_util.py. 2019-07-12 18:19:51 -07:00
test_combinations.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
test_file_system.cc
test_ops_2.cc Make the python default for shape fn be call_cpp_shape_fn. 2016-11-14 10:44:25 -08:00
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 Move combinations.py closer to TF's test_util.py. 2019-07-12 18:19:51 -07:00
traceable_stack_test.py Use generators in traceable_stack and add new peek_top_obj method 2019-03-29 06:21:30 -07:00
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