Searched refs:select_tensor_node (Results 1 – 1 of 1) sorted by relevance
640 auto* select_tensor_node = graph->create(select_tensor_symbol, 1); in ReplaceWithMaybeCopy() local642 select_tensor_node->addInput(n->input(0)); in ReplaceWithMaybeCopy()644 select_tensor_node->addInput(output); in ReplaceWithMaybeCopy()646 replacement.emplace_back(n, new_node, select_tensor_node); in ReplaceWithMaybeCopy()652 auto* const select_tensor_node = std::get<2>(tup); in ReplaceWithMaybeCopy() local655 select_tensor_node->insertBefore(old_node); in ReplaceWithMaybeCopy()657 select_tensor_node->output()->copyMetadata(old_node->output()); in ReplaceWithMaybeCopy()658 old_node->replaceAllUsesWith(select_tensor_node); in ReplaceWithMaybeCopy()