Home
last modified time | relevance | path

Searched defs:tensor_ptr (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/lite/java/native/runtime/
Dms_tensor.cpp24 jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_getShape()
44 jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_getDataType()
55jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_getByteData()
79jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_getLongData()
109jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_getIntData()
139jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_getFloatData()
169jlong tensor_ptr, jbyteArray data, in Java_com_mindspore_lite_MSTensor_setData()
194jlong tensor_ptr, in Java_com_mindspore_lite_MSTensor_setByteBufferData()
223 … jlong JNICALL Java_com_mindspore_lite_MSTensor_size(JNIEnv *env, jobject thiz, jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_size()
234 jlong tensor_ptr) { in Java_com_mindspore_lite_MSTensor_elementsNum()
[all …]
/third_party/mindspore/mindspore/core/base/
Dbase_ref_utils.cc25 auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(ref_list[i]); in IterateFindTensor() local
43 auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(base_ref); in TransformVectorRefToMultiTensor() local
/third_party/mindspore/mindspore/lite/src/
Dtensorlist.cc105 auto tensor_ptr = new (std::nothrow) Tensor(dtype, tensor_shape[i]); in MallocTensorListData() local
129 auto tensor_ptr = this->tensors_[i]; in MallocData() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dcallbacks_ge.cc99 TensorPtr tensor_ptr = GetMeTensorTransformed(graph_id, name, ge_tensor_ptr); in CheckpointSaveCallback() local
168 TensorPtr tensor_ptr = GetMeTensorForSummary(tag_name, ge_tensor_ptr); in SummarySaveCallback() local
Dcallbacks.cc46 auto tensor_ptr = item.second; in SummarySaveCallback() local
Dconvert_utils_py.cc122 auto tensor_ptr = value->cast<tensor::TensorPtr>(); in __anone0198b6c0202() local
242 auto tensor_ptr = value.cast<tensor::TensorPtr>(); in AnyToPyData() local
285 auto tensor_ptr = utils::cast<tensor::TensorPtr>(value); in BaseRefToPyData() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_scalar_to_tensor.cc39 tensor::TensorPtr tensor_ptr = ScalarToTensor(value->cast<ScalarPtr>()); in CreateTensorInput() local
Dconvert_const_input_to_tensor_input.cc37 tensor::TensorPtr tensor_ptr = nullptr; in CreateTensorInput() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dvalue_based_eliminate.cc48 auto tensor_ptr = value->cast<tensor::TensorPtr>(); in IsNodeScalarTrueWith() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_v2_info.cc210 …mindspore::tensor::TensorPtr tensor_ptr = std::make_shared<mindspore::tensor::Tensor>(tensor_data,… in CreateSubOp() local
Dreduce_method_info.cc212 …mindspore::tensor::TensorPtr tensor_ptr = std::make_shared<mindspore::tensor::Tensor>(divisor, dty… in CreateReduceMeanForwardOp() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dfullconnection_int8_tests.cc47 auto tensor_ptr = reinterpret_cast<int *>(tensor->MutableData()); in MakeIntTensor() local
Dmatmul_int8_tests.cc53 auto tensor_ptr = reinterpret_cast<int8_t *>(tensor->MutableData()); in MakeQuantTensor() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dgenerate_graph.cc86 mindspore::tensor::TensorPtr tensor_ptr = std::make_shared<tensor::Tensor>(value, kInt32); in CreateInt32Tensor() local
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dutil.cc196 …GeTensorPtr tensor_ptr = make_shared<GeTensor>(*desc, static_cast<uint8_t *>(tensor->data_c()), da… in ConvertTensor() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc252 auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(base_ref[i]); in SetOutputFlags() local
Danf_runtime_algorithm.cc957 auto tensor_ptr = type_ptr->cast<TensorTypePtr>(); in GetOutputInferDataType() local
2237 auto tensor_ptr = iter_tensor->second; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dfused_batch_norm_fusion.cc71 auto tensor_ptr = value->cast<tensor::TensorPtr>(); in GetFactor() local
/third_party/mindspore/tests/ut/cpp/ir/
Dmeta_tensor_test.cc157 std::shared_ptr<Tensor> tensor_ptr = std::make_shared<Tensor>(tensor); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/debug/
Ddump_proto.cc218 tensor::TensorPtr tensor_ptr = dyn_cast<tensor::Tensor>(val); in SetValueToProto() local
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Dproto_exporter.cc154 tensor::TensorPtr tensor_ptr = dyn_cast<tensor::Tensor>(val); in SetValueToProto() local
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc150 auto tensor_ptr = py::cast<tensor::TensorPtr>(obj); in GetId() local
434 tensor::TensorPtr tensor_ptr = opt::CreateTupleTensor(value_tuple); in ConvertValueTupleToTensor() local
465 tensor::TensorPtr tensor_ptr = nullptr; in ConvertPyObjectToTensor() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc577 tensor::TensorPtr tensor_ptr = opt::CreateTupleTensor(value_tuple); in ConvertValueTupleToTensor() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc519 tensor::TensorPtr tensor_ptr = std::make_shared<tensor::Tensor>(kInt32->type_id(), shp); in CreateSwitchOpParameters() local