Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dkernels.cc96 const tensorflow::AttrValue* GetAttrValue(TF_OpKernelConstruction* ctx, in GetAttrValue() function
282 const tensorflow::AttrValue* attr = GetAttrValue(ctx, attr_name, status); in TF_OpKernelConstruction_GetAttrSize()
368 const tensorflow::AttrValue* attr = GetAttrValue(ctx, attr_name, status); \
Dc_test_util.h122 bool GetAttrValue(TF_Operation* oper, const char* attr_name,
Dc_api.cc649 const tensorflow::AttrValue* GetAttrValue(TF_Operation* oper, in GetAttrValue() function
1235 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrMetadata()
1342 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrString()
1360 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrStringList()
1394 const auto* attr = GetAttrValue(oper, attr_name, status); \
1458 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrTensorShapeProto()
1472 const auto* attr = GetAttrValue(oper, attr_name, status); in TF_OperationGetAttrTensorShapeProtoList()
1517 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.cc481 ASSERT_TRUE(GetAttrValue(feed, "dtype", &attr_value, s)) << TF_Message(s); in TEST()
488 ASSERT_FALSE(GetAttrValue(feed, "missing", &attr_value, s)); in TEST()
523 ASSERT_TRUE(GetAttrValue(add, "T", &attr_value, s)) << TF_Message(s); in TEST()
525 ASSERT_TRUE(GetAttrValue(add, "N", &attr_value, s)) << TF_Message(s); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc186 Tensor GetAttrValue(const NodeDef& node) { in GetAttrValue() function in tensorflow::grappler::__anonb8195e580111::LayoutOptimizerTest
247 auto input_sizes = GetAttrValue(*input_sizes_node); in TEST_F()