Lines Matching refs:kInputTensor

53     Shape input = context->getInputShape(kInputTensor);  in prepare()
65     NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis));  in execute()
66     switch (context->getInputType(kInputTensor)) {  in execute()
68             return eval(context->getInputBuffer<_Float16>(kInputTensor),  in execute()
69                         context->getInputShape(kInputTensor), numGroups, axis,  in execute()
72             return eval(context->getInputBuffer<float>(kInputTensor),  in execute()
73                         context->getInputShape(kInputTensor), numGroups, axis,  in execute()
76             return eval(context->getInputBuffer<uint8_t>(kInputTensor),  in execute()
77                         context->getInputShape(kInputTensor), numGroups, axis,  in execute()
80             return eval(context->getInputBuffer<int8_t>(kInputTensor),  in execute()
81                         context->getInputShape(kInputTensor), numGroups, axis,  in execute()