Searched refs:FindEnterNodeFromTPUExecuteNodeInput (Results 1 – 1 of 1) sorted by relevance
80 xla::StatusOr<Node*> FindEnterNodeFromTPUExecuteNodeInput(Node* input_node) { in FindEnterNodeFromTPUExecuteNodeInput() function164 TF_ASSIGN_OR_RETURN(auto enter_node, FindEnterNodeFromTPUExecuteNodeInput( in ExtractExecuteNodeInfo()607 FindEnterNodeFromTPUExecuteNodeInput(variable_edge->src())); in AddReshardOp()