Home
last modified time | relevance | path

Searched refs:set_kernel_info (Results 1 – 25 of 63) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc76 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
77 relu2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
91 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
147 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
148 relu2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
149 relu3->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
150 relu4->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
151 relu5->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
152 relu6->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
170 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion_test.cc81 cast0->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
82 cast1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
100 layer->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
142 cast0->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
143 cast1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
159 layer->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
201 cast0->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
202 cast1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
218 layer->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
255 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc102 add1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
103 add2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
110 add1->set_kernel_info(nullptr); in TEST_F()
114 add1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
145 add1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
146 add2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
Deliminate_redundant_op_test.cc109 sub->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
110 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
176 sub->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
177 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
253 sub->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
254 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
Dallreduce_fusion_test.cc66 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
107 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
149 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
205 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dinsert_cast_test.cc78 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
81 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
132 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
135 node->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
Dinsert_trans_op_test.cc57 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in GetSingleOutputGraph()
81 max_pool->set_kernel_info(std::make_shared<device::KernelInfo>()); in GetMutilpleOutputGraph()
Dremove_internal_output_test.cc60 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in GetSingleOutputGraph()
89 max_pool->set_kernel_info(std::make_shared<device::KernelInfo>()); in GetMutilpleOutputGraph()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc71 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
114 conv2d_backprop_filter->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
172 dynamic_rnn->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
226 layer_norm->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
271 relu1->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
272 relu2->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
286 cast->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
343 add->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
345 conv2d->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc252 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
271 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
290 pre_add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
370 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
403 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
457 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
474 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
492 pre_add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
513 add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
528 pre_add->set_kernel_info(std::make_shared<KernelInfo>()); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dsplit_umonad.cc53 depend_cnode->set_kernel_info(std::make_shared<device::KernelInfo>()); in ProcessNode()
65 new_cnode->set_kernel_info(cnode->kernel_info_ptr()); in ProcessNode()
Dupdate_state_formatter.cc70 tuple_getitem->set_kernel_info(std::make_shared<device::KernelInfo>()); in ExtendInputsOfUpdateState()
121 mt_node->set_kernel_info(std::make_shared<device::KernelInfo>()); in Run()
126 new_node->set_kernel_info(std::make_shared<device::KernelInfo>()); in Run()
Dgraph_kernel_helper.cc141 new_value_node->set_kernel_info(std::make_shared<device::KernelInfo>()); in ConvertToScalarTensor()
239 parameter->set_kernel_info(vnode->kernel_info_ptr()); in ConvertNonscalarTensorToParameter()
339 fg->parameters()[i]->set_kernel_info(std::make_shared<device::KernelInfo>()); in SetNewKernelInfo()
610 cnode->set_kernel_info(std::make_shared<device::KernelInfo>()); in ResetKernelInfo()
704 cnode->set_kernel_info(kernel_info); in CreateCNode()
870 param->set_kernel_info(std::make_shared<device::KernelInfo>()); in LiteGraph2AnfGraph()
893 value_node->set_kernel_info(std::make_shared<device::KernelInfo>()); in LiteGraph2AnfGraph()
922 mt->set_kernel_info(std::make_shared<device::KernelInfo>()); in LiteGraph2AnfGraph()
Deliminate_redundant_output.cc51 idx_node->set_kernel_info(std::make_shared<device::KernelInfo>()); in SetIndex()
254 make_tuple->set_kernel_info(std::make_shared<device::KernelInfo>()); in ReplaceMakeTuple()
Dsubstitute_dropout.cc53 uniform_real_input[1]->set_kernel_info(std::make_shared<device::KernelInfo>()); in PreProcess()
Ddecrease_transfer_precision.cc161 cnode->set_kernel_info(std::make_shared<device::KernelInfo>()); in Process_Father()
274 cnode->set_kernel_info(std::make_shared<device::KernelInfo>()); in Process_Son()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_grad_split.cc42 bn_update_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfUpdateGrad()
76 bn_reduce_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfReduceGrad()
Dbn_grad_split.cc43 bn_update_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfUpdateGrad()
75 bn_reduce_grad->set_kernel_info(std::make_shared<device::KernelInfo>()); in CreateOutputsOfReduceGrad()
Dbn_split.cc53 bn_training_reduce->set_kernel_info(kernel_info); in CreateOutputsOfBNTrainingReduce()
93 bn_training_update->set_kernel_info(kernel_info); in CreateOutputsOfBNTrainingUpdate()
/third_party/mindspore/mindspore/ccsrc/vm/
Dsegment_runner.cc87 eqv[node]->set_kernel_info(node->kernel_info_ptr()); in RefSubGraphNode()
139 eqv[n]->set_kernel_info(n->kernel_info_ptr()); in TransformSegmentToAnfGraph()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dtransdata_split_test.cc132 transpose->set_kernel_info(kernel_info); in TEST_F()
180 transpose->set_kernel_info(kernel_info); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/enhancer/
Dinsert_tensor_move_for_getnext_test.cc55 get_next->set_kernel_info(std::make_shared<device::KernelInfo>()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc92 kernelptr_first->set_kernel_info(kernel_info_first); in CreateKernelGraph()
122 kernelptr_floor->set_kernel_info(kernel_info_floor); in CreateKernelGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_decoder.cc257 cnode_->set_kernel_info(kernel_info); in CreateKernelInfo()
308 value_node->set_kernel_info(kernel_info); in DecodeValueNode()
348 new_parameter->set_kernel_info(kernel_info); in DecodeParameter()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_cascade.cc91 tensor_move->set_kernel_info(kernel_info); in InsertTensorMove()

123