Lines Matching refs:getInputShape

83     Shape input = context->getInputShape(kInputTensor);  in prepare()
86     Shape indices = context->getInputShape(kInputIndices);  in prepare()
103     NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis));  in execute()
107                         context->getInputShape(kInputTensor), axis,  in execute()
109                         context->getInputShape(kInputIndices),  in execute()
113                         context->getInputShape(kInputTensor), axis,  in execute()
115                         context->getInputShape(kInputIndices),  in execute()
119                         context->getInputShape(kInputTensor), axis,  in execute()
121                         context->getInputShape(kInputIndices),  in execute()
125                         context->getInputShape(kInputTensor), axis,  in execute()
127                         context->getInputShape(kInputIndices),  in execute()
131                         context->getInputShape(kInputTensor), axis,  in execute()
133                         context->getInputShape(kInputIndices),  in execute()