experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/python/framework
History
Ron Shapiro aae0fc015b Support uint32 and uint64 in tf.get_static_value() from within a tf.function() 
Attempting to do so without this change raises an exception:

```
File ".../tensorflow/python/ops/check_ops.py", line 672, in assert_equal
  data, summarize, message, name)
File ".../tensorflow/python/ops/check_ops.py", line 369, in _binary_assert
  x_static = tensor_util.constant_value(x)
File ".../tensorflow/python/framework/tensor_util.py", line 876, in constant_value
  ret = _ConstantValue(tensor, partial)
File ".../tensorflow/python/framework/tensor_util.py", line 681, in _ConstantValue
  return MakeNdarray(tensor.op.get_attr("value"))
File ".../tensorflow/python/framework/tensor_util.py", line 641, in MakeNdarray
  raise TypeError("Unsupported tensor type: %s" % tensor.dtype)
TypeError: Unsupported tensor type: 23
```
PiperOrigin-RevId: 356437863
Change-Id: I51d9adf8727bcf01ac3a862a608fa93e2cb6a88d
2021-02-08 23:51:41 -08:00
..
experimental
PY2 removal cleanup
2021-01-15 16:48:57 -08: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
Add read variable op to function control outputs
2020-10-21 20:28:26 -07:00
auto_control_deps_utils.py
Support ordering of certain collectives.
2020-03-03 16:24:21 -08:00
auto_control_deps.py
Add CollectiveBcastSend/RecvV2 which takes input tensors rather than attributes.
2020-12-17 10:45:19 -08: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
Use __slots__ for small classes
2020-06-28 18:41:22 +02:00
combinations.py
Internal changes only.
2020-09-08 18:59:38 -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.py
add BUILD file for python/util and refactor python/BUILD
2020-12-15 11:43:18 -08:00
config_test.py
[TF2XLA] [NFC] Clarify config.optimizer.set_jit and config.optimizer.get_jit semantics
2021-01-29 16:49:06 -08:00
config.py
[TF2XLA] [NFC] Clarify config.optimizer.set_jit and config.optimizer.get_jit semantics
2021-01-29 16:49:06 -08: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
Small doc-string fixes for tf.constant.
2020-12-09 17:42:57 -08:00
convert_to_constants_test.py
Replace instances of "blacklist" with "denylist" where possible. See Google Developer guidelines at https://developers.google.com/style/word-list#blacklist for more information.
2020-07-20 16:05:25 -07:00
convert_to_constants.py
Merge pull request #45610 from TomWildenhain-Microsoft:patch-1
2021-02-08 12:11:20 -08:00
cpp_shape_inference.proto
Record the type of a variant tensor with its handle data
2020-09-30 17:00:01 -07:00
device_spec_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
device_spec.py
Doc format fixes.
2020-11-03 12:14:05 -08:00
device_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
device.py
Use __slots__ for small classes
2020-06-28 18:41:22 +02:00
dtypes_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
dtypes.cc
Fixes malformatted error string in dtypes.cc
2020-11-05 11:11:49 -08:00
dtypes.py
Fixes tf.bool.as_numpy_dtype to return np.bool_ instead of np.bool (which is the same as Python bool).
2021-01-22 16:39:54 -08:00
error_interpolation_test.py
Fix skipped tests in error_interpolation_test.py
2020-10-26 09:16:32 -07: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 OperatorNotAllowedInGraphError to public for tf.keras to use public API of TensorFlow only. The change won't break any existing usages.
2020-07-20 15:30:23 -07:00
errors_test.py
Add python/lib BUILD and refactor python/BUILD
2020-12-10 12:42:17 -08:00
errors.py
Removing remove_undocumented calls from tensorflow/python.
2018-04-25 11:57:31 -07:00
fast_tensor_util.pyx
Replace deprecated usage of np.asscalar with np.ndarray.item()
2019-01-28 20:00:38 +01:00
file_system_test.py
Automated rollback of commit 2af37fc088a495f2e55421274aff475cfba5e3bf
2020-02-10 11:54:00 -08:00
framework_lib.py
Removed internal calls to @RegisterShape and related APIs
2019-09-18 15:44:15 -07:00
func_graph.py
Tiny typo fix "default_use_recource" -> "default_use_resource"
2021-02-02 06:56:49 -08: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
Set handle data of function inputs and outputs.
2020-12-07 20:55:03 -08:00
function_test.py
Annotate FunctionDef registered from eager runtime to correct annotate tensorflow::Graph's construction context.
2020-12-11 14:41:33 -08:00
function.py
Remove accessing Graph._collections via the protected member in tensorflow/python/framework/function.py.
2020-12-08 10:04:32 -08: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
Adds proto comparators and options for nan-equivalence to graph util.
2021-01-26 09:35:06 -08:00
graph_util_test.py
Adds proto comparators and options for nan-equivalence to graph util.
2021-01-26 09:35:06 -08:00
graph_util.py
Adds proto comparators and options for nan-equivalence to graph util.
2021-01-26 09:35:06 -08:00
importer_test.py
Mark tests that are not compatible with v2 behavior "run_v1_only".
2020-12-01 16:58:28 -08: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
Add an example of an IndexedSlices
2020-12-07 09:31:21 -08:00
is_mlir_bridge_test_false.py
Add plumbing for setting enable_mlir_bridge=False
2020-09-25 11:10:35 -07:00
is_mlir_bridge_test_true.py
Add plumbing for setting enable_mlir_bridge=False
2020-09-25 11:10:35 -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
[TF:XLA] Enable XLA through autojit for all tensorflow/python/kernel_test/ tests.
2019-01-14 02:42:27 -08:00
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
Merge pull request #43610 from Intel-tensorflow:pluggable_device_load
2020-12-08 16:47:36 -08:00
memory_checker_test_helper.cc
systemlibs: unbundle pybind11
2020-04-08 03:38:38 +08:00
memory_checker_test.py
Add C++ loop memory leak test for MemoryChecker
2020-12-09 10:21:43 -08:00
memory_checker.py
Fix a memory leak when creating UninitializedVariables (as SavedModel loading code does)
2020-09-15 10:41:31 -07:00
meta_graph_test.py
Include Ops that are used via PartitionedCalls to MetaGraphDef.MetaInfoDef.stripped_op_list
2020-07-22 19:39:13 -07:00
meta_graph.py
Include Ops that are used via PartitionedCalls to MetaGraphDef.MetaInfoDef.stripped_op_list
2020-07-22 19:39:13 -07: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
fix typos in python directory
2020-10-29 16:21:24 +03:00
op_def_library_test.py
minor spelling tweaks
2020-02-11 15:09:21 +09:00
op_def_library.py
add BUILD file for python/util and refactor python/BUILD
2020-12-15 11:43:18 -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
op_def_util_pybind.cc
Add c++ utility function that converts AttrValue protobufs to corresponding PyObjects.
2020-08-12 13:40:50 -07:00
op_def_util_test.py
Add c++ utility function that converts AttrValue protobufs to corresponding PyObjects.
2020-08-12 13:40:50 -07:00
op_def_util.cc
Narrow dependencies for op_def_util from safe_ptr.h to safe_pyobject_ptr.h
2020-10-12 15:10:05 -07:00
op_def_util.h
Narrow dependencies for op_def_util from safe_ptr.h to safe_pyobject_ptr.h
2020-10-12 15:10:05 -07:00
ops_enable_eager_test.py
TF Framework ops_eager_test missing test case add
2019-03-06 11:24:53 +05:30
ops_test.py
Remove ndarray wrapper from TF Numpy. We return tensors directly.
2021-02-04 19:20:27 -08:00
ops.py
Remove ndarray wrapper from TF Numpy. We return tensors directly.
2021-02-04 19:20:27 -08:00
proto_comparators.cc
Adds proto comparators and options for nan-equivalence to graph util.
2021-01-26 09:35:06 -08:00
proto_test.py
Remove hourglass imports from even more tests
2016-12-16 18:26:25 -08:00
py_context_manager_pybind.cc
Utility class for calling Python context mangers from c++
2020-08-28 14:16:01 -07:00
py_context_manager_test.py
Utility class for calling Python context mangers from c++
2020-08-28 14:16:01 -07:00
py_context_manager.cc
Utility class for calling Python context mangers from c++
2020-08-28 14:16:01 -07:00
py_context_manager.h
Utility class for calling Python context mangers from c++
2020-08-28 14:16:01 -07:00
python_api_dispatcher_test.py
Dispatch handler for Python APIs. For background, see the RFC for [TensorFlow Extension Types](https://github.com/tensorflow/community/pull/269).
2020-10-21 16:10:52 -07:00
python_api_dispatcher_wrapper.cc
Dispatch handler for Python APIs. For background, see the RFC for [TensorFlow Extension Types](https://github.com/tensorflow/community/pull/269).
2020-10-21 16:10:52 -07:00
python_api_dispatcher.cc
Dispatch handler for Python APIs. For background, see the RFC for [TensorFlow Extension Types](https://github.com/tensorflow/community/pull/269).
2020-10-21 16:10:52 -07:00
python_api_dispatcher.h
Dispatch handler for Python APIs. For background, see the RFC for [TensorFlow Extension Types](https://github.com/tensorflow/community/pull/269).
2020-10-21 16:10:52 -07:00
python_api_info_test.py
Rollback of rollback of "Extend basic binary operations to more integral types"
2021-01-28 09:51:49 -08:00
python_api_info_wrapper.cc
Class to hold precomputed information about a TensorFlow Python API, to allow the API to be executed rapidly.
2020-11-09 15:53:35 -08:00
python_api_info.cc
Class to hold precomputed information about a TensorFlow Python API, to allow the API to be executed rapidly.
2020-11-09 15:53:35 -08:00
python_api_info.h
Class to hold precomputed information about a TensorFlow Python API, to allow the API to be executed rapidly.
2020-11-09 15:53:35 -08:00
python_api_parameter_converter_test.py
Type converter for Python API parameters.
2020-11-25 13:02:33 -08:00
python_api_parameter_converter_wrapper.cc
Type converter for Python API parameters.
2020-11-25 13:02:33 -08:00
python_api_parameter_converter.cc
Type converter for Python API parameters.
2020-11-25 13:02:33 -08:00
python_api_parameter_converter.h
Type converter for Python API parameters.
2020-11-25 13:02:33 -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
Merge pull request #40903 from tg-at-google:sign-compare-warning-fixes-batch-6
2020-07-15 16:35:51 -07:00
python_op_gen_internal.h
PR review changes
2020-06-23 23:24:20 +00:00
python_op_gen_main.cc
Replace instances of "whitelist" with "allowlist" where possible. See Google Developer guidelines at https://developers.google.com/style/word-list#blacklist for more information.
2020-07-08 10:13:43 -07:00
python_op_gen_test.cc
Merge pull request #40557 from rahul-kamat:gen-ops-with-type-annotations
2020-06-29 12:04:02 -07:00
python_op_gen_wrapper.cc
systemlibs: unbundle pybind11
2020-04-08 03:38:38 +08:00
python_op_gen.cc
Move the _ThreadLocalData class used by eager Context from Python (eager/context.py) to c++ (pywrap_tfe.h).
2020-09-10 12:50:34 -07:00
python_op_gen.h
Merge pull request #40557 from rahul-kamat:gen-ops-with-type-annotations
2020-06-29 12:04:02 -07:00
python_tensor_converter_test.py
Add PythonTensorConverter class, which can be used in c++ to efficiently convert PyObjects to tensors.
2020-10-26 17:48:14 -07:00
python_tensor_converter_wrapper.cc
Add PythonTensorConverter class, which can be used in c++ to efficiently convert PyObjects to tensors.
2020-10-26 17:48:14 -07:00
python_tensor_converter.cc
Add PythonTensorConverter class, which can be used in c++ to efficiently convert PyObjects to tensors.
2020-10-26 17:48:14 -07:00
python_tensor_converter.h
Add PythonTensorConverter class, which can be used in c++ to efficiently convert PyObjects to tensors.
2020-10-26 17:48:14 -07:00
random_seed_test.py
…
random_seed.py
Change "graph-level" to "global" in docs for consistency.
2020-12-04 17:17:36 -08:00
registry_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
registry.py
Move stack more of the trace mappers and filters to C++. Simplify the API surface of extract_stack, and clean up its docstring.
2020-12-21 06:45:42 -08:00
smart_cond_test.py
Add @run_deprecated_v1 annotation to tests failing in v2
2018-11-29 15:43:25 -08:00
smart_cond.py
[*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:"
2020-12-22 09:24:04 +11:00
sparse_tensor_test.py
SparseTensor: when deserializing from component tensors, propagate the static dense_shape from the typespec.
2020-11-13 14:33:11 -08:00
sparse_tensor.py
add BUILD file for python/util and refactor python/BUILD
2020-12-15 11:43:18 -08:00
subscribe_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
subscribe.py
Merge pull request #40783 from lgeiger:slots
2020-07-09 13:13:04 -07: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
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
tensor_shape_test.py
Removing __hash__ in TensorShape and Dimension. Having __hash__ in Dimension somehow causes the tensor b in b = tf.constant(0, shape=[a.shape[0]]) to be placed on a different graph than a.
2021-02-05 12:16:33 -08:00
tensor_shape.py
Removing __hash__ in TensorShape and Dimension. Having __hash__ in Dimension somehow causes the tensor b in b = tf.constant(0, shape=[a.shape[0]]) to be placed on a different graph than a.
2021-02-05 12:16:33 -08:00
tensor_spec_test.py
remove v1 only decorator
2020-07-29 10:59:13 -07:00
tensor_spec.py
Register TF BoundedTensorSpec for json serialization.
2021-02-05 14:15:35 -08:00
tensor_util_test.py
Support uint32 and uint64 in tf.get_static_value() from within a tf.function()
2021-02-08 23:51:41 -08:00
tensor_util.py
Support uint32 and uint64 in tf.get_static_value() from within a tf.function()
2021-02-08 23:51:41 -08:00
test_combinations_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
test_combinations.py
[*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:"
2020-12-22 09:24:04 +11:00
test_file_system.cc
Moving some filesystems to Transactional API
2020-08-04 23:05:16 -07:00
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
Make tf.test.TestCase respect superclass setUp and tearDown.
2020-10-13 09:11:02 -07:00
test_util.py
Fix leaking registered functions when destroying a loaded model.
2021-01-21 14:14:11 -08:00
tf2_test.py
fixed the breakage. Re-submit plumbing of the enable_v2_behavior() call to C++
2021-01-06 20:35:51 -08:00
tfrt_utils.py
Add a TFRT utility class.
2020-10-07 12:29:14 -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
Move stack more of the trace mappers and filters to C++. Simplify the API surface of extract_stack, and clean up its docstring.
2020-12-21 06:45:42 -08:00
type_spec_test.py
When a pair of namedtuple objects are compared to determine their most specific compatible type, the namedtuple objects can be treated as normal tuples. This can result in failing to create a dataset consisting of nested namedtuples. This CL fixes this problem.
2021-01-19 11:28:20 -08:00
type_spec.py
When a pair of namedtuple objects are compared to determine their most specific compatible type, the namedtuple objects can be treated as normal tuples. This can result in failing to create a dataset consisting of nested namedtuples. This CL fixes this problem.
2021-01-19 11:28:20 -08:00
versions_test.py
Move away from deprecated asserts
2020-06-30 16:10:22 -07:00
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