Home
last modified time | relevance | path

Searched defs:type_ptr (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_py.cc307 TensorPtr TensorPy::MakeTensor(const py::array &input, const TypePtr &type_ptr) { in MakeTensor()
538 …:ConvertBytesToTensor(const py::bytes &bytes_obj, const py::tuple &dims, const TypePtr &type_ptr) { in ConvertBytesToTensor()
610 const TypePtr type_ptr) { in SetDeviceAddress()
712 .def(py::init([](const Tensor &tensor, const TypePtr &type_ptr) { in RegMetaTensor()
721 .def(py::init([](const BaseTensor &tensor, const TypePtr &type_ptr) { in RegMetaTensor()
729 .def(py::init([](const TypePtr &type_ptr, const py::tuple &shape) { in RegMetaTensor()
734 .def(py::init([](const TypePtr &type_ptr, const py::list &shape) { in RegMetaTensor()
740 …py::init([](const py::array &input, const TypePtr &type_ptr) { return TensorPy::MakeTensor(input, … in RegMetaTensor()
742 .def(py::init([](const py::float_ input, const TypePtr &type_ptr) { in RegMetaTensor()
746 .def(py::init([](const py::int_ input, const TypePtr &type_ptr) { in RegMetaTensor()
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dabstract.cc53 auto type_ptr = mindspore::TypeIdToType(mindspore::TypeId(type)); in MSSetAbstract() local
72 auto type_ptr = mindspore::TypeIdToType(mindspore::TypeId(type)); in MSSetMultiAbstract() local
Dutils.h251 auto type_ptr = mindspore::TypeIdToType(type); in GetScalarParam() local
Dtensor.cc121 auto type_ptr = mindspore::TypeIdToType(mindspore::kNumberTypeFloat32); in MSNewTensorScalarFloat32() local
138 auto type_ptr = mindspore::TypeIdToType(mindspore::kNumberTypeInt32); in MSNewTensorScalarInt32() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dmeta_tensor.cc28 MetaTensor::MetaTensor(const TypePtr &type_ptr, const ShapeVector &shape) in MetaTensor()
67 TypePtr MetaTensor::SetDtype(const TypePtr type_ptr) { in SetDtype()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dnode.cc24 auto type_ptr = data->ToAbstract()->BuildType(); local
31 auto type_ptr = data->ToAbstract()->BuildType(); local
/third_party/mindspore/mindspore-src/source/mindspore/core/symbolic_shape/
Dutils.cc28 SymbolPtr GenValueByTensorShape(const ShapeVector &shape, const TypePtr &type_ptr) { in GenValueByTensorShape()
56 SymbolPtr GenValueByShape(const BaseShapePtr &baseshape, const TypePtr &type_ptr) { in GenValueByShape()
85 auto type_ptr = v->type(); in KernelTensorValueToSymbol() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/common/
Dimport_from_meta_graphT.cc52 auto type_ptr = TypeIdToType(type_id); in ConverterConstTensor() local
97 auto type_ptr = TypeIdToType(type_id); in ConvertTensorToAbstractTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dtuple_to_tensor.cc59 tensor::TensorPtr CreateTensorByTupleCast(const std::vector<T> &values, const TypePtr &type_ptr, in CreateTensorByTupleCast()
77 …:TensorPtr CreateTensorWithValueTuple(const ValueSequencePtr &value_tuple, const TypePtr &type_ptr, in CreateTensorWithValueTuple()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dcast_mapper.cc50 TypePtr type_ptr = TypeIdToType(TypeId(dst_type)); in Mapper() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dkvcache_quant_pass.cc65 TypePtr type_ptr = TypeIdToType(dst_type); in NewQuantNode() local
91 TypePtr type_ptr = TypeIdToType(dst_type); in NewAscendAntiQuantNode() local
Dint64_cast_int32_pass.cc110 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in Run() local
Doutput_variable_pass.cc73 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in Run() local
Dinput_and_output_variable_pass.cc150 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in CreateAssign() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/common/
Dutils.cc125 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in GetTypeFromNode() local
145 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in GetTypeFromNode() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/
Darg_min_fusion.cc69 auto type_ptr = GetAttr(kOutputType)->cast<api::TensorTypePtr>()->element(); in get_output_type() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dcustom_mapper.cc40 auto type_ptr = custom_prim->GetAttr("type" + std::to_string(index)); in GetParamFromAttrs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dapply_momentum_scale_fusion.cc44 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
Dapply_momentum_weight_scale_fusion.cc53 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dmindir_adjust.cc170 auto type_ptr = tensor_element->GetTypeTrack(); in AdjustInputDataType() local
200 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in ValueNodeInt64Convert() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dquant_dtype_cast.cc83 …auto type_ptr = mindspore::TypeIdToType(static_cast<TypeId>(GetValue<int64_t>(primitive->GetAttr(k… in InferType() local
Dbatch_isend_irecv.cc122 TypePtr type_ptr = nullptr; in InferType() local
/third_party/mesa3d/src/panfrost/lib/
Dpan_indirect_dispatch.c61 nir_def *type_ptr = nir_iadd(&b, job_hdr_ptr, nir_imm_int64(&b, 4 * 4)); in pan_indirect_dispatch_init() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expander/mindir_adapter/
Dinfer.cc62 auto type_ptr = v->type(); in SetValue() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/dtype/
Dtype.cc207 size_t GetTypeByte(const TypePtr &type_ptr) { in GetTypeByte()

123