Searched refs:svdf_node (Results 1 – 1 of 1) sorted by relevance
103 std::unique_ptr<tensorflow::NodeDef> svdf_node(new NodeDef); in CreateNodes() local104 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 string* weights_feature_input = svdf_node->add_input(); in CreateNodes()114 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 …]