Searched refs:nodes_to_remove (Results 1 – 3 of 3) sorted by relevance
181 std::vector<string> nodes_to_remove; in TEST_P() local182 nodes_to_remove.reserve(noop_nodes.size()); in TEST_P()187 nodes_to_remove.push_back(node->name()); in TEST_P()196 for (const auto &noop_node_name : nodes_to_remove) in TEST_P()
267 std::vector<Node*> nodes_to_remove = {launch}; in BuildXlaLaunchOps() local304 nodes_to_remove.push_back(output_node); in BuildXlaLaunchOps()334 for (Node* node : nodes_to_remove) { in BuildXlaLaunchOps()
790 std::vector<Node*> arg_nodes_to_update, nodes_to_remove; in RemoveUnusedXlaInput() local820 nodes_to_remove.push_back(n); in RemoveUnusedXlaInput()822 for (Node* n : nodes_to_remove) { in RemoveUnusedXlaInput()1379 std::vector<Node*> ret_nodes_to_update, nodes_to_remove; in RemoveUnusedXlaOutput() local1399 nodes_to_remove.push_back(e->src()); in RemoveUnusedXlaOutput()1400 nodes_to_remove.push_back(n); in RemoveUnusedXlaOutput()1402 for (Node* n : nodes_to_remove) { in RemoveUnusedXlaOutput()2697 std::vector<Node*> nodes_to_remove = {replicate}; in BuildTPUReplicateOps() local2722 nodes_to_remove.push_back(in_edges[i]->src()); in BuildTPUReplicateOps()2738 nodes_to_remove.push_back(in_edges[num_replicated_inputs + i]->src()); in BuildTPUReplicateOps()[all …]