Searched refs:blockShapeTensor (Results 1 – 1 of 1) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | spacetobatchnd_builder_test.cpp | 77 std::shared_ptr<NNTensor> blockShapeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveBlockShapeTensor() local
80 blockShapeTensor->SetBuffer(blockShapeValue, sizeof(int64_t) * blockShapeLen); in SaveBlockShapeTensor()
81 m_allTensors.emplace_back(blockShapeTensor); in SaveBlockShapeTensor()
315 std::shared_ptr<NNTensor> blockShapeTensor = TransToNNTensor(OH_NN_INT64, shapeDim, nullptr, variable
317 blockShapeTensor->SetBuffer(nullptr, 0);
318 m_allTensors.emplace_back(blockShapeTensor);
341 std::shared_ptr<NNTensor> blockShapeTensor = TransToNNTensor(OH_NN_INT64, paddingsDim, nullptr, variable
343 blockShapeTensor->SetBuffer(nullptr, 0);
344 m_allTensors.emplace_back(blockShapeTensor);
|