Searched defs:GetNodeAttr (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/util/ |
H A D | mirror_pad_mode.cc | 24 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 D | padding.cc | 24 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 D | hlo_tfgraph_builder_test.cc | 56 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 D | node_def_util.cc | 159 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