Searched refs:shapeTensor (Results 1 - 3 of 3) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | broadcast_to_test.cpp | 53 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(dataType, dim, quantParam, type);
in SaveParamsTensor() local
57 shapeTensor->SetBuffer(shapeValue, sizeof(int64_t) * shapeSize);
in SaveParamsTensor()
58 m_allTensors.emplace_back(shapeTensor);
in SaveParamsTensor()
163 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local
167 shapeTensor->SetBuffer(shapeValue, sizeof(float) * shapeSize);
in HWTEST_F()
168 m_allTensors.emplace_back(shapeTensor);
in HWTEST_F()
199 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
201 m_allTensors.emplace_back(shapeTensor);
in HWTEST_F()
|
| H A D | crop_test.cpp | 62 std::shared_ptr<NNTensor> shapeTensor;
in SetInputAndShape() local
63 shapeTensor = TransToNNTensor(OH_NN_FLOAT32, m_shapeDim, nullptr, OH_NN_TENSOR);
in SetInputAndShape()
64 m_allTensors.emplace_back(shapeTensor);
in SetInputAndShape()
|
| H A D | l2_normalize_test.cpp | 341 std::shared_ptr<NNTensor> shapeTensor = TransToNNTensor(OH_NN_INT8, m_paramDim,
in HWTEST_F() local
343 m_allTensors.emplace_back(shapeTensor);
in HWTEST_F()
|
Completed in 2 milliseconds