Home
last modified time | relevance | path

Searched defs:input_prim (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dgroupnorm_fusion.cc74 bool IsReduceSumNode(const EquivPtr &equiv, const VarPtr &input_prim, const VarPtr &input_axes, in IsReduceSumNode()
91 bool IsReduceMeanNode(const EquivPtr &equiv, const VarPtr &input_prim, const VarPtr &input_axes, in IsReduceMeanNode()
Dnorm_fusion.cc75 bool IsReduceNode(const EquivPtr &equiv, const VarPtr &input_prim, const VarPtr &input_axes, std::v… in IsReduceNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/
Dfold_with_infershape.cc141 auto input_prim = GetValueNode<PrimitivePtr>(input_cnode->input(0)); in CheckCanSpecialFold() local
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dgfx10_shader_ngg.c62 const unsigned input_prim = si_get_input_prim(gs_sel, &shader->key, false); in gfx10_ngg_calculate_subgroup_info() local
Dsi_state_shaders.cpp847 unsigned input_prim = gs->info.base.gs.input_primitive; in gfx9_get_gs_info() local
1082 unsigned input_prim = sel->info.base.gs.input_primitive; in si_shader_gs() local
1507 unsigned input_prim = si_get_input_prim(gs_sel, &shader->key, false); in gfx10_shader_ngg() local
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_tess.c159 const struct draw_prim_info *input_prim, in draw_tess_ctrl_shader_run()
324 const struct draw_prim_info *input_prim, in draw_tess_eval_shader_run()
Ddraw_pt_fetch_shade_pipeline.c51 unsigned input_prim; member
Ddraw_pt_fetch_shade_pipeline_llvm.c56 enum mesa_prim input_prim; member
Ddraw_gs.c580 const struct draw_prim_info *input_prim, in draw_geometry_shader_run()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dmatmul_add_comm_reduction.cc97 auto input_prim = GetCNodePrimitive(cur_input_node); in FindPullDownNode() local
/third_party/mesa3d/src/amd/vulkan/
Dradv_shader_info.c1474 const unsigned input_prim = radv_get_pre_rast_input_topology(es_info, gs_info); in gfx10_get_ngg_info() local
Dradv_shader_info.h143 unsigned input_prim; member