Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc41 EXPECT_EQ(kTfLiteBool, in TEST()
42 typeToTfLiteType<TfLiteTypeToType<kTfLiteBool>::Type>()); in TEST()
57 EXPECT_EQ(kTfLiteBool, typeToTfLiteType<bool>()); in TEST()
62 static_assert(std::is_same<bool, TfLiteTypeToType<kTfLiteBool>::Type>::value, in TEST()
Dportable_type_to_tflitetype.h68 MATCH_TYPE_AND_TFLITE_TYPE(bool, kTfLiteBool);
Doptional_debug_tools.cc66 case kTfLiteBool: in TensorTypeName()
Dutil.cc113 case kTfLiteBool: in GetSizeOfType()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc87 case kTfLiteBool: in GetTensorFlowDataType()
123 return kTfLiteBool; in GetTensorFlowLiteType()
Dutil_test.cc122 EXPECT_EQ(TF_BOOL, GetTensorFlowDataType(kTfLiteBool)); in TEST()
139 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/python/interpreter_wrapper/
Dnumpy.cc59 case kTfLiteBool: in TfLiteTypeToPyArrayType()
98 return kTfLiteBool; in TfLiteTypeFromPyType()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dgather.cc72 case kTfLiteBool: in Prepare()
163 case kTfLiteBool: in Eval()
187 case kTfLiteBool: in Eval()
Delementwise.cc35 return type == kTfLiteBool; in IsLogicalSupportedType()
78 return EvalImpl<bool>(context, node, bool_func, kTfLiteBool); in EvalLogical()
/external/tensorflow/tensorflow/lite/kernels/
Dgather.cc72 case kTfLiteBool: in Prepare()
163 case kTfLiteBool: in Eval()
187 case kTfLiteBool: in Eval()
Dcast.cc92 case kTfLiteBool: in copyToTensor()
126 case kTfLiteBool: in Eval()
Dreverse.cc48 input->type != kTfLiteBool) { in Prepare()
123 case kTfLiteBool: { in Eval()
Dunpack.cc51 input->type != kTfLiteInt16 && input->type != kTfLiteBool) { in Prepare()
122 case kTfLiteBool: { in Eval()
Dselect.cc78 TF_LITE_ENSURE_TYPES_EQ(context, input_condition->type, kTfLiteBool); in SelectPrepare()
144 case kTfLiteBool: \ in SelectEval()
Done_hot.cc135 case kTfLiteBool: in Prepare()
186 case kTfLiteBool: in Eval()
Dconcatenation.cc60 input_type == kTfLiteBool); in Prepare()
183 case kTfLiteBool: in Eval()
Dwhere.cc66 if (cond_tensor->type != kTfLiteBool) { in Prepare()
Dcomparisons.cc60 output->type = kTfLiteBool; in ComparisonPrepareCommon()
165 case kTfLiteBool: in EqualEval()
215 case kTfLiteBool: in NotEqualEval()
Dsubgraph_test_util.cc204 SetupTensor(subgraph, kCondInput, kTfLiteBool); in BuildIfSubgraph()
243 SetupTensor(subgraph, kOutput, kTfLiteBool); in BuildLessEqualCondSubgraph()
475 SetupTensor(subgraph, kOutput, kTfLiteBool); in BuildLessEqualCondSubgraphWithDynamicTensor()
662 ASSERT_EQ(tensor->type, kTfLiteBool); in CheckBoolTensor()
/external/tensorflow/tensorflow/lite/c/
Dc_api_types.h68 kTfLiteBool = 6, enumerator
/external/tensorflow/tensorflow/lite/swift/Sources/
DTensor.swift85 case kTfLiteBool:
/external/tensorflow/tensorflow/lite/tools/serialization/
Denum_mapping.h83 case kTfLiteBool: in TfLiteTypeToSchemaType()
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dtensor_jni.cc92 case kTfLiteBool: in ElementByteSize()
145 case kTfLiteBool: { in WriteOneDimensionalArray()
201 case kTfLiteBool: { in ReadOneDimensionalArray()
372 case kTfLiteBool: { in WriteScalar()
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver.cc348 case kTfLiteBool: in Check()
526 case kTfLiteBool: { in SetInput()
615 case kTfLiteBool: in SetExpectation()
718 case kTfLiteBool: in ReadOutput()

123