Searched defs:ms_tensor (Results 1 – 10 of 10) sorted by relevance
34 … auto ms_tensor = mindspore::MSTensor(std::make_shared<DETensor>(de_tensor)); in __anon30679f650102() local47 auto ms_tensor = mindspore::MSTensor(std::make_shared<DETensor>(tensor)); in __anon30679f650102() local
112 …onvertConstantTensor(nvinfer1::INetworkDefinition *network, const mindspore::MSTensor &ms_tensor) { in ConvertConstantTensor()171 … const mindspore::MSTensor &ms_tensor, size_t expand_shape_size) { in ConvertTensorWithExpandDims()203 nvinfer1::Weights TransposeWeight(const mindspore::MSTensor &ms_tensor, void **pack_weight) { in TransposeWeight()226 nvinfer1::Weights TransposeWeightFP32(const mindspore::MSTensor &ms_tensor, void **pack_weight) { in TransposeWeightFP32()253 nvinfer1::Weights ConvertWeight(const mindspore::MSTensor &ms_tensor) { in ConvertWeight()
107 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CreateTensor() local131 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CreateRefTensor() local152 auto ms_tensor = new (std::nothrow) MSTensor(impl); in CharStringsToTensor() local196 auto ms_tensor = new (std::nothrow) MSTensor(impl); in Clone() local
731 for (auto ms_tensor : after_inputs) { in InitPrintTensorDataCallbackParameter() local736 for (const auto ms_tensor : after_outputs) { in InitPrintTensorDataCallbackParameter() local756 auto ms_tensor = before_inputs.at(i); in InitDumpTensorDataCallbackParameter() local779 auto ms_tensor = after_outputs.at(i); in InitDumpTensorDataCallbackParameter() local
733 for (auto ms_tensor : after_inputs) { in InitPrintTensorDataCallbackParameter() local738 for (auto ms_tensor : after_outputs) { in InitPrintTensorDataCallbackParameter() local758 auto ms_tensor = before_inputs.at(i); in InitDumpTensorDataCallbackParameter() local781 auto ms_tensor = after_outputs.at(i); in InitDumpTensorDataCallbackParameter() local
443 for (auto &ms_tensor : ms_tensors.second) { in RunGraph() local496 for (auto &ms_tensor : ms_tensors.second) { in Train() local528 for (auto &ms_tensor : ms_tensors.second) { in Eval() local551 auto *ms_tensor = in_kernel->out_tensors().at(0); in CompileEvalOutputs() local584 auto *ms_tensor = kernel->out_tensors().at(0); in CompileTrainOutputs() local
329 auto ms_tensor = mindspore::MSTensor(std::make_shared<DETensor>(tensor)); in operator ()() local358 auto ms_tensor = mindspore::MSTensor(std::make_shared<DETensor>(host_output)); in operator ()() local
798 auto ms_tensor = new (std::nothrow) schema::TensorT(); in SetOpOutputNode() local849 auto ms_tensor = new (std::nothrow) schema::TensorT(); in SetOpOutputNode() local
186 auto ms_tensor = output_iter.second; in Java_com_mindspore_lite_LiteSession_getOutputMapByTensor() local
1777 auto ms_tensor = std::make_shared<tensor::Tensor>(data_type, input_shape); in GetModelInputsInfo() local