Searched refs:l2_normalize (Results 1 – 4 of 4) sorted by relevance
35 auto l2_normalize = std::make_shared<L2Normalize>(); in TEST_F() local36 l2_normalize->Init(std::vector<int64_t>{0, 1, 2}); in TEST_F()39 auto abstract = l2_normalize->Infer({tensor_x->ToAbstract()}); in TEST_F()
212 self.l2_normalize = P.L2Normalize()215 out = self.l2_normalize(x)
212 from .l2_normalize import _l2_normalize_tbe
333 …BE", "async_flag": false, "binfile_name": "l2_normalize.so", "compute_cost": 10, "kernel_name": "l…