Searched refs:value_attr (Results 1 – 2 of 2) sorted by relevance
63 const auto& value_attr = node->attr().at("value"); in FilterPartitionedConstNodes() local64 const tensorflow::TensorProto& tensor = value_attr.tensor(); in FilterPartitionedConstNodes()83 const auto& value_attr = node->attr().at("value"); in InferFilterRank() local84 const tensorflow::TensorProto& tensor = value_attr.tensor(); in InferFilterRank()190 const auto& value_attr = const_node_parts[i]->attr().at("value"); in MaybeMergeConstNodes() local191 const tensorflow::TensorProto& tensor = value_attr.tensor(); in MaybeMergeConstNodes()218 const auto& value_attr = const_node_parts[i]->attr().at("value"); in MaybeMergeConstNodes() local219 const tensorflow::TensorProto& tensor = value_attr.tensor(); in MaybeMergeConstNodes()
156 AttrValue value_attr; in BuildConstNode() local157 SetAttrValue(t, &value_attr); in BuildConstNode()158 (*node->mutable_attr())["value"] = value_attr; in BuildConstNode()454 AttrValue value_attr; in NodeWithTensorContent() local455 SetAttrValue(t, &value_attr); in NodeWithTensorContent()456 (*node->mutable_attr())["value"] = value_attr; in NodeWithTensorContent()620 AttrValue value_attr; in TEST() local621 SetAttrValue(2, &value_attr); in TEST()622 (*node.mutable_attr())["Nsparse"] = value_attr; in TEST()628 AttrValue value_attr; in TEST() local[all …]