Lines Matching refs:kInputTensor
31 constexpr uint32_t kInputTensor = 0; variable
51 switch (context->getInputType(kInputTensor)) { in execute()
53 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
54 context->getInputShape(kInputTensor), in execute()
57 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
58 context->getInputShape(kInputTensor), in execute()
68 switch (context->getInputType(kInputTensor)) { in executeAbs()
71 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
72 context->getInputShape(kInputTensor), in executeAbs()
75 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
76 context->getInputShape(kInputTensor), in executeAbs()
80 context->getInputBuffer<int32_t>(kInputTensor), in executeAbs()
81 context->getInputShape(kInputTensor), in executeAbs()
91 OperandType inputType = context->getInputType(kInputTensor); in validate()
103 OperandType inputType = context->getInputType(kInputTensor); in validateAbs()
117 OperandType inputType = context->getInputType(kInputTensor); in validateFloor()
124 const Shape& input = context->getInputShape(kInputTensor); in validateFloor()
135 Shape input = context->getInputShape(kInputTensor); in prepare()
142 Shape input = context->getInputShape(kInputTensor); in prepareFloor()