Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/
Dnnapi_utils.cc36 void ConverTensorQuantSymmToASymm(MSTensor *ms_tensor) { in ConverTensorQuantSymmToASymm()
52 int AddNNAPIOperand(ANeuralNetworksModel *nnapi_model, MSTensor ms_tensor, int idx, int quant_chann… in AddNNAPIOperand()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dtypes.cc152 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CreateTensor() local
193 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CreateTensor() local
280 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CreateRefTensor() local
402 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CharStringsToTensor() local
467 auto ms_tensor = new (std::nothrow) MSTensor(impl); in Clone() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_utils.cc65 nvinfer1::Dims ConvertCudaDims(const TensorInfo &ms_tensor) { in ConvertCudaDims()
96 std::vector<int32_t> ConvertTensorAsIntVector(const TensorInfo &ms_tensor) { in ConvertTensorAsIntVector()
224 nvinfer1::ITensor *ConvertConstantTensor(TensorRTContext *ctx, const TensorInfo &ms_tensor, in ConvertConstantTensor()
283 …ensor *ConvertScalarToITensor(TensorRTContext *ctx, size_t shape_size, const TensorInfo &ms_tensor, in ConvertScalarToITensor()
369 nvinfer1::ITensor *ConvertTensorWithExpandDims(TensorRTContext *ctx, const TensorInfo &ms_tensor, in ConvertTensorWithExpandDims()
418 nvinfer1::ITensor *ConvertConstantTensorWithDims(TensorRTContext *ctx, const TensorInfo &ms_tensor, in ConvertConstantTensorWithDims()
447 nvinfer1::Weights TransposeWeight2D(const TensorInfo &ms_tensor, void **pack_weight) { in TransposeWeight2D()
502 nvinfer1::Weights ConvertWeight(const TensorInfo &ms_tensor) { in ConvertWeight()
792 int ParseData2Vector(const TensorInfo &ms_tensor, std::vector<float> *dst) { in ParseData2Vector()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/benchmark_train/
Dnet_train.cc518 auto ms_tensor = before_inputs.at(i); in InitDumpTensorDataCallbackParameter() local
541 auto ms_tensor = after_outputs.at(i); in InitDumpTensorDataCallbackParameter() local
603 auto ms_tensor = after_inputs.at(i); in InitTimeProfilingCallbackParameter() local
607 auto ms_tensor = after_outputs.at(i); in InitTimeProfilingCallbackParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/native/
Dmodel_parallel_runner.cpp215 auto &ms_tensor = *static_cast<mindspore::MSTensor *>(tensor_pointer); in Java_com_mindspore_ModelParallelRunner_predictZeroCopy() local
294 auto &ms_tensor = *static_cast<mindspore::MSTensor *>(tensor_pointer); in Java_com_mindspore_ModelParallelRunner_predictWithOutputZeroCopy() local
306 auto &ms_tensor = *static_cast<mindspore::MSTensor *>(tensor_pointer); in Java_com_mindspore_ModelParallelRunner_predictWithOutputZeroCopy() local
Dmodel.cpp454 auto &ms_tensor = *static_cast<mindspore::MSTensor *>(tensor_pointer); in Java_com_mindspore_Model_runStep() local
548 auto &ms_tensor = *static_cast<mindspore::MSTensor *>(tensor_pointer); in Java_com_mindspore_Model_resize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dtensor_utils.cc84 for (auto ms_tensor : ms_tensors) { in MSTensorToTensorPtr() local
114 for (auto ms_tensor : ms_tensors) { in MSTensorToTensor() local
Dfunc_graph_utils.cc325 auto ms_tensor = std::make_shared<tensor::Tensor>(static_cast<TypeId>(data_type), shape); in GetFuncGraphInputsInfo() local
348 auto ms_tensor = std::make_shared<tensor::Tensor>(static_cast<TypeId>(data_type), shape); in GetFuncGraphOutputsInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/
Dgraph.cc238 auto ms_tensor = train_coder->output_tensors().at(0); in CompileTrainOutputs() local
272 auto ms_tensor = in_coder->output_tensors().at(0); in CompileEvalOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/graph/
Dgraph_impl.h118 auto ms_tensor = std::make_shared<tensor::Tensor>(data_type, input_shape); in GetModelInputsInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/
Dutils.cc204 auto ms_tensor = after_outputs.at(i); in InnerPredict() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_session.cc503 for (auto &ms_tensor : ms_tensors.second) { in RunGraph() local
556 for (auto &ms_tensor : ms_tensors.second) { in Train() local
588 for (auto &ms_tensor : ms_tensors.second) { in Eval() local
665 auto *ms_tensor = kernel->out_tensors().at(0); in CompileTrainOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/kernel_executor/
Dkernel_executor_impl.cc384 bool KernelExecutorImpl::TensorIsValid(const MSTensor &ms_tensor, const lite::Tensor *lite_tensor) { in TensorIsValid()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/benchmark/
Dbenchmark_unified_api.cc1796 for (auto ms_tensor : after_inputs) { in InitPrintTensorDataCallbackParameter() local
1801 for (auto ms_tensor : after_outputs) { in InitPrintTensorDataCallbackParameter() local
1821 auto ms_tensor = before_inputs.at(i); in InitDumpTensorDataCallbackParameter() local
1844 auto ms_tensor = after_outputs.at(i); in InitDumpTensorDataCallbackParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_apply_centered_rms_prop.cc60 Tensor *ms_tensor = ctx.Input(2); in ValidParam() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/c_api/
Dtensor_c.cc113 auto ms_tensor = static_cast<mindspore::MSTensor *>(tensor); in OH_AI_TensorGetName() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Dexecute.cc394 auto ms_tensor = mindspore::MSTensor(std::make_shared<DETensor>(tensor)); in operator ()() local
422 auto ms_tensor = mindspore::MSTensor(std::make_shared<DETensor>(host_output)); in operator ()() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/llm_engine/
Dllm_engine_plugin.cc644 auto ms_tensor = ConvertGeTensorNoCopy(&ge_tensor); in Predict() local
697 auto ms_tensor = ConvertGeTensorNoCopy(&ge_tensor); in Predict() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Ddecoder_tensorrt.cc57 nvinfer1::ITensor *DecoderTensorRT::CastTensor(TensorRTContext *ctx, const TensorInfo &ms_tensor, in CastTensor()
Dencoder_tensorrt.cc85 nvinfer1::ITensor *EncoderTensorRT::CastTensor(TensorRTContext *ctx, const TensorInfo &ms_tensor, in CastTensor()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/
Danf_exporter.cc1093 auto ms_tensor = new (std::nothrow) schema::TensorT(); in SetOpOutputNode() local
1145 auto ms_tensor = new (std::nothrow) schema::TensorT(); in SetOpOutputNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_ge/
Dge_graph_executor.cc1730 tensor::Tensor ms_tensor(output_info.dtype, output_info.shape); in SyncDeviceOutputsToHost() local
1850 auto ms_tensor = ConvertGeTensorNoCopy(&ge_tensor, graph_id, i); in RunGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_convert.cc518 auto ms_tensor = inputs[ms_real_idx]; in ConvertToAclInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc1109 auto ms_tensor = std::make_shared<tensor::Tensor>(data_type, input_shape); in GetModelInputsInfo() local