Home
last modified time | relevance | path

Searched defs:tensor_info (Results 1 – 25 of 51) sorted by relevance

123

/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dweight_quantizer.cc61 STATUS WeightQuantizer::SetAbstract(const tensor::TensorPtr &tensor_info, const ParameterPtr &param… in SetAbstract()
101 tensor::TensorPtr tensor_info; in DoConvQuantize() local
146 auto tensor_info = std::static_pointer_cast<tensor::Tensor>(param_node->default_param()); in DoMulQuantize() local
229 tensor::TensorPtr tensor_info; in DoGatherQuantize() local
287 tensor::TensorPtr tensor_info; in DoOptimizerQuantize() local
333 tensor::TensorPtr tensor_info; in DoMarkWeightQuantizeIfQuantized() local
353 tensor::TensorPtr tensor_info; in ProcessLstmWeightByIndex() local
/third_party/mindspore/mindspore/lite/tools/common/
Dtensor_util.cc54 tensor::TensorPtr tensor_info = nullptr; in CreateTensorInfo() local
86 auto tensor_info = CreateTensorInfo(nullptr, 0, shape, data_type); in CreateTensorAbstract() local
105 auto tensor_info = CreateTensorInfo(data, data_size, shape, data_type); in SetParameterAbstractAndParam() local
119 int SetTensorData(const tensor::TensorPtr &tensor_info, const void *data, size_t data_size) { in SetTensorData()
136 std::unique_ptr<schema::TensorT> CreateTensorTFromTensorInfo(const tensor::TensorPtr &tensor_info, in CreateTensorTFromTensorInfo()
181 …InitParameterFromTensorInfo(const ParameterPtr &param_node, const tensor::TensorPtr &tensor_info) { in InitParameterFromTensorInfo()
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/parser/
Dmodel_parser_test.cc72 auto tensor_info = std::make_shared<tensor::Tensor>(TypeId::kNumberTypeFloat32, shape); in BuildGraphInputs() local
120 auto tensor_info = std::make_shared<tensor::Tensor>(TypeId::kNumberTypeFloat32, shape); in BuildGraphNodes() local
132 auto tensor_info = std::make_shared<tensor::Tensor>(TypeId::kNumberTypeFloat32, ShapeVector{}); in BuildGraphNodes() local
/third_party/mindspore/tests/ut/data/dump/gpu_dumps/golden/
Dread_tensors_nonexist_node_expected.json4 "tensor_info": { object
22 "tensor_info": { object
40 "tensor_info": { object
Dread_tensors_base_stat_expected.json4 "tensor_info": { object
43 "tensor_info": { object
82 "tensor_info": { object
119 "tensor_info": { object
Dread_tensors_expected.json4 "tensor_info": { object
33 "tensor_info": { object
79 "tensor_info": { object
210 "tensor_info": { object
255 "tensor_info": { object
/third_party/mindspore/tests/ut/python/debugger/gpu_tests/
Dtest_sync_read_tensors_base_stat.py114 …def compare_expect_actual_result(self, tensor_info, tensor_base_data_list, tensor_stat_data_list, … argument
152 …def print_read_tensors(self, tensor_info, tensor_base_data_list, tensor_stat_data_list, test_index… argument
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_given_tensor_fill_parser.cc48 …auto tensor_info = CreateTensorInfo(iter->ints().data(), data_size, shape_vector, kNumberTypeInt64… in ParseInt8GivenIntTensorFill() local
71 auto tensor_info = CreateTensorInfo(iter->s().data(), data_count, shape_vector, kNumberTypeUInt8); in ParseInt8GivenTensorFill() local
Donnx_inputs_adjust.cc97 auto tensor_info = default_value->cast<tensor::TensorPtr>(); in ReplaceInt64ParameterNode() local
121 auto tensor_info = std::dynamic_pointer_cast<tensor::Tensor>(param_node->default_param()); in ValidParameterNode() local
148 auto tensor_info = primitive_c->GetAttr("const_data"); in ReplaceConstant() local
Donnx_constant_parser.cc37 auto tensor_info = std::make_shared<tensor::Tensor>(data_type, shape_vector); in AddDataInfoAttr() local
Donnx_model_parser.cc84 auto tensor_info = CreateTensorInfo(tensor_data, sizeof(int), {1}, kNumberTypeInt32); in CreateConstParamter() local
195 auto tensor_info = std::make_shared<tensor::Tensor>(data_type, shape_vector); in BuildParameterNode() local
878 auto tensor_info = parameter->default_param()->cast<tensor::TensorPtr>(); in BuildCNode() local
1076 auto tensor_info = quant_parameter_node->default_param()->cast<tensor::TensorPtr>(); in CopyTensorQuantParam() local
1199 …auto tensor_info = CreateTensorInfo(tensor_list_data.data(), tensor_list_data.size() * sizeof(int), in AddTensorArrayEdge() local
1296 auto tensor_info = CreateTensorInfo(data, data_size, {1}, type); in BuildParameterNodeForQuantParam() local
Donnx_pad_adjust.cc43 auto tensor_info = lite::CreateTensorInfo(data.data(), size, shape_vector, kNumberTypeInt32); in CreateNewParameter() local
Donnx_node_parser.cc56 const tensor::TensorPtr &tensor_info) { in CopyOnnxTensorData()
/third_party/mindspore/tests/st/debugger/
Dtest_sync_trans_true_read_tensor.py62 def print_read_tensors(tensor_info, tensor_data): argument
Dtest_sync_trans_false_read_tensors.py62 def print_read_tensors(tensor_info, tensor_data): argument
Dtest_async_sink_mode_true_read_tensors.py59 def print_read_tensors(tensor_info, tensor_data): argument
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc534 auto tensor_info = in AddNewBiasNode() local
569 auto tensor_info = param->default_param()->cast<tensor::TensorPtr>(); in GetTensorInfo() local
636 STATUS GetTensorInfoFromAbstract(tensor::TensorPtr *tensor_info, const CNodePtr &cnode, size_t inde… in GetTensorInfoFromAbstract()
892 const tensor::TensorPtr &tensor_info) { in BuildParameterNode()
962 auto tensor_info = lite::CreateTensorInfo(&data, sizeof(int32_t), {1}, kNumberTypeInt32); in BuildIntValueParameterNode() local
984 …auto tensor_info = lite::CreateTensorInfo(data.data(), data.size() * sizeof(int32_t), shape_vector… in BuildIntVecParameterNode() local
1016 auto tensor_info = lite::CreateTensorInfo(data_1d.data(), size, shape_vector, kNumberTypeInt32); in BuildIntVec2DParameterNode() local
1036 auto tensor_info = lite::CreateTensorInfo(&data, sizeof(float), {1}, kNumberTypeFloat32); in BuildFloatValueParameterNode() local
1057 …auto tensor_info = lite::CreateTensorInfo(data.data(), data.size() * sizeof(float), shape_vector, … in BuildFloatVecParameterNode() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc107 …S SetFloatTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_info) { in SetFloatTensorInfo()
142 …S SetInt32TensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_info) { in SetInt32TensorInfo()
177 …S SetInt64TensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_info) { in SetInt64TensorInfo()
220 …US SetBoolTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_info) { in SetBoolTensorInfo()
248 … SetStringTensorInfo(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_info) { in SetStringTensorInfo()
306 …:ConvertConstVariant(const tensorflow::TensorProto &tensor_proto, tensor::TensorPtr *tensor_info) { in ConvertConstVariant()
372 tensor::TensorPtr *tensor_info) { in SetTensorInfoFromType()
404 auto tensor_info = std::make_shared<tensor::Tensor>(type, *shape_vector); in ConvertConstTensor() local
/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc243 tensor::TensorPtr tensor_info = in BuildTensorInfoForFuncGraph() local
261 tensor::TensorPtr tensor_info = BuildTensorInfoForFuncGraph(parameter_proto); in BuildParameterForFuncGraph() local
315 tensor::TensorPtr tensor_info = BuildTensorInfoForFuncGraph(tensor_proto); in BuildInputForFuncGraph() local
494 …tensor::TensorPtr tensor_info = std::make_shared<tensor::Tensor>(kDefaultValueSwitchMap[attr_tenso… in ObtainCNodeAttrInTensorForm() local
588 …tensor::TensorPtr tensor_info = std::make_shared<tensor::Tensor>(kDefaultValueSwitchMap[attr_tenso… in ObtainValueNodeInTensorForm() local
618 …tensor::TensorPtr tensor_info = std::make_shared<tensor::Tensor>(kDefaultValueSwitchMap[attr_tenso… in ObtainValueNodeInTupleTensorForm() local
790 tensor::TensorPtr tensor_info = in GetAbstractForCNode() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dbatchmatmul_fusion.cc43 auto tensor_info = param_input->default_param()->cast<tensor::TensorPtr>(); in GetInputAddr() local
68 auto tensor_info = lite::CreateTensorInfo(nullptr, 0, shape_vector, fc_weight_param->data_type()); in GetRightMatmulInputParamter() local
Dconv_conv_fusion.cc88 …auto tensor_info = lite::CreateTensorInfo(nullptr, 0, {new_bias_size}, up_bias_param->data_type()); in GenNewConvBias() local
137 …auto tensor_info = lite::CreateTensorInfo(nullptr, 0, new_weight_shape, up_weight_param->data_type… in GenNewConvWeight() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dnode_infershape.cc62 auto tensor_info = std::make_shared<tensor::Tensor>(tensor->data_type(), shape_vector); in NewTensorInfo() local
533 auto tensor_info = NewTensorInfo(tensor); in ConvertLiteTensorToAbstract() local
572 …auto tensor_info = std::make_shared<tensor::Tensor>(kNumberTypeInt32, data_shape, data_info.data()… in ConvertTensorListToAbstract() local
Dinfershape_pass.cc286 … auto tensor_info = std::make_shared<tensor::Tensor>((TypeId)data_info.data_type_, shape_vec); in SetSubGraphInput() local
290 auto tensor_info = std::make_shared<tensor::Tensor>((TypeId)data_info.data_type_, shape_vec, in SetSubGraphInput() local
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Dfetch_content.cc36 STATUS GetShapeVectorFromStringTensor(const tensor::TensorPtr &tensor_info, ShapeVector *shape_vect… in GetShapeVectorFromStringTensor()
244 auto tensor_info = std::dynamic_pointer_cast<tensor::Tensor>(param_node->default_param()); in FetchFromDefaultParam() local
400 auto tensor_info = abstract_tensor->GetValueTrack(); in FetchDataFromCNode() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_inputs_adjust.cc120 auto tensor_info = default_value->cast<tensor::TensorPtr>(); in ReplaceInt64ParameterNode() local

123