Searched refs:visitor_func (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph.cc | 321 const VisitorFunction& visitor_func, const CallGraphNode& node, in VisitNodesInternal() argument
331 VisitNodesInternal(visitor_func, GetNode(computation), visited)); in VisitNodesInternal()
334 return visitor_func(node); in VisitNodesInternal()
337 Status CallGraph::VisitNodes(const VisitorFunction& visitor_func, in VisitNodes() argument
344 TF_RETURN_IF_ERROR(VisitNodesInternal(visitor_func, node, &visited)); in VisitNodes()
350 visitor_func, GetNode(module_->entry_computation()), &visited)); in VisitNodes()
|
| D | call_graph.h | 200 Status VisitNodes(const VisitorFunction& visitor_func,
274 const VisitorFunction& visitor_func, const CallGraphNode& node,
|
| D | dfs_hlo_visitor_with_default.h | 335 std::function<Status(HloInstructionPtr)> visitor_func) in FunctionVisitorBase() argument
336 : visitor_func_(std::move(visitor_func)) {} in FunctionVisitorBase()
|