Searched defs:GetNodeAttr (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
H A Dmirror_pad_mode.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, function in namespace:tensorflow
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
H A Dpadding.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, function in namespace:tensorflow
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_tfgraph_builder_test.cc56 static const tensorflow::AttrValue &GetNodeAttr(const tensorflow::NodeDef &node, function in namespace:xla::hlo_graph_dumper::__anon25654
79 auto dims_value = GetNodeAttr(node, "dims");
84 auto shape_value = GetNodeAttr(node, "_output_shapes");
103 EXPECT_EQ(GetNodeAttr(node, "value").s(), "123");
104 EXPECT_EQ(GetNodeAttr(node, "type").s(), "S32");
105 EXPECT_EQ(GetNodeAttr(node, "tf_op_name").s(), "x");
106 EXPECT_EQ(GetNodeAttr(node, "tf_op_type").s(), "y");
/external/tensorflow/tensorflow/core/framework/
H A Dnode_def_util.cc159 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
169 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
264 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, function in namespace:tensorflow
275 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, function in namespace:tensorflow
284 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, function in namespace:tensorflow
301 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.number_attr(), &repeats));
309 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.type_attr(), &dtype));
487 return GetNodeAttr(node_def, arg_def.number_attr(), num);

Completed in 243 milliseconds