Searched refs:AbstractSparseTensor (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | validator.cc | 39 using mindspore::abstract::AbstractSparseTensor;
120 …ptrBase->isa<AbstractSparseTensor>() || ptrBase->isa<abstract::AbstractRefKey>() || ptrBase->isa<A… in ValidateAbstract()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | abstract_value.cc | 1342 TypePtr AbstractSparseTensor::BuildType() const { in BuildType()
1348 AbstractBasePtr AbstractSparseTensor::Clone() const { in Clone()
1350 auto clone = std::make_shared<AbstractSparseTensor>(element()->Clone()); in Clone()
1370 AbstractBasePtr AbstractSparseTensor::Broaden() const { in Broaden()
1372 auto broaden = std::make_shared<AbstractSparseTensor>(element()->Broaden()); in Broaden()
1392 AbstractBasePtr AbstractSparseTensor::BroadenWithShape() const { in BroadenWithShape()
1394 auto broaden = std::make_shared<AbstractSparseTensor>(element()->Broaden()); in BroadenWithShape()
1417 std::string AbstractSparseTensor::ToString() const { in ToString()
|
| D | prim_others.cc | 361 …auto ret = std::make_shared<AbstractSparseTensor>(values->element()->BuildType(), dense_shape_vec); in InferImplMakeSparseTensor()
373 auto sparse_tensor = CheckArg<AbstractSparseTensor>(op_name, args_spec_list, 0); in InferImplSparseTensorGetValues()
383 auto sparse_tensor = CheckArg<AbstractSparseTensor>(op_name, args_spec_list, 0); in InferImplSparseTensorGetIndices()
393 auto sparse_tensor = CheckArg<AbstractSparseTensor>(op_name, args_spec_list, 0); in InferImplSparseTensorGetDenseShape()
|
| D | abstract_value.h | 732 class MS_CORE_API AbstractSparseTensor : public AbstractUndetermined {
734 …explicit AbstractSparseTensor(const AbstractBasePtr &element, const BaseShapePtr &shape = std::mak…
736 AbstractSparseTensor(const TypePtr &element_type, const ShapeVector &shape) in AbstractSparseTensor() function
738 ~AbstractSparseTensor() override = default;
739 MS_DECLARE_PARENT(AbstractSparseTensor, AbstractUndetermined) in MS_DECLARE_PARENT() argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 37 using mindspore::abstract::AbstractSparseTensor;
82 if (t->isa<AbstractSparseTensor>()) { in AdaptAbs()
83 auto abs_sparse = dyn_cast<AbstractSparseTensor>(t); in AdaptAbs()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 460 } else if (abs_base->isa<AbstractSparseTensor>()) { in ConvertAbstractToPython()
461 auto arg = dyn_cast<AbstractSparseTensor>(abs_base); in ConvertAbstractToPython()
|