|
|
|
@@ -16,6 +16,13 @@ |
|
|
|
#include "megbrain/graph/event.h" |
|
|
|
#include "megbrain/utils/thin/hash_table.h" |
|
|
|
|
|
|
|
namespace mgb { |
|
|
|
class NumRangeCheckerError final : public MegBrainError { |
|
|
|
public: |
|
|
|
using MegBrainError::MegBrainError; |
|
|
|
}; |
|
|
|
} // namespace mgb |
|
|
|
|
|
|
|
namespace mgb { |
|
|
|
|
|
|
|
/*! |
|
|
|
@@ -47,11 +54,7 @@ namespace mgb { |
|
|
|
std::string format_msg(const HostTensorND &hv, float range); |
|
|
|
|
|
|
|
public: |
|
|
|
class Error final: public MegBrainError { |
|
|
|
public: |
|
|
|
using MegBrainError::MegBrainError; |
|
|
|
}; |
|
|
|
|
|
|
|
using Error = NumRangeCheckerError; |
|
|
|
NumRangeChecker(cg::ComputingGraph *graph, float range); |
|
|
|
}; |
|
|
|
} |
|
|
|
|