Searched refs:specialized_func_node (Results 1 – 1 of 1) sorted by relevance
498 NodeDef* specialized_func_node) { in RemovePushedDownConstInputs() argument504 const auto& inputs = specialized_func_node->input(); in RemovePushedDownConstInputs()510 specialized_func_node->clear_input(); in RemovePushedDownConstInputs()511 for (const auto& keep : keep_inputs) specialized_func_node->add_input(keep); in RemovePushedDownConstInputs()524 specialized_func_node->add_input(ctrl); in RemovePushedDownConstInputs()533 NodeDef* specialized_func_node) { in RemovePushedDownConstInputTypes() argument542 auto* attr = specialized_func_node->mutable_attr(); in RemovePushedDownConstInputTypes()560 NodeDef* specialized_func_node) { in RemoveUnusedOutputsTypes() argument570 auto* attr = specialized_func_node->mutable_attr(); in RemoveUnusedOutputsTypes()585 NodeDef* specialized_func_node) { in UpdateSpecializedFunctionCallSite() argument[all …]