Searched defs:call_builder (Results 1 – 7 of 7) sorted by relevance
58 typedef struct call_builder struct64 } call_builder; typedef
224 NodeBuilder call_builder(call_node_name, "PartitionedCall"); in BuildPartitionedCallOp() local300 NodeBuilder call_builder(call_node_name, "StatefulPartitionedCall"); in BuildStatefulPartitionedCallOp() local
386 XlaBuilder call_builder("call"); in TEST_F() local
1781 NodeDefBuilder call_builder(absl::StrCat("oc_call_", func_call_node_name), in BuildHostGraphForFuncCallNode() local
2670 auto call_builder = HloComputation::Builder("Call"); in TEST_P() local2741 auto call_builder = HloComputation::Builder("Call"); in TEST_P() local
1402 auto call_builder = HloComputation::Builder(TestName() + "_call"); in TEST_F() local
4955 HloComputation::Builder call_builder(TestName() + ".Call"); in TEST_F() local