Lines Matching refs:getInputShape

54     Shape input = context->getInputShape(kInputTensor);  in prepare()
57     Shape indices = context->getInputShape(kInputIndices);  in prepare()
74     NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis));  in execute()
78                         context->getInputShape(kInputTensor), axis,  in execute()
80                         context->getInputShape(kInputIndices),  in execute()
84                         context->getInputShape(kInputTensor), axis,  in execute()
86                         context->getInputShape(kInputIndices),  in execute()
90                         context->getInputShape(kInputTensor), axis,  in execute()
92                         context->getInputShape(kInputIndices),  in execute()
96                         context->getInputShape(kInputTensor), axis,  in execute()
98                         context->getInputShape(kInputIndices),  in execute()
102                         context->getInputShape(kInputTensor), axis,  in execute()
104                         context->getInputShape(kInputIndices),  in execute()