Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dslice_builder_test.cpp58 std::shared_ptr<NNTensor> axesTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxesTensor() local
61 axesTensor->SetBuffer(axesValue, sizeof(int64_t)); in SaveAxesTensor()
62 m_allTensors.emplace_back(axesTensor); in SaveAxesTensor()
181 std::shared_ptr<NNTensor> axesTensor = TransToNNTensor(OH_NN_FLOAT32, paramsDim, in HWTEST_F() local
185 axesTensor->SetBuffer(axesValue, sizeof(float)); in HWTEST_F()
186 m_allTensors.emplace_back(axesTensor); in HWTEST_F()
227 std::shared_ptr<NNTensor> axesTensor = TransToNNTensor(OH_NN_INT64, paramsDim, in HWTEST_F() local
229 m_allTensors.emplace_back(axesTensor); in HWTEST_F()

Completed in 2 milliseconds