Lines Matching refs:kInputTensor

74     switch (context->getInputType(kInputTensor)) {  in execute()
76 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
77 context->getInputShape(kInputTensor), in execute()
80 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
81 context->getInputShape(kInputTensor), in execute()
91 switch (context->getInputType(kInputTensor)) { in executeAbs()
94 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
95 context->getInputShape(kInputTensor), in executeAbs()
98 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
99 context->getInputShape(kInputTensor), in executeAbs()
103 context->getInputBuffer<int32_t>(kInputTensor), in executeAbs()
104 context->getInputShape(kInputTensor), in executeAbs()
113 const auto tensorType = context->getInputType(kInputTensor); in executeRsqrt()
116 return compute<float, _Float16>(frsqrt, context->getInputBuffer<_Float16>(kInputTensor), in executeRsqrt()
117 context->getInputShape(kInputTensor), in executeRsqrt()
120 return compute<float, float>(frsqrt, context->getInputBuffer<float>(kInputTensor), in executeRsqrt()
121 context->getInputShape(kInputTensor), in executeRsqrt()
124 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt()
129 context->getInputBuffer<uint8_t>(kInputTensor), in executeRsqrt()
130 context->getInputShape(kInputTensor), in executeRsqrt()
134 const Shape inShape = context->getInputShape(kInputTensor); in executeRsqrt()
139 context->getInputBuffer<int8_t>(kInputTensor), in executeRsqrt()
140 context->getInputShape(kInputTensor), in executeRsqrt()
150 Shape input = context->getInputShape(kInputTensor); in prepare()
157 Shape input = context->getInputShape(kInputTensor); in prepareFloor()