Lines Matching refs:getInputShape
180 Shape input = context->getInputShape(kInputTensor); in prepare()
181 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
182 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare()
229 context->getInputShape(kInputTensor), in execute()
231 context->getInputShape(kRoiTensor), in execute()
233 context->getInputShape(kBatchSplitTensor), in execute()
241 context->getInputShape(kInputTensor), in execute()
243 context->getInputShape(kRoiTensor), in execute()
245 context->getInputShape(kBatchSplitTensor), in execute()
253 context->getInputShape(kInputTensor), in execute()
255 context->getInputShape(kRoiTensor), in execute()
257 context->getInputShape(kBatchSplitTensor), in execute()
265 context->getInputShape(kInputTensor), in execute()
267 context->getInputShape(kRoiTensor), in execute()
269 context->getInputShape(kBatchSplitTensor), in execute()