Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dstrided_slice_builder_test.cpp79 std::shared_ptr<NNTensor> ellipsisMaskTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveEllipsisMaskTensor() local
82 ellipsisMaskTensor->SetBuffer(ellipsisMaskValue, sizeof(int64_t)); in SaveEllipsisMaskTensor()
83 m_allTensors.emplace_back(ellipsisMaskTensor); in SaveEllipsisMaskTensor()
427 std::shared_ptr<NNTensor> ellipsisMaskTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, variable
429 ellipsisMaskTensor->SetBuffer(nullptr, 0);
430 m_allTensors.emplace_back(ellipsisMaskTensor);