Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc52 auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape_vector); in CreateOmParameter() local
331 auto abstract_tensor = lite::CreateTensorAbstract(graph_output_dims_[j], data_type); in SetMultiOutputs() local
352 auto abstract_tensor = lite::CreateTensorAbstract(graph_output_dims_[0], type); in SetCustomOutputs() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dunused_cast_node_remove_pass.cc52 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in Run() local
Dnode_infershape.cc401 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(elements[i]); in GetCNodeOutputTensors() local
417 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->abstract()); in GetCNodeOutputTensors() local
Dadd_tensor_array.cc150 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in Process() local
Dinfershape_pass.cc345 auto abstract_tensor = abstract_base->cast<abstract::AbstractTensorPtr>(); in SetSubGraphAbstract() local
Ddecrease_transpose_algo.cc593 auto abstract_tensor = abstract_base->cast<abstract::AbstractTensorPtr>(); in SetSubGraphAbstract() local
/third_party/mindspore/mindspore/lite/tools/converter/ops/
Dwhile.cc76 auto abstract_tensor = lite::CreateTensorAbstract(shape, build_type_ptr->type_id()); in WhileInfer() local
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Dfetch_content.cc101 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in GetDataTypeAndShape() local
119 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(valueAbstract); in FetchFromTensorValue() local
381 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in FetchDataFromCNode() local
Danf_exporter.cc837 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(elements[i]); in SetOpOutputNode() local
856 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->abstract()); in SetOpOutputNode() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc187 auto abstract_tensor = CreateTensorAbstract(shape_vector, data_type); in BuildParameterNode() local
213 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in BuildOpOutputs() local
224 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in BuildOpOutputs() local
306 auto abstract_tensor = CreateTensorAbstract(shape_vector, data_type); in ConvertGraphInputs() local
1205 auto abstract_tensor = tensor_info->ToAbstract(); in AddTensorArrayEdge() local
1283 auto abstract_tensor = CreateTensorAbstract({}, type); in BuildParameterNodeForQuantParam() local
Donnx_inputs_adjust.cc79 auto abstract_tensor = param_node->abstract()->cast<abstract::AbstractTensorPtr>(); in ReplaceInt64ParameterNode() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_inputs_adjust.cc99 auto abstract_tensor = param_node->abstract()->cast<abstract::AbstractTensorPtr>(); in ReplaceInt64ParameterNode() local
Dtflite_model_parser.cc445 auto abstract_tensor = CreateTensorAbstract(shape_vector, dtype); in ConvertGraphInputs() local
750 auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape_vector); in ConvertOutputTensor() local
765 auto abstract_tensor = CreateTensorAbstract(shape_vector, GetTfliteDataType(tensor->type)); in ConvertOutputTensor() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc111 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->abstract()); in GetTypeFromNode() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc130 auto abstract_tensor = CreateFakeAbstractTensor(); in CreateMultipleOutputsOfAnfNode() local
Dconv2d_info.cc114 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(input_node_abstract); in CheckIfSplit() local
/third_party/mindspore/mindspore/lite/tools/common/
Dtensor_util.cc186 auto abstract_tensor = tensor_info->ToAbstract(); in InitParameterFromTensorInfo() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dweight_quantizer.cc78 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in SetAbstract() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc461 auto abstract_tensor = CreateTensorAbstract(shape, type); in ConvertParameter() local
870 auto abstract_tensor = CreateTensorAbstract({}, type); in ConvertOutputTensor() local
880 auto abstract_tensor = CreateTensorAbstract({}, kNumberTypeFloat32); in ConvertOutputTensor() local
/third_party/mindspore/mindspore/core/ops/grad/
Dconv2d_backprop_filter.cc75 auto abstract_tensor = filter_size->cast<abstract::AbstractTensorPtr>(); in Conv2DBackpropFilterInferShape() local
Dconv2d_backprop_input.cc104 auto abstract_tensor = input_size->cast<abstract::AbstractTensorPtr>(); in Conv2DBackpropInputInferShape() local
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_adjust.cc190 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(value_node->abstract()); in ValueNodeInt64Convert() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconstant_folding_fusion.cc65 … auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->input(i)->abstract()); in GetCNodeInputTensors() local
Dtf_lstm_cell_fusion.cc387 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(hidden_param->abstract()); in CreateLSTMNode() local
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc648 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in GetTensorInfoFromAbstract() local
1149 auto abstract_tensor = abstract->cast<abstract::AbstractTensorPtr>(); in FetchShapeFromAbstract() local
1215 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in GetDataTypeFromAnfNode() local

12