Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DTranspose.cpp87 if (context->isOmittedInput(kPermTensor)) { in prepare()
91 const Shape& permShape = context->getInputShape(kPermTensor); in prepare()
92 const int32_t* permData = context->getInputBuffer<int32_t>(kPermTensor); in prepare()
120 context->getInputBuffer<int32_t>(kPermTensor), in execute()
121 context->getInputShape(kPermTensor), in execute()
127 context->getInputBuffer<int32_t>(kPermTensor), in execute()
128 context->getInputShape(kPermTensor), in execute()
134 context->getInputBuffer<int32_t>(kPermTensor), in execute()
135 context->getInputShape(kPermTensor), in execute()
141 context->getInputBuffer<int32_t>(kPermTensor), in execute()
[all …]
/packages/modules/NeuralNetworks/common/types/operations/include/
DTranspose.h28 constexpr uint32_t kPermTensor = 1; variable