Lines Matching refs:kInputTensor
30 constexpr uint32_t kInputTensor = 0; variable
48 switch (context->getInputType(kInputTensor)) { in execute()
50 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
51 context->getInputShape(kInputTensor), in execute()
54 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
55 context->getInputShape(kInputTensor), in execute()
65 switch (context->getInputType(kInputTensor)) { in executeAbs()
68 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
69 context->getInputShape(kInputTensor), in executeAbs()
72 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
73 context->getInputShape(kInputTensor), in executeAbs()
77 context->getInputBuffer<int32_t>(kInputTensor), in executeAbs()
78 context->getInputShape(kInputTensor), in executeAbs()
88 OperandType inputType = context->getInputType(kInputTensor); in validate()
100 OperandType inputType = context->getInputType(kInputTensor); in validateAbs()
113 OperandType inputType = context->getInputType(kInputTensor); in validateFloor()
120 const Shape& input = context->getInputShape(kInputTensor); in validateFloor()
129 Shape input = context->getInputShape(kInputTensor); in prepare()
136 Shape input = context->getInputShape(kInputTensor); in prepareFloor()