Lines Matching refs:getInputShape

86     Shape input = context->getInputShape(kInputTensor);  in prepare()
89     Shape indices = context->getInputShape(kInputIndices);  in prepare()
106     NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis));  in execute()
110                         context->getInputShape(kInputTensor), axis,  in execute()
112                         context->getInputShape(kInputIndices),  in execute()
116                         context->getInputShape(kInputTensor), axis,  in execute()
118                         context->getInputShape(kInputIndices),  in execute()
122                         context->getInputShape(kInputTensor), axis,  in execute()
124                         context->getInputShape(kInputIndices),  in execute()
128                         context->getInputShape(kInputTensor), axis,  in execute()
130                         context->getInputShape(kInputIndices),  in execute()
134                         context->getInputShape(kInputTensor), axis,  in execute()
136                         context->getInputShape(kInputIndices),  in execute()