Lines Matching refs:kInputTensor
30 constexpr uint32_t kInputTensor = 0; variable
65 OperandType inputType = context->getInputType(kInputTensor); in validate()
83 Shape input = context->getInputShape(kInputTensor); in prepare()
103 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
104 switch (context->getInputType(kInputTensor)) { in execute()
106 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
107 context->getInputShape(kInputTensor), axis, in execute()
112 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
113 context->getInputShape(kInputTensor), axis, in execute()
118 return eval(context->getInputBuffer<int32_t>(kInputTensor), in execute()
119 context->getInputShape(kInputTensor), axis, in execute()
124 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
125 context->getInputShape(kInputTensor), axis, in execute()
130 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
131 context->getInputShape(kInputTensor), axis, in execute()