Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DSlice.cpp53 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t* beginData, in evalGeneric() argument
64 beginIndex[i] = static_cast<uint32_t>(beginData[i]); in evalGeneric()
123 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare() local
129 const int32_t sliceBegin = beginData[i]; in prepare()
134 NN_RET_CHECK_LE(beginData[i], getSizeOfDimension(inputShape, i)); in prepare()
H A DStridedSlice.cpp53 bool compute(const T* inputData, const Shape& inputShape, const int32_t* beginData, in compute() argument
67 starts.emplace_back(beginData[idx]); in compute()
163 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare() local
181 : ClampedIndex(beginData[idx], dim, positiveStride); in prepare()
/aosp12/packages/modules/NeuralNetworks/common/include/
H A DOperations.h121 const int32_t* beginData, const int32_t* endData,