Searched refs:kSizeTensor (Results 1 – 1 of 1) sorted by relevance
35 constexpr uint32_t kSizeTensor = 2; variable114 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()119 const int32_t* sizeData = context->getInputBuffer<int32_t>(kSizeTensor); in prepare()146 context->getInputBuffer<int32_t>(kSizeTensor), in execute()147 context->getInputShape(kSizeTensor), in execute()155 context->getInputBuffer<int32_t>(kSizeTensor), in execute()156 context->getInputShape(kSizeTensor), in execute()164 context->getInputBuffer<int32_t>(kSizeTensor), in execute()165 context->getInputShape(kSizeTensor), in execute()173 context->getInputBuffer<int32_t>(kSizeTensor), in execute()[all …]