Home
last modified time | relevance | path

Searched refs:set_input_value (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dtranspose_test.cc67 transpose->set_input_value(val); in SetUp()
Dreshape_test.cc67 reshape->set_input_value(val); in SetUp()
Dreduce_method_test.cc67 reduce_sum->set_input_value(val); in SetUp()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc183 (void)assign_iter_num->set_input_ref(*var_iter_num).set_input_value(*const_iter_num); in InitLoopVar()
185 (void)assign_loop_cond->set_input_ref(*var_loop_cond).set_input_value(*const_loop_cond); in InitLoopVar()
187 (void)assign_one->set_input_ref(*var_one).set_input_value(*const_one); in InitLoopVar()
189 (void)assign_zero->set_input_ref(*var_zero).set_input_value(*const_zero); in InitLoopVar()
378 (void)assign_op->set_input_ref(*init_var).set_input_value(*param_op); in InitParamWithData()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h156 void set_input_value(const std::vector<ValuePtr> &input_value) { input_value_ = input_value; } in set_input_value() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc335 operator_info->set_input_value(input_value); in CreateTheOperatorInfo()
Dstep_parallel.cc1913 (*operator_).set_input_value(input_value); in ExtractInformation()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc323 op_info->set_input_value(input_value); in CreateOpInfo()