Searched refs:SetAttrs (Results 1 – 22 of 22) sorted by relevance
220 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapCommon()236 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapActivation()258 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapActivationGrad()280 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapReduce()308 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapConv2D()361 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrPool()397 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrPoolGrad()423 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapAdder()454 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMapLayerNorm()509 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrSlice()
100 op_ptr->SetAttrs(attrs); in Emit()110 op_ptr->SetAttrs(attrs); in Op()
89 void SetAttrs(const DAttrs &attrs) { attrs_ = attrs; } in SetAttrs() function
39 dst_prim->SetAttrs(src_prim->attrs()); in Mapper()
32 dst_prim->SetAttrs(src_prim->attrs()); in Mapper()
42 dst_prim->SetAttrs(src_prim->attrs()); in Mapper()
47 dst_prim->SetAttrs(src_prim->attrs()); in Mapper()
52 dst_prim->SetAttrs(src_prim->attrs()); in Mapper()
60 dst_prim->SetAttrs(src_prim->attrs()); in Mapper()
163 dst_prim->SetAttrs(src_prim->attrs()); in MoveAttrMap()
83 prim->SetAttrs({
68 Primitive &SetAttrs(const std::unordered_map<std::string, ValuePtr> &attrs) { in SetAttrs() function
372 prim.SetAttrs({ in TEST_F()386 prim.SetAttrs({ in TEST_F()
135 prim->SetAttrs(origin_prim->attrs()); in ClonePrimitive()
304 prim->SetAttrs(attrs); in operator()343 prim->SetAttrs(attrs); in operator()
360 (void)new_prim->SetAttrs(prim->attrs()); in NewRecomputedNode()
67 (void)Primitive::SetAttrs(adapter->attrs_); in PrimitivePy()
77 prim->SetAttrs(attrs); in SetMiniStepOpDoMirrorLabel()96 prim->SetAttrs(attrs); in SetAllReduceRecomputeFlag()1750 (void)prim->SetAttrs(attrs_temp); in SetVirtualDatasetStrategy()1784 (void)prim->SetAttrs(attrs_temp); in SetVirtualDatasetStrategy()1853 (void)prim->SetAttrs(attrs); in GenerateBatchParallelStrategy()
362 prim->SetAttrs(attrs); in AddCommOpFusionType()380 prim->SetAttrs(attrs); in AddCommOpMeanFlag()
514 (void)new_prim->SetAttrs(end_prim->attrs()); in LastStageEndNode()
537 (void)to_primitive->SetAttrs(from_primitive->attrs()); in CopyNodeAttrs()