Searched refs:tensor_names (Results 1 – 4 of 4) sorted by relevance
88 tensor_names = part.split(',')90 g = graph_in.extract_subgraph(graph_name, tensor_names)91 assert len(g.ops) == len(tensor_names)93 all_tensors += tensor_names
157 auto tensor_names = session->GetOutputTensorNames(); in TEST_F() local160 EXPECT_EQ(tensor_names.size(), 1); in TEST_F()161 …EXPECT_EQ(tensor_names.at(0), "Default/network-WithLossCell/_backbone-LeNet5/fc3-Dense/BiasAdd-op1… in TEST_F()178 auto tensor_names = session->GetOutputTensorNames(); in TEST_F() local180 …EXPECT_EQ(tensor_names.at(0), "Default/network-WithLossCell/_backbone-LeNet5/max_pool2d-MaxPool2d/… in TEST_F()
408 def extract_subgraph(self, graph_name, tensor_names, difference=False): argument411 outputs = set(tensor_names)
198 auto tensor_names = model->GetOutputTensorNames(); in GetOutputByTensorName() local199 for (const auto &tensor_name : tensor_names) { in GetOutputByTensorName()