Home
last modified time | relevance | path

Searched refs:GetGraphInputTensorShape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/converter/
Dconverter_context.h101 std::vector<int64_t> GetGraphInputTensorShape(const std::string &tensor_name) const { in GetGraphInputTensorShape() function
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc312 …std::vector<int64_t> shape = ConverterContext::GetInstance()->GetGraphInputTensorShape(layer.name(… in ConvertGraphInputsOfLayer()
338 ConverterContext::GetInstance()->GetGraphInputTensorShape(caffe_model_.input(i)); in ConvertGraphInputsOfShape()
369 …std::vector<int64_t> shape = ConverterContext::GetInstance()->GetGraphInputTensorShape(caffe_model… in ConvertGraphInputsOfDim()
/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()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc435 shape = ConverterContext::GetInstance()->GetGraphInputTensorShape(node.name()); in ConvertParameter()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc296 …std::vector<int64_t> shape_vector = ConverterContext::GetInstance()->GetGraphInputTensorShape(inpu… in ConvertGraphInputs()