Lines Matching refs:input2
146 const TfLiteTensor* input2, TfLiteTensor* output, in PrepareInt16SubOp() argument
157 TF_LITE_ENSURE_EQ(context, input2->params.zero_point, 0); in PrepareInt16SubOp()
167 CheckedLog2(input2->params.scale, &input2_scale_log2_rounded); in PrepareInt16SubOp()
198 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in Prepare() local
201 TF_LITE_ENSURE_EQ(context, input1->type, input2->type); in Prepare()
202 output->type = input2->type; in Prepare()
204 data->requires_broadcast = !HaveSameShapes(input1, input2); in Prepare()
209 context, input1, input2, &output_size)); in Prepare()
215 TF_LITE_ENSURE_OK(context, Prepare8BitSubOp(context, input1, input2, output, in Prepare()
218 TF_LITE_ENSURE_OK(context, PrepareInt16SubOp(context, input1, input2, in Prepare()
228 const TfLiteTensor* input2, TfLiteTensor* output) { in EvalSub() argument
237 GetTensorData<data_type>(input1), GetTensorShape(input2), \ in EvalSub()
238 GetTensorData<data_type>(input2), GetTensorShape(output), \ in EvalSub()
275 const TfLiteTensor* input1, const TfLiteTensor* input2, in EvalQuantized() argument
292 GetTensorShape(input1), GetTensorShape(input2), &op_params); in EvalQuantized()
296 GetTensorData<data_type>(input1), GetTensorShape(input2), \ in EvalQuantized()
297 GetTensorData<data_type>(input2), GetTensorShape(output), \ in EvalQuantized()
348 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in Eval() local
352 EvalSub<kernel_type>(context, node, params, data, input1, input2, output); in Eval()
355 EvalQuantized<kernel_type>(context, node, params, data, input1, input2, in Eval()