Home
last modified time | relevance | path

Searched refs:Broaden (Results 1 – 25 of 32) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/grad/
Dbatch_norm_grad.cc61 auto dx = input_args[kInputIndex1]->Broaden(); in BatchNormGradInfer()
62 auto dscale = input_args[kInputIndex2]->Broaden(); in BatchNormGradInfer()
63 auto reserve_1 = input_args[kInputIndex3]->Broaden(); in BatchNormGradInfer()
64 auto reserve_2 = input_args[kInputIndex4]->Broaden(); in BatchNormGradInfer()
Dlayer_norm_grad.cc31 auto x_backprob = input_args[kInputIndex0]->Broaden(); in LayerNormGradInfer()
32 auto gamma_backprob = input_args[kInputIndex4]->Broaden(); in LayerNormGradInfer()
33 auto beta_backprob = input_args[kInputIndex4]->Broaden(); in LayerNormGradInfer()
Dflatten_grad.cc35 auto ret = input_x->Broaden(); in FlattenGradInfer()
/third_party/mindspore/mindspore/core/abstract/
Dabstract_value.h82 virtual AbstractBasePtr Broaden() const;
125 AbstractBasePtr Broaden() const override;
145 AbstractBasePtr Broaden() const override { return Clone(); } in Broaden() function
160 AbstractBasePtr Broaden() const override { return Clone(); } in Broaden() function
188 AbstractBasePtr Broaden() const override { return Clone(); } in Broaden() function
216 AbstractBasePtr Broaden() const override { in Broaden() function
245 AbstractBasePtr Broaden() const override;
335 AbstractBasePtr Broaden() const override;
406 …AbstractBasePtr Broaden() const override { return std::make_shared<AbstractTuple>(ElementsBroaden(… in Broaden() function
435 …AbstractBasePtr Broaden() const override { return std::make_shared<AbstractList>(ElementsBroaden()… in Broaden() function
[all …]
Dabstract_value.cc136 AbstractBasePtr AbstractBase::Broaden() const { in Broaden() function in mindspore::abstract::AbstractBase
158 AbstractBasePtr AbstractScalar::Broaden() const { in Broaden() function in mindspore::abstract::AbstractScalar
162 return AbstractBase::Broaden(); in Broaden()
166 return AbstractBase::Broaden(); in Broaden()
321 AbstractBasePtr broadend = ele->Broaden(); in ElementsBroaden()
473 AbstractBasePtr AbstractSlice::Broaden() const { in Broaden() function in mindspore::abstract::AbstractSlice
477 AbstractBasePtr start = start_->Broaden(); in Broaden()
478 AbstractBasePtr stop = stop_->Broaden(); in Broaden()
479 AbstractBasePtr step = step_->Broaden(); in Broaden()
632 AbstractBasePtr AbstractTensor::Broaden() const { in Broaden() function in mindspore::abstract::AbstractTensor
[all …]
Dprim_nn.cc102 AbstractBasePtr ret = input_tensor->Broaden(); in InferImplPooling()
119 AbstractBasePtr ret = d_out->Broaden(); in InferImplPoolingGrad()
195 auto dx = args_spec_list[dx_index]->Broaden(); in InferImplFusedSparseAdam()
196 auto dscale = args_spec_list[dscale_index]->Broaden(); in InferImplFusedSparseAdam()
197 auto dbias = args_spec_list[dbias_index]->Broaden(); in InferImplFusedSparseAdam()
394 AbstractBasePtr ret = args_spec_list[0]->Broaden(); in InferImplBiasAddGrad()
406 return args_spec_list[0]->Broaden(); in InferImplHSigmoid()
419 return args_spec_list[1]->Broaden(); in InferImplHSigmoidGrad()
428 args_list.push_back(args_spec_list[i]->Broaden()); in InferImplBpropCut()
455 elements.push_back(args_spec_list[i]->Clone()->Broaden()); in InferImplSparseApplyFtrl()
[all …]
Ddshape.h53 virtual void Broaden() {} in Broaden() function
80 void Broaden() override;
Dprim_debug.cc37 return std::make_shared<AbstractTuple>(AbstractBasePtrList({tensor_value->Broaden()})); in InferImplDebug()
Danalysis_context.cc168 auto ret_spec = scalar_spec->Broaden(); in SpecializeKey()
174 return arg->Broaden(); in SpecializeKey()
Dprim_maths.cc38 AbstractBasePtr dx = input_x->Broaden(); in InferImplMinOrMaxGrad()
39 AbstractBasePtr dy = input_y->Broaden(); in InferImplMinOrMaxGrad()
51 return inp->Clone()->Broaden(); in InferImplSqrt()
65 return out->Broaden(); in InferImplSqrtGrad()
Ddshape.cc104 void Shape::Broaden() { in Broaden() function in mindspore::abstract::Shape
Dprim_others.cc183 auto depends = args_spec_list[0]->Broaden(); // Avoid eliminating the dependent node. in InferImplDepend()
199 return args_spec_list[0]->Broaden(); in InferImplUpdateState()
579 return args_spec_list[0]->Broaden(); in InferImplLoad()
Dprim_arrays.cc124 AbstractTensorPtr ret = dyn_cast<AbstractTensor>(tensor_base->Broaden()); in InferImplStack()
648 return args_spec_list[1]->Broaden(); in InferImplDynamicAssign()
824 return args_spec_list[0]->Broaden(); in InferImplMapUniform()
863 auto output = input_x->Broaden(); in InferImplSplit()
997 AbstractTensorPtr ret = dyn_cast<AbstractTensor>(tensor_base->Broaden()); in InferImplConcat()
1014 return args_spec_list[0]->Broaden(); in InferImplRange()
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Ddata_test.cc170 AbstractBasePtr s2 = s1->Broaden(); in TEST_F()
181 AbstractBasePtr s2 = s1->Broaden(); in TEST_F()
188 AbstractBasePtr f2 = f1->Broaden(); in TEST_F()
192 AbstractBasePtr l2 = l1.Broaden(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkpynative.cc118 abstract::AbstractBasePtrList args_spec{node->abstract()->Broaden()}; in BuildZerosLikeNode()
127 abstract::AbstractBasePtrList args_spec{out->ToAbstract()->Broaden()}; in BuildZerosLikeValue()
161 abstract::AbstractBasePtrList args_spec{out->ToAbstract()->Broaden()}; in BuildOnesLikeValue()
219 auto arg = out_->ToAbstract()->Broaden(); in AccumulateDout()
698 din->set_abstract(input_adjoint_iter->second->out()->ToAbstract()->Broaden()); in BackPropagate()
764 v_node->set_abstract(adjoint->op_args()[i]->ToAbstract()->Broaden()); in BackPropagateOneCNodeWithBPropFuncGraph()
768 out_node->set_abstract(adjoint->out()->ToAbstract()->Broaden()); in BackPropagateOneCNodeWithBPropFuncGraph()
805 v_node->set_abstract(adjoint->op_args()[i]->ToAbstract()->Broaden()); in BackPropagateOneCNodeWithFPropFuncGraph()
982 sens_param->set_abstract(last_node_adjoint_iter->second->out()->ToAbstract()->Broaden()); in SetSensAndWeights()
1089 k_node->set_abstract(iter->second->out()->ToAbstract()->Broaden()); in BuildKNode()
/third_party/mindspore/mindspore/core/ops/
Dl2_normalize.cc55 return input_args[0]->Broaden(); in L2NormalizeInfer()
Dassign.cc43 return args_spec_list[1]->Broaden(); in InferImplAssign()
Ddepth_to_space.cc74 auto ret = input_x->Broaden(); in DepthToSpaceInfer()
Dbatch_to_space.cc80 auto ret = input_args[0]->Broaden(); in BatchToSpaceInfer()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpy_pass.cc107 para_node->set_abstract(default_value->ToAbstract()->Broaden()); in BuildNewParameter()
279 para_node->set_abstract(default_value->ToAbstract()->Broaden()); in Run()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprim.h151 … [](const AbstractBasePtr &arg) -> AbstractBasePtr { return arg->Broaden(); }); in NormalizeArgs()
Dstatic_analysis.cc251 out = out->Broaden(); in EvalValueNode()
719 return std::make_shared<EvalResult>(out_specs[0]->Broaden(), std::make_shared<AttrValueMap>()); in ProcessEvalResults()
788 auto broadAbstract = result->abstract()->Broaden(); in ExecEvaluator()
1017 a = a->Broaden(); in FromValueInside()
Devaluator.cc282 return arg->Broaden(); in BroadenArgs()
407 … abstract = std::make_shared<AbstractRef>(abs_ref->ref_key()->Broaden(), abs_ref); in Run()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc92 return abstract->Broaden(); in Build()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dmap.cc329 return arg->Broaden(); in NormalizeArgs()

12