HomeSort by relevance Sort by last modified time
    Searched defs:GetNodeAttr (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/util/
mirror_pad_mode.cc 24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
padding.cc 24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_tfgraph_builder_test.cc 56 static const tensorflow::AttrValue &GetNodeAttr(const tensorflow::NodeDef &node,
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/
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,
275 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
284 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
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 114 milliseconds