Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/src/
DReverse.cpp39 NN_RET_CHECK_EQ(context->getInputType(kInputAxisTensor), OperandType::TENSOR_INT32); in validate()
40 const Shape inputAxisTensorShape = context->getInputShape(kInputAxisTensor); in validate()
43 << "Input tensor #" << kInputAxisTensor << " must have 1 dimension"; in validate()
46 << "Input tensor #" << kInputAxisTensor << " dimension must be 1 but is " << dim0; in validate()
/packages/modules/NeuralNetworks/common/cpu_operations/
DReverse.cpp49 const Shape axisShape = context->getInputShape(kInputAxisTensor); in prepare()
52 const int32_t axisDimension = (context->getInputBuffer<int32_t>(kInputAxisTensor))[0]; in prepare()
68 tflite::reference_ops::Reverse((context->getInputBuffer<int32_t>(kInputAxisTensor))[0], in reverse()
/packages/modules/NeuralNetworks/common/types/operations/include/
DReverse.h29 constexpr uint32_t kInputAxisTensor = 1; variable