Searched refs:sub_graph (Results 1 – 3 of 3) sorted by relevance
116 for (auto& sub_graph : model->subgraphs) { in InjectStatsToFullyConnected() local117 for (auto& op : sub_graph->operators) { in InjectStatsToFullyConnected()122 auto& input_tensor = sub_graph->tensors[op->inputs[0]]; in InjectStatsToFullyConnected()128 auto& output_tensor = sub_graph->tensors[op->outputs[0]]; in InjectStatsToFullyConnected()
185 sub_graph = graph_util.extract_sub_graph(graph_def, ["n3"])186 self.assertEqual("n1", sub_graph.node[0].name)187 self.assertEqual("n2", sub_graph.node[1].name)188 self.assertEqual("n3", sub_graph.node[2].name)189 self.assertEqual("n5", sub_graph.node[3].name)
626 TF_ASSIGN_OR_RETURN(auto sub_graph, in ConvertLibFunction()636 GraphToFunctionDef(*sub_graph, function_name, control_ret, &func_def)); in ConvertLibFunction()