Searched refs:kPrimMemCpyAsync (Results 1 – 3 of 3) sorted by relevance
62 if (main_name == prim::kPrimTensorMove->name() || main_name == prim::kPrimMemCpyAsync->name()) { in CheckEqualKernelBuildInfo()
202 {prim::kPrimMemCpyAsync, {InferImplMemCpyAsync, nullptr, true}}, in GetPrimitiveToEvalImplMap()
428 inline const PrimitivePtr kPrimMemCpyAsync = std::make_shared<Primitive>("memcpy_async"); variable