Searched refs:kPrimGatherD (Results 1 – 6 of 6) sorted by relevance
35 …GatherD() : PrimitiveC(prim::kPrimGatherD->name()) { InitIOName({"x", "dim", "index"}, {"output"})… in GatherD()
73 REGISTER_PRIMITIVE_EVAL_IMPL(GatherD, prim::kPrimGatherD, GatherDInfer, nullptr, true);
48 if (AnfAlgo::GetCNodeName(cnode) == prim::kPrimGatherD->name()) { in Process()
43 Register(prim::kPrimGatherD->name(), {1}); in ConstInputToAttrInfoRegistry()
179 inline const PrimitivePtr kPrimGatherD = std::make_shared<Primitive>("GatherD"); variable
518 if (device_target != kGPUDevice && op_run_info->op_name == prim::kPrimGatherD->name()) { in ConstructInputTensor()