Searched refs:visitor_func (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph.cc | 318 const VisitorFunction& visitor_func, const CallGraphNode& node, in VisitNodesInternal() argument
328 VisitNodesInternal(visitor_func, GetNode(computation), visited)); in VisitNodesInternal()
331 return visitor_func(node); in VisitNodesInternal()
334 Status CallGraph::VisitNodes(const VisitorFunction& visitor_func, in VisitNodes() argument
341 TF_RETURN_IF_ERROR(VisitNodesInternal(visitor_func, node, &visited)); in VisitNodes()
347 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 | 248 std::function<Status(HloInstructionPtr)> visitor_func) in FunctionVisitorBase() argument
249 : visitor_func_(std::move(visitor_func)) {} in FunctionVisitorBase()
|
| D | hlo_computation.h | 315 Status Accept(const std::function<Status(HloInstruction*)>& visitor_func);
317 const std::function<Status(const HloInstruction*)>& visitor_func) const;
|
| D | hlo_computation.cc | 873 const std::function<Status(HloInstruction*)>& visitor_func) { in Accept() argument
874 FunctionVisitor visitor(visitor_func); in Accept()
879 const std::function<Status(const HloInstruction*)>& visitor_func) const { in Accept()
880 ConstFunctionVisitor visitor(visitor_func); in Accept()
|
| D | hlo_instruction.h | 1025 Status Accept(const std::function<Status(HloInstruction*)>& visitor_func);
1027 const std::function<Status(const HloInstruction*)>& visitor_func) const;
|
| D | hlo_instruction.cc | 2825 const std::function<Status(HloInstruction*)>& visitor_func) { in Accept() argument
2826 FunctionVisitor visitor(visitor_func); in Accept()
2831 const std::function<Status(const HloInstruction*)>& visitor_func) const { in Accept()
2832 ConstFunctionVisitor visitor(visitor_func); in Accept()
|