Home
last modified time | relevance | path

Searched refs:HloEvaluatorTypedVisitor (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_typed_visitor_bool.cc21 template class HloEvaluatorTypedVisitor<bool>; variable
Dhlo_evaluator_typed_visitor_float.cc21 template class HloEvaluatorTypedVisitor<float>; variable
Dhlo_evaluator_typed_visitor_double.cc21 template class HloEvaluatorTypedVisitor<double>; variable
Dhlo_evaluator_typed_visitor_complex128.cc21 template class HloEvaluatorTypedVisitor<complex128>; variable
Dhlo_evaluator_typed_visitor_int32.cc20 template class HloEvaluatorTypedVisitor<int32_t>; variable
Dhlo_evaluator_typed_visitor_uint32.cc20 template class HloEvaluatorTypedVisitor<uint32_t>; variable
Dhlo_evaluator_typed_visitor_uint16.cc20 template class HloEvaluatorTypedVisitor<uint16_t>; variable
Dhlo_evaluator_typed_visitor_int8.cc20 template class HloEvaluatorTypedVisitor<int8_t>; variable
Dhlo_evaluator_typed_visitor_complex64.cc21 template class HloEvaluatorTypedVisitor<complex64>; variable
Dhlo_evaluator_typed_visitor_bfloat16.cc21 template class HloEvaluatorTypedVisitor<bfloat16, float>; variable
Dhlo_evaluator_typed_visitor_uint64.cc21 template class HloEvaluatorTypedVisitor<uint64_t>; variable
Dhlo_evaluator_typed_visitor_uint8.cc20 template class HloEvaluatorTypedVisitor<uint8_t>; variable
Dhlo_evaluator_typed_visitor_int64.cc21 template class HloEvaluatorTypedVisitor<int64_t>; variable
Dhlo_evaluator_typed_visitor_int16.cc20 template class HloEvaluatorTypedVisitor<int16_t>; variable
Dhlo_evaluator_typed_visitor_half.cc21 template class HloEvaluatorTypedVisitor<Eigen::half, float>; variable
Dhlo_evaluator_typed_visitor.h134 class HloEvaluatorTypedVisitor : public DfsHloVisitorWithDefault {
158 explicit HloEvaluatorTypedVisitor(HloEvaluator* p) : parent_(p) {}
2495 extern template class HloEvaluatorTypedVisitor<bool>;
2496 extern template class HloEvaluatorTypedVisitor<uint8_t>;
2497 extern template class HloEvaluatorTypedVisitor<uint32_t>;
2498 extern template class HloEvaluatorTypedVisitor<uint64_t>;
2499 extern template class HloEvaluatorTypedVisitor<int8_t>;
2500 extern template class HloEvaluatorTypedVisitor<int32_t>;
2501 extern template class HloEvaluatorTypedVisitor<int64_t>;
2502 extern template class HloEvaluatorTypedVisitor<Eigen::half, float>;
[all …]
Dhlo_evaluator.cc719 std::make_unique<HloEvaluatorTypedVisitor<bool>>(this); in HloEvaluator()
721 std::make_unique<HloEvaluatorTypedVisitor<uint8_t>>(this); in HloEvaluator()
723 std::make_unique<HloEvaluatorTypedVisitor<uint16_t>>(this); in HloEvaluator()
725 std::make_unique<HloEvaluatorTypedVisitor<uint32_t>>(this); in HloEvaluator()
727 std::make_unique<HloEvaluatorTypedVisitor<uint64_t>>(this); in HloEvaluator()
729 std::make_unique<HloEvaluatorTypedVisitor<int8_t>>(this); in HloEvaluator()
731 std::make_unique<HloEvaluatorTypedVisitor<int16_t>>(this); in HloEvaluator()
733 std::make_unique<HloEvaluatorTypedVisitor<int32_t>>(this); in HloEvaluator()
735 std::make_unique<HloEvaluatorTypedVisitor<int64_t>>(this); in HloEvaluator()
737 std::make_unique<HloEvaluatorTypedVisitor<Eigen::half, float>>(this); in HloEvaluator()
[all …]
Dhlo_evaluator.h254 friend class HloEvaluatorTypedVisitor; variable