Lines Matching refs:kInputTensor
33 constexpr uint32_t kInputTensor = 0; variable
66 auto inputType = context->getInputType(kInputTensor); in validate()
72 const Shape& inputShape = context->getInputShape(kInputTensor); in validate()
86 Shape input = context->getInputShape(kInputTensor); in prepare()
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
99 switch (context->getInputType(kInputTensor)) { in execute()
101 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
102 context->getInputShape(kInputTensor), numGroups, axis, in execute()
105 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
106 context->getInputShape(kInputTensor), numGroups, axis, in execute()
109 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
110 context->getInputShape(kInputTensor), numGroups, axis, in execute()
113 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
114 context->getInputShape(kInputTensor), numGroups, axis, in execute()