| /third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/native/ |
| D | ms_tensor.cpp | 24 jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getShape()
47 …jint JNICALL Java_com_mindspore_MSTensor_getDataType(JNIEnv *env, jobject thiz, jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getDataType()
58 jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getByteData()
89 jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getLongData()
125 jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getIntData()
161 … jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getFloatData()
197 … jlong tensor_ptr) { in Java_com_mindspore_MSTensor_getFloat16Data()
244 … jlong tensor_ptr, jbyteArray data, in Java_com_mindspore_MSTensor_setByteData()
275 … jlong tensor_ptr, jfloatArray data, in Java_com_mindspore_MSTensor_setFloatData()
307 … jlong tensor_ptr, jintArray data, in Java_com_mindspore_MSTensor_setIntData()
[all …]
|
| D | model_parallel_runner.cpp | 143 auto tensor_ptr = std::make_unique<mindspore::MSTensor>(tensor); in GetParallelInOrOutTensors() local
238 auto tensor_ptr = std::make_unique<mindspore::MSTensor>(tensor); in Java_com_mindspore_ModelParallelRunner_predictZeroCopy() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/base/ |
| D | base_ref_utils.cc | 25 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-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | mbuf_receive_manager.cc | 39 bool CopyDataToTensor(const uint8_t *src_addr, mindspore::tensor::TensorPtr tensor_ptr, const size_… in CopyDataToTensor()
93 auto tensor_ptr = std::make_shared<mindspore::tensor::Tensor>(type_id, tensor_shape); in ConvertDataItemToTensorPtr() local
113 auto tensor_ptr = ConvertDataItemToTensorPtr(item); in ProcessFullTensor() local
183 …auto tensor_ptr = std::make_shared<mindspore::tensor::Tensor>(type_id, sliced_tensor_->tensor_shap… in FinishSliceTensor() local
|
| D | tensorprint_utils.cc | 40 auto tensor_ptr = std::get<mindspore::tensor::TensorPtr>(data_elem); in OutputReceiveData2StdOut() local
98 auto tensor_ptr = std::get<mindspore::tensor::TensorPtr>(data_elem); in OutputReceiveData2PbFile() local
|
| D | tensordump_utils.cc | 28 void SaveTensor2NPY(std::string file_name, mindspore::tensor::TensorPtr tensor_ptr) { in SaveTensor2NPY()
143 auto tensor_ptr = std::get<mindspore::tensor::TensorPtr>(data_elem); in AsyncSaveDatasetToNpyFile() local
|
| D | tensorsummary_utils.cc | 52 auto tensor_ptr = std::get<mindspore::tensor::TensorPtr>(data_elem); in SummaryReceiveData() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ps_roi_pooling_grad_v2_gpu_kernel.cc | 158 for (auto tensor_ptr : inputs) { in Resize() local
164 for (auto tensor_ptr : outputs) { in Resize() local
201 for (auto tensor_ptr : outputs) { in Resize() local
|
| D | ps_roi_pooling_v2_gpu_kernel.cc | 107 for (auto tensor_ptr : inputs) { in Resize() local
111 for (auto tensor_ptr : outputs) { in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | ones_like.cc | 35 auto tensor_ptr = TensorConstructUtils::CreateOnesTensor(type, shape_vec, true); in InferValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | callbacks_ge.cc | 95 TensorPtr tensor_ptr = GetMeTensorTransformed(graph_id, name, ge_tensor_ptr); in CheckpointSaveCallback() local
163 TensorPtr tensor_ptr = GetMeTensorForSummary(tag_name, ge_tensor_ptr); in SummarySaveCallback() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/ |
| D | tensorlist.cc | 107 auto tensor_ptr = new (std::nothrow) Tensor(dtype, tensor_shape[i]); in MallocTensorListData() local
131 auto tensor_ptr = this->tensors_[i]; in MallocData() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | convert_const_scalar_to_tensor.cc | 42 tensor::TensorPtr tensor_ptr = ScalarToTensor(value->cast<ScalarPtr>()); in ConvertTensorInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | histogram_fixed_width_fusion.cc | 61 auto tensor_ptr = ScalarToTensor(attr_value->cast<ScalarPtr>()); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/ |
| D | tensor_utils.cc | 91 auto tensor_ptr = std::make_shared<mindspore::tensor::Tensor>(type_id, shape, ref_tensor_data); in MSTensorToTensorPtr() local
156 auto tensor_ptr = std::make_shared<mindspore::tensor::Tensor>(type_id, shape, data, data_size); in TensorToTensorPtr() local
166 … [](mindspore::tensor::TensorPtr tensor_ptr) { return mindspore::tensor::Tensor(*tensor_ptr); }); in TensorPtrToTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/ |
| D | callbacks.cc | 45 auto tensor_ptr = item.second; in SummarySaveCallback() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ps_roi_pooling_grad_cpu_kernel.cc | 245 for (auto tensor_ptr : inputs) { in Resize() local
251 for (auto tensor_ptr : outputs) { in Resize() local
288 for (auto tensor_ptr : outputs) { in Resize() local
|
| D | ps_roi_pooling_v2_cpu_kernel.cc | 186 for (auto tensor_ptr : inputs) { in Resize() local
192 for (auto tensor_ptr : outputs) { in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/ |
| D | infer.cc | 610 auto tensor_ptr = py::cast<mindspore::tensor::TensorPtr>(ptr); in GetShapeForStubTensor() local
632 auto tensor_ptr = py::cast<mindspore::tensor::TensorPtr>(ptr); in GetDTypeForStubTensor() local
646 auto tensor_ptr = pyObj.cast<mindspore::tensor::MetaTensorPtr>(); in InferShape() local
665 auto tensor_ptr = pyObj.cast<mindspore::tensor::MetaTensorPtr>(); in InferDType() local
685 auto tensor_ptr = pyObj.cast<mindspore::tensor::MetaTensorPtr>(); in InferRank() local
698 auto tensor_ptr = pyObj.cast<mindspore::tensor::MetaTensorPtr>(); in InferSize() local
|
| D | strategy.cc | 151 auto tensor_ptr = py::cast<mindspore::tensor::MetaTensorPtr>(obj); in TensorComputable() local
180 auto tensor_ptr = pyObj.cast<mindspore::tensor::TensorPtr>(); in StubTensorComputable() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | value_based_eliminate.cc | 52 auto tensor_ptr = value->cast<tensor::TensorPtr>(); in IsNodeScalarTrueWith() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | slice_grad_unify_mindir.cc | 57 auto tensor_ptr = ori_value->cast<tensor::TensorPtr>(); in GetTensorValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hcom_util.cc | 50 [](KernelTensor *tensor_ptr) { return ConvertHcclType(tensor_ptr->dtype_id()); }); in GetHcomDataType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_attr_and_input_convert_regist.cc | 101 tensor::TensorPtr tensor_ptr = nullptr; in ConvertAttrToInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | pattern_matcher.h | 511 auto tensor_ptr = dyn_cast<tensor::Tensor>(value); in IsTensorConstant() local
548 auto tensor_ptr = dyn_cast<tensor::Tensor>(value); in IsTensorScalarConstant() local
564 tensor::TensorPtr tensor_ptr = dyn_cast<tensor::Tensor>(value); in GetPointerToTensorData() local
|