Home
last modified time | relevance | path

Searched refs:EagerNode (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_executor.h46 class EagerNode {
48 EagerNode() {} in EagerNode() function
50 virtual ~EagerNode() {} in ~EagerNode()
72 class AsyncEagerNode : public EagerNode {
74 using EagerNode::EagerNode; // Lift EagerNode constructors.
110 Status SyncExecute(EagerNode* node);
116 Status AddOrExecute(std::unique_ptr<EagerNode> node);
164 std::unique_ptr<EagerNode> node;
Dcopy_to_device_node.h27 class CopyToDeviceNode : public EagerNode {
31 : EagerNode(), src_(src), dst_(dst), dstd_(dstd), ctx_(ctx) { in CopyToDeviceNode()
Dexecute_node.h72 class ExecuteNode : public EagerNode {
81 : EagerNode(), in ExecuteNode()
Deager_executor.cc87 Status EagerExecutor::SyncExecute(EagerNode* node) { in SyncExecute()
114 Status EagerExecutor::AddOrExecute(std::unique_ptr<EagerNode> node) { in AddOrExecute()
Dexecute.cc772 std::unique_ptr<EagerNode> node(new eager::RemoteExecuteNode( in EagerRemoteExecute()
1018 std::unique_ptr<EagerNode> node(new CopyToDeviceNode(h, *result, dstd, *ctx)); in LocalEagerCopyToDevice()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.h166 class ClientTensorHandleDeleteNode : public EagerNode {
171 : tensorflow::EagerNode(), in ClientTensorHandleDeleteNode()
Dremote_tensor_handle_data.cc54 std::unique_ptr<EagerNode> node( in DestroyRemoteTensorHandle()