Home
last modified time | relevance | path

Searched refs:MeTensorPtr (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/tests/ut/cpp/transform/
Dgraph_runner_test.cc76 std::shared_ptr<std::vector<MeTensorPtr>> DoExecGraph(const std::vector<MeTensorPtr> &inputs) { in DoExecGraph()
100 std::vector<MeTensorPtr> me_outputs = TransformUtil::ConvertGeTensors(ge_outputs, request_dims); in DoExecGraph()
102 return std::make_shared<std::vector<MeTensorPtr>>(me_outputs); in DoExecGraph()
140 std::map<string, MeTensorPtr> dict; in TEST_F()
142 MeTensorPtr init_tensor_ptr = MakeTensor(kF32, list0); in TEST_F()
151 MeTensorPtr me_tensor_ptr = MakeTensor(kF32, list1); in TEST_F()
154 MeTensorPtr input_ptr = MakeTensor(kF32, list2); in TEST_F()
155 std::vector<MeTensorPtr> me_inputs; in TEST_F()
157 std::vector<MeTensorPtr> me_outputs; in TEST_F()
180 std::map<std::string, MeTensorPtr> dict; in TEST_F()
[all …]
Dtransform_base_test.h41 MeTensorPtr MakeTensor(const TypePtr& t, std::initializer_list<int64_t> shp);
Dconvert_test.cc717 MeTensorPtr me_tensor_ptr = TransformUtil::ConvertGeTensor(ge_tensor_ptr); in TEST_F()
762 MeTensorPtr input_ptr1 = MakeTensor(kF32, list0); in TEST_F()
763 MeTensorPtr input_ptr2 = MakeTensor(kF32, list1); in TEST_F()
764 MeTensorPtr input_ptr3 = MakeTensor(kF32, list2); in TEST_F()
765 std::vector<MeTensorPtr> me_inputs; in TEST_F()
816 std::vector<MeTensorPtr> me_outputs = TransformUtil::ConvertGeTensors(ge_tensors, request_dims); in TEST_F()
Dtransform_base_test.cc100 MeTensorPtr MakeTensor(const TypePtr& t, std::initializer_list<int64_t> shp) { in MakeTensor()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dutil.h87 static GeTensorPtr ConvertTensor(const MeTensorPtr &tensor, const std::string &format);
96 static std::vector<GeTensorPtr> ConvertInputTensors(const std::vector<MeTensorPtr> &me_tensors,
105 static MeTensorPtr ConvertGeTensor(const GeTensorPtr &tensor);
114 static MeTensorPtr ConvertGeTensor(GeTensorPtr ge_tensor, const ShapeVector &request_dims);
122 static std::vector<MeTensorPtr> ConvertGeTensors(const std::vector<GeTensorPtr> &ge_tensors,
130 static std::vector<MeTensorPtr> ConvertGeTensors(const std::vector<GeTensorPtr> &ge_tensors);
139 …static MeTensorPtr GenerateMeTensor(const GeTensorPtr &ge_tensor, const ShapeVector &me_dims, cons…
Dutil.cc147 std::vector<GeTensorPtr> TransformUtil::ConvertInputTensors(const std::vector<MeTensorPtr> &me_tens… in ConvertInputTensors()
175 GeTensorPtr TransformUtil::ConvertTensor(const MeTensorPtr &tensor, const std::string &format) { in ConvertTensor()
203 std::vector<MeTensorPtr> TransformUtil::ConvertGeTensors(const std::vector<GeTensorPtr> &ge_tensors, in ConvertGeTensors()
205 std::vector<MeTensorPtr> outputs; in ConvertGeTensors()
208 MeTensorPtr me_tensor_ptr = nullptr; in ConvertGeTensors()
226 std::vector<MeTensorPtr> TransformUtil::ConvertGeTensors(const std::vector<GeTensorPtr> &ge_tensors… in ConvertGeTensors()
227 std::vector<MeTensorPtr> outputs; in ConvertGeTensors()
230 MeTensorPtr me_tensor_ptr = ConvertGeTensor(ge_tensors[index]); in ConvertGeTensors()
344 MeTensorPtr TransformUtil::GenerateMeTensor(const GeTensorPtr &ge_tensor, const ShapeVector &me_dim… in GenerateMeTensor()
373 MeTensorPtr TransformUtil::ConvertGeTensor(const GeTensorPtr &ge_tensor) { in ConvertGeTensor()
[all …]
Dgraph_runner.h51 …RunGraph(const RunOptions &options, const std::vector<MeTensorPtr> &inputs, std::vector<MeTensorPt…
Dgraph_runner.cc167 Status GraphRunner::RunGraph(const RunOptions &options, const std::vector<MeTensorPtr> &inputs, in RunGraph()
168 std::vector<MeTensorPtr> *const outputs) { in RunGraph()
Dtypes.h42 using MeTensorPtr = std::shared_ptr<MeTensor>; variable
Dop_adapter_util.cc33 auto me_tensor = value->cast<MeTensorPtr>(); in ConvertAnyUtil()
Dconvert.cc1851 …GeTensorPtr ge_tensor = transform::TransformUtil::ConvertTensor(vec[i]->cast<MeTensorPtr>(), kOpFo… in TryConvertValueNodeToMultiConst()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute_ge.cc40 using MeTensorPtr = mindspore::tensor::TensorPtr; typedef
57 MeTensorPtr ConvertPyObjToTensor(const py::object &obj) { in ConvertPyObjToTensor()
58 MeTensorPtr me_tensor_ptr = nullptr; in ConvertPyObjToTensor()
60 me_tensor_ptr = py::cast<MeTensorPtr>(obj); in ConvertPyObjToTensor()
97 MeTensorPtr me_tensor_ptr = ConvertPyObjToTensor(op_inputs[i]); in SetInputsForSingleOpGraph()
180 MeTensorPtr me_tensor_ptr = ConvertPyObjToTensor(op_inputs[i]); in ToTensorPtr()
223 std::vector<MeTensorPtr> ConvertOutputTensors(const OpExecInfoPtr &op_exec_info, in ConvertOutputTensors()
225 std::vector<MeTensorPtr> outputs; in ConvertOutputTensors()
307 std::vector<MeTensorPtr> graph_outputs = ConvertOutputTensors(op_exec_info, ge_outputs); in RunOpInGE()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_ge.cc46 using mindspore::transform::MeTensorPtr;
397 std::shared_ptr<py::object> DoExecGraph(const FuncGraphPtr &graph, const std::vector<MeTensorPtr> &… in DoExecGraph()
424 std::vector<MeTensorPtr> me_outputs = TransformUtil::ConvertGeTensors(ge_outputs); in DoExecGraph()