Lines Matching refs:kInputTensor
33 constexpr uint32_t kInputTensor = 0; variable
88 const OperandType inputType = context->getInputType(kInputTensor); in validate()
106 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
140 switch (context->getInputType(kInputTensor)) { in execute()
142 return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
143 context->getInputShape(kInputTensor), in execute()
151 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
152 context->getInputShape(kInputTensor), in execute()
160 return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor), in execute()
161 context->getInputShape(kInputTensor), in execute()
169 return evalGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
170 context->getInputShape(kInputTensor), in execute()
178 return evalGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
179 context->getInputShape(kInputTensor), in execute()