Searched refs:b_builder (Results 1 – 2 of 2) sorted by relevance
387 NodeBuilder b_builder("B", "SymbolicGradient", in TEST() local391 b_builder.Attr("f", b_name_attr); in TEST()392 b_builder.Attr("Tin", {DT_FLOAT, DT_FLOAT}); in TEST()393 b_builder.Attr("Tout", {DT_FLOAT}); in TEST()394 b_builder.Input({a, a}); in TEST()395 Node* b = builder.opts().FinalizeBuilder(&b_builder); in TEST()
1614 auto b_builder = HloComputation::Builder(TestName() + "_b"); in TEST_F() local1615 auto b_param = b_builder.AddInstruction( in TEST_F()1617 auto b_call = b_builder.AddInstruction( in TEST_F()1619 auto b_computation = b_builder.Build(); in TEST_F()