Home
last modified time | relevance | path

Searched defs:shape_vector (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_given_tensor_fill_parser.cc42 ShapeVector shape_vector(shape.begin(), shape.end()); in ParseInt8GivenIntTensorFill() local
70 ShapeVector shape_vector(shape.begin(), shape.end()); in ParseInt8GivenTensorFill() local
85 std::vector<int64_t> shape_vector; in Parse() local
Donnx_constant_parser.cc36 …std::vector<int64_t> shape_vector(onnx_const_tensor.dims().begin(), onnx_const_tensor.dims().end()… in AddDataInfoAttr() local
Donnx_pad_adjust.cc36 ShapeVector shape_vector; in CreateNewParameter() local
Donnx_inputs_adjust.cc193 auto shape_vector = shape_ptr->shape(); in ReplaceTransposeWithGraphInput() local
Donnx_model_parser.cc186 std::vector<int64_t> shape_vector(tensor.dims().begin(), tensor.dims().end()); in BuildParameterNode() local
296 …std::vector<int64_t> shape_vector = ConverterContext::GetInstance()->GetGraphInputTensorShape(inpu… in ConvertGraphInputs() local
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Dfetch_content.cc36 …apeVectorFromStringTensor(const tensor::TensorPtr &tensor_info, ShapeVector *shape_vector, size_t … in GetShapeVectorFromStringTensor()
90 …GetDataTypeAndShape(const ParameterPtr &param_node, TypeId *data_type, ShapeVector *shape_vector) { in GetDataTypeAndShape()
127 auto shape_vector = utils::cast<abstract::ShapePtr>(abstract_tensor->BuildShape())->shape(); in FetchFromTensorValue() local
236 ShapeVector shape_vector; in FetchFromDefaultParam() local
389 auto shape_vector = utils::cast<abstract::ShapePtr>(abstract_tensor->BuildShape())->shape(); in FetchDataFromCNode() local
/third_party/mindspore/mindspore/lite/test/common/
Dimport_from_meta_graphT.cc53 std::vector<int64_t> shape_vector(shape.begin(), shape.end()); in ConverterConstTensor() local
98 std::vector<int64_t> shape_vector; in ConvertTensorToAbstractTensor() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dgroup_depthwise_op_convert_pass.cc114 std::vector<int64_t> shape_vector; in Run() local
Dnode_infershape.cc61 std::vector<int64_t> shape_vector(shape.begin(), shape.end()); in NewTensorInfo() local
551 std::vector<int64_t> shape_vector(shape.begin(), shape.end()); in ConvertTensorListToAbstract() local
Dslice_prepose_pass.cc88 std::vector<int64_t> shape_vector; in GetDefaultParamShape() local
368 …reposePass::CreateReshapeCNode(const FuncGraphPtr &graph, const std::vector<int64_t> &shape_vector, in CreateReshapeCNode()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc110 ShapeVector shape_vector{}; in SetFloatTensorInfo() local
145 ShapeVector shape_vector{}; in SetInt32TensorInfo() local
180 ShapeVector shape_vector{}; in SetInt64TensorInfo() local
223 ShapeVector shape_vector{}; in SetBoolTensorInfo() local
250 ShapeVector shape_vector{}; in SetStringTensorInfo() local
395 std::vector<int64_t> *shape_vector) { in ConvertConstTensor()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc436 …std::vector<int64_t> shape_vector = ConverterContext::GetInstance()->GetGraphInputTensorShape(tens… in ConvertGraphInputs() local
625 std::vector<int64_t> shape_vector; in ConvertConstTensor() local
746 std::vector<int64_t> shape_vector; in ConvertOutputTensor() local
762 std::vector<int64_t> shape_vector; in ConvertOutputTensor() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc62 STATUS GetShapeVectorFromCNode(const mindspore::CNodePtr &cnode, std::vector<int64_t> *shape_vector… in GetShapeVectorFromCNode()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc533 std::vector<int64_t> shape_vector = {kernel_num}; in AddNewBiasNode() local
900 std::vector<int64_t> shape_vector; in BuildParameterNode() local
983 std::vector<int64_t> shape_vector{static_cast<int64_t>(data.size())}; in BuildIntVecParameterNode() local
1006 std::vector<int64_t> shape_vector; in BuildIntVec2DParameterNode() local
1056 std::vector<int64_t> shape_vector{static_cast<int64_t>(data.size())}; in BuildFloatVecParameterNode() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc337 std::vector<int64_t> shape_vector = in ConvertGraphInputsOfShape() local
511 std::vector<int64_t> shape_vector; in ConvertBlobs() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc40 std::vector<int64_t> shape_vector; in CreateFakeAbstractTensor() local
/third_party/mindspore/mindspore/lite/tools/common/
Dtensor_util.cc165 auto &shape_vector = src_tensor->shape(); in UpdateTensorTFromTensorInfo() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dbatchmatmul_fusion.cc67 std::vector<int64_t> shape_vector(rmatmul_input_shape.begin(), rmatmul_input_shape.end()); in GetRightMatmulInputParamter() local
Dconstant_folding_fusion.cc143 std::vector<int64_t> shape_vector; in CreateNewParamter() local
Dtf_bidirection_gru_fusion.cc436 std::vector<int64_t> shape_vector(shape.begin(), shape.end()); in AddDefaultParameter() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc51 ShapeVector shape_vector = {static_cast<int64_t>(om_data.DataSize())}; in CreateOmParameter() local
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_adjust.cc204 auto shape_vector = utils::cast<abstract::ShapePtr>(abstract_tensor->BuildShape())->shape(); in ValueNodeInt64Convert() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc399 std::vector<int64_t> shape_vector; in CreateOutputsOfSplitWithOverlap() local