Searched refs:AttrValueHasType (Results 1 – 12 of 12) 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 | 213 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, ATTR_TYPE)); \
223 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "list(" ATTR_TYPE ")")); \
239 Status s = AttrValueHasType(*attr_value, ATTR_TYPE); \
254 Status s = AttrValueHasType(*attr_value, "list(" ATTR_TYPE ")"); \
337 Status s = AttrValueHasType(*attr_value, "string"); in GetNodeAttrString()
350 Status s = AttrValueHasType(*attr_value, "list(string)"); in TryGetNodeAttr()
367 Status s = AttrValueHasType(*attr_value, "list(shape)"); in TryGetNodeAttr()
382 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "list(type)")); in GetNodeAttr()
393 TF_RETURN_IF_ERROR(AttrValueHasType(*attr_value, "tensor")); in GetNodeAttr()
404 Status s = AttrValueHasType(*attr_value, "tensor"); in TryGetNodeAttr()
[all …]
|
| D | kernel_def_util.cc | 85 Status s = AttrValueHasType(*attr_value, type_str); \ in KernelAttrsMatch()
123 if (!AttrValueHasType(*attr_value, "list(type)").ok()) { in KernelAttrsMatch()
|
| D | attr_value_util.h | 42 Status AttrValueHasType(const AttrValue& attr_value, StringPiece type);
|
| D | op_def_util.cc | 91 TF_RETURN_WITH_CONTEXT_IF_ERROR(AttrValueHasType(attr_value, attr.type()), in ValidateAttrValue()
327 AttrValueHasType(attr.allowed_values(), list_type), " for attr '", in ValidateOpDef()
|
| D | attr_value_util.cc | 299 Status AttrValueHasType(const AttrValue& attr_value, StringPiece type) { in AttrValueHasType() function
|
| D | function.cc | 126 Status status = AttrValueHasType(*v, a.type()); in ValidateSignatureWithAttrs()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 147 TF_RETURN_IF_ERROR(AttrValueHasType(attr_tmp_, ATTR_TYPE)); \
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_ops_test.cc | 162 TF_EXPECT_OK(AttrValueHasType(*kernel_attr, "string")); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util.cc | 202 Status s = AttrValueHasType(*attr_value, "string"); in GetXlaClusterForNode()
|
| D | encapsulate_subgraphs_pass.cc | 681 TF_RETURN_IF_ERROR(AttrValueHasType(node_attr.second, "string")); in GetFunctionNameAttr()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 376 tensorflow::AttrValueHasType(*attr, "list(" attr_type ")"); \
|