Searched refs:ShapeVerifier (Results 1 – 2 of 2) sorted by relevance
85 Status ShapeVerifier::Preprocess(HloInstruction* hlo) { in Preprocess()100 Status ShapeVerifier::HandleElementwiseUnary(HloInstruction* hlo) { in HandleElementwiseUnary()104 Status ShapeVerifier::HandleElementwiseBinary(HloInstruction* hlo) { in HandleElementwiseBinary()108 Status ShapeVerifier::HandleClamp(HloInstruction* clamp) { in HandleClamp()112 Status ShapeVerifier::HandleSelect(HloInstruction* select) { in HandleSelect()116 Status ShapeVerifier::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect()120 Status ShapeVerifier::HandleConcatenate(HloInstruction* concatenate) { in HandleConcatenate()130 Status ShapeVerifier::HandleConvert(HloInstruction* convert) { in HandleConvert()136 Status ShapeVerifier::HandleBitcastConvert(HloInstruction* convert) { in HandleBitcastConvert()142 Status ShapeVerifier::HandleCopy(HloInstruction* copy) { in HandleCopy()[all …]
30 class ShapeVerifier : public DfsHloVisitor {32 ShapeVerifier(bool layout_sensitive, bool allow_mixed_precision) in ShapeVerifier() function182 virtual std::unique_ptr<ShapeVerifier> GetVerifier() const = 0;210 std::unique_ptr<ShapeVerifier> GetVerifier() const override { in GetVerifier()211 return absl::make_unique<ShapeVerifier>(layout_sensitive_, in GetVerifier()