Searched refs:const_node_parts (Results 1 – 2 of 2) sorted by relevance
52 std::vector<const NodeDef*>* const_node_parts) { in FilterPartitionedConstNodes() argument59 const_node_parts->push_back(node); in FilterPartitionedConstNodes()69 sort(const_node_parts->begin(), const_node_parts->end(), in FilterPartitionedConstNodes()150 std::vector<const NodeDef*> const_node_parts; in CreateConstNode() local151 FilterPartitionedConstNodes(const_pattern, nodes_, &const_node_parts); in CreateConstNode()153 if (const_node_parts.empty()) return; in CreateConstNode()160 MaybeMergeConstNodes(const_node_parts, transpose_tensor_value, merged_node); in CreateConstNode()165 const std::vector<const NodeDef*>& const_node_parts, in MaybeMergeConstNodes() argument168 merged_node->set_name(const_node_parts[0]->name()); in MaybeMergeConstNodes()170 merged_node->set_device(const_node_parts[0]->device()); in MaybeMergeConstNodes()[all …]
55 const std::vector<const tensorflow::NodeDef*>& const_node_parts,