Home
last modified time | relevance | path

Searched refs:DfsHloVisitor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h175 tensorflow::gtl::FlatMap<PrimitiveType, std::unique_ptr<DfsHloVisitor>,
Dhlo_computation.cc663 template Status HloComputation::Accept(DfsHloVisitor* visitor) const;
667 DfsHloVisitor* visitor, in AcceptWithOperandOrder()
715 DfsHloVisitor*, const std::vector<const HloInstruction*>&) const;
Dhlo_verifier.h28 class ShapeVerifier : public DfsHloVisitor {
Dhlo_computation.h287 DfsHloVisitor* visitor,
Ddfs_hlo_visitor.h352 using DfsHloVisitor = DfsHloVisitorBase<HloInstruction*>; variable
Dhlo_instruction.h661 Status AcceptWithOperandOrder(DfsHloVisitor* visitor,
678 Status AcceptOrdered(DfsHloVisitor* visitor,
Dhlo_instruction.cc2688 template Status HloInstruction::Visit(DfsHloVisitor* visitor);
2809 template Status HloInstruction::Accept(DfsHloVisitor*, bool, bool);
2813 DfsHloVisitor* visitor, const CompareFunction& operand_order, in AcceptWithOperandOrder()
2877 DfsHloVisitor* visitor, const std::vector<const HloInstruction*>& order) { in AcceptOrdered()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc189 return DfsHloVisitor::Postprocess(hlo); in Postprocess()