Home
last modified time | relevance | path

Searched refs:new_prim (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dminmax_grad.h73 auto new_prim = std::make_shared<Primitive>(prim->name()); in operator() local
74 new_prim->set_attr("grad_x", MakeValue(true)); in operator()
75 new_prim->set_attr("grad_y", MakeValue(true)); in operator()
78 new_prim->set_attr("grad_y", MakeValue(false)); in operator()
81 new_prim->set_attr("grad_x", MakeValue(false)); in operator()
85 args.push_back(NewValueNode(new_prim)); in operator()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc185 auto new_prim = std::make_shared<ops::Conv2DFusion>(); in CopyConvPrim() local
186 MS_CHECK_TRUE_MSG(new_prim != nullptr, nullptr, "create Conv2DFusion return nullptr"); in CopyConvPrim()
187 new_prim->set_pad(ori_conv_prim->get_pad()); in CopyConvPrim()
188 new_prim->set_in_channel(ori_conv_prim->get_in_channel()); in CopyConvPrim()
189 new_prim->set_out_channel(ori_conv_prim->get_out_channel()); in CopyConvPrim()
190 new_prim->set_dilation(ori_conv_prim->get_dilation()); in CopyConvPrim()
191 new_prim->set_format(ori_conv_prim->get_format()); in CopyConvPrim()
192 new_prim->set_group(ori_conv_prim->get_group()); in CopyConvPrim()
193 new_prim->set_kernel_size(ori_conv_prim->get_kernel_size()); in CopyConvPrim()
195 new_prim->set_pad_mode(PAD); in CopyConvPrim()
[all …]
/third_party/mesa3d/src/mesa/drivers/dri/i965/
Dbrw_primitive_restart.c175 struct _mesa_prim new_prim; in vbo_sw_primitive_restart_common_start() local
200 new_prim = prims[0]; in vbo_sw_primitive_restart_common_start()
205 new_prim.count = indirect_params[0]; in vbo_sw_primitive_restart_common_start()
206 new_prim.start = indirect_params[2]; in vbo_sw_primitive_restart_common_start()
207 new_prim.basevertex = indirect_params[3]; in vbo_sw_primitive_restart_common_start()
213 new_ib.count = new_prim.count; in vbo_sw_primitive_restart_common_start()
215 prims = &new_prim; in vbo_sw_primitive_restart_common_start()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc359 auto new_prim = std::make_shared<Primitive>(prim::kPrimAllGather->name()); in NewRecomputedNode() local
360 (void)new_prim->SetAttrs(prim->attrs()); in NewRecomputedNode()
361 new_prim->set_attr(kAttrFusion, MakeValue(fusion_id + fusion_id_increasement_size)); in NewRecomputedNode()
362 new_prim->set_prim_type(prim->prim_type()); in NewRecomputedNode()
363 new_prim->set_instance_name(instance_name); in NewRecomputedNode()
364 auto value_node = NewValueNode(new_prim); in NewRecomputedNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dmomentum_lossscale_fusion.cc77 auto new_prim = std::make_shared<Primitive>(kFusedMulApplyMomentumOpName); in Process() local
82 std::vector<AnfNodePtr> new_node_inputs{NewValueNode(new_prim), in Process()
/third_party/mesa3d/src/gallium/drivers/svga/
Dsvga_draw_elements.c94 src_sbuf->translated_indices.new_prim == gen_prim && in translate_indices()
158 src_sbuf->translated_indices.new_prim = gen_prim; in translate_indices()
Dsvga_resource_buffer.h220 enum pipe_prim_type orig_prim, new_prim; member
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc158 auto new_prim = std::make_shared<Primitive>(*prim); in UpdateFracNZReduceOp() local
159 auto new_prim_node = NewValueNode(new_prim); in UpdateFracNZReduceOp()
162 auto axis_value = new_prim->GetAttr(kAttrAxis); in UpdateFracNZReduceOp()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.cc795 auto new_prim = [](const std::string &name) { return std::make_shared<Primitive>(name); }; in OpListFilter() local
798 …std::transform(enable_ops_only.begin(), enable_ops_only.end(), std::back_inserter(*ops), new_prim); in OpListFilter()
801 … (void)std::transform(enable_ops.begin(), enable_ops.end(), std::back_inserter(*ops), new_prim); in OpListFilter()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc513 auto new_prim = GetValueNode(value_node)->cast<PrimitivePtr>(); in LastStageEndNode() local
514 (void)new_prim->SetAttrs(end_prim->attrs()); in LastStageEndNode()