Searched defs:NodeMatcher (Results 1 – 2 of 2) sorted by relevance
110 struct NodeMatcher : public ::testing::MatcherInterface<const Node*> { struct111 bool MatchAndExplain( in MatchAndExplain()227 void DescribeTo(::std::ostream* os) const override { in DescribeTo()303 bool MatchAndExplainInput(const Node* node, int input_idx, in MatchAndExplainInput()330 absl::optional<string> op;331 absl::optional<string> name;332 absl::optional<string> assigned_device;333 absl::optional<Tensor> constant_value;334 absl::optional<std::vector<::testing::Matcher<OutEdge>>> input_matchers;336 control_dep_set;[all …]
27 struct NodeMatcher { struct28 explicit NodeMatcher(Node* node) : node_(node) {} in NodeMatcher() argument30 Node* node() const { return node_; } in node()31 const Operator* op() const { return node()->op(); } in op()32 IrOpcode::Value opcode() const { return node()->opcode(); } in opcode()34 bool HasProperty(Operator::Property property) const { in HasProperty()37 Node* InputAt(int index) const { return node()->InputAt(index); } in InputAt()39 bool Equals(const Node* node) const { return node_ == node; } in Equals()