Searched refs:AttrValueHasType (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | attr_value_util_test.cc | 64 EXPECT_TRUE(AttrValueHasType(V(123), "int").ok()); in TEST()
65 EXPECT_TRUE(AttrValueHasType(V(1.2), "float").ok()); in TEST()
66 EXPECT_TRUE(AttrValueHasType(V(DT_FLOAT), "type").ok()); in TEST()
67 EXPECT_TRUE(AttrValueHasType(F("f", {}), "func").ok()); in TEST()
68 EXPECT_TRUE(AttrValueHasType(Fs({{"f", {}}, {"g", {}}}), "list(func)").ok()); in TEST()
71 EXPECT_FALSE(AttrValueHasType(V(123), "func").ok()); in TEST()
72 EXPECT_FALSE(AttrValueHasType(V(1.2), "int").ok()); in TEST()
73 EXPECT_FALSE(AttrValueHasType(V(DT_FLOAT), "shape").ok()); in TEST()
74 EXPECT_FALSE(AttrValueHasType(F("f", {}), "string").ok()); in TEST()
75 EXPECT_FALSE(AttrValueHasType(P("T"), "float").ok()); in TEST()
[all …]
|
| D | node_def_util.cc | 215 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, ATTR_TYPE)); \
225 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "list(" ATTR_TYPE ")")); \
240 Status s = AttrValueHasType(*attr_value, ATTR_TYPE); \
255 Status s = AttrValueHasType(*attr_value, "list(" ATTR_TYPE ")"); \
309 Status s = AttrValueHasType(*attr_value, "string"); in GetNodeAttrString()
320 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "list(type)")); in GetNodeAttr()
331 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "tensor")); in GetNodeAttr()
340 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "func")); in GetNodeAttr()
|
| D | kernel_def_util.cc | 55 if (!AttrValueHasType(*found, "list(type)").ok()) { in KernelAttrsMatch()
|
| D | attr_value_util.h | 42 Status AttrValueHasType(const AttrValue& attr_value, StringPiece type);
|
| D | op_def_util.cc | 90 TF_RETURN_WITH_CONTEXT_IF_ERROR(AttrValueHasType(attr_value, attr.type()), in ValidateAttrValue()
313 AttrValueHasType(attr.allowed_values(), list_type), " for attr '", in ValidateOpDef()
|
| D | attr_value_util.cc | 276 Status AttrValueHasType(const AttrValue& attr_value, StringPiece type) { in AttrValueHasType() function
|
| D | function.cc | 111 Status status = AttrValueHasType(*v, a.type()); in ValidateSignatureWithAttrs()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util.cc | 209 Status s = AttrValueHasType(*attr_value, "string"); in GetXlaClusterForNode()
|
| D | encapsulate_subgraphs_pass.cc | 1392 TF_RETURN_IF_ERROR(AttrValueHasType(node_attr.second, "string")); in GetFunctionNameAttr()
1396 TF_RETURN_IF_ERROR(AttrValueHasType(node_attr.second, "string")); in GetFunctionNameAttr()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_ops_test.cc | 162 TF_EXPECT_OK(AttrValueHasType(*kernel_attr, "string")); in TEST()
|