Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DSlice.cpp37 constexpr uint32_t kSizeTensor = 2; variable
119 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()
124 const int32_t* sizeData = context->getInputBuffer<int32_t>(kSizeTensor); in prepare()
151 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
152 context->getInputShape(kSizeTensor), in execute()
160 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
161 context->getInputShape(kSizeTensor), in execute()
169 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
170 context->getInputShape(kSizeTensor), in execute()
179 context->getInputShape(kSizeTensor), in execute()
[all …]