Lines Matching refs:kInputTensor
35 constexpr uint32_t kInputTensor = 0; variable
90 const OperandType inputType = context->getInputType(kInputTensor); in validate()
111 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
145 switch (context->getInputType(kInputTensor)) { in execute()
147 return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
148 context->getInputShape(kInputTensor), in execute()
156 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
165 return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor), in execute()
166 context->getInputShape(kInputTensor), in execute()
174 return evalGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
175 context->getInputShape(kInputTensor), in execute()
183 return evalGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
184 context->getInputShape(kInputTensor), in execute()