Searched refs:DataTypeFromString (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| 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 …]
|
| D | types.cc | 116 bool DataTypeFromString(StringPiece sp, DataType* dt) { in DataTypeFromString() function
120 if (DataTypeFromString(sp, &non_ref) && !IsRefType(non_ref)) { in DataTypeFromString()
|
| D | op_def_builder.cc | 214 VERIFY(DataTypeFromString(type_string, &dt), in FinalizeAttr()
351 if (DataTypeFromString(type_or_attr, &dt)) { in FinalizeInputOrOutput()
|
| D | types.h | 230 bool DataTypeFromString(StringPiece sp, DataType* dt);
|
| D | op_def_util.cc | 256 VALIDATE(!DataTypeFromString(attr.name(), &dt), "Attr can't have name ", in ValidateOpDef()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | strip_unused_nodes.cc | 45 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
66 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_rewriter_transform.cc | 65 CHECK(DataTypeFromString(data_type_string, &data_type)) in RewriteQuantizedStrippedModelForHexagon()
|
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| D | tf_driver.cc | 77 CHECK(DataTypeFromString(input_layer_type[i], &input_types_[i])); in TfDriver()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 122 CHECK(DataTypeFromString(input_types_strs.at(i), &data_type)) in PlaceShapeType()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 494 CHECK(DataTypeFromString(input_layer_types[n], &input.data_type)) in Main()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 77 CHECK(DataTypeFromString(type, &dtype)); in Recv()
|