Lines Matching refs:kInputTensor
30 constexpr uint32_t kInputTensor = 0; variable
63 auto inputType = context->getInputType(kInputTensor); in validate()
69 const Shape& inputShape = context->getInputShape(kInputTensor); in validate()
83 Shape input = context->getInputShape(kInputTensor); in prepare()
95 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
96 switch (context->getInputType(kInputTensor)) { in execute()
98 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
99 context->getInputShape(kInputTensor), numGroups, axis, in execute()
102 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
103 context->getInputShape(kInputTensor), numGroups, axis, in execute()
106 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
107 context->getInputShape(kInputTensor), numGroups, axis, in execute()
110 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
111 context->getInputShape(kInputTensor), numGroups, axis, in execute()