Home
last modified time | relevance | path

Searched refs:abs_tensor (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/core/ir/
Dmeta_tensor_extends.cc36 auto abs_tensor = std::make_shared<abstract::AbstractTensor>(dtype, tensor_shape); in ToAbstract() local
43 abs_tensor = std::make_shared<abstract::AbstractRef>(abs_ref_key, abs_tensor); in ToAbstract()
45 abs_tensor->set_value(shared_from_base<MetaTensor>()); in ToAbstract()
47 return abs_tensor; in ToAbstract()
Dtensor.cc593 auto abs_tensor = std::make_shared<abstract::AbstractTensor>(dtype, tensor_shape); in ToAbstract() local
599 abs_tensor = std::make_shared<abstract::AbstractRef>(abs_ref_key, abs_tensor); in ToAbstract()
601 abs_tensor->set_value(shared_from_base<Tensor>()); in ToAbstract()
603 return abs_tensor; in ToAbstract()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc195 auto abs_tensor = abs->cast<abstract::AbstractTensorPtr>(); in SetOutput() local
196 auto ele = abs_tensor->element(); in SetOutput()
/third_party/mindspore/mindspore/core/abstract/
Dabstract_value.h659 auto abs_tensor = AbstractTensor::Clone()->cast<AbstractTensorPtr>(); in Clone() local
660 if (abs_tensor == nullptr) { in Clone()
663 return std::make_shared<AbstractRef>(ref_key_->Clone(), abs_tensor); in Clone()
672 auto abs_tensor = AbstractTensor::Broaden()->cast<AbstractTensorPtr>(); in Broaden() local
673 if (abs_tensor == nullptr) { in Broaden()
676 return std::make_shared<AbstractRef>(ref_key_->Broaden(), abs_tensor); in Broaden()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.cc695 auto abs_tensor = std::make_shared<abstract::AbstractTensor>(out_type, abs_shape); in CreateCNode() local
696 cnode->set_abstract(abs_tensor); in CreateCNode()
698 auto abs_tensor = std::make_shared<abstract::AbstractTensor>(out_type, out_info.shape); in CreateCNode() local
699 cnode->set_abstract(abs_tensor); in CreateCNode()
/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils_py.cc402 auto abs_tensor = dyn_cast<abstract::AbstractTensor>(tensor); in SetValueRange() local
403 abs_tensor->set_value_range(min_value, max_value); in SetValueRange()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc541 auto abs_tensor = new_tensor->ToAbstract(); in CreateOutputNodeParam() local
542 new_param->set_abstract(abs_tensor); in CreateOutputNodeParam()