Home
last modified time | relevance | path

Searched refs:blockShapeTensor (Results 1 - 1 of 1) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dspacetobatchnd_builder_test.cpp77 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, in HWTEST_F() local
317 blockShapeTensor->SetBuffer(nullptr, 0); in HWTEST_F()
318 m_allTensors.emplace_back(blockShapeTensor); in HWTEST_F()
341 std::shared_ptr<NNTensor> blockShapeTensor = TransToNNTensor(OH_NN_INT64, paddingsDim, nullptr, in HWTEST_F() local
343 blockShapeTensor->SetBuffer(nullptr, 0); in HWTEST_F()
344 m_allTensors.emplace_back(blockShapeTensor); in HWTEST_F()

Completed in 1 milliseconds