Home
last modified time | relevance | path

Searched refs:AllocateTensorBuffer (Results 1 – 17 of 17) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Ddevice.h66 virtual void* AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) = 0;
67 virtual void* AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) = 0;
H A Dhdi_device_v1_0.h62 void* AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) override;
63 void* AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) override;
H A Dhdi_device_v2_0.h62 void* AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) override;
63 void* AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) override;
H A Dhdi_device_v2_1.h62 void* AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) override;
63 void* AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) override;
H A Dhdi_device_v1_0.cpp391 void* HDIDeviceV1_0::AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) in AllocateTensorBuffer() function in OHOS::NeuralNetworkRuntime::HDIDeviceV1_0
396 void* HDIDeviceV1_0::AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) in AllocateTensorBuffer() function in OHOS::NeuralNetworkRuntime::HDIDeviceV1_0
H A Dnnexecutor.cpp625 void* inputBuffer = m_device->AllocateTensorBuffer(length, inputTensor); in SetInput()
778 …void* deviceOutputBuffer = m_device->AllocateTensorBuffer(length, m_outputTensorDescs[index].first… in SetOutput()
853 … void* deviceInputBuffer = m_device->AllocateTensorBuffer(length, m_inputTensorDescs[index].first); in CreateInputMemory()
918 …void* deviceOutputBuffer = m_device->AllocateTensorBuffer(length, m_outputTensorDescs[index].first… in CreateOutputMemory()
H A Dhdi_device_v2_0.cpp428 void* HDIDeviceV2_0::AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) in AllocateTensorBuffer() function in OHOS::NeuralNetworkRuntime::HDIDeviceV2_0
433 void* HDIDeviceV2_0::AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) in AllocateTensorBuffer() function in OHOS::NeuralNetworkRuntime::HDIDeviceV2_0
H A Dhdi_device_v2_1.cpp423 void* HDIDeviceV2_1::AllocateTensorBuffer(size_t length, std::shared_ptr<TensorDesc> tensor) in AllocateTensorBuffer() function in OHOS::NeuralNetworkRuntime::HDIDeviceV2_1
428 void* HDIDeviceV2_1::AllocateTensorBuffer(size_t length, std::shared_ptr<NNTensor> tensor) in AllocateTensorBuffer() function in OHOS::NeuralNetworkRuntime::HDIDeviceV2_1
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_executor/
H A Dnn_executor_test.cpp77 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
78 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));
158 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_outputTensorDescs[m_in…
173 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_inputTensorDescs[m_ind…
1659 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_inputTensorDescs[m_ind…
1706 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_inputTensorDescs[m_ind…
1815 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_inputTensorDescs[m_ind…
1923 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_outputTensorDescs[m_in…
1970 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_outputTensorDescs[m_in…
2079 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_outputTensorDescs[m_in…
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_compiled_cache/
H A Dnn_compiled_cache_test.cpp67 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
68 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_backend/
H A Dnn_backend_test.cpp65 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
66 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_compiler/
H A Dnn_compiler_test.cpp68 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
69 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_tensor/
H A Dnn_tensor_test.cpp73 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
74 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp293 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
294 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));
1857 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_inputTensorDescs[0].fi…
2037 …EXPECT_CALL(*((MockIDevice *) device.get()), AllocateTensorBuffer(length, m_outputTensorDescs[0].f…
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/hdi_device/
H A Dhdi_device_test.cpp1348 void* ret = hdiDevice->AllocateTensorBuffer(length, tensor);
1366 void* ret = hdiDevice->AllocateTensorBuffer(length, tensor);
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/hdi_device/
H A Dhdi_device_test.cpp1490 void* ret = hdiDevice->AllocateTensorBuffer(length, tensor);
1508 void* ret = hdiDevice->AllocateTensorBuffer(length, tensor);
2862 void* ret = hdiDevice->AllocateTensorBuffer(length, tensor);
2880 void* ret = hdiDevice->AllocateTensorBuffer(length, tensor);
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_core_test/
H A Dneural_network_core_test.cpp131 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<TensorDesc>));
132 MOCK_METHOD2(AllocateTensorBuffer, void*(size_t, std::shared_ptr<NNTensor>));