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()
|