Searched refs:b_builder (Results 1 – 2 of 2) sorted by relevance
321 NodeBuilder b_builder("B", "SymbolicGradient", in TEST() local325 b_builder.Attr("f", b_name_attr); in TEST()326 b_builder.Attr("Tin", {DT_FLOAT, DT_FLOAT}); in TEST()327 b_builder.Attr("Tout", {DT_FLOAT}); in TEST()328 b_builder.Input({a, a}); in TEST()329 Node* b = builder.opts().FinalizeBuilder(&b_builder); in TEST()
1442 auto b_builder = HloComputation::Builder(TestName() + "_b"); in TEST_F() local1443 auto b_param = b_builder.AddInstruction( in TEST_F()1445 auto b_call = b_builder.AddInstruction( in TEST_F()1447 auto b_computation = b_builder.Build(); in TEST_F()