STT-tensorflow

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

add benchmarks_test option to capture profiles

2020-07-20 17:27:49 -07:00

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

…

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

Move away from deprecated asserts

2020-06-30 16:10:22 -07:00

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

Initialize variables in tf.function without conds when possible

2019-01-31 14:17:19 -08:00

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

Internal change

2020-07-17 20:18:01 -07:00

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

…

lift_to_graph_test.py

Fix a bug in lift_to_graph.py.

2020-06-11 02:32:40 -07:00

lift_to_graph.py

Fix a bug in lift_to_graph.py.

2020-06-11 02:32:40 -07:00

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

Use trace.Trace instead of traceme.TraceMe in python

2020-06-12 19:48:10 -07:00

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

Use __slots__ for small classes

2020-06-28 18:41:22 +02:00

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

…

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

…

wrap_function.py

Internal symbol name change.

2021-01-05 14:15:45 -08:00