Searched defs:ProfileOp (Results 1 – 1 of 1) sorted by relevance
1523 struct ProfileOp : public Node { struct1525 ProfileOp(Graph* graph, std::function<void(std::vector<IValue>&)> callback) in ProfileOp() argument1531 const std::function<void(std::vector<IValue>&)>& getCallback() const { in getCallback()1535 void setCallback(std::function<void(std::vector<IValue>&)> callback) { in setCallback()1539 bool hasSeenTensor() const { in hasSeenTensor()1543 void setHasSeenTensor(bool has_seen_tensor) { in setHasSeenTensor()1549 bool has_seen_tensor_ = false;