Searched refs:ge_tensor_ptr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | callbacks_ge.cc | 71 const std::shared_ptr<ge::Tensor> &ge_tensor_ptr) { in GetMeTensorTransformed() argument
84 return TransformUtil::ConvertGeTensor(ge_tensor_ptr, *parameter_shape_ptr); in GetMeTensorTransformed()
95 std::shared_ptr<ge::Tensor> ge_tensor_ptr = std::make_shared<ge::Tensor>(item.second); in CheckpointSaveCallback() local
99 TensorPtr tensor_ptr = GetMeTensorTransformed(graph_id, name, ge_tensor_ptr); in CheckpointSaveCallback()
121 … GetMeTensorForSummary(const std::string &name, const std::shared_ptr<ge::Tensor> &ge_tensor_ptr) { in GetMeTensorForSummary() argument
138 return TransformUtil::ConvertGeTensor(ge_tensor_ptr, shape); in GetMeTensorForSummary()
144 return TransformUtil::ConvertGeTensor(ge_tensor_ptr); in GetMeTensorForSummary()
150 return TransformUtil::ConvertGeTensor(ge_tensor_ptr); in GetMeTensorForSummary()
167 std::shared_ptr<ge::Tensor> ge_tensor_ptr = std::make_shared<ge::Tensor>(item.second); in SummarySaveCallback() local
168 TensorPtr tensor_ptr = GetMeTensorForSummary(tag_name, ge_tensor_ptr); in SummarySaveCallback()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | graph_runner.cc | 181 auto ge_tensor_ptr = TransformUtil::ConvertTensor(it, kOpFormat_NCHW); in RunGraph() local
182 if (ge_tensor_ptr != nullptr) { in RunGraph()
183 ge_inputs.emplace_back(ge_tensor_ptr); in RunGraph()
|
| D | util.cc | 163 auto ge_tensor_ptr = TransformUtil::ConvertTensor(me_tensors[index], format); in ConvertInputTensors() local
164 if (ge_tensor_ptr != nullptr) { in ConvertInputTensors()
165 ge_tensors.emplace_back(ge_tensor_ptr); in ConvertInputTensors()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 635 auto ge_tensor_ptr = TransformUtil::ConvertTensor(me_tensor_ptr, kOpFormat_NCHW); in TEST_F() local
636 ASSERT_EQ(ge_tensor_ptr->GetTensorDesc().GetFormat(), GeFormat::FORMAT_NCHW); in TEST_F()
637 ASSERT_EQ(ge_tensor_ptr->GetTensorDesc().GetDataType(), GeDataType::DT_FLOAT); in TEST_F()
641 ASSERT_EQ(ge_dims[i], ge_tensor_ptr->GetTensorDesc().GetShape().GetDims()[i]); in TEST_F()
643 for (i = 0; i < ge_tensor_ptr->GetTensorDesc().GetShape().GetShapeSize(); i++) { in TEST_F()
644 ASSERT_EQ(data[i], (reinterpret_cast<float*>(ge_tensor_ptr->GetData()))[i]); in TEST_F()
704 GeTensorPtr ge_tensor_ptr = in TEST_F() local
706 GeTensor& ge_tensor = *ge_tensor_ptr; in TEST_F()
717 MeTensorPtr me_tensor_ptr = TransformUtil::ConvertGeTensor(ge_tensor_ptr); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute_ge.cc | 181 auto ge_tensor_ptr = transform::TransformUtil::ConvertTensor(me_tensor_ptr, kOpFormat_NCHW); in ToTensorPtr() local
182 if (ge_tensor_ptr == nullptr) { in ToTensorPtr()
186 inputs->push_back(ge_tensor_ptr); in ToTensorPtr()
|