Searched refs:kBatchesTensor (Results 1 – 1 of 1) sorted by relevance
199 constexpr uint32_t kBatchesTensor = 2; variable233 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()284 context->getInputShape(kBatchesTensor), in execute()296 context->getInputShape(kBatchesTensor), in execute()309 context->getInputShape(kBatchesTensor), in execute()342 constexpr uint32_t kBatchesTensor = 2; variable762 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()824 context->getInputBuffer<int32_t>(kBatchesTensor), in execute()825 context->getInputShape(kBatchesTensor), in execute()869 context->getInputShape(kBatchesTensor), in execute()[all …]