Home
last modified time | relevance | path

Searched refs:kTfLiteBool (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc77 case kTfLiteBool: in GetTensorFlowDataType()
101 return kTfLiteBool; in GetTensorFlowLiteType()
Dutil_test.cc118 EXPECT_EQ(TF_BOOL, GetTensorFlowDataType(kTfLiteBool)); in TEST()
129 EXPECT_EQ(kTfLiteBool, GetTensorFlowLiteType(TF_BOOL)); in TEST()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.cc187 case kTfLiteBool: in Check()
299 case kTfLiteBool: { in SetInput()
335 case kTfLiteBool: in SetExpectation()
403 case kTfLiteBool: in ReadOutput()
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver.cc187 case kTfLiteBool: in Check()
298 case kTfLiteBool: { in SetInput()
341 case kTfLiteBool: in SetExpectation()
406 case kTfLiteBool: in ReadOutput()
/external/tensorflow/tensorflow/lite/kernels/
Dselect.cc43 TF_LITE_ENSURE(context, input_condition->type == kTfLiteBool); in SelectPrepare()
83 case kTfLiteBool: \ in SelectEval()
Dcomparisons.cc42 input1->type != kTfLiteString || input1->type != kTfLiteBool); in ComparisonPrepare()
45 output->type = kTfLiteBool; in ComparisonPrepare()
132 case kTfLiteBool: in EqualEval()
168 case kTfLiteBool: in NotEqualEval()
Dcast.cc87 case kTfLiteBool: in copyToTensor()
115 case kTfLiteBool: in Eval()
Delementwise.cc33 return type == kTfLiteBool; in IsLogicalSupportedType()
75 return EvalImpl<bool>(context, node, bool_func, kTfLiteBool); in EvalLogical()
Done_hot.cc131 case kTfLiteBool: in Prepare()
175 case kTfLiteBool: in Eval()
Dsubgraph_test_util.cc166 SetupTensor(subgraph, kCondInput, kTfLiteBool); in BuildIfSubgraph()
208 SetupTensor(subgraph, kOutput, kTfLiteBool); in BuildLessEqualCondSubgraph()
400 ASSERT_EQ(tensor->type, kTfLiteBool); in CheckBoolTensor()
Dreduce.cc198 case kTfLiteBool: in InitializeTemporaries()
199 temp_sum->type = kTfLiteBool; in InitializeTemporaries()
231 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteBool); in PrepareAny()
513 case kTfLiteBool: in EvalGeneric()
Dwhere.cc61 if (cond_tensor->type != kTfLiteBool) { in Prepare()
Dlogical.cc61 if (type != kTfLiteBool) { in Prepare()
Dtile.cc185 case kTfLiteBool: in Eval()
Dif.cc54 TF_LITE_ENSURE_EQ(context, cond->type, kTfLiteBool); in Prepare()
Dslice.cc214 case kTfLiteBool: in Eval()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dpython_utils.cc47 case kTfLiteBool: in TfLiteTypeToPyArrayType()
74 return kTfLiteBool; in TfLiteTypeFromPyArray()
/external/tensorflow/tensorflow/lite/experimental/swift/Sources/
DTensor.swift90 case kTfLiteBool:
/external/tensorflow/tensorflow/lite/experimental/writer/
Denum_mapping.h73 case kTfLiteBool: in TfLiteTypeToSchemaType()
/external/tensorflow/tensorflow/lite/
Doptional_debug_tools.cc53 case kTfLiteBool: in TensorTypeName()
/external/tensorflow/tensorflow/lite/c/
Dc_api_internal_test.cc86 EXPECT_EQ(type_name(kTfLiteBool), "BOOL"); in TEST()
Dc_api_internal.c169 case kTfLiteBool: in TfLiteTypeGetName()
Dc_api_internal.h203 kTfLiteBool = 6, enumerator
/external/tensorflow/tensorflow/lite/experimental/micro/
Dsimple_tensor_allocator.cc41 case kTfLiteBool: in TfLiteTypeSizeOf()
/external/libtextclassifier/utils/
Dtflite-model-executor.h99 case kTfLiteBool: in SetInput()

12