Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dcrop_test.cpp79 std::shared_ptr<NNTensor> offsetTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveOffset() local
81 offsetTensor->SetBuffer(offsetValue, sizeof(int64_t)); in SaveOffset()
82 m_allTensors.emplace_back(offsetTensor); in SaveOffset()
216 std::shared_ptr<NNTensor> offsetTensor = TransToNNTensor(OH_NN_FLOAT32, m_offsetDim, variable
221 offsetTensor->SetBuffer(offsetValue, sizeof(float) * offsetSize);
222 m_allTensors.emplace_back(offsetTensor);