Searched refs:DataTypeFromString (Results 1 – 13 of 13) 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 | 115 bool DataTypeFromString(StringPiece sp, DataType* dt) { in DataTypeFromString() function
119 if (DataTypeFromString(sp, &non_ref) && !IsRefType(non_ref)) { in DataTypeFromString()
|
| D | op_def_builder.cc | 221 VERIFY(DataTypeFromString(type_string, &dt), in FinalizeAttr()
368 if (DataTypeFromString(type_or_attr, &dt)) { in FinalizeInputOrOutput()
|
| D | types.h | 229 bool DataTypeFromString(StringPiece sp, DataType* dt);
|
| D | op_def_util.cc | 283 VALIDATE(!DataTypeFromString(attr.name(), &dt), "Attr can't have name ", in ValidateOpDef()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | strip_unused_nodes.cc | 43 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
64 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_rewriter_transform.cc | 70 CHECK(DataTypeFromString(data_type_string, &data_type)) in RewriteQuantizedStrippedModelForHexagon()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 55 if (!DataTypeFromString(parts[0], &data_type)) return tensor_prop; in GetTensorProperties()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_testspec.cc | 48 CHECK(DataTypeFromString(input_layer_type[i], &type)); in GenerateInputValues()
|
| D | tf_driver.cc | 120 CHECK(DataTypeFromString(input_layer_type[i], &input_types_[i])); in TfDriver()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform.cc | 128 CHECK(DataTypeFromString(input_types_strs.at(i), &data_type)) in PlaceShapeType()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 54 CHECK(DataTypeFromString(type, &dtype)); in Recv()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 534 CHECK(DataTypeFromString(input_layer_types[n], &input.data_type)) in Main()
|