| .. |
|
benchmarks
|
Enable passing TFRT python tests.
|
2020-10-15 17:52:05 -07:00 |
|
memory_tests
|
Internal change only.
|
2020-09-10 09:48:17 -07:00 |
|
backprop_test.py
|
Rolls back cl/338294004 and part of cl/338299497
|
2020-10-21 19:58:10 -07:00 |
|
backprop_util.py
|
Rolls back cl/338294004 and part of cl/338299497
|
2020-10-21 19:58:10 -07:00 |
|
backprop.py
|
Return None, not ndarray wrapping a None from tape gradient
|
2020-08-10 18:03:06 -07:00 |
|
benchmarks_test_base.py
|
add benchmarks_test option to capture profiles
|
2020-07-20 17:27:49 -07:00 |
|
benchmarks_test.py
|
Enable passing TFRT python tests.
|
2020-10-15 17:52:05 -07:00 |
|
BUILD
|
Increase shard count for test that times out
|
2020-10-16 17:36:41 -07:00 |
|
cancellation_test.py
|
|
|
|
cancellation.py
|
Use __slots__ for small classes
|
2020-06-28 18:41:22 +02:00 |
|
context_test.py
|
Implement more context methods for tfrt and turn on context_test.
|
2020-09-27 12:34:17 -07:00 |
|
context.py
|
Set a timeout to check health RPC
|
2020-10-21 13:02:25 -07: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
|
throwTypeError -> ThrowTypeError
|
2020-04-16 10:46:01 -07:00 |
|
def_function_test_cpu_only.py
|
Fallback tf.function with compile_with_xla==true to current TF. And disable a few function tests that are not supported by tfrt.
|
2020-10-13 14:41:16 -07:00 |
|
def_function_test.py
|
Disallow saving if the function cannot be used for inference
|
2020-10-15 21:08:51 -07:00 |
|
def_function_xla_jit_test.py
|
Fix DefFunctionTest.testConstantOnWrongDevice to not be flakey
|
2020-10-02 10:13:05 -07:00 |
|
def_function_xla_test.py
|
|
|
|
def_function.py
|
Internal change
|
2020-10-09 11:12:24 -07:00 |
|
device_placement_test.py
|
Refactor SelectDevice method to allow us to reuse this method for TFRT device placement. This can allow TFRT to make the same decision as current TF.
|
2020-07-10 17:26:08 -07: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
|
Convert IndexedSlices to dense tensors for forwardprop's tf.function cases
|
2020-09-22 18:26:01 -07:00 |
|
forwardprop_util.py
|
|
|
|
forwardprop.py
|
Merge pull request #41628 from abhichou4:exp/accumulator
|
2020-07-31 12:15:11 -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
|
Add read variable op to function control outputs
|
2020-10-21 20:28:26 -07:00 |
|
function.cc
|
resolve name conflict for now
|
2020-08-25 17:44:10 -07:00 |
|
function.py
|
Rolls back cl/338294004 and part of cl/338299497
|
2020-10-21 19:58:10 -07:00 |
|
gradient_input_output_exclusions_test.py
|
|
|
|
gradient_input_output_exclusions.py
|
Fix the string representation of string literal subscripts to be x['a'] instead of x[a].
|
2020-05-25 17:12:18 -07: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
|
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
|
allow profiler_client.py v2 (our rpc client) to provide ProfilerOptions,
|
2020-04-13 14:24:44 -07:00 |
|
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
|
Make TFE_TensorHandleCache aware of TFE_Context.
|
2020-10-01 12:31:59 -07: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
|
Avoid dereferencing a null pointer when the operation is missing.
|
2020-10-20 09:28:07 -07:00 |
|
pywrap_tfe_test.py
|
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 |
|
pywrap_tfe.h
|
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 |
|
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
|
Flip the ordering of test context managers so that parameterized testcases can be correctly processed.
|
2020-03-06 14:40:15 -08:00 |
|
remote_test.py
|
Remove experimental mirroring policy APIs.
|
2020-08-25 16:00:49 -07:00 |
|
remote.py
|
Infer local ip in connect_to_cluster.
|
2020-10-28 18:25:44 -07:00 |
|
tape_test.py
|
Do not create an extra GradientTape in custom_gradient/recompute_grad. The extra tape leads the tf.function gradient code to believe that the user intends to compute higher order derivatives. This requires generating a forward function with all possible side outputs which is expensive. This also doesn't work well with control flow and causes the added test to fail.
|
2020-04-17 19:58:53 -07:00 |
|
tape.py
|
Use __slots__ for small classes
|
2020-06-28 18:41:22 +02:00 |
|
tensor_test.py
|
Support more DTypes in c_api_tfrt.
|
2020-10-02 11:27:59 -07: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 |
