Searched refs:kTfLiteUInt8 (Results 1 – 25 of 29) sorted by relevance
12
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | concatenation.cc | 61 input_type == kTfLiteFloat32 || input_type == kTfLiteUInt8); in Prepare() 70 if (input_type == kTfLiteUInt8) { in Prepare() 90 if (input_type == kTfLiteUInt8) { in Prepare() 126 case kTfLiteUInt8: in Eval()
|
D | space_to_depth.cc | 52 data_type == kTfLiteFloat32 || data_type == kTfLiteUInt8 || in Prepare() 94 case kTfLiteUInt8: in Eval()
|
D | pooling.cc | 102 if (input->type == kTfLiteUInt8) { in GenericPrepare() 244 case kTfLiteUInt8: in AverageEval() 266 case kTfLiteUInt8: in MaxEval() 287 case kTfLiteUInt8: in L2Eval()
|
D | activations.cc | 75 if (input->type == kTfLiteUInt8) { in SigmoidPrepare() 109 if (input->type == kTfLiteUInt8) { in SoftmaxPrepare() 215 case kTfLiteUInt8: { in SigmoidEval() 320 case kTfLiteUInt8: { in SoftmaxEval()
|
D | gather.cc | 52 case kTfLiteUInt8: in Prepare() 97 case kTfLiteUInt8: in Eval()
|
D | depthwise_conv.cc | 104 data_type == kTfLiteFloat32 || data_type == kTfLiteUInt8); in Prepare() 110 if (data_type == kTfLiteUInt8) { in Prepare() 245 case kTfLiteUInt8: in Eval()
|
D | split.cc | 80 input_type == kTfLiteFloat32 || input_type == kTfLiteUInt8); in Prepare() 136 case kTfLiteUInt8: { in Eval()
|
D | conv.cc | 130 data_type == kTfLiteFloat32 || data_type == kTfLiteUInt8); in Prepare() 142 if (data_type == kTfLiteUInt8) { in Prepare() 428 case kTfLiteUInt8: in Eval()
|
D | transpose.cc | 122 case kTfLiteUInt8: in Eval()
|
D | hashtable_lookup.cc | 80 TF_LITE_ENSURE_EQ(context, hits->type, kTfLiteUInt8); in Prepare()
|
D | pad.cc | 136 case kTfLiteUInt8: in Eval()
|
D | space_to_batch_nd.cc | 133 case kTfLiteUInt8: in Eval()
|
D | batch_to_space_nd.cc | 136 case kTfLiteUInt8: in Eval()
|
D | mul.cc | 158 } else if (output->type == kTfLiteUInt8) { in Eval()
|
D | add.cc | 174 } else if (output->type == kTfLiteUInt8) { in Eval()
|
D | mean.cc | 206 case kTfLiteUInt8: in Eval()
|
D | fully_connected.cc | 252 case kTfLiteUInt8: in Eval()
|
D | topk_v2.cc | 205 case kTfLiteUInt8: in Eval()
|
/external/tensorflow/tensorflow/contrib/lite/testing/ |
D | tflite_driver.cc | 76 case kTfLiteUInt8: in Check() 190 case kTfLiteUInt8: { in SetInput() 221 case kTfLiteUInt8: in SetExpectation()
|
/external/tensorflow/tensorflow/contrib/lite/java/src/main/native/ |
D | tensor_jni.cc | 65 case kTfLiteUInt8: { in writeOneDimensionalArray() 113 case kTfLiteUInt8: { in readOneDimensionalArray() 161 case kTfLiteUInt8: in elementByteSize()
|
/external/tensorflow/tensorflow/contrib/lite/examples/label_image/ |
D | label_image.cc | 158 case kTfLiteUInt8: in RunInference() 194 case kTfLiteUInt8: in RunInference()
|
/external/tensorflow/tensorflow/contrib/lite/ |
D | optional_debug_tools.cc | 45 case kTfLiteUInt8: in TensorTypeName()
|
D | context.h | 131 kTfLiteUInt8 = 3, enumerator
|
D | interpreter_test.cc | 109 {kTfLiteUInt8, sizeof(uint8_t)}, in TEST() 143 {kTfLiteUInt8, sizeof(uint8_t), reinterpret_cast<const char*>(uint8s)}, in TEST() 182 {kTfLiteUInt8}, in TEST() 214 interpreter.SetTensorParametersReadWrite(i, kTfLiteUInt8, "", {sizes[i]}, in TEST()
|
D | interpreter.h | 49 return kTfLiteUInt8;
|
12