Searched refs:kPrimReduceAny (Results 1 – 4 of 4) sorted by relevance
54 Register(prim::kPrimReduceAny->name(), {1}); in ConstInputToAttrInfoRegistry()
63 } else if (kernel_name_ == prim::kPrimReduceAny->name()) { in InitKernel()
224 {prim::kPrimReduceAny, {InferImplReduceFunc, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
475 inline const PrimitivePtr kPrimReduceAny = std::make_shared<Primitive>("ReduceAny"); variable