Searched refs:kPrimScatterAdd (Results 1 – 3 of 3) sorted by relevance
32 {prim::kPrimScatterAdd->name(), &ScatterArithmeticCPUKernel<T>::ScatterAdd}, in InitComputeFunc()
118 {prim::kPrimScatterAdd, {InferImplScatterAdd, nullptr, true}}, in GetPrimitiveToEvalImplMap()
226 inline const PrimitivePtr kPrimScatterAdd = std::make_shared<Primitive>("ScatterAdd"); variable