Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTranspose.cpp38 constexpr uint32_t kPermTensor = 1; variable
111 if (context->isOmittedInput(kPermTensor)) { in prepare()
115 const Shape& permShape = context->getInputShape(kPermTensor); in prepare()
116 const int32_t* permData = context->getInputBuffer<int32_t>(kPermTensor); in prepare()
144 context->getInputBuffer<int32_t>(kPermTensor), in execute()
145 context->getInputShape(kPermTensor), in execute()
151 context->getInputBuffer<int32_t>(kPermTensor), in execute()
152 context->getInputShape(kPermTensor), in execute()
158 context->getInputBuffer<int32_t>(kPermTensor), in execute()
159 context->getInputShape(kPermTensor), in execute()
[all …]