Searched refs:DataTypeFromString (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | types_test.cc | 61 EXPECT_TRUE(DataTypeFromString(DataTypeString(dt_e), &dt_e2)); in TEST()
63 EXPECT_TRUE(DataTypeFromString(DataTypeString(dt_e_ref), &dt_e2_ref)); in TEST()
85 TEST(TypesTest, DataTypeFromString) { in TEST() argument
87 ASSERT_TRUE(DataTypeFromString("int32", &dt)); in TEST()
89 ASSERT_TRUE(DataTypeFromString("int32_ref", &dt)); in TEST()
91 EXPECT_FALSE(DataTypeFromString("int32_ref_ref", &dt)); in TEST()
92 EXPECT_FALSE(DataTypeFromString("foo", &dt)); in TEST()
93 EXPECT_FALSE(DataTypeFromString("foo_ref", &dt)); in TEST()
94 ASSERT_TRUE(DataTypeFromString("int64", &dt)); in TEST()
96 ASSERT_TRUE(DataTypeFromString("int64_ref", &dt)); in TEST()
[all …]
|
| H A D | types.cc | 156 bool DataTypeFromString(StringPiece sp, DataType* dt) { in DataTypeFromString() function
160 if (DataTypeFromString(sp, &non_ref) && !IsRefType(non_ref)) { in DataTypeFromString()
|
| H A D | op_def_builder.cc | 223 VERIFY(DataTypeFromString(type_string, &dt), in FinalizeAttr()
370 if (DataTypeFromString(type_or_attr, &dt)) { in FinalizeInputOrOutput()
|
| H A D | types.h | 233 bool DataTypeFromString(StringPiece sp, DataType* dt);
|
| H A D | op_def_util.cc | 283 VALIDATE(!DataTypeFromString(attr.name(), &dt), "Attr can't have name ", in ValidateOpDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | strip_unused_nodes.cc | 43 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
64 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/ |
| H A D | cost_utils.cc | 55 if (!DataTypeFromString(parts[0], &data_type)) return tensor_prop; in GetTensorProperties()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
| H A D | generate_testspec.cc | 59 CHECK(DataTypeFromString(input_layer_type[i], &type)); in GenerateInputValues()
|
| H A D | tf_driver.cc | 121 CHECK(DataTypeFromString(input_layer_type[i], &input_types_[i])); in TfDriver()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tfrt_op_kernel.cc | 212 DataTypeFromString(absl::StripAsciiWhitespace(name_type[1]), &data_type); in ParseInputOutputSpec()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | testlib.cc | 54 CHECK(DataTypeFromString(type, &dtype)); in Recv()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/ |
| H A D | benchmark_model.cc | 538 CHECK(DataTypeFromString(input_layer_types[n], &input.data_type)) in Main()
|