Lines Matching refs:context
86 Result<Version> validate(const IOperationValidationContext* context) { in validate() argument
87 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
88 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
90 const OperandType inputType = context->getInputType(kInputTensor); in validate()
104 context, {inputType, OperandType::TENSOR_INT32, OperandType::TENSOR_INT32})); in validate()
105 NN_RET_CHECK(validateOutputTypes(context, {inputType})); in validate()
110 bool prepare(IOperationExecutionContext* context) { in prepare() argument
111 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
115 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare()
119 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare()
123 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()
124 const int32_t* sizeData = context->getInputBuffer<int32_t>(kSizeTensor); in prepare()
126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
139 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
142 bool execute(IOperationExecutionContext* context) { in execute() argument
144 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
145 switch (context->getInputType(kInputTensor)) { in execute()
147 return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
148 context->getInputShape(kInputTensor), in execute()
149 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
150 context->getInputShape(kBeginTensor), in execute()
151 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
152 context->getInputShape(kSizeTensor), in execute()
153 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
156 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
158 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
159 context->getInputShape(kBeginTensor), in execute()
160 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
161 context->getInputShape(kSizeTensor), in execute()
162 context->getOutputBuffer<float>(kOutputTensor), in execute()
163 context->getOutputShape(kOutputTensor)); in execute()
165 return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor), in execute()
166 context->getInputShape(kInputTensor), in execute()
167 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
168 context->getInputShape(kBeginTensor), in execute()
169 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
170 context->getInputShape(kSizeTensor), in execute()
171 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
172 context->getOutputShape(kOutputTensor)); in execute()
174 return evalGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
175 context->getInputShape(kInputTensor), in execute()
176 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
177 context->getInputShape(kBeginTensor), in execute()
178 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
179 context->getInputShape(kSizeTensor), in execute()
180 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
181 context->getOutputShape(kOutputTensor)); in execute()
183 return evalGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
184 context->getInputShape(kInputTensor), in execute()
185 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
186 context->getInputShape(kBeginTensor), in execute()
187 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
188 context->getInputShape(kSizeTensor), in execute()
189 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
190 context->getOutputShape(kOutputTensor)); in execute()