Searched refs:prim_conv (Results 1 – 5 of 5) sorted by relevance
123 auto prim_conv = GetValueNode<PrimitivePtr>(conv_cnode->input(0)); in CheckCanFusion() local124 MS_ASSERT(prim_conv != nullptr); in CheckCanFusion()134 if (prim_conv->GetAttr(ops::kOutChannel) == nullptr) { in CheckCanFusion()137 auto out_channel = GetValue<int64_t>(prim_conv->GetAttr(ops::kOutChannel)); in CheckCanFusion()
369 static const unsigned prim_conv[] = { in si_conv_pipe_prim() local386 assert(mode < ARRAY_SIZE(prim_conv)); in si_conv_pipe_prim()387 return prim_conv[mode]; in si_conv_pipe_prim()1032 static const int prim_conv[] = { in si_conv_prim_to_gs_out() local1050 assert(mode < ARRAY_SIZE(prim_conv)); in si_conv_prim_to_gs_out()1052 return prim_conv[mode]; in si_conv_prim_to_gs_out()
50 static const int prim_conv[] = { in r300_translate_primitive() local66 unsigned hwprim = prim_conv[prim]; in r300_translate_primitive()
139 static const unsigned prim_conv[] = { in r600_conv_pipe_prim() local157 assert(prim < ARRAY_SIZE(prim_conv)); in r600_conv_pipe_prim()158 return prim_conv[prim]; in r600_conv_pipe_prim()163 static const int prim_conv[] = { in r600_conv_prim_to_gs_out() local181 assert(mode < ARRAY_SIZE(prim_conv)); in r600_conv_prim_to_gs_out()183 return prim_conv[mode]; in r600_conv_prim_to_gs_out()
1706 …PrimitivePtr prim_conv = dyn_cast<Primitive>((dyn_cast<ValueNode>(conv_node->input(kZeroNum)))->va… in ExportMergeConv() local1708 (*node_map_ptr)[node] = ExportPrimitive(func_graph, node_map_ptr, prim_conv, inputs, graph_proto); in ExportMergeConv()