Home
last modified time | relevance | path

Searched refs:MakeValue (Results 1 – 25 of 464) sorted by relevance

12345678910>>...19

/third_party/mindspore/tests/ut/cpp/operator/
Dcc_implementations_test.cc34 list.push_back(MakeValue(static_cast<int64_t>(1))); in TEST_F()
35 list.push_back(MakeValue(static_cast<int64_t>(2))); in TEST_F()
39 list.push_back(MakeValue(1.0f)); in TEST_F()
40 list.push_back(MakeValue(1.5f)); in TEST_F()
44 list.push_back(MakeValue(3.0)); in TEST_F()
45 list.push_back(MakeValue(0.5)); in TEST_F()
49 list.push_back(MakeValue(INT64_MAX)); in TEST_F()
50 list.push_back(MakeValue(static_cast<int64_t>(2))); in TEST_F()
59 list.push_back(MakeValue(INT64_MIN)); in TEST_F()
60 list.push_back(MakeValue(static_cast<int64_t>(-1))); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_attrs_process.cc37 AnfAlgo::SetNodeAttr("input_names", MakeValue(input_names), anf_node); in SetAkgAttrsForFour2Five()
38 AnfAlgo::SetNodeAttr("output_names", MakeValue(output_names), anf_node); in SetAkgAttrsForFour2Five()
47 AnfAlgo::SetNodeAttr("dst_type", MakeValue(dst_type), anf_node); in SetAkgAttrsForFour2Five()
54 AnfAlgo::SetNodeAttr("input_names", MakeValue(input_names), anf_node); in SetAkgAttrsForFive2Four()
55 AnfAlgo::SetNodeAttr("output_names", MakeValue(output_names), anf_node); in SetAkgAttrsForFive2Four()
63 AnfAlgo::SetNodeAttr("shape4d", MakeValue(shape_transform), anf_node); in SetAkgAttrsForFive2Four()
64 AnfAlgo::SetNodeAttr("output_format", MakeValue(kOpFormat_NCHW), anf_node); in SetAkgAttrsForFive2Four()
73 AnfAlgo::SetNodeAttr("dstType", MakeValue(dst_type), anf_node); in SetAkgAttrsForFive2Four()
81 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForCast()
82 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForCast()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dcc_implementations.cc210 …return MakeValue(sum); …
214 …return MakeValue(sum); …
218 …return MakeValue(sum); …
222 …return MakeValue(sum); …
226 …return MakeValue(sum); …
230 …return MakeValue(sum); …
234 …return MakeValue(sum); …
238 …return MakeValue(sum); …
242 …return MakeValue(sum); …
246 …return MakeValue(sum); …
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_gru_v2_grad_fission.cc106 AnfAlgo::SetNodeAttr("t_state", MakeValue(SizeToLong(cur_t)), gru_v2_hidden_grad_cell_op); in CreateGRUV2HiddenGradCellNode()
107 AnfAlgo::SetNodeAttr("gate_order", MakeValue(gate_order), gru_v2_hidden_grad_cell_op); in CreateGRUV2HiddenGradCellNode()
149 AnfAlgo::SetNodeAttr("transpose_x1", MakeValue(false), matmul_node); in AddTLoopNode()
150 AnfAlgo::SetNodeAttr("transpose_x2", MakeValue(true), matmul_node); in AddTLoopNode()
182 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(t_size)), concat_t_node); in AddTConcatNode()
183 …AnfAlgo::SetNodeAttr(kAttrDynInputSizes, MakeValue(std::vector<int64_t>{SizeToLong(t_size)}), conc… in AddTConcatNode()
184 AnfAlgo::SetNodeAttr(kAttrAxis, MakeValue(static_cast<int64_t>(0)), concat_t_node); in AddTConcatNode()
231 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(SizeToLong(0)), split_v); in AddHSplitNode()
232 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(SizeToLong(kSplitVOutputNum)), split_v); in AddHSplitNode()
233 AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(split_list), split_v); in AddHSplitNode()
[all …]
Ddynamic_rnn_grad_fission_v2.cc63 AnfAlgo::SetNodeAttr("forget_bias", MakeValue(1.0f), basic_lstm_cell_c_state_grad); in CreateTLoopNode()
64 AnfAlgo::SetNodeAttr("activation", MakeValue("Tanh"), basic_lstm_cell_c_state_grad); in CreateTLoopNode()
72 AnfAlgo::SetNodeAttr("transpose_x1", MakeValue(false), matmul); in CreateTLoopNode()
73 AnfAlgo::SetNodeAttr("transpose_x2", MakeValue(true), matmul); in CreateTLoopNode()
86 MakeValue(std::vector<int64_t>{ in CreateTLoopNode()
90 … AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(static_cast<int64_t>(kAttrSplitDimValue)), split_v); in CreateTLoopNode()
91 … AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(static_cast<int64_t>(kAttrNumSplitValue)), split_v); in CreateTLoopNode()
110 AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(size_split), lstm_split); in CreateLSTMSPlitV()
111 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(static_cast<int64_t>(0)), lstm_split); in CreateLSTMSPlitV()
112 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(SizeToLong(num_split_x)), lstm_split); in CreateLSTMSPlitV()
[all …]
Dgather_v2_ds_fission.cc72 auto padding_value = MakeValue(padding_vector); in CreatePad()
76 auto last_padding_value = MakeValue(last_padding_vector); in CreatePad()
80 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), pad); in CreatePad()
81 AnfAlgo::SetNodeAttr(kAttrInputIsDynamicShape, MakeValue(true), pad); in CreatePad()
82 AnfAlgo::SetNodeAttr(kAttrOutputIsDynamicShape, MakeValue(true), pad); in CreatePad()
104 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), gather_v2); in CreateGatherV2Ds()
105 AnfAlgo::SetNodeAttr(kAttrInputIsDynamicShape, MakeValue(true), gather_v2); in CreateGatherV2Ds()
107 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), gather_v2); in CreateGatherV2Ds()
109 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), gather_v2); in CreateGatherV2Ds()
124 AnfAlgo::SetNodeAttr(kAttrBegin, MakeValue(Convert2Long(offsets)), slice); in CreateSlice()
[all …]
/third_party/mindspore/mindspore/core/ops/
Dresize.cc45 (void)this->AddAttr(kFormat, MakeValue(swi)); in set_format()
50 (void)this->AddAttr(kMethod, MakeValue(swi)); in set_method()
53 void Resize::set_new_height(const int64_t new_height) { (void)this->AddAttr(kNewHeight, MakeValue(n… in set_new_height()
55 void Resize::set_new_width(const int64_t new_width) { (void)this->AddAttr(kNewWidth, MakeValue(new_… in set_new_width()
58 (void)this->AddAttr(kPreserveAspectRatio, MakeValue(preserve_aspect_ratio)); in set_preserve_aspect_ratio()
63 (void)this->AddAttr(kCoordinateTransformMode, MakeValue(swi)); in set_coordinate_transform_mode()
66 void Resize::set_cubic_coeff(const float cubic_coeff) { (void)this->AddAttr(kCubicCoeff, MakeValue(… in set_cubic_coeff()
69 (void)this->AddAttr(kExcludeOutside, MakeValue(exclude_outside)); in set_exclude_outside()
73 (void)this->AddAttr(kExtrapolationValue, MakeValue(extrapolation_value)); in set_extrapolation_value()
78 (void)this->AddAttr(kNearestMode, MakeValue(swi)); in set_nearest_mode()
Dproposal.cc26 …_feat_stride(const float feat_stride) { (void)this->AddAttr(kFeatStride, MakeValue(feat_stride)); } in set_feat_stride()
33 void Proposal::set_base_size(const float base_size) { (void)this->AddAttr(kBaseSize, MakeValue(base… in set_base_size()
40 void Proposal::set_min_size(const float min_size) { (void)this->AddAttr(kMinSize, MakeValue(min_siz… in set_min_size()
47 void Proposal::set_ratio(const std::vector<float> &ratio) { (void)this->AddAttr(kRatio, MakeValue(r… in set_ratio()
54 void Proposal::set_scale(const std::vector<float> &scale) { (void)this->AddAttr(kScale, MakeValue(s… in set_scale()
62 (void)this->AddAttr(kPreNmsTopn, MakeValue(pre_nms_topn)); in set_pre_nms_topn()
71 (void)this->AddAttr(kPostNmsTopn, MakeValue(post_nms_topn)); in set_post_nms_topn()
79 void Proposal::set_nms_thresh(const float nms_thresh) { (void)this->AddAttr(kNmsThresh, MakeValue(n… in set_nms_thresh()
Dprior_box.cc26 (void)this->AddAttr(kMinSizes, MakeValue(min_sizes)); in set_min_sizes()
32 (void)this->AddAttr(kMaxSizes, MakeValue(max_sizes)); in set_max_sizes()
41 (void)this->AddAttr(kAspectRatios, MakeValue(aspect_ratios)); in set_aspect_ratios()
47 (void)this->AddAttr(kVariances, MakeValue(variances)); in set_variances()
56 (void)this->AddAttr(kImageSizeW, MakeValue(image_size_w)); in set_image_size_w()
65 (void)this->AddAttr(kImageSizeH, MakeValue(image_size_h)); in set_image_size_h()
73 void PriorBox::set_step_w(const float step_w) { (void)this->AddAttr(kStepW, MakeValue(step_w)); } in set_step_w()
80 void PriorBox::set_step_h(const float step_h) { (void)this->AddAttr(kStepH, MakeValue(step_h)); } in set_step_h()
87 void PriorBox::set_clip(const bool clip) { (void)this->AddAttr(kClip, MakeValue(clip)); } in set_clip()
94 void PriorBox::set_flip(const bool flip) { (void)this->AddAttr(kFlip, MakeValue(flip)); } in set_flip()
[all …]
Dadder.cc36 void Adder::set_in_channel(const int64_t in_channel) { (void)this->AddAttr(kInChannel, MakeValue(in… in set_in_channel()
43 void Adder::set_out_channel(const int64_t out_channel) { (void)this->AddAttr(kOutChannel, MakeValue… in set_out_channel()
51 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
61 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
69 void Adder::set_stride(const std::vector<int64_t> &stride) { (void)this->AddAttr(kStride, MakeValue… in set_stride()
76 …_list(const std::vector<int64_t> &pad_list) { (void)this->AddAttr(kPadList, MakeValue(pad_list)); } in set_pad_list()
83 …tion(const std::vector<int64_t> &dilation) { (void)this->AddAttr(kDilation, MakeValue(dilation)); } in set_dilation()
90 void Adder::set_group(const int64_t group) { (void)this->AddAttr(kGroup, MakeValue(group)); } in set_group()
99 (void)this->AddAttr(kFormat, MakeValue(swi)); in set_format()
Dsplit_with_overlap.cc34 …ap::set_ratio(const std::vector<int64_t> &ratio) { (void)this->AddAttr(kRatio, MakeValue(ratio)); } in set_ratio()
37 (void)this->AddAttr(kExtendTop, MakeValue(extend_top)); in set_extend_top()
41 (void)this->AddAttr(kExtendBottom, MakeValue(extend_bottom)); in set_extend_bottom()
45 (void)this->AddAttr(kNumberSplit, MakeValue(number_split)); in set_number_split()
48 void SplitWithOverlap::set_split_dim(int64_t split_dim) { (void)this->AddAttr(kSplitDim, MakeValue(… in set_split_dim()
50 …Overlap::set_split_stride(int64_t stride) { (void)this->AddAttr(kSplitStride, MakeValue(stride)); } in set_split_stride()
52 void SplitWithOverlap::set_pad_top(int64_t pad_top) { (void)this->AddAttr(kPadTop, MakeValue(pad_to… in set_pad_top()
55 (void)this->AddAttr(kTransFormat, MakeValue(trans_format)); in set_trans_format()
Dconv2d_transpose.cc47 … MakeValue(CheckAndConvertUtils::CheckInteger(kInChannel, in_channel, kGreaterThan, 0, name()))); in set_in_channel()
52 … MakeValue(CheckAndConvertUtils::CheckInteger(kOutChannel, out_channel, kGreaterThan, 0, name()))); in set_out_channel()
61 (void)AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
70 (void)AddAttr(kStride, MakeValue(stride)); in set_stride()
77 (void)AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
90 (void)AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
96 (void)AddAttr(kPad, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kPad, pad, name()))); in set_pad()
100 …(void)AddAttr(kMode, MakeValue(CheckAndConvertUtils::CheckInteger(kMode, mode, kEqual, 1, name()))… in set_mode()
104 …(void)AddAttr(kGroup, MakeValue(CheckAndConvertUtils::CheckInteger(kGroup, group, kGreaterThan, 0,… in set_group()
109 (void)AddAttr(kFormat, MakeValue(f)); in set_format()
[all …]
Ddetection_post_process.cc42 (void)this->AddAttr(kInputSize, MakeValue(inputSize)); in set_input_size()
50 …cess::set_scale(const std::vector<float> &scale) { (void)this->AddAttr(kScale, MakeValue(scale)); } in set_scale()
57 (void)this->AddAttr(kNmsIouThreshold, MakeValue(NmsIouThreshold)); in set_nms_iou_threshold()
65 (void)this->AddAttr(kNmsScoreThreshold, MakeValue(NmsScoreThreshold)); in set_nms_score_threshold()
73 (void)this->AddAttr(kMaxDetections, MakeValue(MaxDetections)); in set_max_detections()
78 (void)this->AddAttr(kDetectionsPerClass, MakeValue(DetectionsPerClass)); in set_detections_per_class()
86 (void)this->AddAttr(kMaxClassesPerDetection, MakeValue(MaxClassesPerDetection)); in set_max_classes_per_detection()
93 (void)this->AddAttr(kNumClasses, MakeValue(NumClasses)); in set_num_classes()
97 (void)this->AddAttr(kUseRegularNms, MakeValue(UseRegularNms)); in set_use_regular_nms()
105 (void)this->AddAttr(kOutQuantized, MakeValue(OutQuantized)); in set_out_quantized()
[all …]
Daffine.cc30 (void)this->AddAttr(kAffineContext, MakeValue(context)); in set_context()
33 void Affine::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kAffineOutputDim, MakeValue(o… in set_output_dim()
35 void Affine::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpose_a))… in set_transpose_a()
37 void Affine::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpose_b))… in set_transpose_b()
40 (void)this->AddAttr(kActivationType, MakeValue(static_cast<int64_t>(activation_type))); in set_activation_type()
/third_party/mindspore/mindspore/core/ops/grad/
Dde_conv2d_grad_filter.cc43 (void)this->AddAttr(kInChannel, MakeValue(in_channel)); in set_in_channel()
53 (void)this->AddAttr(kOutChannel, MakeValue(out_channel)); in set_out_channel()
63 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
74 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
84 (void)this->AddAttr(kPadList, MakeValue(pad_list)); in set_pad_list()
94 (void)this->AddAttr(kStride, MakeValue(stride)); in set_stride()
104 (void)this->AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
113 void DeConv2DGradFilter::set_group(const int64_t group) { (void)this->AddAttr(kGroup, MakeValue(gro… in set_group()
123 (void)this->AddAttr(kFormat, MakeValue(swi)); in set_format()
134 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
[all …]
Dlstm_grad.cc36 (void)AddAttr(kInput_size, MakeValue(input_size)); in set_input_size()
41 (void)AddAttr(kHidden_size, MakeValue(hidden_size)); in set_hidden_size()
46 (void)AddAttr(kNumLayers, MakeValue(num_layers)); in set_num_layers()
49 void LSTMGrad::set_has_bias(const bool has_bias) { (void)AddAttr(kHasBias, MakeValue(has_bias)); } in set_has_bias()
56 (void)AddAttr(kDropout, MakeValue(dropout)); in set_dropout()
62 …bidirectional(const bool bidirectional) { (void)AddAttr(kBidirectional, MakeValue(bidirectional));… in set_bidirectional()
68 (void)AddAttr(kNumDirections, MakeValue(num_directions)); in set_num_directions()
71 void LSTMGrad::set_zoneout_cell(float zoneout_cell) { (void)AddAttr(kZoneoutCell, MakeValue(zoneout… in set_zoneout_cell()
75 void LSTMGrad::set_zoneout_hidden(float zoneout_hidden) { (void)AddAttr(kZoneoutHidden, MakeValue(z… in set_zoneout_hidden()
Dgroup_conv2d_grad_input.cc44 (void)this->AddAttr(kInChannel, MakeValue(in_channel)); in set_in_channel()
54 (void)this->AddAttr(kOutChannel, MakeValue(out_channel)); in set_out_channel()
64 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
75 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
85 (void)this->AddAttr(kPadList, MakeValue(pad_list)); in set_pad_list()
95 (void)this->AddAttr(kStride, MakeValue(stride)); in set_stride()
105 (void)this->AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
114 void GroupConv2DGradInput::set_group(const int64_t &group) { (void)this->AddAttr(kGroup, MakeValue(… in set_group()
123 (void)this->AddAttr(kInputShape, MakeValue(input_shape)); in set_input_shape()
134 (void)this->AddAttr(kFormat, MakeValue(swi)); in set_format()
[all …]
Dpooling_grad.cc38 (void)this->AddAttr(kPoolMode, MakeValue(swi)); in set_pool_mode()
46 …set_window(const std::vector<int64_t> &window) { (void)this->AddAttr(kWindow, MakeValue(window)); } in set_window()
54 …set_stride(const std::vector<int64_t> &stride) { (void)this->AddAttr(kStride, MakeValue(stride)); } in set_stride()
64 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
74 (void)this->AddAttr(kPadList, MakeValue(pad_list)); in set_pad_list()
85 (void)this->AddAttr(kRoundMode, MakeValue(swi)); in set_round_mode()
96 (void)this->AddAttr(kFormat, MakeValue(swi)); in set_format()
105 void PoolingGrad::set_global(const bool global) { (void)this->AddAttr(kGlobal, MakeValue(global)); } in set_global()
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Ddp_algo_test.cc173 ValuePtr transpose_a_0 = MakeValue(false); in SetUp()
174 ValuePtr transpose_b_0 = MakeValue(false); in SetUp()
183 ValuePtr transpose_a_1 = MakeValue(false); in SetUp()
184 ValuePtr transpose_b_1 = MakeValue(false); in SetUp()
193 ValuePtr transpose_a_2 = MakeValue(false); in SetUp()
194 ValuePtr transpose_b_2 = MakeValue(false); in SetUp()
203 ValuePtr transpose_a_3 = MakeValue(false); in SetUp()
204 ValuePtr transpose_b_3 = MakeValue(false); in SetUp()
213 ValuePtr transpose_a_4 = MakeValue(false); in SetUp()
214 ValuePtr transpose_b_4 = MakeValue(false); in SetUp()
[all …]
Dedge_costmodel_test.cc63 ValuePtr transpose_a_1 = MakeValue(false); in SetUp()
64 ValuePtr transpose_b_1 = MakeValue(false); in SetUp()
72 ValuePtr transpose_a_2 = MakeValue(false); in SetUp()
73 ValuePtr transpose_b_2 = MakeValue(false); in SetUp()
81 ValuePtr transpose_a_3 = MakeValue(false); in SetUp()
82 ValuePtr transpose_b_3 = MakeValue(false); in SetUp()
90 ValuePtr transpose_a_4 = MakeValue(false); in SetUp()
91 ValuePtr transpose_b_4 = MakeValue(false); in SetUp()
99 ValuePtr transpose_a_5 = MakeValue(false); in SetUp()
100 ValuePtr transpose_b_5 = MakeValue(true); in SetUp()
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc92 conv2d->AddAttr("stride", MakeValue(static_cast<int64_t>(2))); in TEST_F()
93 conv2d->AddAttr("pad", MakeValue(static_cast<int64_t>(0))); in TEST_F()
94 conv2d->AddAttr("dilation", MakeValue(static_cast<int64_t>(0))); in TEST_F()
117 prim->AddAttr("T", MakeValue(static_cast<int64_t>(0))); in TEST_F()
138 batch_norm->AddAttr("epsilon", MakeValue(0.001f)); in TEST_F()
139 batch_norm->AddAttr("momentum", MakeValue(0.1f)); in TEST_F()
176 prim->AddAttr("stride", MakeValue(list)); in TEST_F()
177 prim->AddAttr("pad", MakeValue(static_cast<int64_t>(0))); in TEST_F()
178 prim->AddAttr("pad_mode", MakeValue(std::string("pad"))); in TEST_F()
179 prim->AddAttr("dilation", MakeValue(static_cast<int64_t>(1))); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.cc35 ValuePtr param_value = MakeValue(shape); in SkipRedisReshapeOP()
49 ValuePtr ptr = MakeValue(shape); in ReshapeOP()
55 ValuePtr param_value = MakeValue(shape); in ReshapeOP()
64 ValuePtr attr_value = MakeValue(value); in CreateStridedSliceOp()
72 ValuePtr param_begin_value = MakeValue(begin); in CreateStridedSliceOp()
74 ValuePtr param_end_value = MakeValue(end); in CreateStridedSliceOp()
77 ValuePtr param_strides_value = MakeValue(strides); in CreateStridedSliceOp()
155 ValuePtr attr_value = MakeValue(group_name); in AllGatherOP()
169 ValuePtr attr_value = MakeValue(concat_dim); in ConcatOP()
185 ValuePtr attr_value_axis = MakeValue(DEFAULT); in SplitOP()
[all …]
/third_party/mindspore/mindspore/core/ops/fusion/
Dactivation.cc26 void Activation::set_alpha(const float alpha) { (void)this->AddAttr(kAlpha, MakeValue(alpha)); } in set_alpha()
28 void Activation::set_min_val(const float min_val) { (void)this->AddAttr(kMinVal, MakeValue(min_val)… in set_min_val()
30 void Activation::set_max_val(const float max_val) { (void)this->AddAttr(kMaxVal, MakeValue(max_val)… in set_max_val()
34 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
57 void Activation::set_approximate(bool approximate) { (void)this->AddAttr(kApproximate, MakeValue(ap… in set_approximate()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc50 AnfAlgo::SetNodeAttr(kAttrVisited, MakeValue(true), reshape); in CreateReshapeNode()
51 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(dst_shape), reshape); in CreateReshapeNode()
69 AnfAlgo::SetNodeAttr(kAttrSrcFormat, MakeValue(input_format), trans_node); in SetTransNodeAttr()
70 AnfAlgo::SetNodeAttr(kAttrDstFormat, MakeValue(output_format), trans_node); in SetTransNodeAttr()
99 AnfAlgo::SetNodeAttr(kAttrGroups, MakeValue(fz_group), out_trans); in SetGroupAttr()
100 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(fz_group), out_trans); in SetGroupAttr()
102 AnfAlgo::SetNodeAttr(kAttrGroups, MakeValue(fz_group), in_trans); in SetGroupAttr()
103 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(fz_group), in_trans); in SetGroupAttr()
311 AnfAlgo::SetNodeAttr(kAttrPerm, MakeValue(perm), trans_node); in NewTransOpNode()
314 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), trans_node); in NewTransOpNode()
[all …]
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc142 std::vector<ValuePtr> elements = {MakeValue(v1), MakeValue(v2)}; in Make_Manager()
152 ValuePtr transpose_a = MakeValue(false); in Make_Manager()
153 ValuePtr transpose_b = MakeValue(false); in Make_Manager()
156 prim1->AddAttr("instance_name", MakeValue("matmul1")); in Make_Manager()
161 std::vector<ValuePtr> elements2 = {MakeValue(v3), MakeValue(v4)}; in Make_Manager()
178 prim2->AddAttr("instance_name", MakeValue("matmul2")); in Make_Manager()
182 prim1->set_attr("strategy", MakeValue(static_cast<int64_t>(0))); in Make_Manager()
186 std::vector<ValuePtr> elements_t = {MakeValue(static_cast<int64_t>(0))}; in Make_Manager()
194 std::vector<ValuePtr> elements_t2 = {MakeValue(vt1), MakeValue(vt2)}; in Make_Manager()
225 ValuePtr val1 = MakeValue(v1); in TEST_F()
[all …]

12345678910>>...19