Searched refs:kPrimTensorMove (Results 1 – 2 of 2) sorted by relevance
62 if (main_name == prim::kPrimTensorMove->name() || main_name == prim::kPrimMemCpyAsync->name()) { in CheckEqualKernelBuildInfo()
394 inline const PrimitivePtr kPrimTensorMove = std::make_shared<Primitive>("TensorMove"); variable