MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TensorToObj

Function TensorToObj

tensorflow/lite/delegates/gpu/cl/api.cc:356–361  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

354};
355
356TensorObject TensorToObj(const Tensor& tensor) {
357 if (tensor.StorageType() == TensorStorageType::BUFFER) {
358 return OpenClBuffer{tensor.GetMemoryPtr()};
359 }
360 return OpenClTexture{tensor.GetMemoryPtr()};
361}
362
363// Responsible for creating new tensor objects.
364class TensorTieFactory {

Callers 1

NewTensorTieMethod · 0.85

Calls 2

StorageTypeMethod · 0.80
GetMemoryPtrMethod · 0.45

Tested by

no test coverage detected