Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dstrided_slice_builder_test.cpp101 … std::shared_ptr<NNTensor> shrinkAxisMaskTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveShrinkAxisTensor() local
104 shrinkAxisMaskTensor->SetBuffer(shrinkAxisMaskValue, sizeof(int64_t)); in SaveShrinkAxisTensor()
105 m_allTensors.emplace_back(shrinkAxisMaskTensor); in SaveShrinkAxisTensor()