Home
last modified time | relevance | path

Searched defs:abstract_tensor (Results 1 – 25 of 40) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/om/
Dom_model_parser.cc86 auto abstract_tensor = CreateTensorAbstract(shape_vector, data_type); in CreateGraphInputs() local
159 auto abstract_tensor = CreateTensorAbstract(dims, type); in SetCustomOutputs() local
219 …auto abstract_tensor = CreateTensorAbstract(custom_output_info.shape, custom_output_info.data_type… in SetMultiOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/
Danf_util.cc127 auto abstract_tensor = api::utils::cast<api::AbstractTensorPtr>(abstract_base); in GetShapeVectorFromParameter() local
261 auto abstract_tensor = abstract->cast<api::AbstractTensorPtr>(); in FetchShapeFromAbstract() local
282 auto abstract_tensor = abstract->cast<api::AbstractTensorPtr>(); in FetchTypeIdFromAbstract() local
429 auto abstract_tensor = tensor_info->ToAbstract(); in InitParameterFromTensorInfo() local
724 auto abstract_tensor = api::utils::cast<api::AbstractTensorPtr>(abstract_base); in GetDataTypeAndShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/
Dfetch_content.cc115 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in GetDataTypeAndShape() local
133 auto abstract_tensor = valueAbstract->cast<abstract::AbstractTensorPtr>(); in FetchFromTensorValue() local
413 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in FetchDataFromCNode() local
483 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in FetchDataFromAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/common/
Dutils.cc120 auto abstract_tensor = cnode->abstract()->cast<abstract::AbstractTensorPtr>(); in GetTypeFromNode() local
140 auto abstract_tensor = abstract_base->cast<abstract::AbstractTensorPtr>(); in GetTypeFromNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/third_party/
Dthird_party_model_parser.cc128 auto abstract_tensor = CreateTensorAbstract(shapes[i], dtypes[i]); in BuildGraphInputs() local
207 auto abstract_tensor = CreateTensorAbstract(shapes[i], dtypes[i]); in BuildGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dcustom_creator.cc156 auto abstract_tensor = api::MakeShared<api::AbstractTensor>(kNumberTypeUInt8, shape_vector); in CreateOmParameter() local
341 auto abstract_tensor = CreateTensorAbstract(shape_vector, kDataTypeMap.at(output_info.type)); in SetCustomSingleOutput() local
385 auto abstract_tensor = CreateTensorAbstract(shape_vector, kDataTypeMap.at(output_info.type)); in SetCustomMultiOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/
Dpytorch_model_parser.cc177 auto abstract_tensor = CreateTensorAbstract(input_shape, data_type); in ConvertGraphInputs() local
413 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in BuildOpOutputs() local
424 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in BuildOpOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/ops/
Dwhile.cc79 auto abstract_tensor = lite::CreateTensorAbstract(shape, build_type_ptr->type_id()); in WhileInfer() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dmindir_adjust.cc166 auto abstract_tensor = abstract->cast<abstract::AbstractTensorPtr>(); in AdjustInputDataType() local
190 auto abstract_tensor = value_node->abstract()->cast<abstract::AbstractTensorPtr>(); in ValueNodeInt64Convert() local
Dcast_op_adjust.cc35 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in GetInOutDataTypeValue() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dint64_cast_int32_pass.cc108 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in Run() local
Dinput_data_type_trans_pass.cc54 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in CastDataType() local
Doutput_variable_pass.cc72 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(make_tuple_input->abstract()); in Run() local
Dlite_tensor_extractor.cc474 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(elements[i]); in GetCNodeOutputTensors() local
492 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->abstract()); in GetCNodeOutputTensors() local
Dinput_and_output_variable_pass.cc149 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in CreateAssign() local
Dadd_tensor_array.cc158 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dcustom_ascend_utils.cc86 auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape_vector); in CreateOmParameter() local
111 auto abstract_tensor = FuncGraphUtils::GetAbstractFromNode(outputs_[0]); in SetCustomOutputs() local
123 auto abstract_tensor = FuncGraphUtils::GetAbstractFromNode(outputs_[j]); in SetCustomOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/fusion_inout_test/
Dfusion_inout_test.cc80 auto abstract_tensor = tensor_info->ToAbstract(); in AddParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc258 auto abstract_tensor = CreateTensorAbstract(shape_vector, data_type); in BuildParameterNode() local
294 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in BuildOpOutputs() local
305 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in BuildOpOutputs() local
405 auto abstract_tensor = CreateTensorAbstract(shape_vector, data_type); in ConvertGraphInputs() local
1435 auto abstract_tensor = tensor_info->ToAbstract(); in AddTensorArrayEdge() local
1514 auto abstract_tensor = CreateTensorAbstract({}, type); in BuildParameterNodeForQuantParam() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/
Dtflite_inputs_adjust.cc101 auto abstract_tensor = param_node->abstract()->cast<abstract::AbstractTensorPtr>(); in ReplaceInt64ParameterNode() local
Dtflite_model_parser.cc525 auto abstract_tensor = CreateTensorAbstract(shape_vector, dtype); in ConvertGraphInputs() local
843 auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape_vector); in ConvertOutputTensor() local
859 auto abstract_tensor = CreateTensorAbstract(shape_vector, GetTfliteDataType(tensor->type)); in ConvertOutputTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc140 auto abstract_tensor = CreateFakeAbstractTensor(); in CreateMultipleOutputsOfAnfNode() local
Dconv2d_info.cc117 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(input_node_abstract); in CheckIfSplit() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dconst_output_eliminate.h74 auto abstract_tensor = const_data->ToAbstract(); in operator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dshape_unify_mindir.cc108 auto abstract_tensor = anf_node->abstract()->cast<abstract::AbstractTensorPtr>(); in CreateTensorToScalar() local

12