Searched refs:specialized_func_node (Results 1 – 1 of 1) sorted by relevance
586 NodeDef* specialized_func_node) { in RemovePushedDownConstInputs() argument592 const auto& inputs = specialized_func_node->input(); in RemovePushedDownConstInputs()599 specialized_func_node->clear_input(); in RemovePushedDownConstInputs()600 for (const auto& keep : keep_inputs) specialized_func_node->add_input(keep); in RemovePushedDownConstInputs()613 specialized_func_node->add_input(ctrl); in RemovePushedDownConstInputs()622 NodeDef* specialized_func_node) { in RemovePushedDownConstInputTypes() argument631 auto* attr = specialized_func_node->mutable_attr(); in RemovePushedDownConstInputTypes()650 NodeDef* specialized_func_node) { in RemoveUnusedOutputsTypes() argument659 auto* attr = specialized_func_node->mutable_attr(); in RemoveUnusedOutputsTypes()675 NodeDef* specialized_func_node) { in UpdateSpecializedFunctionCallSite() argument[all …]