Searched refs:TypedVisitor (Results 1 – 2 of 2) sorted by relevance
200 class HloEvaluator::TypedVisitor : public DfsHloVisitorWithDefault { class in xla::HloEvaluator202 explicit TypedVisitor(HloEvaluator* p) : parent_(p) {} in TypedVisitor() function in xla::HloEvaluator::TypedVisitor2038 typed_visitors_[PRED] = MakeUnique<TypedVisitor<bool>>(this); in HloEvaluator()2039 typed_visitors_[U8] = MakeUnique<TypedVisitor<uint8>>(this); in HloEvaluator()2043 typed_visitors_[U32] = MakeUnique<TypedVisitor<uint32>>(this); in HloEvaluator()2044 typed_visitors_[U64] = MakeUnique<TypedVisitor<uint64>>(this); in HloEvaluator()2045 typed_visitors_[S8] = MakeUnique<TypedVisitor<int8>>(this); in HloEvaluator()2049 typed_visitors_[S32] = MakeUnique<TypedVisitor<int32>>(this); in HloEvaluator()2050 typed_visitors_[S64] = MakeUnique<TypedVisitor<int64>>(this); in HloEvaluator()2051 typed_visitors_[F16] = MakeUnique<TypedVisitor<Eigen::half, float>>(this); in HloEvaluator()[all …]
121 class TypedVisitor;