Searched refs:GeTensor (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter_util.cc | 30 GeTensor ConvertAnyUtil(const ValuePtr &value, const AnyTraits<mindspore::tensor::Tensor> &) { in ConvertAnyUtil()
35 return ge_tensor == nullptr ? GeTensor() : *ge_tensor; in ConvertAnyUtil()
131 GeTensor VectorToTensorUtil(const ValuePtr &value) { in VectorToTensorUtil()
137 return GeTensor(GeTensorDesc(ge::Shape({0}))); in VectorToTensorUtil()
147 return GeTensor(*desc, reinterpret_cast<uint8_t *>(data.data()), data.size() * sizeof(int32_t)); in VectorToTensorUtil()
155 return GeTensor(*desc, reinterpret_cast<uint8_t *>(data.data()), data.size() * sizeof(int64_t)); in VectorToTensorUtil()
163 return GeTensor(*desc, reinterpret_cast<uint8_t *>(data.data()), data.size() * sizeof(float)); in VectorToTensorUtil()
172 return GeTensor(*desc, static_cast<uint8_t *>(data.data()), data.size() * sizeof(uint8_t)); in VectorToTensorUtil()
177 return GeTensor(); in VectorToTensorUtil()
180 GeTensor ConvertAnyUtil(const ValuePtr &value, const AnyTraits<AnyValue>) { in ConvertAnyUtil()
[all …]
|
| D | graph_runner.cc | 129 std::vector<GeTensor> ge_inputs; in RunGraph()
130 std::vector<GeTensor> ge_outputs; in RunGraph()
162 … [](const GeTensor &ge_tensor) { return std::make_shared<GeTensor>(ge_tensor); }); in RunGraph()
|
| D | types.h | 35 using GeTensor = ge::Tensor; variable
47 using GeTensorPtr = std::shared_ptr<GeTensor>;
|
| D | op_adapter_util.h | 32 GeTensor ConvertAnyUtil(const ValuePtr &value, const AnyTraits<mindspore::tensor::Tensor> &traits);
61 GeTensor ConvertAnyUtil(const ValuePtr &value, const AnyTraits<AnyValue>);
|
| D | op_adapter.h | 301 …static GeTensor ConvertAny(const ValuePtr &value, const AnyTraits<mindspore::tensor::Tensor> &trai… in ConvertAny()
422 static GeTensor ConvertAny(const ValuePtr &value, const AnyTraits<AnyValue> anyTraitsValue) { in ConvertAny()
|
| D | convert.cc | 163 …(void)const_iter_num->set_attr_value(GeTensor(desc, reinterpret_cast<uint8_t *>(&value), sizeof(in… in InitLoopVar()
167 …(void)const_loop_cond->set_attr_value(GeTensor(desc, reinterpret_cast<uint8_t *>(&value), sizeof(i… in InitLoopVar()
171 …(void)const_one->set_attr_value(GeTensor(desc, reinterpret_cast<uint8_t *>(&value), sizeof(int64_t… in InitLoopVar()
175 …(void)const_zero->set_attr_value(GeTensor(desc, reinterpret_cast<uint8_t *>(&value), sizeof(int64_… in InitLoopVar()
|
| D | util.cc | 196 …GeTensorPtr tensor_ptr = make_shared<GeTensor>(*desc, static_cast<uint8_t *>(tensor->data_c()), da… in ConvertTensor()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute_ge.h | 32 using GeTensor = ge::Tensor; variable
33 using GeTensorPtr = std::shared_ptr<GeTensor>;
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 705 …std::make_shared<GeTensor>(desc, reinterpret_cast<uint8_t*>(data.data()), data.size() * sizeof(DTY… in TEST_F()
706 GeTensor& ge_tensor = *ge_tensor_ptr; in TEST_F()
797 …std::make_shared<GeTensor>(desc1, reinterpret_cast<uint8_t*>(data1.data()), data1.size() * sizeof(… in TEST_F()
799 …std::make_shared<GeTensor>(desc2, reinterpret_cast<uint8_t*>(data2.data()), data2.size() * sizeof(… in TEST_F()
801 …std::make_shared<GeTensor>(desc3, reinterpret_cast<uint8_t*>(data3.data()), data3.size() * sizeof(… in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 174 ge::GeTensorPtr ge_tensor = std::make_shared<ge::GeTensor>( in NewConstantOp()
|