Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DSqueeze.cpp39 const int32_t* squeezeDims = context->getInputBuffer<int32_t>(kSqueezeDims); in prepare()
41 const Shape squeezeDimsShape = context->getInputShape(kSqueezeDims); in prepare()
53 if (context->isOmittedInput(kSqueezeDims)) { in prepare()
/packages/modules/NeuralNetworks/common/types/operations/include/
DSqueeze.h26 constexpr uint32_t kSqueezeDims = 1; variable