Searched refs:add_node_def (Results 1 – 15 of 15) sorted by relevance
35 function_def->add_node_def()->set_op("Mul"); in TEST_F()56 function_def->add_node_def()->set_op("TPUPartitionedCall"); in TEST_F()
601 NodeDef* func_def_node = func->add_node_def(); in MakeFunctionDef()
294 function_def->add_node_def()->set_op(op); in TEST()341 b->add_node_def()->set_op("A"); in TEST()342 c->add_node_def()->set_op("B"); in TEST()344 b->add_node_def()->set_op("B"); in TEST()345 c->add_node_def()->set_op("C"); in TEST()
169 NodeDef* node_def = fdef->add_node_def(); in FillFunctionBody()
1832 *(fdef.add_node_def()) = n.ToNodeDef(); in Create()1897 NodeDef* n = fdef.add_node_def(); in Define()
1457 NodeDef* ndef = fdef2.add_node_def(); in TEST()
65 set_batch_node_attribute(num_batch_threads, function_def->add_node_def()); in AddBatchOp()
53 NodeDef* node = function_def->add_node_def(); in AddFakeSinks()
118 NodeDef* node = fd->add_node_def(); in AddNode()
130 auto* new_node = function.add_node_def(); in TEST()
409 auto* if_node = fused_function->add_node_def(); in LazyConjunctionNodes()
368 auto function_node = branch_func->add_node_def(); in AddBranch()
635 const NodeDef& add_node_def, int port_id, in IsCpuCompatibleDataType() argument648 !HaveSameDataType(&add_node_def, bias_add_node_def) || in IsCpuCompatibleDataType()
610 *error_proto.mutable_function(0)->add_node_def() = in TEST_F()
1403 NodeDef* node = fn.add_node_def(); in TEST_F()1406 node = fn.add_node_def(); in TEST_F()