Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dconstant_of_shape_test.cpp56 std::shared_ptr<NNTensor> dataTypeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveDataType() local
58 dataTypeTensor->SetBuffer(dataTypeValue, sizeof(int64_t)); in SaveDataType()
59 m_allTensors.emplace_back(dataTypeTensor); in SaveDataType()
179 std::shared_ptr<NNTensor> dataTypeTensor = TransToNNTensor(OH_NN_FLOAT32, m_dataTypeDim, variable
182 dataTypeTensor->SetBuffer(dataTypeValue, sizeof(float));
183 m_allTensors.emplace_back(dataTypeTensor);
255 std::shared_ptr<NNTensor> dataTypeTensor = TransToNNTensor(OH_NN_INT64, m_dataTypeDim, variable
257 m_allTensors.emplace_back(dataTypeTensor);
H A Dcrop_test.cpp270 std::shared_ptr<NNTensor> dataTypeTensor = TransToNNTensor(OH_NN_INT64, m_axisDim, variable
272 m_allTensors.emplace_back(dataTypeTensor);