Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DSlice.cpp34 constexpr uint32_t kBeginTensor = 1; variable
110 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
118 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()
144 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
145 context->getInputShape(kBeginTensor), in execute()
153 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
154 context->getInputShape(kBeginTensor), in execute()
162 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
163 context->getInputShape(kBeginTensor), in execute()
171 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
[all …]
DStridedSlice.cpp37 constexpr uint32_t kBeginTensor = 1; variable
92 context->getInputBuffer<int32_t>(kBeginTensor), in executeTyped()
144 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
160 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()