| .. |
|
benchmarks
|
PY2 removal cleanup
|
2021-01-15 16:48:57 -08:00 |
|
memory_tests
|
PY2 removal cleanup
|
2021-01-15 16:48:57 -08:00 |
|
backprop_test.py
|
tape.batch_jacobian: don't make zeros with the wrong dtype if gradients are disconnected
|
2020-12-14 20:12:45 -08:00 |
|
backprop_util.py
|
Internal symbol name change.
|
2021-01-05 14:15:45 -08:00 |
|
backprop.py
|
Typo: add missing space
|
2021-01-13 21:00:04 -08:00 |
|
benchmarks_test_base.py
|
|
|
|
benchmarks_test.py
|
Rationale:
|
2020-11-03 12:55:59 -08:00 |
|
BUILD
|
PY2 removal cleanup
|
2021-01-15 16:48:57 -08:00 |
|
cancellation_test.py
|
Add a SWIG wrapper for the tensorflow::CancellationManager class.
|
2019-07-03 08:28:39 -07:00 |
|
cancellation.py
|
Use __slots__ for small classes
|
2020-06-28 18:41:22 +02:00 |
|
context_test.py
|
Allow collecting graph via RunMetadata in TFRT.
|
2020-10-26 12:40:18 -07:00 |
|
context.py
|
Clear kernel cache when reseting TPU system.
|
2021-01-08 14:47:20 -08:00 |
|
core_test.py
|
Clean up a few disable_tfrt annotations.
|
2020-09-28 13:28:19 -07:00 |
|
core.py
|
Export the Eager 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.
|
2019-12-17 19:40:40 -08:00 |
|
custom_device_test.py
|
|
|
|
custom_device_testutil.cc
|
|
|
|
def_function_test_cpu_only.py
|
Rationale:
|
2020-11-03 12:55:59 -08:00 |
|
def_function_test.py
|
Internal change
|
2021-01-07 23:26:21 -08:00 |
|
def_function_xla_jit_test.py
|
[TF2XLA] [NFC] Test case for uninitialized tensor array due to dynamic size
|
2020-12-22 19:57:53 -08:00 |
|
def_function_xla_test.py
|
|
|
|
def_function.py
|
Reduce retracing warnings.
|
2020-12-08 21:37:33 -08:00 |
|
device_placement_test.py
|
Run placer before importing function to mlir.
|
2020-11-24 17:18:21 -08:00 |
|
execute.py
|
|
|
|
executor.py
|
Use __slots__ for small classes
|
2020-06-28 18:41:22 +02:00 |
|
forwardprop_test.py
|
Merge pull request #45386 from ROCmSoftwarePlatform:google_upstream_rocm_misc_update_201203
|
2020-12-09 07:38:17 -08:00 |
|
forwardprop_util.py
|
|
|
|
forwardprop.py
|
Remove ndarray wrapper from TF Numpy. We return tensors directly.
|
2020-10-24 15:21:32 -07:00 |
|
function_argument_naming_test.py
|
Clean up a few disable_tfrt annotations.
|
2020-09-28 13:28:19 -07:00 |
|
function_defun_collection_test.py
|
Move away from deprecated asserts
|
2020-06-30 16:10:22 -07:00 |
|
function_gradients_test.py
|
Remove unnecessary eval() calls
|
2020-07-10 22:35:43 -07:00 |
|
function_test.py
|
Makes experimental_implements work correctly with GradientTape. Before it would erroneously keep experimental_implements attribute in forward/backward of eager versions of the tf.function, which would cause a runtime crash.
|
2021-01-15 10:26:33 -08:00 |
|
function.cc
|
resolve name conflict for now
|
2020-08-25 17:44:10 -07:00 |
|
function.py
|
Makes experimental_implements work correctly with GradientTape. Before it would erroneously keep experimental_implements attribute in forward/backward of eager versions of the tf.function, which would cause a runtime crash.
|
2021-01-15 10:26:33 -08:00 |
|
gradient_input_output_exclusions_test.py
|
|
|
|
gradient_input_output_exclusions.py
|
Upgrade to gast 0.4.
|
2021-01-06 10:17:47 -08:00 |
|
graph_only_ops_test.py
|
remove v1 decorator
|
2020-07-29 11:53:37 -07:00 |
|
graph_only_ops.py
|
|
|
|
imperative_grad.py
|
Export the Eager 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.
|
2019-12-17 19:40:40 -08:00 |
|
lift_to_graph_test.py
|
|
|
|
lift_to_graph.py
|
|
|
|
monitoring_test.py
|
Make eager/monitoring_test.py less flaky.
|
2020-06-08 14:44:00 -07:00 |
|
monitoring.py
|
[*.py,tensorflow/cc/framework/cc_op_gen.cc] Rename "Arguments:" to "Args:"
|
2020-12-22 09:24:04 +11:00 |
|
ops_test.py
|
- Handle error metadata properly in c_api_tfrt.
|
2020-10-16 11:59:33 -07:00 |
|
profiler_client_test.py
|
|
|
|
profiler_client.py
|
|
|
|
profiler_test.py
|
|
|
|
profiler.py
|
|
|
|
pywrap_gradient_exclusions.cc
|
Create a V2 Op to stop the gradient when the input is out of range.
|
2020-10-12 10:39:36 -07:00 |
|
pywrap_gradient_exclusions.h
|
|
|
|
pywrap_tensor_conversion.cc
|
Make TFE_TensorHandleCache aware of TFE_Context.
|
2020-10-01 12:31:59 -07:00 |
|
pywrap_tensor_conversion.h
|
Make TFE_TensorHandleCache aware of TFE_Context.
|
2020-10-01 12:31:59 -07:00 |
|
pywrap_tensor_test_util.cc
|
Export TFE_TensorHandleToNumpy in pywrap_tensor.h so that in case be used in python binding for unified API.
|
2020-09-08 20:09:11 -07:00 |
|
pywrap_tensor_test.py
|
Export TFE_TensorHandleToNumpy in pywrap_tensor.h so that in case be used in python binding for unified API.
|
2020-09-08 20:09:11 -07:00 |
|
pywrap_tensor.cc
|
Update TFE_Py_TensorShapeSlice to use abstract interface APIs instead of TF_* APIs.
|
2020-11-17 09:29:24 -08:00 |
|
pywrap_tensor.h
|
1. Add helper for casting EagerTensor* to AbstractTensorHandle*. This allows us to directly work with EagerTensors and will also allow using tf.data with unified APIs.
|
2020-09-14 17:40:33 -07:00 |
|
pywrap_tfe_src.cc
|
Simplify C++ tape APIs to match the proposal in https://github.com/tensorflow/community/pull/335
|
2020-12-03 14:28:48 -08:00 |
|
pywrap_tfe_test.py
|
Fix bug where int attributes >= 2**31 caused on exception on some platforms.
|
2020-11-20 12:50:28 -08:00 |
|
pywrap_tfe.h
|
Internal refactoring change.
|
2020-10-26 07:12:49 -07:00 |
|
remote_benchmarks_test.py
|
Remove experimental mirroring policy APIs.
|
2020-08-25 16:00:49 -07:00 |
|
remote_cloud_tpu_test.py
|
[TF2XLA] Update the test to match removed XLA:CPU device
|
2020-08-11 17:56:10 -07:00 |
|
remote_cluster_test.py
|
Use Unavailable error for non-existing server context since this indicates the remote server has restarted.
|
2020-10-21 17:32:30 -07:00 |
|
remote_execution_test.py
|
Remove the internal flag of lazy_remote_inputs_copy.
|
2021-01-07 10:42:36 -08:00 |
|
remote_test.py
|
Fix a flaky test
|
2021-01-11 13:25:07 -08:00 |
|
remote.py
|
Infer local ip in connect_to_cluster.
|
2020-10-28 17:52:06 -07:00 |
|
tape_test.py
|
|
|
|
tape.py
|
|
|
|
tensor_test.py
|
Update TFE_Py_TensorShapeSlice to use abstract interface APIs instead of TF_* APIs.
|
2020-11-17 09:29:24 -08:00 |
|
test.py
|
Add tensorflow/python/platform/__init__.py for clearer error msg.
|
2018-11-15 12:24:30 -08:00 |
|
wrap_function_device_test.py
|
Add wrap_function tests with explicit device placement.
|
2020-09-28 12:38:47 -07:00 |
|
wrap_function_test.py
|
Replace == with is in test comparing variable identity.
|
2019-08-15 13:37:05 -07:00 |
|
wrap_function.py
|
Internal symbol name change.
|
2021-01-05 14:15:45 -08:00 |
