Searched refs:m_tensor (Results 1 – 6 of 6) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/neural_network_runtime_test/ |
H A D | neural_network_runtime_test.cpp | 200 m_tensor.dataType = OH_NN_INT32; in SetTensor() 201 m_tensor.dimensionCount = 0; in SetTensor() 202 m_tensor.dimensions = nullptr; in SetTensor() 203 m_tensor.quantParam = nullptr; in SetTensor() 204 m_tensor.type = OH_NN_TENSOR; in SetTensor() 236 m_tensor = {OH_NN_FLOAT32, 2, inputDims, nullptr, OH_NN_TENSOR}; in SetInputAndOutput() 1218 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR}; 1223 … OH_NN_ReturnCode ret = OH_NNExecutor_SetInput(nnExecutor, inputIndex, &m_tensor, buffer, length); 1435 m_tensor = {OH_NN_FLOAT32, 2, inputDims, nullptr, OH_NN_TENSOR}; 1871 …OH_NN_ReturnCode ret = OH_NNExecutor_SetInputWithMemory(nnExecutor, inputIndex, &m_tensor, memory); [all …]
|
H A D | neural_network_runtime_test.h | 42 OH_NN_Tensor m_tensor; variable
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/ |
H A D | neural_network_runtime_test.cpp | 201 m_tensor.dataType = OH_NN_INT32; in SetTensor() 202 m_tensor.dimensionCount = 0; in SetTensor() 203 m_tensor.dimensions = nullptr; in SetTensor() 204 m_tensor.quantParam = nullptr; in SetTensor() 205 m_tensor.type = OH_NN_TENSOR; in SetTensor() 237 m_tensor = {OH_NN_FLOAT32, 2, inputDims, nullptr, OH_NN_TENSOR}; in SetInputAndOutput() 1271 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR}; 1301 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR}; 1363 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR}; 1393 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR}; [all …]
|
H A D | neural_network_runtime_test.h | 42 OH_NN_Tensor m_tensor; variable
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_core_test/ |
H A D | neural_network_core_test.h | 42 OH_NN_Tensor m_tensor; variable
|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | lstm_test.cpp | 67 std::shared_ptr<NNTensor> m_tensor {}; member in OHOS::NeuralNetworkRuntime::UnitTest::LSTMBuilderTest
|