| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/ |
| D | basic_tokenizer_op.cc | 148 std::shared_ptr<Tensor> cur_input; in Compute() local
158 cur_input = processed_tensor; in Compute()
160 RETURN_IF_NOT_OK(nfd_normalize_->Compute(cur_input, &processed_tensor)); in Compute()
161 cur_input = processed_tensor; in Compute()
162 RETURN_IF_NOT_OK(replace_accent_chars_->Compute(cur_input, &processed_tensor)); in Compute()
167 cur_input = processed_tensor; in Compute()
168 RETURN_IF_NOT_OK(replace_control_chars_->Compute(cur_input, &processed_tensor)); in Compute()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_cast_cpu.cc | 78 auto cur_input = AnfAlgo::GetInputNode(cnode, input_index); in InsertCast() local
79 MS_EXCEPTION_IF_NULL(cur_input); in InsertCast()
80 … if (cur_input->isa<Parameter>() && AnfAlgo::IsParameterWeight(cur_input->cast<ParameterPtr>())) { in InsertCast()
88 …AddCastOpNodeToGraph(func_graph, cur_input, dev_fmt, origin_type, device_type, origin_shape, devic… in InsertCast()
111 auto cur_input = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(used_node), used_node_index); in InsertCastForGraphOutput() local
115 …AddCastOpNodeToGraph(func_graph, cur_input, dev_fmt, device_type, infer_type, origin_shape, infer_… in InsertCastForGraphOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | decrease_compute_precision.cc | 85 auto cur_input = AnfAlgo::GetInputNode(cnode, input_index); in InsertCastForGraphKernel() local
86 if (HasAbstractMonad(cur_input)) { in InsertCastForGraphKernel()
95 if (cur_input->isa<ValueNode>()) { in InsertCastForGraphKernel()
96 ori_dtype = cur_input->cast<ValueNodePtr>()->value()->cast<tensor::TensorPtr>()->data_type(); in InsertCastForGraphKernel()
100 if (cur_input->isa<ValueNode>()) { in InsertCastForGraphKernel()
101 auto valuePtr = cur_input->cast<ValueNodePtr>(); in InsertCastForGraphKernel()
107 (void)mng->Replace(cur_input, value_node); in InsertCastForGraphKernel()
109 …auto cast = AddCastCNode(func_graph, cur_input, dev_fmt, ori_dtype, new_dtype, ori_shape, new_dtyp… in InsertCastForGraphKernel()
112 ShapeVector out_shape = GetShape(cur_input); in InsertCastForGraphKernel()
118 (void)mng->Replace(cur_input, cast); in InsertCastForGraphKernel()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | tanh_int8.cc | 52 int8_t *cur_input = in_ptr_ + task_id * thread_stride_; in DoActivation() local
55 TanhInt8(cur_input, cur_output, current_size, &tanh_quant_); in DoActivation()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | pad_fp32.cc | 87 int cur_input = 1; in InitMirrorPadBlock() local
90 if (cur_input > 1) { in InitMirrorPadBlock()
91 input_separate_dims.emplace_back(cur_input); in InitMirrorPadBlock()
98 cur_input = 1; in InitMirrorPadBlock()
101 if (cur_input != 1 || cur_output != 1) { in InitMirrorPadBlock()
102 input_separate_dims.emplace_back(cur_input); in InitMirrorPadBlock()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_pad_for_nms_with_mask.cc | 66 auto cur_input = AnfAlgo::GetInputNode(cnode, input_idx); in Process() local
73 auto pad = InsertPadToGraph(func_graph, cur_input, origin_type, origin_shape); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 419 auto cur_input = AnfAlgo::GetInputNode(cnode, input_index); in InsertCastForInput() local
420 MS_EXCEPTION_IF_NULL(cur_input); in InsertCastForInput()
421 if (HasAbstractMonad(cur_input)) { in InsertCastForInput()
423 new_inputs.push_back(cur_input); in InsertCastForInput()
430 auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(cur_input, 0); in InsertCastForInput()
449 …AddCastOpNodeToGraph(func_graph, cur_input, dev_fmt, origin_type, device_type, origin_shape, infer… in InsertCastForInput()
455 new_inputs.push_back(cur_input); in InsertCastForInput()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 1302 auto &cur_input = inputs_cnode.at(j); in InsertEventHcomDependCommonBak() local
1303 MS_LOG(INFO) << "The index:" << j << " input, name:" << AnfAlgo::GetCNodeName(cur_input); in InsertEventHcomDependCommonBak()
1305 auto pre_stream_id = AnfAlgo::GetStreamId(cur_input); in InsertEventHcomDependCommonBak()
1307 auto it = std::find(cnodes.begin(), cnodes.end(), cur_input); in InsertEventHcomDependCommonBak()
1310 << " can't find input node:" << AnfAlgo::GetCNodeName(cur_input); in InsertEventHcomDependCommonBak()
1334 auto &cur_input = input_cnodes.at(i); in GetLastInputCnode() local
1335 auto stream_id = AnfAlgo::GetStreamId(cur_input); in GetLastInputCnode()
1336 auto cur_index = GetIndexByKey(graph_ptr, cur_input.get()); in GetLastInputCnode()
1338 …MS_LOG_EXCEPTION << "The input node:" << AnfAlgo::GetCNodeName(cur_input) << " is not found in gra… in GetLastInputCnode()
1342 result[stream_id] = std::make_pair(cur_input, cur_index); in GetLastInputCnode()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | transpose_strategy.cc | 104 std::vector<int> cur_input; in TransformAttrByAxes() local
117 cur_input.push_back(origin_input[index]); in TransformAttrByAxes()
121 …auto param_node = BuildIntVecParameterNode(func_graph, cur_input, cnode->input(input_index)->fulln… in TransformAttrByAxes()
|