Lines Matching refs:kInputTensor

54     Shape input = context->getInputShape(kInputTensor);  in prepare()
74     NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis));  in execute()
75     switch (context->getInputType(kInputTensor)) {  in execute()
77             return eval(context->getInputBuffer<_Float16>(kInputTensor),  in execute()
78                         context->getInputShape(kInputTensor), axis,  in execute()
83             return eval(context->getInputBuffer<float>(kInputTensor),  in execute()
84                         context->getInputShape(kInputTensor), axis,  in execute()
89             return eval(context->getInputBuffer<int32_t>(kInputTensor),  in execute()
90                         context->getInputShape(kInputTensor), axis,  in execute()
95             return eval(context->getInputBuffer<uint8_t>(kInputTensor),  in execute()
96                         context->getInputShape(kInputTensor), axis,  in execute()
101             return eval(context->getInputBuffer<int8_t>(kInputTensor),  in execute()
102                         context->getInputShape(kInputTensor), axis,  in execute()