Home
last modified time | relevance | path

Searched refs:GetAttrValue (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_test_util.h122 bool GetAttrValue(TF_Operation* oper, const char* attr_name,
Dc_api.cc647 const tensorflow::AttrValue* GetAttrValue(TF_Operation* oper, in GetAttrValue() function
1233 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrMetadata()
1340 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrString()
1358 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrStringList()
1391 const auto* attr = GetAttrValue(oper, attr_name, status); \
1455 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrTensorShapeProto()
1469 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrTensorShapeProtoList()
1514 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrValueProto()
Dc_test_util.cc424 bool GetAttrValue(TF_Operation* oper, const char* attr_name, in GetAttrValue() function
Dc_api_test.cc478 ASSERT_TRUE(GetAttrValue(feed, "dtype", &attr_value, s)) << TF_Message(s); in TEST()
485 ASSERT_FALSE(GetAttrValue(feed, "missing", &attr_value, s)); in TEST()
520 ASSERT_TRUE(GetAttrValue(add, "T", &attr_value, s)) << TF_Message(s); in TEST()
522 ASSERT_TRUE(GetAttrValue(add, "N", &attr_value, s)) << TF_Message(s); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc180 Tensor GetAttrValue(const NodeDef& node) { in GetAttrValue() function in tensorflow::grappler::__anone834b9fc0111::LayoutOptimizerTest
241 auto input_sizes = GetAttrValue(*input_sizes_node); in TEST_F()