Skip to content

SeetaResearch / Dragon

Go to a project
Commit 53d5742d by Ting PAN
Options

add DragonBoard

1 parent 93943fc8
Inline Side-by-side
Showing with 5 additions and 1 deletions
.gitmodules 0 → 100644
[submodule "DragonBoard"]
path = DragonBoard
url = git://github.com/neopenx/DragonBoard.git
Dragon/modules/python/dragon.cc
......@@ -281,7 +281,7 @@ PyObject* FetchTensorCC(PyObject* self, PyObject* args) {
TypeId type_id = CTypeToFetcher(tensor->meta().id());
CHECK(type_id != 0) << "\nTensor(" << tensor->name()
<< ") has not been computed yet.";
unique_ptr<TensorFetcherBase> fetcher(createFetcher(type_id));
unique_ptr<TensorFetcherBase> fetcher(CreateFetcher(type_id));
if (fetcher.get()) {
return fetcher->Fetch(*tensor); // copy the tensor data to a numpy object
} else {
......
DragonBoard @ c4e649f6
Subproject commit c4e649f66e59b39d07a1a857c72e721ebbdb93ae
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!