Searched refs:mutation_filter (Results 1 – 2 of 2) sorted by relevance
15 std::optional<std::function<bool(Node*)>> mutation_filter = std::nullopt)16 : mutation_filter_(std::move(mutation_filter)), in mutation_filter_()75 std::optional<std::function<bool(Node*)>> mutation_filter = std::nullopt);
237 const auto& mutation_filter = *mutation_filter_; in RemoveTensorMutation() local238 if (!mutation_filter(node)) { in RemoveTensorMutation()360 std::optional<std::function<bool(Node*)>> mutation_filter) { in RemoveTensorMutation() argument361 MutationRemover mr(graph, std::move(mutation_filter)); in RemoveTensorMutation()