Home
last modified time | relevance | path

Searched refs:const_pattern (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_svdf.cc51 const std::string& const_pattern, in FilterPartitionedConstNodes() argument
58 if (StrContains(node->name(), const_pattern) && node->op() == "Const") { in FilterPartitionedConstNodes()
101 for (const std::string& const_pattern : const_node_patterns_) { in CreateNodes() local
102 CreateConstNode(const_pattern); in CreateNodes()
149 void SvdfCluster::CreateConstNode(const std::string& const_pattern) { in CreateConstNode() argument
152 FilterPartitionedConstNodes(const_pattern, nodes_, &const_node_parts); in CreateConstNode()
157 StrContains(const_pattern, "SVDF_weights_feature"); in CreateConstNode()
278 for (const std::string& const_pattern : node_patterns) { in CreateCluster() local
279 cluster->AddConstNodePattern(const_pattern); in CreateCluster()
H A Dresolve_svdf.h39 void AddConstNodePattern(const std::string& const_pattern) { in AddConstNodePattern() argument
40 const_node_patterns_.push_back(const_pattern); in AddConstNodePattern()
49 void CreateConstNode(const std::string& const_pattern);