Searched refs:merged_node (Results 1 – 2 of 2) sorted by relevance
159 std::unique_ptr<tensorflow::NodeDef> merged_node(new NodeDef); in CreateConstNode() local160 MaybeMergeConstNodes(const_node_parts, transpose_tensor_value, merged_node); in CreateConstNode()161 new_nodes_.push_back(std::move(merged_node)); in CreateConstNode()167 const std::unique_ptr<tensorflow::NodeDef>& merged_node) { in MaybeMergeConstNodes() argument168 merged_node->set_name(const_node_parts[0]->name()); in MaybeMergeConstNodes()169 merged_node->set_op("Const"); in MaybeMergeConstNodes()170 merged_node->set_device(const_node_parts[0]->device()); in MaybeMergeConstNodes()171 (*merged_node->mutable_attr())["dtype"].set_type( in MaybeMergeConstNodes()184 (*merged_node->mutable_attr())["value"].mutable_tensor(); in MaybeMergeConstNodes()
57 const std::unique_ptr<tensorflow::NodeDef>& merged_node);