Searched defs:map_defun_node (Results 1 – 3 of 3) sorted by relevance
69 void UpdateMapDefunAttrs(FunctionBody* map_defun_fn, Node* map_defun_node) { in UpdateMapDefunAttrs()78 Status AddMapDefunOutput(FunctionBody* map_defun_fn, Node* map_defun_node, in AddMapDefunOutput()102 FunctionBody* map_defun_fn, Node* map_defun_node) { in RemoveMapDefunOutput()333 const NodeDef& map_defun_node, in Vectorize()368 const NodeDef& map_defun_node) { in Initialize()644 const NodeDef& map_defun_node, FunctionDefLibrary* lib, in VectorizeMapDefun()
70 Node* map_defun_node; in WrapFunctionWithMapDefun() local100 const NodeDef& map_defun_node = outer.node_def(0); in WrapAndVectorize() local252 auto map_defun_node = vectorized->node_def( in TEST() local477 const NodeDef& map_defun_node = vectorized->node_def(0); in TEST() local
70 NodeDef* map_defun_node = vectorized_func->mutable_node_def()->Add(); in CreateMapDefunWrapper() local119 const NodeDef& map_defun_node = vectorized_func->node_def(0); in AddVectorizedFunction() local