| .. |
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
lift_to_graph.py
|
|
|
|
monitoring_test.py
|
|
|
|
monitoring.py
|
|
|
|
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
|
|
|
