| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | quantize.cc | 57 input->type == kTfLiteInt16 || in Prepare()
61 output->type == kTfLiteInt16 || in Prepare()
64 if ((input->type == kTfLiteInt16 && output->type == kTfLiteInt8) || in Prepare()
66 (input->type == kTfLiteInt16 && output->type == kTfLiteInt16) || in Prepare()
67 (input->type == kTfLiteInt16 && output->type == kTfLiteInt32)) { in Prepare()
|
| D | softmax_common.cc | 36 input->type == kTfLiteInt16) { in CalculateSoftmaxParams()
40 } else if (input->type == kTfLiteInt16) { in CalculateSoftmaxParams()
46 if (output->type == kTfLiteInt16) { in CalculateSoftmaxParams()
60 if (input->type == kTfLiteInt16) { in CalculateSoftmaxParams()
108 if (input->type == kTfLiteInt16) { in SoftmaxPrepare()
120 if (output->type == kTfLiteInt16) { in SoftmaxPrepare()
123 input->type == kTfLiteInt16); in SoftmaxPrepare()
129 if (input->type == kTfLiteInt16) { in SoftmaxPrepare()
|
| D | leaky_relu.cc | 77 output->type == kTfLiteInt16) { in LeakyReluPrepare()
90 if (input->type == kTfLiteInt16 && output->type == kTfLiteInt16) { in LeakyReluPrepare()
125 case kTfLiteInt16: { in LeakyReluEval()
|
| D | quantize_common.cc | 51 case kTfLiteInt16: in EvalQuantizeReference()
64 } else if (input->type == kTfLiteInt16) { in EvalQuantizeReference()
74 case kTfLiteInt16: in EvalQuantizeReference()
|
| D | softmax.cc | 40 if (output->type == kTfLiteInt16) { in SoftmaxQuantized()
80 case kTfLiteInt16: { in SoftmaxEval()
|
| D | gather.cc | 69 case kTfLiteInt16: in Prepare()
157 case kTfLiteInt16: in Eval()
181 case kTfLiteInt16: in Eval()
|
| D | dequantize.cc | 61 input->type == kTfLiteInt16); in Prepare()
102 case kTfLiteInt16: in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | quantize.cc | 119 output->type == kTfLiteInt16); in Prepare()
122 if (input->type == kTfLiteInt16) { in Prepare()
124 output->type == kTfLiteInt16 || in Prepare()
139 if (input->type == kTfLiteInt16) { in Prepare()
142 if (output->type == kTfLiteInt16) { in Prepare()
180 case kTfLiteInt16: in Eval()
190 case kTfLiteInt16: { in Eval()
201 case kTfLiteInt16: in Eval()
|
| D | mul.cc | 101 output->type == kTfLiteInt16) { in Prepare()
171 input1->type == kTfLiteInt16)) { in EvalQuantized()
201 } else if (input1->type == kTfLiteInt16) { in EvalQuantized()
234 } else if (input1->type == kTfLiteInt16 && input2->type == kTfLiteInt16 && in EvalQuantized()
281 output->type == kTfLiteInt16) { in Eval()
|
| D | add.cc | 135 if (input1->type == kTfLiteInt16 && input2->type == kTfLiteInt16 && in Prepare()
136 output->type == kTfLiteInt16) { in Prepare()
198 } else if (output->type == kTfLiteInt16) { in Prepare()
323 } else if (output->type == kTfLiteInt16) { in EvalAddQuantized()
348 } else if (output->type == kTfLiteInt16) { in EvalAddQuantized()
390 output->type == kTfLiteInt16) { in Eval()
|
| D | sub.cc | 98 output->type == kTfLiteInt16); in PrepareGeneralSubOp()
107 } else if (output->type == kTfLiteInt16) { in PrepareGeneralSubOp()
136 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()
467 output->type == kTfLiteInt16) { in Eval()
|
| D | batch_matmul.cc | 314 if (lhs_data->type == kTfLiteInt8 || lhs_data->type == kTfLiteInt16) { in Prepare()
333 if (lhs_data->type == kTfLiteInt16) { in Prepare()
341 lhs_data->type == kTfLiteInt16); in Prepare()
344 rhs_data->type == kTfLiteInt16); in Prepare()
420 } else if (tensor_in->type == kTfLiteInt16) { in TransposeRowsColumns()
597 } else if (lhs->type == kTfLiteInt16 && rhs->type == kTfLiteInt16) { in EvalQuantized()
616 if (rhs->type == kTfLiteInt8 || rhs->type == kTfLiteInt16) { in GetTempRhs()
631 if (lhs->type == kTfLiteInt8 || lhs->type == kTfLiteInt16) { in GetTempLhs()
704 case kTfLiteInt16: in Eval()
|
| D | activations.cc | 284 input->type == kTfLiteInt16) { in ReluPrepare()
290 if (input->type == kTfLiteInt16) { in ReluPrepare()
361 output->type == kTfLiteInt16) { in LeakyReluPrepare()
374 if (input->type == kTfLiteInt16 && output->type == kTfLiteInt16) { in LeakyReluPrepare()
424 if (input->type == kTfLiteInt16) { in TanhPrepare()
530 } else if (input->type == kTfLiteInt16) { in SigmoidPrepare()
536 if (input->type == kTfLiteInt16) { in SigmoidPrepare()
591 if (output->type == kTfLiteInt16) { in SoftmaxPrepare()
594 input->type == kTfLiteInt16); in SoftmaxPrepare()
615 case kTfLiteInt16: in SoftmaxPrepare()
[all …]
|
| D | dequantize.cc | 60 op_context.input->type == kTfLiteInt16 || in Prepare()
63 if (op_context.input->type == kTfLiteInt16) { in Prepare()
|
| D | concatenation.cc | 58 input_type == kTfLiteInt8 || input_type == kTfLiteInt16 || in Prepare()
101 if (input_type == kTfLiteInt16) { in Prepare()
180 case kTfLiteInt16: in Eval()
|
| D | gather.cc | 69 case kTfLiteInt16: in Prepare()
157 case kTfLiteInt16: in Eval()
181 case kTfLiteInt16: in Eval()
|
| D | depthwise_conv.cc | 126 data_type == kTfLiteInt8 || data_type == kTfLiteInt16); in Prepare()
130 filter->type == data_type || data_type == kTfLiteInt16); in Prepare()
133 if (data_type == kTfLiteInt16) { in Prepare()
146 } else if (data_type == kTfLiteInt16) { in Prepare()
561 case kTfLiteInt16: in EvalImpl()
584 case kTfLiteInt16: in Eval()
585 return EvalImpl<kernel_type, kTfLiteInt16>(context, node); in Eval()
|
| D | reverse.cc | 47 input->type != kTfLiteInt16 && input->type != kTfLiteInt64 && in Prepare()
105 case kTfLiteInt16: { in Eval()
|
| D | cast.cc | 83 case kTfLiteInt16: in copyToTensor()
119 case kTfLiteInt16: in Eval()
|
| D | split.cc | 88 input_type == kTfLiteInt8 || input_type == kTfLiteInt16 || in Prepare()
152 case kTfLiteInt16: { in Eval()
|
| D | unpack.cc | 51 input->type != kTfLiteInt16 && input->type != kTfLiteBool) { in Prepare()
126 case kTfLiteInt16: { in Eval()
|
| /external/tensorflow/tensorflow/lite/ |
| D | type_to_tflitetype_test.cc | 29 EXPECT_EQ(kTfLiteInt16, in TEST()
30 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt16>::Type>()); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 67 case kTfLiteInt16: in GetTensorFlowDataType()
105 return kTfLiteInt16; in GetTensorFlowLiteType()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | softmax.cc | 40 if (output->type == kTfLiteInt16) { in SoftmaxQuantized()
88 case kTfLiteInt16: { in SoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | numpy.cc | 47 case kTfLiteInt16: in TfLiteTypeToPyArrayType()
88 return kTfLiteInt16; in TfLiteTypeFromPyType()
|