|
|
|
@@ -72,10 +72,10 @@ static std::map<string, string> tbe_func_adapter_map = { |
|
|
|
{"lamb_next_mv_with_decay_v1", "lamb_next_m_v_with_decay_v1"}, |
|
|
|
{"lamb_next_mv", "lamb_next_m_v"}, |
|
|
|
{"split", "split_d"}, |
|
|
|
{"resize_nearest_neighbor", "resize_nearest_neighbor_d"}, |
|
|
|
{"resize_nearest_neighbor_grad", "resize_nearest_neighbor_grad_d"}, |
|
|
|
{"resize_nearest_neighbor", "resize_nearest_neighbor_v2_d"}, |
|
|
|
{"resize_nearest_neighbor_grad", "resize_nearest_neighbor_v2_grad_d"}, |
|
|
|
{"pad", "pad_d"}, |
|
|
|
{"adam", "apply_adam"}}; |
|
|
|
{"adam", "apply_adam_d"}}; |
|
|
|
|
|
|
|
void TbeAdapter::NormalizeFuncName(std::string *func_name) { |
|
|
|
if (func_name == nullptr) { |
|
|
|
|