We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1f5f963 commit acd44bfCopy full SHA for acd44bf
1 file changed
sherpa-onnx/csrc/onnx-utils.cc
@@ -189,8 +189,8 @@ Ort::Value View(Ort::Value *v) {
189
auto type_and_shape = v->GetTensorTypeAndShapeInfo();
190
std::vector<int64_t> shape = type_and_shape.GetShape();
191
192
- auto memory_info =
193
- Ort::MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeDefault);
+ auto memory_info = v->GetTensorMemoryInfo();
+
194
switch (type_and_shape.GetElementType()) {
195
case ONNX_TENSOR_ELEMENT_DATA_TYPE_INT32:
196
return Ort::Value::CreateTensor(
0 commit comments