Home
last modified time | relevance | path

Searched refs:GeTensorPtr (Results 1 – 12 of 12) sorted by relevance

/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…
214 static std::string PrintGeTensor(const GeTensorPtr ge_tensor);
Dutil.cc147 std::vector<GeTensorPtr> TransformUtil::ConvertInputTensors(const std::vector<MeTensorPtr> &me_tens… in ConvertInputTensors()
149 std::vector<GeTensorPtr> ge_tensors; in ConvertInputTensors()
175 GeTensorPtr TransformUtil::ConvertTensor(const MeTensorPtr &tensor, const std::string &format) { in ConvertTensor()
196 …GeTensorPtr tensor_ptr = make_shared<GeTensor>(*desc, static_cast<uint8_t *>(tensor->data_c()), da… in ConvertTensor()
203 std::vector<MeTensorPtr> TransformUtil::ConvertGeTensors(const std::vector<GeTensorPtr> &ge_tensors, in ConvertGeTensors()
226 std::vector<MeTensorPtr> TransformUtil::ConvertGeTensors(const std::vector<GeTensorPtr> &ge_tensors… 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()
388 MeTensorPtr TransformUtil::ConvertGeTensor(const GeTensorPtr ge_tensor, const ShapeVector &request_… in ConvertGeTensor()
403 std::string TransformUtil::PrintGeTensor(const GeTensorPtr ge_tensor) { in PrintGeTensor()
Dgraph_runner.cc109 Status GraphRunner::RunGraph(const RunOptions &options, const std::vector<GeTensorPtr> &inputs, in RunGraph()
110 std::vector<GeTensorPtr> *outputs) { in RunGraph()
133 [](const GeTensorPtr &i) { return *i; }); in RunGraph()
169 std::vector<GeTensorPtr> ge_inputs; in RunGraph()
190 std::vector<GeTensorPtr> ge_outputs; in RunGraph()
Dgraph_runner.h52 …RunGraph(const RunOptions &options, const std::vector<GeTensorPtr> &inputs, std::vector<GeTensorPt…
Dtypes.h47 using GeTensorPtr = std::shared_ptr<GeTensor>; variable
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.cc77 bool SetInputsForSingleOpGraph(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &i… in SetInputsForSingleOpGraph()
121 bool BuildSingleOpGraph(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &inputs, in BuildSingleOpGraph()
170 void ToTensorPtr(const OpExecInfoPtr op_exec_info, std::vector<GeTensorPtr> *const inputs) { in ToTensorPtr()
190 …Code ConvertAttributes(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &inputs) { in ConvertAttributes()
224 const std::vector<GeTensorPtr> &ge_tensors) { in ConvertOutputTensors()
279 std::vector<GeTensorPtr> inputs{}; in RunOpInGE()
290 std::vector<GeTensorPtr> ge_inputs; in RunOpInGE()
291 std::vector<GeTensorPtr> ge_outputs; in RunOpInGE()
Dpynative_execute_ge.h33 using GeTensorPtr = std::shared_ptr<GeTensor>; variable
39 bool BuildSingleOpGraph(const OpExecInfoPtr &op_exec_info, const std::vector<GeTensorPtr> &inputs,
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_ge.cc45 using mindspore::transform::GeTensorPtr;
51 std::vector<GeTensorPtr> ge_tensors; in DoExecNonInputGraph()
52 std::vector<GeTensorPtr> ge_outputs; in DoExecNonInputGraph()
281 std::vector<GeTensorPtr> ge_tensors = TransformUtil::ConvertInputTensors(inputs, kOpFormat_NCHW); in RunGEInitGraph()
288 std::vector<GeTensorPtr> ge_outputs; in RunGEInitGraph()
399 std::vector<GeTensorPtr> ge_tensors = TransformUtil::ConvertInputTensors(inputs, kOpFormat_NCHW); in DoExecGraph()
404 std::vector<GeTensorPtr> ge_outputs; in DoExecGraph()
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc704 GeTensorPtr ge_tensor_ptr = in TEST_F()
770 …std::vector<GeTensorPtr> ge_tensors = TransformUtil::ConvertInputTensors(me_inputs, kOpFormat_NCHW… in TEST_F()
796 GeTensorPtr ge_tensor_ptr1 = in TEST_F()
798 GeTensorPtr ge_tensor_ptr2 = in TEST_F()
800 GeTensorPtr ge_tensor_ptr3 = in TEST_F()
803 std::vector<GeTensorPtr> ge_tensors; in TEST_F()
Dgraph_runner_test.cc77 …std::vector<GeTensorPtr> ge_tensor_ptrs = TransformUtil::ConvertInputTensors(inputs, kOpFormat_NCH… in DoExecGraph()
79 std::vector<GeTensorPtr> ge_outputs; in DoExecGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/
Dop_tiling_adapter.cc174 ge::GeTensorPtr ge_tensor = std::make_shared<ge::GeTensor>( in NewConstantOp()