Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dassert_test.cpp53 std::shared_ptr<NNTensor> summarizeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
56 summarizeTensor->SetBuffer(summarizeValue, sizeof(int64_t)); in SaveParamsTensor()
57 m_allTensors.emplace_back(summarizeTensor); in SaveParamsTensor()
174 std::shared_ptr<NNTensor> summarizeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
177 summarizeTensor->SetBuffer(summarizeValue, sizeof(float));
178 m_allTensors.emplace_back(summarizeTensor);
209 std::shared_ptr<NNTensor> summarizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
211 m_allTensors.emplace_back(summarizeTensor);