STT-tensorflow

benchmarks

Merge pull request #45907 from jpodivin:reportarg

2020-12-21 21:07:25 -08:00

memory_tests

Internal change only.

2020-09-10 09:48:17 -07: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

Add pfor converters and basic type information to optionals

2020-11-19 09:25:24 -08:00

backprop.py

add BUILD file for python/util and refactor python/BUILD

2020-12-15 11:43:18 -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

disable tap for failed test

2020-12-16 14:43:26 -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

Merge pull request #43610 from Intel-tensorflow:pluggable_device_load

2020-12-08 16:47:36 -08:00

core_test.py

Clean up a few disable_tfrt annotations.

2020-09-28 13:28:19 -07:00

core.py

…

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

Reduce retracing warnings.

2020-12-08 21:37:33 -08:00

def_function_xla_jit_test.py

[TF2XLA] Support asserts as no-ops for tf.function(jit_compile=True)

2020-12-17 18:53:10 -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

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

Internal change

2020-12-11 13:32:13 -08:00

function.cc

resolve name conflict for now

2020-08-25 17:44:10 -07:00

function.py

add BUILD file for python/util and refactor python/BUILD

2020-12-15 11:43:18 -08:00

gradient_input_output_exclusions_test.py

Make sure the gradient_input_output_exclusions_test is resilient to windows line endings.

2020-02-13 09:13:58 -08:00

gradient_input_output_exclusions.py

…

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

Use __slots__ for small classes

2020-06-28 18:41:22 +02: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

profiler session allow more options to be specified at python level. such as host trace level and python tracer enable.

2020-04-01 14:39:29 -07:00

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

…

remote_test.py

Fix a data race in resolve when the device of a TensorHandle is unknown at creation time.

2020-12-14 18:17:54 -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 clean-up: move signature handling code that's specific to WrapFunction into WrapFunction subclass.

2020-04-11 09:43:35 -07:00