| 113 | } |
| 114 | |
| 115 | std::unique_ptr<ITensorHandle> RefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo, |
| 116 | const bool isMemoryManaged) const |
| 117 | { |
| 118 | if (isMemoryManaged) |
| 119 | { |
| 120 | return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager); |
| 121 | } |
| 122 | else |
| 123 | { |
| 124 | return std::make_unique<RefTensorHandle>(tensorInfo); |
| 125 | } |
| 126 | } |
| 127 | |
| 128 | std::unique_ptr<ITensorHandle> RefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo, |
| 129 | DataLayout dataLayout, |