Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc50 const std::string& const_pattern, in FilterPartitionedConstNodes() argument
57 if (StrContains(node->name(), const_pattern) && node->op() == "Const") { in FilterPartitionedConstNodes()
100 for (const std::string& const_pattern : const_node_patterns_) { in CreateNodes() local
101 CreateConstNode(const_pattern); in CreateNodes()
148 void SvdfCluster::CreateConstNode(const std::string& const_pattern) { in CreateConstNode() argument
151 FilterPartitionedConstNodes(const_pattern, nodes_, &const_node_parts); in CreateConstNode()
156 StrContains(const_pattern, "SVDF_weights_feature"); in CreateConstNode()
277 for (const std::string& const_pattern : node_patterns) { in CreateCluster() local
278 cluster->AddConstNodePattern(const_pattern); in CreateCluster()
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);