Searched refs:call_builder (Results 1 – 3 of 3) sorted by relevance
1250 NodeDefBuilder call_builder(absl::StrCat("oc_call_", func_call_node_name), in BuildHostGraphForFuncCallNode() local1252 call_builder.Input(key_placeholder->name(), 0, DT_STRING); in BuildHostGraphForFuncCallNode()1253 call_builder.Attr("device_ordinal", device_ordinal_value); in BuildHostGraphForFuncCallNode()1254 call_builder.Attr(kXlaHasHostTransferAttrName, true); in BuildHostGraphForFuncCallNode()1256 TF_RETURN_IF_ERROR(call_builder.Finalize(&call_def)); in BuildHostGraphForFuncCallNode()
1211 auto call_builder = HloComputation::Builder(TestName() + "_call"); in TEST_F() local1212 auto call_param = call_builder.AddInstruction( in TEST_F()1214 auto call_root = call_builder.AddInstruction( in TEST_F()1216 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F()
3774 HloComputation::Builder call_builder(TestName() + ".Call"); in TEST_F() local3775 HloInstruction* zero = call_builder.AddInstruction( in TEST_F()3777 HloInstruction* one = call_builder.AddInstruction( in TEST_F()3779 call_builder.AddInstruction( in TEST_F()3783 m->AddEntryComputation(call_builder.Build()); in TEST_F()