adapt ResizeNearestNeighbourV2 with grad and ApplyAdam

pull/946/head
zhaozhenlong 5 years ago
parent 2b3192259d
commit bd9cebb80a

@ -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_with_decay_v1", "lamb_next_m_v_with_decay_v1"},
{"lamb_next_mv", "lamb_next_m_v"}, {"lamb_next_mv", "lamb_next_m_v"},
{"split", "split_d"}, {"split", "split_d"},
{"resize_nearest_neighbor", "resize_nearest_neighbor_d"}, {"resize_nearest_neighbor", "resize_nearest_neighbor_v2_d"},
{"resize_nearest_neighbor_grad", "resize_nearest_neighbor_grad_d"}, {"resize_nearest_neighbor_grad", "resize_nearest_neighbor_v2_grad_d"},
{"pad", "pad_d"}, {"pad", "pad_d"},
{"adam", "apply_adam"}}; {"adam", "apply_adam_d"}};
void TbeAdapter::NormalizeFuncName(std::string *func_name) { void TbeAdapter::NormalizeFuncName(std::string *func_name) {
if (func_name == nullptr) { if (func_name == nullptr) {

Loading…
Cancel
Save