Home
last modified time | relevance | path

Searched refs:kTfLiteInt16 (Results 1 – 25 of 83) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc62 if (op_sig.inputs.at(0).type == kTfLiteInt16 && in GetBuiltinOperatorVersion()
63 op_sig.inputs.at(1).type == kTfLiteInt16 && in GetBuiltinOperatorVersion()
64 op_sig.outputs.at(1).type == kTfLiteInt16) { in GetBuiltinOperatorVersion()
89 if (op_sig.inputs.at(0).type == kTfLiteInt16 && in GetBuiltinOperatorVersion()
90 op_sig.inputs.at(1).type == kTfLiteInt16 && in GetBuiltinOperatorVersion()
91 op_sig.outputs.at(1).type == kTfLiteInt16) { in GetBuiltinOperatorVersion()
148 if (op_sig.inputs.at(0).type == kTfLiteInt16 && in GetBuiltinOperatorVersion()
149 op_sig.inputs.at(1).type == kTfLiteInt16 && in GetBuiltinOperatorVersion()
150 op_sig.outputs.at(0).type == kTfLiteInt16) { in GetBuiltinOperatorVersion()
200 if (op_sig.inputs.at(0).type == kTfLiteInt16) { in GetBuiltinOperatorVersion()
[all …]
Dop_version_test.cc153 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in SimpleVersioningTestExtended()
246 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in TEST()
322 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16, 5); in TEST()
327 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in TEST()
329 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16, 4); in TEST()
421 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16); in TEST()
435 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in TEST()
436 .outputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in TEST()
448 .inputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in TEST()
449 .outputs = CreateOpSignatureTensorSpecs(kTfLiteInt16), in TEST()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dtable.cc42 input->type == kTfLiteInt8 || input->type == kTfLiteInt16); in Prepare()
44 output->type == kTfLiteInt8 || output->type == kTfLiteInt16); in Prepare()
47 if (input->type == kTfLiteInt16) { in Prepare()
50 if (output->type == kTfLiteInt16) { in Prepare()
58 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteInt16); in Prepare()
88 case kTfLiteInt16: in EvalTable()
110 case kTfLiteInt16: in Eval()
Dquantize.cc123 output->type == kTfLiteInt16); in Prepare()
126 if (input->type == kTfLiteInt16) { in Prepare()
128 output->type == kTfLiteInt16 || in Prepare()
132 context, output->type == kTfLiteInt8 || output->type == kTfLiteInt16); in Prepare()
146 if (input->type == kTfLiteInt16 && output->type == kTfLiteInt16) { in Prepare()
195 case kTfLiteInt16: in Eval()
221 case kTfLiteInt16: in Eval()
246 case kTfLiteInt16: in Eval()
259 case kTfLiteInt16: { in Eval()
270 case kTfLiteInt16: in Eval()
Dsub.cc99 output->type == kTfLiteInt16); in PrepareGeneralSubOp()
108 } else if (output->type == kTfLiteInt16) { in PrepareGeneralSubOp()
137 op_params->left_shift = output->type == kTfLiteInt16 ? 15 : 20; in PrepareGeneralSubOp()
261 if (input1->type == kTfLiteInt16 && input2->type == kTfLiteInt16 && in Prepare()
262 output->type == kTfLiteInt16) { in Prepare()
291 } else if (output->type == kTfLiteInt16) { in Prepare()
457 output->type == kTfLiteInt16) { in Eval()
Dadd.cc135 if (input1->type == kTfLiteInt16 && input2->type == kTfLiteInt16 && in Prepare()
136 output->type == kTfLiteInt16) { in Prepare()
198 } else if (output->type == kTfLiteInt16) { in Prepare()
337 } else if (output->type == kTfLiteInt16) { in EvalAddQuantized()
366 } else if (output->type == kTfLiteInt16) { in EvalAddQuantized()
409 output->type == kTfLiteInt16) { in Eval()
Dmul.cc103 output->type == kTfLiteInt16) { in Prepare()
193 input1->type == kTfLiteInt16)) { in EvalQuantized()
223 } else if (input1->type == kTfLiteInt16) { in EvalQuantized()
256 } else if (input1->type == kTfLiteInt16 && input2->type == kTfLiteInt16 && in EvalQuantized()
308 output->type == kTfLiteInt16) { in Eval()
Dactivations.cc274 input->type == kTfLiteInt16) { in ReluPrepare()
280 if (input->type == kTfLiteInt16) { in ReluPrepare()
351 output->type == kTfLiteInt16) { in LeakyReluPrepare()
364 if (input->type == kTfLiteInt16 && output->type == kTfLiteInt16) { in LeakyReluPrepare()
414 if (input->type == kTfLiteInt16) { in TanhPrepare()
520 } else if (input->type == kTfLiteInt16) { in SigmoidPrepare()
526 if (input->type == kTfLiteInt16) { in SigmoidPrepare()
589 } else if (input->type == kTfLiteInt16 && output->type == kTfLiteInt16) { in SoftmaxPrepare()
621 case kTfLiteInt16: in SoftmaxPrepare()
631 } else if (input->type == kTfLiteInt16) { in SoftmaxPrepare()
[all …]
Dbatch_matmul.cc317 if ((lhs_data->type == kTfLiteInt8 || lhs_data->type == kTfLiteInt16) && in Prepare()
337 if (lhs_data->type == kTfLiteInt16) { in Prepare()
345 lhs_data->type == kTfLiteInt16); in Prepare()
348 rhs_data->type == kTfLiteInt16); in Prepare()
424 } else if (tensor_in->type == kTfLiteInt16) { in TransposeRowsColumns()
652 } else if (lhs->type == kTfLiteInt16 && rhs->type == kTfLiteInt16) { in EvalQuantized()
671 if (rhs->type == kTfLiteInt8 || rhs->type == kTfLiteInt16) { in GetTempRhs()
686 if (lhs->type == kTfLiteInt8 || lhs->type == kTfLiteInt16) { in GetTempLhs()
759 case kTfLiteInt16: in Eval()
Ddequantize.cc60 op_context.input->type == kTfLiteInt16 || in Prepare()
63 if (op_context.input->type == kTfLiteInt16) { in Prepare()
Dreduce.cc218 case kTfLiteInt16: in InitializeTemporaries()
244 if (op_context.input->type == kTfLiteInt16) { in PrepareSimple()
292 op_context.input->type == kTfLiteInt16) { in PrepareMeanOrSum()
301 if (op_context.input->type == kTfLiteInt16) { in PrepareMeanOrSum()
336 if (op_context.input->type == kTfLiteInt16) { in PrepareProd()
351 op_context.input->type == kTfLiteInt16) && in PrepareProd()
478 case kTfLiteInt16: in InitializeMeanOutput()
594 case kTfLiteInt16: { in EvalMean()
719 input->type == kTfLiteInt16) { in EvalType()
837 case kTfLiteInt16: in EvalGeneric()
[all …]
Dconcatenation.cc102 case kTfLiteInt16: in EvalImpl()
138 input_type == kTfLiteInt8 || input_type == kTfLiteInt16 || in Prepare()
185 if (input_type == kTfLiteInt16) { in Prepare()
Ddepthwise_conv.cc128 data_type == kTfLiteInt8 || data_type == kTfLiteInt16); in Prepare()
132 filter->type == data_type || data_type == kTfLiteInt16); in Prepare()
135 if (data_type == kTfLiteInt16) { in Prepare()
148 } else if (data_type == kTfLiteInt16) { in Prepare()
567 case kTfLiteInt16: in EvalImpl()
590 case kTfLiteInt16: in Eval()
591 return EvalImpl<kernel_type, kTfLiteInt16>(context, node); in Eval()
Dreverse.cc47 input->type != kTfLiteInt16 && input->type != kTfLiteInt64 && in Prepare()
105 case kTfLiteInt16: { in Eval()
Dgather.cc69 case kTfLiteInt16: in Prepare()
197 case kTfLiteInt16: in Eval()
228 case kTfLiteInt16: in Eval()
Dsplit.cc89 input_type == kTfLiteInt8 || input_type == kTfLiteInt16 || in Prepare()
153 case kTfLiteInt16: { in Eval()
Dreverse_sequence.cc47 input->type != kTfLiteUInt8 && input->type != kTfLiteInt16 && in Prepare()
143 case kTfLiteInt16: { in Eval()
Dunpack.cc51 input->type != kTfLiteInt16 && input->type != kTfLiteBool) { in Prepare()
126 case kTfLiteInt16: { in Eval()
Dfill.cc99 if (value->type == kTfLiteInt16) { in Prepare()
146 case kTfLiteInt16: in Eval()
Dtranspose_conv.cc153 input_type == kTfLiteInt16) { in AllocateTemporaryTensorsIfRequired()
272 input->type == kTfLiteInt8 || input->type == kTfLiteInt16); in Prepare()
282 } else if (input->type == kTfLiteInt16) { in Prepare()
294 if (input->type == kTfLiteInt16) { in Prepare()
350 input->type == kTfLiteInt16) { in Prepare()
357 if (input->type == kTfLiteInt16 && bias && bias->type == kTfLiteInt64) { in Prepare()
698 case kTfLiteInt16: { in Eval()
Delementwise.cc58 return type == kTfLiteFloat32 || type == kTfLiteInt8 || type == kTfLiteInt16; in IsAbsSupportedType()
95 (input->type == kTfLiteInt16 && in GenericPrepare()
117 if (input->type == kTfLiteInt16) { in GenericPrepare()
233 case kTfLiteInt16: in AbsEval()
Dpack.cc52 input0->type != kTfLiteInt16 && input0->type != kTfLiteInt64) { in Prepare()
130 case kTfLiteInt16: { in Eval()
/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc29 EXPECT_EQ(kTfLiteInt16, in TEST()
30 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt16>::Type>()); in TEST()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc79 case kTfLiteInt16: in GetTensorFlowDataType()
119 return kTfLiteInt16; in GetTensorFlowLiteType()
154 case kTfLiteInt16: in TfLiteTypeToTfTypeName()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dnumpy.cc49 case kTfLiteInt16: in TfLiteTypeToPyArrayType()
90 return kTfLiteInt16; in TfLiteTypeFromPyType()

1234