Home
last modified time | relevance | path

Searched refs:kTfLiteFloat16 (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc242 0, TfLiteType::kTfLiteFloat16, "t0", dims, quantization, false), in InterpreterFp16()
246 2, TfLiteType::kTfLiteFloat16, "t2", dims, quantization, false), in InterpreterFp16()
345 TfLiteType::kTfLiteFloat16); in TEST()
347 TfLiteType::kTfLiteFloat16); in TEST()
862 0, TfLiteType::kTfLiteFloat16, "t0", dims, quantization, false), in InterpreterMultiNode()
866 1, TfLiteType::kTfLiteFloat16, "t1", dims, quantization, false), in InterpreterMultiNode()
870 2, TfLiteType::kTfLiteFloat16, "t2", dims, quantization, false), in InterpreterMultiNode()
972 TfLiteType::kTfLiteFloat16); in TEST()
974 TfLiteType::kTfLiteFloat16); in TEST()
1058 TfLiteType::kTfLiteFloat16); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc59 } else if (weights_->type == kTfLiteFloat16) { in FillCoreMLWeights()
77 } else if (bias_->type == kTfLiteFloat16) { in FillCoreMLBias()
139 return type == kTfLiteFloat32 || type == kTfLiteFloat16; in IsFloatType()
Dconvolution_op_builder.cc176 } else if (weights_->type == kTfLiteFloat16) { in TransposeKernelWeights()
197 } else if (bias_->type == kTfLiteFloat16) { in FillCoreMLBias()
/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc49 EXPECT_EQ(kTfLiteFloat16, in TEST()
50 typeToTfLiteType<TfLiteTypeToType<kTfLiteFloat16>::Type>()); in TEST()
Dportable_type_to_tflitetype.h69 MATCH_TYPE_AND_TFLITE_TYPE(TfLiteFloat16, kTfLiteFloat16);
Doptional_debug_tools.cc74 case kTfLiteFloat16: in TensorTypeName()
Dutil.cc128 case kTfLiteFloat16: in GetSizeOfType()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil_test.cc106 EXPECT_EQ(dst.type, kTfLiteFloat16); in TEST()
112 EXPECT_EQ(TF_HALF, GetTensorFlowDataType(kTfLiteFloat16)); in TEST()
128 EXPECT_EQ(kTfLiteFloat16, GetTensorFlowLiteType(TF_HALF)); in TEST()
Dutil.cc63 case kTfLiteFloat16: in GetTensorFlowDataType()
101 return kTfLiteFloat16; in GetTensorFlowLiteType()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dnumpy.cc39 case kTfLiteFloat16: in TfLiteTypeToPyArrayType()
80 return kTfLiteFloat16; in TfLiteTypeFromPyType()
/external/tensorflow/tensorflow/lite/c/
Dc_api_types.h72 kTfLiteFloat16 = 10, enumerator
Dcommon_test.cc83 EXPECT_EQ(type_name(kTfLiteFloat16), "FLOAT16"); in TEST()
Dcommon.c220 case kTfLiteFloat16: in TfLiteTypeGetName()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc345 1, kTfLiteFloat16, "", {1}, quant, in SetUp()
351 4, kTfLiteFloat16, "", {1}, quant, in SetUp()
357 7, kTfLiteFloat16, "", {1}, quant, in SetUp()
363 10, kTfLiteFloat16, "", {1}, quant, in SetUp()
Dutils.cc224 if (dequantize_input.type == kTfLiteFloat16 && in IsNodeSupported()
/external/tensorflow/tensorflow/lite/swift/Sources/
DTensor.swift95 case kTfLiteFloat16:
/external/tensorflow/tensorflow/lite/tools/serialization/
Denum_mapping.h65 case kTfLiteFloat16: in TfLiteTypeToSchemaType()
/external/tensorflow/tensorflow/lite/kernels/
Ddequantize.h81 case kTfLiteFloat16: { in DequantizeImpl()
Ddequantize.cc61 op_context.input->type == kTfLiteFloat16); in Prepare()
Ddensify.cc85 case kTfLiteFloat16: in Eval()
Dmultinomial.cc104 case kTfLiteFloat16: in MultinomialSample()
Dnumeric_verify.cc96 op_context.input->type == kTfLiteFloat16); in Prepare()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions_test.cc155 EXPECT_EQ(kTfLiteFloat16, type); in TEST_F()
/external/tensorflow/tensorflow/lite/micro/
Dmemory_helpers.cc51 case kTfLiteFloat16: in TfLiteTypeSizeOf()
Dmemory_helpers_test.cc120 tflite::TfLiteTypeSizeOf(kTfLiteFloat16, &size)); in TF_LITE_MICRO_TEST()

12