Searched refs:EagerNode (Results 1 – 7 of 7) sorted by relevance
43 class EagerNode {45 explicit EagerNode(uint64 id);47 virtual ~EagerNode() {} in ~EagerNode()82 void Add(EagerNode* node);112 std::queue<EagerNode*> node_queue_ GUARDED_BY(node_queue_mutex_);
20 EagerNode::EagerNode(tensorflow::uint64 id) : id(id) {} in EagerNode() function in tensorflow::EagerNode42 void EagerExecutor::Add(EagerNode* node) { in Add()112 std::unique_ptr<EagerNode> curr_node; in Run()
26 class CopyToDeviceNode : public EagerNode {29 : EagerNode(ctx->NextId()), in CopyToDeviceNode()
32 class ExecuteNode : public EagerNode {40 : EagerNode(id), in ExecuteNode()
149 void ExecutorAdd(EagerNode* node) { executor_.Add(node); } in ExecutorAdd()
512 EagerNode* node = new ExecuteNode(id, ctx, op->Inputs(), kernel, in EagerLocalExecute()
29 class RemoteExecuteNode : public tensorflow::EagerNode {37 : tensorflow::EagerNode(id), in RemoteExecuteNode()50 : tensorflow::EagerNode(id), in RemoteExecuteNode()