| .. |
|
parallel_device
|
[rollback]Add ordering_token input to collective to leverage AutoControlDependency to order them
|
2020-10-21 19:43:47 -07:00 |
|
abstract_context.h
|
- Integrate C++ tape with op building APIs via TapeContext and TapeOperation which delegate calls to a parent execution context and record operations on the tape. Please see gradients_test.cc for usage.
|
2020-09-23 16:12:39 -07:00 |
|
abstract_function.h
|
Implement MLIR and graph tracing using abstract interfaces shared with immediate execution mode.
|
2020-06-24 21:40:46 -07:00 |
|
abstract_operation.h
|
- Integrate C++ tape with op building APIs via TapeContext and TapeOperation which delegate calls to a parent execution context and record operations on the tape. Please see gradients_test.cc for usage.
|
2020-09-23 16:12:39 -07:00 |
|
abstract_tensor_handle.h
|
Make AbstractTensorHandle RefCounted so that we can use it for refcounting tensors under a GradientTape.
|
2020-07-22 16:30:38 -07:00 |
|
BUILD
|
Use Unavailable error for non-existing server context since this indicates the remote server has restarted.
|
2020-10-21 17:32:30 -07:00 |
|
c_api_cluster_test.cc
|
Fix c_api_remote_test tsan flakiness.
|
2020-05-29 13:08:17 -07:00 |
|
c_api_debug_test.cc
|
Add Tensor & TensorHandle C APIs taking a Context
|
2020-04-06 15:10:09 -07:00 |
|
c_api_debug.cc
|
[TF2XLA] Remove unused and broken call to enable XLA, remove unneeded build macro
|
2020-08-27 11:25:21 -07:00 |
|
c_api_distributed_test.cc
|
Update function runtime to not run function optimization passes for component functions.
|
2020-10-08 11:17:43 -07:00 |
|
c_api_experimental_test.cc
|
Add functions that return device type and ID for eager.
|
2020-10-17 18:39:48 -07:00 |
|
c_api_experimental.cc
|
Add functions that return device type and ID for eager.
|
2020-10-17 18:39:48 -07:00 |
|
c_api_experimental.h
|
Add functions that return device type and ID for eager.
|
2020-10-17 18:39:48 -07:00 |
|
c_api_internal.h
|
Remove experimental mirroring policy APIs.
|
2020-08-25 16:00:49 -07:00 |
|
c_api_remote_function_test.cc
|
Temporarily revert the WaitReady change for handles with unknown devices.
|
2020-08-16 20:33:27 -07:00 |
|
c_api_remote_test_util.cc
|
Support remote multi-device functions with outputs on any remote devices.
|
2020-08-13 16:41:56 -07:00 |
|
c_api_remote_test_util.h
|
Separate function tests from c_api_remote_test
|
2020-08-11 17:14:22 -07:00 |
|
c_api_remote_test.cc
|
Separate function tests from c_api_remote_test
|
2020-08-11 17:14:22 -07:00 |
|
c_api_test_util.cc
|
Support savedmodels that reference assets.
|
2020-09-18 14:29:39 -07:00 |
|
c_api_test_util.h
|
Support savedmodels that reference assets.
|
2020-09-18 14:29:39 -07:00 |
|
c_api_test.cc
|
partial support for list attr in TFE_Op, this unblocks const-folding for ops like depthwise_conv in mlir.
|
2020-08-30 22:02:04 -07:00 |
|
c_api_unified_experimental_eager.cc
|
Implement MLIR and graph tracing using abstract interfaces shared with immediate execution mode.
|
2020-06-24 21:40:46 -07:00 |
|
c_api_unified_experimental_graph.cc
|
1. Do not cache the tracing impl string in c_api_unified_experimental instead actively lookup the factory fn and store a pointer to that in default_factory. This avoid keeping a copy of the string passed in from python. As a result SetDefaultTracingEngine now returns a Status which needed to be plumbed through.
|
2020-09-08 21:12:23 -07:00 |
|
c_api_unified_experimental_internal.h
|
1. Do not cache the tracing impl string in c_api_unified_experimental instead actively lookup the factory fn and store a pointer to that in default_factory. This avoid keeping a copy of the string passed in from python. As a result SetDefaultTracingEngine now returns a Status which needed to be plumbed through.
|
2020-09-08 21:12:23 -07:00 |
|
c_api_unified_experimental_test.cc
|
1. Do not cache the tracing impl string in c_api_unified_experimental instead actively lookup the factory fn and store a pointer to that in default_factory. This avoid keeping a copy of the string passed in from python. As a result SetDefaultTracingEngine now returns a Status which needed to be plumbed through.
|
2020-09-08 21:12:23 -07:00 |
|
c_api_unified_experimental.cc
|
1. Do not cache the tracing impl string in c_api_unified_experimental instead actively lookup the factory fn and store a pointer to that in default_factory. This avoid keeping a copy of the string passed in from python. As a result SetDefaultTracingEngine now returns a Status which needed to be plumbed through.
|
2020-09-08 21:12:23 -07:00 |
|
c_api_unified_experimental.h
|
1. Do not cache the tracing impl string in c_api_unified_experimental instead actively lookup the factory fn and store a pointer to that in default_factory. This avoid keeping a copy of the string passed in from python. As a result SetDefaultTracingEngine now returns a Status which needed to be plumbed through.
|
2020-09-08 21:12:23 -07:00 |
|
c_api.cc
|
Use Unavailable error for non-existing server context since this indicates the remote server has restarted.
|
2020-10-21 17:32:30 -07:00 |
|
c_api.h
|
Update lint check for ContextDevicePlacementPolicy since it has been moved.
|
2020-09-28 11:53:45 -07:00 |
|
custom_device_test.cc
|
Give custom devices the option to do type-based dispatch for ops with no explicit placement
|
2020-08-27 16:39:24 -07:00 |
|
custom_device_testutil.cc
|
Give custom devices the option to do type-based dispatch for ops with no explicit placement
|
2020-08-27 16:39:24 -07:00 |
|
custom_device_testutil.h
|
Give custom devices the option to do type-based dispatch for ops with no explicit placement
|
2020-08-27 16:39:24 -07:00 |
|
dlpack.cc
|
Fix multiple vulnerabilities in tf.experimental.dlpack.to_dlpack.
|
2020-09-18 19:19:43 -07:00 |
|
dlpack.h
|
fix
|
2020-03-28 08:41:24 +00:00 |
|
gradient_checker_test.cc
|
Fix tests that fail with tf32
|
2020-10-21 13:22:05 -07:00 |
|
gradient_checker.cc
|
Rename SparseSoftmaxCrossEntropyLoss to SparseSoftmaxCrossEntropyWithLogits to match up with the op name.
|
2020-09-17 17:53:59 -07:00 |
|
gradient_checker.h
|
fixing style nits in gradient checker and gradients_util
|
2020-09-10 19:52:19 +00:00 |
|
gradients_internal.h
|
First pass at implementing a C++ API for gradients built on top of abstract interfaces.
|
2020-07-10 15:03:08 -07:00 |
|
gradients_test.cc
|
Replace new Tape(...) with std::make_unique<Tape> and remove explicit delete (NFC)
|
2020-10-20 20:51:09 -07:00 |
|
gradients_util.cc
|
deleted unnecessary dependencies
|
2020-09-16 17:36:04 +00:00 |
|
gradients_util.h
|
Rename SparseSoftmaxCrossEntropyLoss to SparseSoftmaxCrossEntropyWithLogits to match up with the op name.
|
2020-09-17 17:53:59 -07:00 |
|
gradients.cc
|
Update signature of VSpace::BuildOnesLike.
|
2020-09-30 11:27:21 -07:00 |
|
gradients.h
|
Update signature of VSpace::BuildOnesLike.
|
2020-09-30 11:27:21 -07:00 |
|
immediate_execution_context.h
|
Implement more context methods for tfrt and turn on context_test.
|
2020-09-27 12:34:17 -07:00 |
|
immediate_execution_operation.h
|
[TF2XLA] Remove unused and broken call to enable XLA, remove unneeded build macro
|
2020-08-27 11:25:21 -07:00 |
|
immediate_execution_tensor_handle.h
|
Add functions that return device type and ID for eager.
|
2020-10-17 18:39:48 -07:00 |
|
mnist_gradients_test.cc
|
Merge pull request #44211 from reedwm:tf32_fixes
|
2020-10-21 17:24:44 -07:00 |
|
mnist_gradients_testutil.cc
|
Replace C++ gradient helpers with TapeContext in mnist_gradients_testutil.
|
2020-09-28 16:03:48 -07:00 |
|
mnist_gradients_testutil.h
|
Replace C++ gradient helpers with TapeContext in mnist_gradients_testutil.
|
2020-09-28 16:03:48 -07:00 |
|
tape.h
|
Update signature of VSpace::BuildOnesLike.
|
2020-09-30 11:27:21 -07:00 |
|
tfe_cancellation_manager_internal.h
|
Attempting to split c_api_internal.h into separate files.
|
2020-04-21 11:04:58 -07:00 |
|
tfe_context_internal.h
|
Split Abstract interfaces into Abstract and ImmediateExecution interfaces.
|
2020-06-17 12:39:13 -07:00 |
|
tfe_executor_internal.h
|
Attempting to split c_api_internal.h into separate files.
|
2020-04-21 11:04:58 -07:00 |
|
tfe_monitoring_internal.h
|
Attempting to split c_api_internal.h into separate files.
|
2020-04-21 11:04:58 -07:00 |
|
tfe_op_attrs_internal.h
|
Less pointer indirection for TFE_OpAttrs, add TFE_OpGetAttrs
|
2020-05-04 09:24:03 -07:00 |
|
tfe_op_internal.h
|
Split Abstract interfaces into Abstract and ImmediateExecution interfaces.
|
2020-06-17 12:39:13 -07:00 |
|
tfe_tensor_debug_info_internal.h
|
Attempting to split c_api_internal.h into separate files.
|
2020-04-21 11:04:58 -07:00 |
|
tfe_tensorhandle_internal.h
|
Split Abstract interfaces into Abstract and ImmediateExecution interfaces.
|
2020-06-17 12:39:13 -07:00 |
|
tracing_utils.cc
|
- Integrate C++ tape with op building APIs via TapeContext and TapeOperation which delegate calls to a parent execution context and record operations on the tape. Please see gradients_test.cc for usage.
|
2020-09-23 16:12:39 -07:00 |
|
tracing_utils.h
|
[Cleanup] Add a tracing::MaybeSetOpName utility to reduce the verbosity of C++ ops.
|
2020-09-23 09:18:26 -07:00 |
