Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf.cc103 std::unique_ptr<tensorflow::NodeDef> svdf_node(new NodeDef); in CreateNodes() local
104 svdf_node->set_op("Svdf"); in CreateNodes()
105 svdf_node->set_name(name_); in CreateNodes()
106 svdf_node->set_device(device_); in CreateNodes()
109 svdf_node->add_input(inputs_[0]); in CreateNodes()
113 std::string* weights_feature_input = svdf_node->add_input(); in CreateNodes()
114 std::string* weights_time_input = svdf_node->add_input(); in CreateNodes()
117 bias_input = svdf_node->add_input(); in CreateNodes()
141 (*svdf_node->mutable_attr())["ActivationFunction"].set_s(activation_function); in CreateNodes()
142 (*svdf_node->mutable_attr())["Rank"].set_i(rank); in CreateNodes()
[all …]